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

---
 tools/fileDataProvider.h | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'tools/fileDataProvider.h')

diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index 567c8e2..972a4c5 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -38,11 +38,20 @@ public:
     void readEntries();
     int getfSize() { if(!read) readEntries(); return d.getfSize();};
     int getpSize();
+    void shuffle(std::vector<DataEntry*> &dat);
     virtual void getAllDataSet(RidList &out);
     virtual int getDataSet(DataList &out);
     virtual int open(){fin.open(fname); eof=false;return 0;};
     virtual int close(){fin.close(); d.clear();return 0;};
     void take(int n,std::vector<DataEntry*> &a,std::vector<DataEntry*> &b);
+    void getRidVector(std::vector<DataEntry*> &rid){
+        if (!read)
+            readEntries();
+        rid.clear();
+        std::vector<DataEntry*> &dat = d.getData();
+        for (int i=0;i<dat.size();++i)
+            rid.push_back(dat[i]);
+    }
 };
 
 #endif
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2