added functions to return a string of array without brackets for obj exporting from a network

pranathivemuri
1 parent 54259237
Showing 1 changed file with 18 additions and 10 deletions Show diff stats
stim/visualization/network.h
@@ -577,26 +577,34 @@ class network{
 		} 
 		ofs.close();
 	}
-	void 
-	myfunction (T i) {  // function:
-    std::cout << ' ' << i;
-    }
-	///exports the graph.
+	void removeCharsFromString(std::string &str, char* charsToRemove ) {
+	   for ( unsigned int i = 0; i < strlen(charsToRemove); ++i ) {
+		  str.erase( remove(str.begin(), str.end(), charsToRemove[i]), str.end() );
+	   }
+	}
+	///exports the network graph to obj
 	void
 	to_obj()
 	{
 		std::ofstream ofs;
 		ofs.open("Graph.obj", std::ofstream::out | std::ofstream::app);
+		int num;
+		num = V.size();
+		string *strArray = new string[num];
 		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";
-			  
+			std::string str;
+			str = V[i].str();
+			removeCharsFromString(str, "[],");
+			ofs << "v " << str << "\n";
+			removeCharsFromString(str," ");
+			strArray[i] = str;
 		}
 		for(unsigned int i = 0; i < E.size(); i++)
 		{
-             ofs << "l " << E[i]->str() << "\n";
+			 std::string str;
+			 str = E[i]->strObj(strArray, num);
+             ofs << "l " << str << "\n";
 		} 
 		ofs.close();
 	}