X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOGimmick.cpp;h=ced62fa28882e1d5ca1b9e6f6fd49dc631a3a642;hb=f6eb8241261ed6a79abc3c73ada5a94349239141;hp=64099e589c651e2bbf37fe4fefa1667cfcc7ef0a;hpb=4ef5bb7efa72c0d9a016f270b5a56e8302f6e209;p=creaImageIO.git

diff --git a/src2/creaImageIOGimmick.cpp b/src2/creaImageIOGimmick.cpp
index 64099e5..ced62fa 100644
--- a/src2/creaImageIOGimmick.cpp
+++ b/src2/creaImageIOGimmick.cpp
@@ -1,21 +1,19 @@
 #include <creaImageIOGimmick.h>
 
 #include <creaImageIOSystem.h>
-#include <creaImageIOImageFinder.h>
 
 #include <boost/filesystem.hpp>
 #include <boost/algorithm/string.hpp>
 
 namespace creaImageIO
 {
-  
+
+
   //==============================================================
   Gimmick::Gimmick()
+    : mImageAdder(0)
   {    
-    crea::MessageManager::RegisterMessageType("Gimmick!",
-					      "Gimmick",1);
-    crea::MessageManager::RegisterMessageType("Gimmick! DEBUG",
-					      "Gimmick",0);
+    RegisterGimmickMessageTypes();
   }
   //==============================================================
 
@@ -39,6 +37,9 @@ namespace creaImageIO
 
     // Create local database handler
     mLocalDatabase = new SQLiteTreeHandler(GetLocalDatabasePath());
+    // Add it to the TreeHandlerMap
+    mTreeHandlerMap["Local database"] = mLocalDatabase;
+
     // Create or open local database
     if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
       {
@@ -154,7 +155,7 @@ namespace creaImageIO
   /// Sets message level
   void Gimmick::SetMessageLevel(int l)
   {
-    crea::MessageManager::SetMessageLevel("Gimmick!",l);
+    SetGimmickMessageLevel(l);
   }
   //========================================================================
 
@@ -162,42 +163,57 @@ namespace creaImageIO
   /// Sets message level
   void Gimmick::SetDebugMessageLevel(int l)
   {
-    crea::MessageManager::SetMessageLevel("Gimmick! DEBUG",l);
+    SetGimmickDebugMessageLevel(l);
   }
   //========================================================================
 
   //========================================================================
   /// Add a file to the local database
-  void Gimmick::AddFileToLocalDatabase(const std::string& f)
-  {
-    ImageFinder finder(mLocalDatabase);
-    if (finder.IsHandledFile(f)) 
+  TreeHandler* Gimmick::GetTreeHandler(const std::string& name) const 
+  {  
+    TreeHandlerMapType::const_iterator i;
+    i = GetTreeHandlerMap().find(name);
+    if ( i == GetTreeHandlerMap().end() )
       {
-	finder.AddFile(f);
+	GimmickError("TreeHandler '"<<name<<"' does not exist");
       }
-    else
-      {
-	GimmickError("File '"<<f<<"' does not exist or is not handled");
-      }    
+    return i->second;
+  }
+
+
+  //========================================================================
+  /// Add the files to the tree handler
+  void Gimmick::AddFiles(const std::string& d, 
+			const std::vector<std::string>& filenames)
+  {
+    GimmickMessage(2,"Adding files to '"<<d<<"'"<<std::endl);
+ 
+    mImageAdder.SetTreeHandler(GetTreeHandler(d));
+    mImageAdder.AddFiles(filenames);
+
   }
   //========================================================================
 
   //========================================================================
   /// Add a dir to the local database
-  void Gimmick::AddDirToLocalDatabase(const std::string& f, bool recurse)
+  void Gimmick::AddDir(const std::string& d, const std::string& f, 
+		       bool recurse)
   {
-    ImageFinder finder(mLocalDatabase);
-    
-    finder.AddDirectory(f,recurse);
+    GimmickMessage(2,"Adding dir '"<<f<<"' to '"<<d<<"' recurse:"
+		   <<recurse<<std::endl);
+
+    mImageAdder.SetTreeHandler(GetTreeHandler(d));
+    mImageAdder.AddDirectory(f,recurse);
     
   }
+
   //========================================================================
 
   //========================================================================
   /// 
-  void Gimmick::PrintLocalDatabase()
+  void Gimmick::Print(const std::string& d)
   {
-    mLocalDatabase->GetTree().Print();
+    GetTreeHandler(d)->GetTree().Print();
   }
   //========================================================================