add save to cellset.h

Laila Saadatifard
1 parent f74c9f43
Showing 1 changed file with 20 additions and 0 deletions Show diff stats
stim/biomodels/cellset.h
@@ -167,6 +167,26 @@ public:
 		cells.push_back(newcell);						//push the new memory entry into the cell array
 	}
+void save(std::string filename){
+
+		
+		size_t ncells = cells.size();	// get the number of cells
+		std::ofstream file(filename);	//open a file to store the cell's coordinates
+		if (file.is_open()) {
+		
+			file << "x y z radius\n";	//write the file header
+			for (size_t c=0; c < ncells; c++){   //for each cell
+				if (cells[c][ip[3]] != NULL)	//check if for the current cell, radius has been assigned
+					file << cells[c][ip[0]]  << delim << cells[c][ip[1]] << delim << cells[c][ip[2]] << delim << cells[c][ip[3]] << '\n' ;
+				else							//check if for the current cell, radius has not been assigned, set radius to zero
+					file << cells[c][ip[0]]  << delim << cells[c][ip[1]] << delim << cells[c][ip[2]] << delim << 0 << '\n' ;
+			}
+			
+		}
+			file.close();
+
+	}
+	
 };		//end class cellset
 };		//end namespace stim