1 files changed, 4 insertions, 1 deletions
diff --git a/main.cpp b/main.cpp
index cf1dd14..434437f 100644
--- a/main.cpp
+++ b/main.cpp
@@ -110,7 +110,10 @@ int main(int argc, char **argv) {
     }
     else return 0;
     DataProvider* dp;
-    dp = new FileDP(vm["feature"].as<std::string>());
+    if (vm["feature"].as<std::string>().find(".rid") == std::string::npos)
+        dp = new FileDP(vm["feature"].as<std::string>());
+    else
+        dp = new RidFileDP(vm["feature"].as<std::string>());
     mainf(*dp);
     delete dp;
     return 0;