RidFile tested

1 files changed, 5 insertions, 1 deletions

diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index c4f6a4a..8ebda20 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -82,9 +82,11 @@ public:
             }
             pos = 0;
             qid = 1;
+            read = true;
         }
         out.clear();
         fsize = d.getfSize();
+        out.setfSize(fsize);
         std::vector<DataEntry*> & dat = d.getData();
         for (int i=0;i<d.getSize();++i)
             if (i!=pos)
@@ -94,6 +96,7 @@ public:
                     e = new DataEntry;
                     e->rank=1;
                     dat[i]->qid=std::to_string(qid);
+                    dat[i]->rank=qid;
                 }
                 else
                 {
@@ -103,12 +106,13 @@ public:
                 e->feature.resize(d.getfSize());
                 e->qid=std::to_string(qid);
                 for (int j = 0; j < fsize; ++j) {
-                    e->feature(i) = fabs(dat[i]->feature(j) -dat[pos]->feature(j));
+                    e->feature(j) = fabs(dat[i]->feature(j) -dat[pos]->feature(j));
                 }
                 out.addEntry(e);
             }
         dat[pos]->qid=std::to_string(qid);
         ++qid;
+        dat[pos]->rank=qid;
         while (pos<dat.size() && dat[pos]->rank!=-1)
             ++pos;
         if (pos==d.getSize())