]> Creatis software - creaImageIO.git/blobdiff - src2/creaImageIOGimmick.cpp
bug correction on addfiles .
[creaImageIO.git] / src2 / creaImageIOGimmick.cpp
index 64099e589c651e2bbf37fe4fefa1667cfcc7ef0a..de6b11dd169ea626d5255a2a64b5f2813607ad10 100644 (file)
@@ -1,21 +1,24 @@
 #include <creaImageIOGimmick.h>
 
 #include <creaImageIOSystem.h>
-#include <creaImageIOImageFinder.h>
-
 #include <boost/filesystem.hpp>
 #include <boost/algorithm/string.hpp>
+//#include "io.h"
+#ifndef PATH_MAX // If not defined yet : do it 
+#  define PATH_MAX 2048
+#endif
 
 namespace creaImageIO
 {
-  
+
+
   //==============================================================
   Gimmick::Gimmick()
+    : mImageAdder(0)
   {    
-    crea::MessageManager::RegisterMessageType("Gimmick!",
-                                             "Gimmick",1);
-    crea::MessageManager::RegisterMessageType("Gimmick! DEBUG",
-                                             "Gimmick",0);
+    RegisterGimmickMessageTypes();
+       mSettings=0;
+       mSynchronizer=0;
   }
   //==============================================================
 
@@ -24,58 +27,122 @@ namespace creaImageIO
   //==============================================================
   Gimmick::~Gimmick()
   {
-    
+        if(mSettings!=0)
+         {
+         mSettings->writeSettingsFile();
+      delete mSettings;
+         }
+         if(mSynchronizer!=0)
+         {
+         delete mSynchronizer;
+         }
   }
   //==============================================================
   
+  //==============================================================
+  void Gimmick::Initialize(const std::string& path)
+  {
+         Initialize();
+  }
 
   //==============================================================
   void Gimmick::Initialize()
   {
+       std::string i_nameDB = "Local database";
     // Create the UserSettings dir if does not exist
     CreateUserSettingsDirectory();
     // Sets the current directory to the home dir
     mCurrentDirectory =  GetHomeDirectory();
+       mSynchronizer= new Synchronizer(GetUserSettingsDirectory()+"Shared/gimmick/");
 
-    // Create local database handler
-    mLocalDatabase = new SQLiteTreeHandler(GetLocalDatabasePath());
-    // Create or open local database
-    if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
-      {
-       std::string mess = "Local database '";
-       mess += GetLocalDatabasePath();
-       mess += "' does not exist : creating it";
-       GimmickMessage(1,mess<<std::endl);
-       
-       // CREATING DEFAULT DB STRUCTURE
-       mLocalDatabase->GetTree().GetDescriptor().CreateDefault();
+       mSettings  = new Settings(mCurrentDirectory);
        
-       if ( ! mLocalDatabase->Create(true) )
-         {
-           GimmickError("ERROR CREATING '"<<GetLocalDatabasePath()<<"'");
-         }
-       mLocalDatabase->SetAttribute(0,"Name","Local database");
-      }
-    else 
-      {
-       /// Open and test it
-       GimmickMessage(1,"Opening local database '"
-                      <<GetLocalDatabasePath()<<"' "
-                      <<std::endl);
-       if ( ! mLocalDatabase->Open(true) )
-         {
-           GimmickError("ERROR OPENING '"<<GetLocalDatabasePath()<<"'");
-         }
+
+       std::string dbpath = GetLocalDatabasePath();
+    // Create or open local database
+       std::string dpath= mCurrentDirectory + "/.gimmick/Shared/gimmick/localdatabase_Descriptor.txt";
+       boost::algorithm::replace_all( dpath,
+                                      INVALID_FILE_SEPARATOR , 
+                                      VALID_FILE_SEPARATOR);
+       mLocalDatabase = createDB(i_nameDB, dpath, dbpath);
+    // Add it to the TreeHandlerMap
+    mTreeHandlerMap[i_nameDB] = mLocalDatabase;
+    
+       //Add additional DB from user Settings
+       addDBSettings();
+
        
+  }
+
+   ///////////////////////////////////////////////////////////////////////
+   // add DB to TreeHandler Map                                                                                        //
+   // @param i_name : DB name                                                                              //
+   // @param i_location : DB location                                                              //
+   // return : -                                                                                                               //
+  ///////////////////////////////////////////////////////////////////////
+       void Gimmick::addDB(const std::string &i_name, 
+                           const std::string &i_location)
+       {
+               if(mTreeHandlerMap.find(i_name) == mTreeHandlerMap.end())
+               {
+                       mTreeHandlerMap[i_name] = new SQLiteTreeHandler(i_location);
+                       mTreeHandlerMap[i_name]->Open(true);
+                       mSettings->addDB(i_location);
+               }
+       }
+
+  ///////////////////////////////////////////////////////////////////////////
+  // create a DB from a attributes descriptor file for medical images      //
+  // @param i_name : DB name                                                                                      //
+  // @param i_locDesc : location of descriptor file                                               //
+  // @param i_locDB : location of DB                                                                      //
+  // return : the SQLiteTreeHandler object on DB                                                  //
+       /////////////////////////////////////////////////////////////////////////
+       SQLiteTreeHandler *Gimmick::createDB(const std::string &i_name,
+                                            const std::string &i_locDesc,
+                                            const std::string &i_locDB)
+  {
+      SQLiteTreeHandler *sqlTreeH = new SQLiteTreeHandler(i_locDB);
+    // Create or open local database
+    if (! boost::filesystem::exists(i_locDB) )
+     {
+         std::string mess = "Local database '";
+         mess += i_locDB;
+         mess += "' does not exist : creating it";
+         GimmickMessage(1,mess<<std::endl);
+         
+                // CREATING DB STRUCTURE
+         sqlTreeH->GetTree().GetDescriptor().createDescriptorfromFile(i_locDesc);
+         if ( ! sqlTreeH->Create(true) )
+                {
+                       GimmickError("ERROR CREATING '"<<i_locDB<<"'");
+         }
+         sqlTreeH->SetAttribute(0,"Name",i_name);
+        }
+        else 
+        {
+               /// Open and test it
+               GimmickMessage(1,"Opening local database '" <<i_locDB<< "' " << std::endl);
+        if ( !sqlTreeH->Open(true) )
+               {
+                       GimmickError("ERROR OPENING '"<<i_locDB<<"'");
+               }
       }
+        return sqlTreeH;
   }
-  //================================================================
 
 
   //==============================================================
   void Gimmick::Finalize()
   {
-    delete mLocalDatabase;
+        
+         // delete SQLiteTreeHandler Object
+          for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
+           it!= mTreeHandlerMap.end(); ++it)
+          {
+                  delete it->second;
+          }
   }
   //==============================================================
 
@@ -118,19 +185,22 @@ namespace creaImageIO
   }
   //================================================================
 
