diff options
Diffstat (limited to 'tools/fileDataProvider.h')
| -rw-r--r-- | tools/fileDataProvider.h | 10 |
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 |