load text file to a network load_txt function written

pranathivemuri
1 parent fe0c7539
Showing 1 changed file with 55 additions and 2 deletions Show diff stats
stim/biomodels/network.h
@@ -372,8 +372,61 @@ public:
 	unsigned nmags(){
 		return E[0].nmags();
 	}
-
-
+	// split a string in text by the character sep
+	stim::vec<T> split(std::string &text, char sep) 
+	{
+		stim::vec<T> tokens;
+		std::size_t start = 0, end = 0;
+		while ((end = text.find(sep, start)) != std::string::npos) {
+		tokens.push_back(atof(text.substr(start, end - start).c_str()));
+		start = end + 1;
+		}
+		tokens.push_back(atof(text.substr(start).c_str()));
+		return tokens;
+	}
+	// load a network in text file to a network class
+	void load_txt(std::string filename)
+	{
+		std::vector <std::string> file_contents;
+		std::ifstream file(filename);
+		std::string line;
+		std::vector<unsigned> id2vert;	//this list stores the vertex ID associated with each network vertex
+		//for each line in the text file, store them as strings in file_contents
+		while (std::getline(file, line))
+		{
+			std::stringstream ss(line);
+			file_contents.push_back(ss.str());
+		}
+		unsigned int numEdges = atoi(file_contents[0].c_str()); //number of edges in the network
+		unsigned int I = atoi(file_contents[1].c_str()) ;		//calculate the number of points3d on the first edge
+		unsigned int count = 1; unsigned int k = 2; // count is global counter through the file contents, k is for the vertices on the edges
+		// for each edge in the network.
+		for (unsigned int i = 0; i < numEdges; i ++ )
+		{
+			// pre allocate a position vector p with number of points3d on the edge p
+			std::vector< stim::vec<T> > p(0, I);
+			// for each point on the nth edge
+		  for (unsigned int j = k; j < I + k; j++)
+		  {
+			 // split the points3d of floats with separator space and form a float3 position vector out of them
+			  p.push_back(split(file_contents[j], ' '));
+		  }
+			count += p.size() + 1; // increment count to point at the next edge in the network
+			I = atoi(file_contents[count].c_str()); // read in the points3d at the next edge and convert it to an integer
+			k = count + 1;
+			edge new_edge = p; // create an edge with a vector of points3d  on the edge
+			E.push_back(new_edge); // push the edge into the network
+		}
+		unsigned int numVertices = atoi(file_contents[count].c_str()); // this line in the text file gives the number of distinct vertices
+		count = count + 1; // this line of text file gives the first verrtex
+		// push each vertex into V
+		for (unsigned int i = 0; i < numVertices; i ++)
+		{
+			vertex new_vertex = split(file_contents[count], ' ');
+			V.push_back(new_vertex);
+			count += atoi(file_contents[count + 1].c_str()) + 2; // Skip number of edge ids + 2 to point to the next vertex
+		}
+	} // end load_txt function
 };		//end stim::network class
 };		//end stim namespace
 #endif