+
   //================================================================
   const std::string& Gimmick::GetLocalDatabasePath()
   {
     if (mLocalDatabasePath.size()==0) 
       {
        mLocalDatabasePath = GetUserSettingsDirectory();
-       mLocalDatabasePath += "local_database.sqlite3";
+       mLocalDatabasePath += "Shared/gimmick/local_database.sqlite3";
        boost::algorithm::replace_all( mLocalDatabasePath,
                                       INVALID_FILE_SEPARATOR , 
                                       VALID_FILE_SEPARATOR);
       }
     return mLocalDatabasePath;    
   }
+
+
   //========================================================================
 
   //========================================================================
@@ -146,6 +216,36 @@ namespace creaImageIO
            GimmickError("ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
          }
       }
+
+       std::string setDir=GetUserSettingsDirectory();
+       boost::algorithm::replace_all( setDir,
+                                      INVALID_FILE_SEPARATOR , 
+                                      VALID_FILE_SEPARATOR);
+       setDir+="Shared/";
+       boost::filesystem::create_directory( setDir );
+       setDir+="gimmick/";
+       boost::filesystem::create_directory( setDir );
+       setDir+="localdatabase_Descriptor.txt";
+
+       if(!boost::filesystem::is_regular(setDir))
+       {
+               char name[PATH_MAX];
+//EED          int err = GetBinaryDirectory(name, PATH_MAX);
+               crea::System::GetAppPath(name,PATH_MAX);
+               std::cout<<name<<std::endl;
+               
+               std::string path=name;
+               path=path.substr(0,path.size()-1);
+               path=path.substr(0,path.find_last_of("/"));
+               //Creating directories
+               path+="/bin/Shared/gimmick/localdatabase_Descriptor.txt";
+               std::cout<<"From: "<<path<<std::endl;
+               std::cout<<"To: "<<setDir<<std::endl;
+               boost::algorithm::replace_all( path,
+                                               INVALID_FILE_SEPARATOR , 
+                                               VALID_FILE_SEPARATOR);
+               boost::filesystem::copy_file(path,setDir);
+       }
   }
   //========================================================================
 
@@ -154,7 +254,7 @@ namespace creaImageIO
   /// Sets message level
   void Gimmick::SetMessageLevel(int l)
   {
-    crea::MessageManager::SetMessageLevel("Gimmick!",l);
+    SetGimmickMessageLevel(l);
   }
   //========================================================================
 
