improved error messages, defaulted to read-only ENVI files

David Mayerich
1 parent b66bbf3f
Showing 6 changed files with 56 additions and 23 deletions Show diff stats
stim/envi/agilent_binary.h
stim/envi/bil.h
stim/envi/binary.h
stim/envi/bip.h
stim/envi/bsq.h
stim/envi/envi.h
@@ -225,7 +225,7 @@ public:
 //#ifdef CUDA_FOUND
 	/// Perform an FFT and return a binary file with bands in the specified range
-	agilent_binary<T> fft(double band_min, double band_max, double ELWN = 15798, int UDR = 2){
+	agilent_binary<T> fft(double band_min, double band_max, double ELWN = 15798, int UDR = 2, int device = 0){
 		auto total_start = std::chrono::high_resolution_clock::now();
 		auto start = std::chrono::high_resolution_clock::now();
@@ -240,6 +240,29 @@ public:
        // std::cout << "Transpose data: " << diff.count() << " s\n";
 		start = std::chrono::high_resolution_clock::now();
+		if (device >= 0) {														//if a CUDA device is specified
+			int dev_count;
+			HANDLE_ERROR(cudaGetDeviceCount(&dev_count));						//get the number of CUDA devices
+			std::cout << "Number of CUDA devices: " << dev_count << std::endl;		//output the number of CUDA devices
+			cudaDeviceProp prop;
+			std::cout << "CUDA devices----" << std::endl;
+			for (int d = 0; d < dev_count; d++) {									//for each CUDA device
+				cudaGetDeviceProperties(&prop, d);									//get the property of the first device
+																					//float cc = prop.major + prop.minor / 10.0f;						//calculate the compute capability
+				std::cout << d << ":  [" << prop.major << "." << prop.minor << "]      " << prop.name << std::endl;	//display the device information
+																													//if(cc > best_device_cc){
+																													//	best_device_cc = cc;										//if this is better than the previous device, use it
+																													//	best_device_id = d;
+																													//}
+			}
+			if (dev_count > 0 && dev_count > device) {							//if the first device is not an emulator
+				cudaGetDeviceProperties(&prop, device);						//get the property of the requested CUDA device
+				if (prop.major != 9999) {
+					std::cout << "Using device " << device << std::endl;
+					HANDLE_ERROR(cudaSetDevice(device));
+				}
+			}
+		}
 		cufftHandle plan;															//allocate space for a cufft plan
 		cufftReal* gpu_data;														//create a pointer to the data
 		size_t batch = R[0] * R[1];													//calculate the batch size (X * Y)
@@ -55,11 +55,12 @@ public:
 			  unsigned long long Y,
 			  unsigned long long B,
 			  unsigned long long header_offset,
-			  std::vector<double> wavelengths){
+			  std::vector<double> wavelengths,
+			  stim::iotype io = stim::io_in){
 		w = wavelengths;
-		return open(filename, vec<unsigned long long>(X, B, Y), header_offset);
+		return open(filename, vec<unsigned long long>(X, B, Y), header_offset, io);
 	}
@@ -143,6 +143,8 @@ public:
  *  @param T is the data type used to store data to disk (generally float or double)
  *  @param D is the dimension of the data (default 3)
  */
+
+enum iotype {io_in, io_out};
 template< typename T, unsigned int D = 3 >
 class binary{
@@ -205,19 +207,21 @@ protected:
 	/// Private helper file that opens a specified binary file.
 	/// @param filename is the name of the binary file to stream
-	bool open_file(std::string filename){
+	bool open_file(std::string filename, stim::iotype io = io_in){
 		//open the file as binary for reading and writing
-		file.open(filename.c_str(), std::ios::in | std::ios::out | std::ios::binary);
-
-		//if the file isn't open, the user may only have read access
-		if(!file.is_open()){
-			std::cout<<"class STIM::BINARY - failed to open file, trying for read only"<<std::endl;
+		if(io == io_in)
 			file.open(filename.c_str(), std::ios::in | std::ios::binary);
-			if(!file.is_open()){
-				std::cout<<"               still unable to load the file"<<std::endl;
+		else if (io == io_out) {
+			file.open(filename.c_str(), std::ios::out | std::ios::binary);
+			if (!file.is_open()) {
+				std::cout << "stim::binary ERROR - unable to open file for writing: " << filename << std::endl;
 				return false;
 			}
 		}
+		else {
+			std::cout << "stim::binary ERROR - unrecognized IO format" << std::endl;
+			return false;
+		}
 		//if the file is successful
 		if(file){
@@ -275,14 +279,14 @@ public:
 	/// @param filename is the name of the binary file
 	/// @param r is a STIM vector specifying the size of the binary file along each dimension
 	/// @param h is the length (in bytes) of any header file (default zero)
-	bool open(std::string filename, vec<unsigned long long> r, unsigned long long h = 0){
+	bool open(std::string filename, vec<unsigned long long> r, unsigned long long h = 0, stim::iotype io = stim::io_in){
 		for(unsigned long long i = 0; i < D; i++)		//set the dimensions of the binary file object
 			R[i] = r[i];
 		header = h;				//save the header size
-		if(!open_file(filename)) return false;	//open the binary file
+		if(!open_file(filename), io) return false;	//open the binary file
 		//reset();
@@ -65,14 +65,15 @@ public:
 			  unsigned long long Y,
 			  unsigned long long B,
 			  unsigned long long header_offset,
-			  std::vector<double> wavelengths){
+			  std::vector<double> wavelengths,
+			  stim::iotype io = stim::io_in){
 		//copy the wavelengths to the BSQ file structure
 		w = wavelengths;
 		//copy the offset to the structure
 		offset = header_offset;
-		return open(filename, vec<unsigned long long>(B, X, Y), header_offset);
+		return open(filename, vec<unsigned long long>(B, X, Y), header_offset, io);
 	}
@@ -64,14 +64,15 @@ public:
 			  unsigned long long Y,
 			  unsigned long long B,
 			  unsigned long long header_offset,
-			  std::vector<double> wavelengths){
+			  std::vector<double> wavelengths,
+			  stim::iotype io = stim::io_in){
 		//copy the wavelengths to the BSQ file structure
 		w = wavelengths;
 		//copy the wavelengths to the structure
 		offset = header_offset;
-		return open(filename, vec<unsigned long long>(X, Y, B), header_offset);
+		return open(filename, vec<unsigned long long>(X, Y, B), header_offset, io);
 	}
 	/// Retrieve a single band (based on index) and stores it in pre-allocated memory.
@@ -363,7 +363,7 @@ public:
 	}
 	/// Open a previously opened ENVI file
-	bool open(){
+	bool open(stim::iotype io = stim::io_in){
 		//load the file
 		if(header.interleave == envi_header::BSQ) {		//if the infile is bsq file
@@ -451,7 +451,7 @@ public:
 	/// @param filename is the name of the ENVI binary file
 	/// @param header is an ENVI header structure
-	bool open(std::string filename, stim::envi_header h){
+	bool open(std::string filename, stim::envi_header h, stim::iotype io = stim::io_in){
 		header = h;							//store the header
@@ -459,7 +459,7 @@ public:
 		allocate();
-		return open();						//open the ENVI file;
+		return open(io);						//open the ENVI file;
 	}
@@ -468,18 +468,21 @@ public:
 	/// @param filename is the name of the ENVI binary file
 	/// @param headername is the name of the ENVI header file
-	bool open(std::string filename, std::string headername){
+	bool open(std::string filename, std::string headername, stim::iotype io = stim::io_in){
 		//allocate memory
 		//allocate();
 		stim::envi_header h;
-		h.load(headername);
+		if (!h.load(headername)) {
+			std::cout << "Error loading header file: " << headername << std::endl;
+			return false;
+		}
 		//load the header
 		//header.load(headername);
-		return open(filename, h);
+		return open(filename, h, io);
 	}
 	/// Normalize a hyperspectral ENVI file given a band number and threshold.