From 44018ad44d7d0d8196f16402bd1fa6c1c10de8ad Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Tue, 16 Jun 2015 11:34:46 +0800
Subject: fscore

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

(limited to 'tools/fileDataProvider.h')

diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index 972a4c5..0ab1948 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -29,13 +29,20 @@ class RidFileDP:public DataProvider
 private:
     std::string fname;
     std::ifstream fin;
+    std::vector<double> mask;
     DataList d;
     bool read;
+    bool maskinit;
     int pos;
     int qid;
 public:
-    RidFileDP(std::string fn=""):fname(fn){read=false;};
+    RidFileDP(std::string fn=""):fname(fn),read(false),maskinit(false){};
     void readEntries();
+    void datmask(std::vector<double> &m){
+        mask.resize(m.size());
+        for (int i=0;i<m.size();++i)
+            mask[i]=m[i];
+        maskinit=true;}
     int getfSize() { if(!read) readEntries(); return d.getfSize();};
     int getpSize();
     void shuffle(std::vector<DataEntry*> &dat);
@@ -52,6 +59,7 @@ public:
         for (int i=0;i<dat.size();++i)
             rid.push_back(dat[i]);
     }
+    static void seed();
 };
 
 #endif
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2