@@ -162,44 +262,182 @@ 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)) 
+  /// Returns the tree handler with the given name
+  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.SetCurrentDatabase(d);
+    mImageAdder.SetTreeHandler(GetTreeHandler(d));
+       mImageAdder.SetSynchronizer(mSynchronizer);
+       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);
+
+       TreeHandler * handler=GetTreeHandler(d);
+       mImageAdder.SetCurrentDatabase(d);
+    mImageAdder.SetTreeHandler(handler);
+       mImageAdder.SetSynchronizer(mSynchronizer);
+    mImageAdder.AddDirectory(f,recurse);  
+  }
+
+  //========================================================================
+
+  //========================================================================
+  void Gimmick::RemoveFile(const std::string& d, 
+                          tree::Node* node)
+  {
+         mImageAdder.SetCurrentDatabase(d);
+         mImageAdder.SetSynchronizer(mSynchronizer);
+         mImageAdder.RemoveFile(node);
+  }
+  //========================================================================
+
+  //========================================================================
+  void Gimmick::CopyFiles(const std::vector<std::string>& filenames, const std::string& d )
+  {
+         TreeHandler * handler=GetTreeHandler(d);
+         mImageAdder.SetCurrentDatabase(d);
+         mImageAdder.SetTreeHandler(handler);
+         mImageAdder.SetSynchronizer(mSynchronizer);
+         mImageAdder.CopyFiles(filenames, mSettings->getValue(SETTINGS_COPY_PATH));
   }
+
   //========================================================================
+  std::string Gimmick::Synchronize(const std::string& d, bool repair, bool checkAttributes)
+  {
+         TreeHandler * handler=GetTreeHandler(d);
+         mImageAdder.SetCurrentDatabase(d);
+         mImageAdder.SetTreeHandler(handler);
+         mImageAdder.SetSynchronizer(mSynchronizer);
+         return mImageAdder.Synchronize(repair, checkAttributes);
+  }
 
   //========================================================================
   /// 
-  void Gimmick::PrintLocalDatabase()
+  void Gimmick::Print(const std::string& d)
+  {
+    GetTreeHandler(d)->GetTree().Print();
+  }
+  //========================================================================
+
+  void Gimmick::GetSetting(const std::string& name, std::string& value)
+  {
+         value = mSettings->getValue(name);
+  }
+  //========================================================================
+
+  //========================================================================
+
+  void Gimmick::GetAttributes(const std::string& d, 
+         const std::string& filename, 
+         const std::vector<std::string>& params, 
+         std::vector<std::string>& results)
+  {
+         TreeHandler * handler=GetTreeHandler(d);
+         mImageAdder.SetCurrentDatabase(d);
+         mImageAdder.SetTreeHandler(handler);
+         mImageAdder.SetSynchronizer(mSynchronizer);
+         mImageAdder.GetAttributes(params, filename, results);
+  }
+  //========================================================================
+
+  //========================================================================
+
+  void Gimmick::UpdateSetting(const std::string& name, const std::string& value)
+  {
+         mSettings->updateSetting(name,value);
+         mSettings->writeSettingsFile();
+  }
+  //========================================================================
+
+  void Gimmick::DeleteDrive(const std::string& drive)
+  {
+         for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
+           it!= mTreeHandlerMap.end(); ++it)
+          {
+                  mImageAdder.SetTreeHandler(it->second);
+                  mImageAdder.DeleteDriveFromMainDB(drive);
+          }
+         mImageAdder.SetSynchronizer(mSynchronizer);
+         mImageAdder.DeleteDriveFromOtherDB(drive);
+  }
+
+  //========================================================================
+  void Gimmick::EditField(tree::Node* node, const std::string& d, const std::string& name, const std::string& key, const std::string& val)
   {
-    mLocalDatabase->GetTree().Print();
+         TreeHandler * handler=GetTreeHandler(d);
+         mImageAdder.SetCurrentDatabase(d);
+         mImageAdder.SetTreeHandler(handler);
+         mImageAdder.EditField(node,name,key,val);
   }
   //========================================================================
 
 
+  /////////////////////////////////////////////////////////////////////////
+  // add DB from Settings file                                                                              //
+  // @param : -                                                                                                                         //
+  // return : -                                                                                                                         //
+  /////////////////////////////////////////////////////////////////////////
+  void Gimmick::addDBSettings()
+  {
+
+        std::string pathSettings = mSettings->getValue(SETTINGS_DBPATH);
+        
+        // split to find all paths
+        std::vector<std::string> paths;
+        std::string separator = ";";
+        std::string::size_type last_pos = pathSettings.find_first_not_of(separator);
+        //find first separator
+        std::string::size_type pos = pathSettings.find_first_of(separator, last_pos);
+        while(std::string::npos != pos || std::string::npos != last_pos)
+        {
+               paths.push_back(pathSettings.substr(last_pos, pos - last_pos));
+               last_pos = pathSettings.find_first_not_of(separator, pos);
+               pos = pathSettings.find_first_of(separator, last_pos);
+        }
+
+        std::vector<std::string>::iterator it_path = paths.begin();
+        for(; it_path != paths.end(); ++it_path)
+        {
+                pos = it_path->find_last_of("\\");
+                last_pos = it_path->find_last_of(".");
+                std::string name = it_path->substr(pos +1, last_pos -pos-1 );
+                addDB(name, it_path->c_str());
+        }
+
+  }
+       
 }