fscore

1 files changed, 9 insertions, 1 deletions

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