added ability to save a single image

David Mayerich
1 parent 8b7be670
Showing 1 changed file with 14 additions and 14 deletions Show diff stats
grids/image_stack.h
@@ -58,6 +58,17 @@ public:
 		}
 	}
+	void save_image(std::string file_name, unsigned int i){
+
+		//create an image
+		stim::image<T> I;
+
+		//retrieve the interlaced data from the image - store it in the grid
+		I.set_interleaved(&ptr[ i * R[0] * R[1] * R[2] ], R[1], R[2], R[0]);
+
+		I.save(file_name);
+	}
+
 	void save_images(std::string file_mask){
 		stim::filename file_path(file_mask);
@@ -71,22 +82,11 @@ public:
 		//create a list of file names
 		std::vector<std::string> file_list = stim::wildcards::increment(file_path.str(), 0, R[3]-1, 1);
-		for(int i=0; i<R[3]; i++){
-
-			//create an image
-			stim::image<T> I;
-
-			//retrieve the interlaced data from the image - store it in the grid
-			I.set_interleaved(&ptr[ i * R[0] * R[1] * R[2] ], R[1], R[2], R[0]);
-
-			I.save(file_list[i]);
-
-
-			std::cout<<file_list[i]<<std::endl;
-		}
+		for(int i=0; i<R[3]; i++)
+			save_image(file_list[i], i);
+	}
-	}