From 01ea2597d922774ff641d6a2eacae22979d68802 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Fri, 22 May 2015 11:31:25 +0800
Subject: added shuffle option

---
 split.cpp | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'split.cpp')

diff --git a/split.cpp b/split.cpp
index be80545..e774ea9 100644
--- a/split.cpp
+++ b/split.cpp
@@ -43,6 +43,7 @@ int main(int argc, char **argv)
     desc.add_options()
             ("help,h", "produce help message")
             ("query,Q", "Query person count")
+            ("shuffle,s","shuffle input")
             ("count,c", po::value<int>(), "take number")
             ("take,a", po::value<string>(), "set output rid file 1(taken)")
             ("left,b", po::value<string>(), "set output rid file 2(left)")
@@ -68,9 +69,16 @@ int main(int argc, char **argv)
     vector<DataEntry*> a;
     vector<DataEntry*> b;
     dp.open();
-    dp.take(vm["count"].as<int>(),a,b);
-    outputRid(a,dp.getfSize(),vm["take"].as<string>());
-    outputRid(b,dp.getfSize(),vm["left"].as<string>());
+    if (vm.count("shuffle")) {
+        dp.getRidVector(a);
+        dp.shuffle(a);
+        outputRid(a, dp.getfSize(), vm["take"].as<string>());
+    }
+    else {
+        dp.take(vm["count"].as<int>(), a, b);
+        outputRid(a, dp.getfSize(), vm["take"].as<string>());
+        outputRid(b, dp.getfSize(), vm["left"].as<string>());
+    }
     dp.close();
     return 0;
 }
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2