From 4f0acec64ad8edcea4269b4121b2ddb2215d3503 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Mon, 27 Apr 2015 11:04:58 +0800
Subject: RidFileDP & qid to string

---
 main.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'main.cpp')

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;
-- 
cgit v1.2.3-70-g09d2