1
0
Fork 0

A couple updates to the FGArray API.

This commit is contained in:
curt 2005-11-10 16:26:59 +00:00
parent 4e20e26d04
commit dde5a64a75
2 changed files with 59 additions and 3 deletions

View file

@ -45,14 +45,20 @@ SG_USING_STD(cout);
SG_USING_STD(endl); SG_USING_STD(endl);
TGArray::TGArray( void ) { TGArray::TGArray( void ):
array_in(NULL),
fitted_in(NULL)
{
// cout << "class TGArray CONstructor called." << endl; // cout << "class TGArray CONstructor called." << endl;
in_data = new int[ARRAY_SIZE_1][ARRAY_SIZE_1]; in_data = new int[ARRAY_SIZE_1][ARRAY_SIZE_1];
// out_data = new float[ARRAY_SIZE_1][ARRAY_SIZE_1]; // out_data = new float[ARRAY_SIZE_1][ARRAY_SIZE_1];
} }
TGArray::TGArray( const string &file ) { TGArray::TGArray( const string &file ):
array_in(NULL),
fitted_in(NULL)
{
// cout << "class TGArray CONstructor called." << endl; // cout << "class TGArray CONstructor called." << endl;
in_data = new int[ARRAY_SIZE_1][ARRAY_SIZE_1]; in_data = new int[ARRAY_SIZE_1][ARRAY_SIZE_1];
// out_data = new float[ARRAY_SIZE_1][ARRAY_SIZE_1]; // out_data = new float[ARRAY_SIZE_1][ARRAY_SIZE_1];
@ -171,6 +177,41 @@ TGArray::parse( SGBucket& b ) {
} }
// write an Array file
bool TGArray::write( const string root_dir, SGBucket& b ) {
// generate output file name
string base = b.gen_base_path();
string path = root_dir + "/" + base;
#ifdef _MSC_VER
fg_mkdir( path.c_str() );
#else
string command = "mkdir -p " + path;
system( command.c_str() );
#endif
string array_file = path + "/" + b.gen_index_str() + ".arr.new.gz";
cout << "array_file = " << array_file << endl;
// write the file
gzFile fp;
if ( (fp = gzopen( array_file.c_str(), "wb9" )) == NULL ) {
cout << "ERROR: cannot open " << array_file << " for writing!" << endl;
return false;
}
cout << "origin = " << originx << ", " << originy << endl;
gzprintf( fp, "%d %d\n", (int)originx, (int)originy );
gzprintf( fp, "%d %d %d %d\n", cols, (int)col_step, rows, (int)row_step );
for ( int i = 0; i < cols; ++i ) {
for ( int j = 0; j < rows; ++j ) {
gzprintf( fp, "%d ", (int)in_data[i][j] );
}
gzprintf( fp, "\n" );
}
gzclose(fp);
return true;
}
// do our best to remove voids by picking data from the nearest neighbor. // do our best to remove voids by picking data from the nearest neighbor.

View file

@ -86,7 +86,16 @@ public:
~TGArray( void ); ~TGArray( void );
// open an Array file (use "-" if input is coming from stdin) // open an Array file (use "-" if input is coming from stdin)
bool open ( const string& file_based ); bool open ( const string& file_base );
// return if array was successfully opened or not
inline bool is_open() {
if ( array_in != NULL ) {
return array_in->is_open();
} else {
return false;
}
}
// close a Array file // close a Array file
bool close(); bool close();
@ -94,6 +103,9 @@ public:
// parse a Array file // parse a Array file
bool parse( SGBucket& b ); bool parse( SGBucket& b );
// write an Array file
bool write( const string root_dir, SGBucket& b );
// do our best to remove voids by picking data from the nearest // do our best to remove voids by picking data from the nearest
// neighbor. // neighbor.
void remove_voids(); void remove_voids();
@ -126,6 +138,9 @@ public:
inline int get_array_elev( int col, int row ) { inline int get_array_elev( int col, int row ) {
return in_data[col][row]; return in_data[col][row];
} }
inline void set_array_elev( int col, int row, int val ) {
in_data[col][row] = val;
}
inline Point3D get_fitted_pt( int i ) { inline Point3D get_fitted_pt( int i ) {
return fitted_list[i]; return fitted_list[i];
} }