add node function added and corrected edge to string function

pranathivemuri
1 parent 0b5fdc0d
Showing 1 changed file with 32 additions and 5 deletions Show diff stats
stim/visualization/network.h
@@ -488,6 +488,12 @@ class network{
 		edge *a = new edge(pos,radii);
 		E.push_back(a);
 	}
+	void
+	addNode(stim::vec<T> nodes)
+	{
+			node *a = new node(nodes);
+			V.push_back(nodes);
+	}
  
 	///adds an edge from two std::lists with positions and radii.
 	/// NOT NECESSARY.
@@ -505,7 +511,7 @@ class network{
 		std::stringstream ss;
 		for(unsigned int i = 0; i < E.size(); i++)
 		{
-			ss << i << ": " << E[i].str() << std::endl;
+			ss << i << ": " << E[i]->str() << std::endl;
 		}
 		return(ss.str());
 	}
@@ -566,14 +572,35 @@ class network{
 		ofs << "edgedef>node1 VARCHAR, node2 VARCHAR, weight INT, length FLOAT, av_radius FLOAT \n";
 		for(int i = 0; i < E.size(); i++)
 		{
-			float len;
 			ofs << E[i]->Node1 << "," << E[i]->Node2 << "," <<E[i]->n_pts()
-			 << ","<< E[i]->length() << "," << E[i]->radius(len) << "\n";
+			 << ","<< E[i]->length() << "," << E[i]->average_radius() << "\n";
+		} 
+		ofs.close();
+	}
+	void 
+	myfunction (T i) {  // function:
+    std::cout << ' ' << i;
+    }
+	///exports the graph.
+	void
+	to_obj()
+	{
+		std::ofstream ofs;
+		ofs.open("Graph.obj", std::ofstream::out | std::ofstream::app);
+		for(int i = 0; i < V.size(); i++)
+		{
+			std::vector<T> myvector;
+			myvector = V[i]
+			ofs << "v " << for_each (myvector.begin(), myvector.end(), myfunction) << "\n";
+			  
+		}
+		for(unsigned int i = 0; i < E.size(); i++)
+		{
+             ofs << "l " << E[i]->str() << "\n";
 		} 
 		ofs.close();
 	}	
-};
  
 };
-
+};
 #endif