X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOGimmick.cpp;h=2ed5b0c72f115c9e4bbdb593b93e67cf57220c58;hb=f5399bbd662ec8584bfa8fe0c68985679d6c9c4c;hp=21d9cd4d01a966e8dd5d60a23f309bfcff8d2763;hpb=fcfe92562365cb738cf919d095d2f8d3608ec389;p=creaImageIO.git

diff --git a/src2/creaImageIOGimmick.cpp b/src2/creaImageIOGimmick.cpp
index 21d9cd4..2ed5b0c 100644
--- a/src2/creaImageIOGimmick.cpp
+++ b/src2/creaImageIOGimmick.cpp
@@ -1,8 +1,6 @@
 #include <creaImageIOGimmick.h>
 
 #include <creaImageIOSystem.h>
-#include <creaImageIOImageFinder.h>
-
 #include <boost/filesystem.hpp>
 #include <boost/algorithm/string.hpp>
 
@@ -12,8 +10,11 @@ namespace creaImageIO
 
   //==============================================================
   Gimmick::Gimmick()
+    : mImageAdder(0)
   {    
     RegisterGimmickMessageTypes();
+	mSettings=0;
+	mSynchronizer=0;
   }
   //==============================================================
 
@@ -22,7 +23,15 @@ namespace creaImageIO
   //==============================================================
   Gimmick::~Gimmick()
   {
-    
+	 if(mSettings!=0)
+	  {
+	  mSettings->writeSettingsFile();
+      delete mSettings;
+	  }
+	  if(mSynchronizer!=0)
+	  {
+	  delete mSynchronizer;
+	  }
   }
   //==============================================================
   
@@ -30,53 +39,126 @@ namespace creaImageIO
   //==============================================================
   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());
+
+	mSettings  = new Settings(mCurrentDirectory);
+	
 
-    // Create local database handler
-    mLocalDatabase = new SQLiteTreeHandler(GetLocalDatabasePath());
+	std::string dbpath = GetLocalDatabasePath();
+    // Create or open local database
+	mLocalDatabase = createDB(i_nameDB, mCurrentDirectory + "\\.gimmick\\localdatabase_Descriptor.txt", dbpath);
     // Add it to the TreeHandlerMap
-    mTreeHandlerMap["Local database"] = mLocalDatabase;
+    mTreeHandlerMap[i_nameDB] = mLocalDatabase;
+    
+	//Add additional DB from user Settings
+	addDBSettings();
 
+	// Creates files and directories database
+    mTimestampDatabase = new TimestampDatabaseHandler(GetTimestampDatabasePath());
     // Create or open local database
-    if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
+    if (! boost::filesystem::exists( GetTimestampDatabasePath() ) )
       {
-	std::string mess = "Local database '";
-	mess += GetLocalDatabasePath();
+	std::string mess = "Timestamp database '";
+	mess += GetTimestampDatabasePath();
 	mess += "' does not exist : creating it";
 	GimmickMessage(1,mess<<std::endl);
 	
-	// CREATING DEFAULT DB STRUCTURE
-	mLocalDatabase->GetTree().GetDescriptor().CreateDefault();
-	
-	if ( ! mLocalDatabase->Create(true) )
+	if ( ! mTimestampDatabase->Create() )
 	  {
-	    GimmickError("ERROR CREATING '"<<GetLocalDatabasePath()<<"'");
+	    GimmickError("ERROR CREATING '"<<GetTimestampDatabasePath()<<"'");
 	  }
-	mLocalDatabase->SetAttribute(0,"Name","Local database");
-      }
+	
+     }
     else 
       {
 	/// Open and test it
-	GimmickMessage(1,"Opening local database '"
-		       <<GetLocalDatabasePath()<<"' "
+	GimmickMessage(1,"Opening Timestamp database '"
+		       <<GetTimestampDatabasePath()<<"' "
 		       <<std::endl);
-	if ( ! mLocalDatabase->Open(true) )
+	if ( ! mTimestampDatabase->Open() )
 	  {
-	    GimmickError("ERROR OPENING '"<<GetLocalDatabasePath()<<"'");
+	    GimmickError("ERROR OPENING '"<<GetTimestampDatabasePath()<<"'");
 	  }
 	
       }
+
+  }
+
+   ///////////////////////////////////////////////////////////////////////
+   // 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;
+	   }
+	delete mTimestampDatabase;
   }
   //==============================================================
 
