Changeset 6216 for geovis

Timestamp:

Mar 29, 2016 8:52:47 AM (8 years ago)

Author:

ldelgass

Message:

add args for collection, filename to idatatest

File:

• geovis/trunk/idatatest.cpp (modified) (3 diffs)

geovis/trunk/idatatest.cpp

@@ -2 +2 @@
 #include <cstdlib>
 #include <iostream>
+#include <sstream>
 #include <fstream>
 
@@ -22 +23 @@
     }
 
+    int collection = 3;
+    if (argc > 1) {
+        collection = atoi(argv[1]); 
+    }
+
     std::cout << std::endl << std::endl;
     std::vector<DirectoryItem> dir;
-    getContents(3, "", dir);
+    getContents(collection, "", dir);
     std::cout << std::endl << std::endl;
     for (std::vector<DirectoryItem>::iterator itr = dir.begin();
@@ -37 +43 @@
 
     std::cout << std::endl << std::endl;
+    const char *filename = NULL;
+    if (argc > 2) {
+        filename = argv[2];
+    } else if (collection == 3) {
+        filename = "station.csv";
+    } else {
+        iDataCleanup();
+        return 0;
+    }
+
+    std::ostringstream oss;
+    oss << "//" << filename;
     Buffer buff;
-    getFile(3, "//station.csv", &buff);
+    getFile(collection, oss.str().c_str(), &buff);
     std::cout << std::endl << std::endl;
-    std::cout << "station.csv size: " << buff.size << std::endl;
-    std::ofstream file("station.csv");
+    std::cout << filename << " size: " << buff.size << std::endl;
+    std::ofstream file(filename);
     file.write((char *)buff.data, buff.size);
     file.close();