@@ -132,6 +214,22 @@ namespace creaImageIO
       }
     return mLocalDatabasePath;    
   }
+
+  //================================================================
+
+  //================================================================
+  const std::string& Gimmick::GetTimestampDatabasePath()
+  {
+    if (mTimestampDatabasePath.size()==0) 
+      {
+	mTimestampDatabasePath = GetUserSettingsDirectory();
+	mTimestampDatabasePath += "timestamp_database.sqlite3";
+	boost::algorithm::replace_all( mTimestampDatabasePath,
+				       INVALID_FILE_SEPARATOR , 
+				       VALID_FILE_SEPARATOR);
+      }
+    return mTimestampDatabasePath;    
+  }
   //========================================================================
 
   //========================================================================
@@ -168,7 +266,7 @@ namespace creaImageIO
   //========================================================================
 
   //========================================================================
-  /// Add a file to the local database
+  /// Returns the tree handler with the given name
   TreeHandler* Gimmick::GetTreeHandler(const std::string& name) const 
   {  
     TreeHandlerMapType::const_iterator i;
@@ -180,22 +278,27 @@ namespace creaImageIO
     return i->second;
   }
 
+  //========================================================================
+  ///Returns the timestamp database handler
+  TimestampDatabaseHandler* Gimmick::GetTimestampDatabase() const 
+  {  
+    return mTimestampDatabase;
+  }
+
 
   //========================================================================
-  /// Add a file to the local database
-  void Gimmick::AddFile(const std::string& d, const std::string& f)
+  /// Add the files to the tree handler
+  void Gimmick::AddFiles(const std::string& d, 
+			const std::vector<std::string>& filenames)
   {
-    GimmickMessage(2,"Adding file '"<<f<<"' to '"<<d<<"'"<<std::endl);
+    GimmickMessage(2,"Adding files to '"<<d<<"'"<<std::endl);
  
-    ImageFinder finder(GetTreeHandler(d));
-    if (finder.IsHandledFile(f)) 
-      {
-	finder.AddFile(f);
-      }
-    else
-      {
-	GimmickError("File '"<<f<<"' does not exist or is not handled");
-      }    
+	mImageAdder.SetCurrentDatabase(d);
+    mImageAdder.SetTreeHandler(GetTreeHandler(d));
+	mImageAdder.SetTimestampHandler(mTimestampDatabase);
+	mImageAdder.SetSynchronizer(mSynchronizer);
+    mImageAdder.AddFiles(filenames);
+	
   }
   //========================================================================
 
@@ -207,13 +310,51 @@ namespace creaImageIO
     GimmickMessage(2,"Adding dir '"<<f<<"' to '"<<d<<"' recurse:"
 		   <<recurse<<std::endl);
 
-    ImageFinder finder(GetTreeHandler(d));
-    
-    finder.AddDirectory(f,recurse);
-    
+	TreeHandler * handler=GetTreeHandler(d);
+	mImageAdder.SetCurrentDatabase(d);
+    mImageAdder.SetTreeHandler(handler);
+	mImageAdder.SetTimestampHandler(mTimestampDatabase);
+	mImageAdder.SetSynchronizer(mSynchronizer);
+    mImageAdder.AddDirectory(f,recurse);  
   }
+
   //========================================================================
 
+  //========================================================================
+  void Gimmick::RemoveFile(const std::string& d, 
+			   tree::Node* node)
+  {
+	  mImageAdder.SetCurrentDatabase(d);
+	  mImageAdder.SetSynchronizer(mSynchronizer);
+	  mTimestampDatabase->RemoveNode("PATH",node,d);
+	  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.SetTimestampHandler(mTimestampDatabase);
+	  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.SetTimestampHandler(mTimestampDatabase);
+	  mImageAdder.SetSynchronizer(mSynchronizer);
+	  return mImageAdder.Synchronize(repair, checkAttributes);
+  }
+
   //========================================================================
   /// 
   void Gimmick::Print(const std::string& d)
@@ -221,6 +362,38 @@ namespace creaImageIO
     GetTreeHandler(d)->GetTree().Print();
   }
   //========================================================================
+  /////////////////////////////////////////////////////////////////////////
+  // 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());
+	 }
+
+  }
+	
 }