X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOGimmick.cpp;h=615f1f041f90b50a41886121d516a11c8df72dea;hb=14cf4a18b7517a3f7e86ee3e756043a3431b87db;hp=a9f72644144b8171a78cf9fbea4e736e3246bd48;hpb=f08880c40f6304be640c7f8702bbbe96f8d50b40;p=creaImageIO.git

diff --git a/src2/creaImageIOGimmick.cpp b/src2/creaImageIOGimmick.cpp
index a9f7264..615f1f0 100644
--- a/src2/creaImageIOGimmick.cpp
+++ b/src2/creaImageIOGimmick.cpp
@@ -1,85 +1,155 @@
-#include <creaImageIOGimmick.h>
 
-#include <creaMessageManager.h>
-#include <creaImageIOImageFinder.h>
+#include <creaImageIOGimmick.h>
 
+#include <creaImageIOSystem.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
+#include <creaImageIOGimmick.h>
+#ifdef _DEBUG
+#define new DEBUG_NEW
+#endif
+
 
 namespace creaImageIO
 {
-  
   //==============================================================
   Gimmick::Gimmick()
+    : mImageAdder(0)
   {    
-    crea::MessageManager::RegisterMessageType("Gimmick!",
-					      "Gimmick",1);
+  RegisterGimmickMessageTypes();
+	mSettings=0;
+	mSynchronizer=0;
+	mLocalDescpName = "localdatabase_Descriptor.dscp";
+	mLocalDBName = "Local database";
   }
   //==============================================================
 
 
-
   //==============================================================
   Gimmick::~Gimmick()
   {
-    
+
+	 if(mSettings!=0)
+	  {
+	  	mSettings->writeSettingsFile();
+		delete mSettings;
+	  }
+	if(mSynchronizer!=0)
+	  {
+	  	delete mSynchronizer;
+	  }
   }
   //==============================================================
   
+  //==============================================================
+  void Gimmick::Initialize(const std::string i_namedescp, const std::string i_namedb)
+  {
+	  mLocalDescpName = i_namedescp;
+	  mLocalDBName = i_namedb;
+	  Initialize();
+  }
 
   //==============================================================
-  bool Gimmick::Initialize()
+  void Gimmick::Initialize()
   {
+	std::string i_nameDB = mLocalDBName;
     // Create the UserSettings dir if does not exist
-    if (!CreateUserSettingsDirectory()) return false;
+    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());
+    mSettings  = new Settings(mCurrentDirectory);
+	
+    std::string dbpath = GetLocalDatabasePath();
     // Create or open local database
-    if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
-      {
-	creaMessage("Gimmick!",1,
-		    "[Gimmick!] Local database '"<<GetLocalDatabasePath()<<"' "
-		    << "does not exist : creating it"<<std::endl);
+    std::string dpath= mCurrentDirectory + "/.gimmick/Shared/gimmick/" + mLocalDescpName;
+    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();	
+  }
 
-	// CREATING DEFAULT DB STRUCTURE
-	mLocalDatabase->GetTree().GetDescriptor().CreateDefault();
-	
-	if ( ! mLocalDatabase->Create(true) )
-	  {
-	    creaMessage("Gimmick!",1,
-			"[Gimmick!] !! ERROR CREATING '"<<GetLocalDatabasePath()<<"'");
-	    return false;
-	  }
-	mLocalDatabase->SetAttribute(0,"Name","Local database");
-      }
-    else 
-      {
-	/// Open and test it
-	creaMessage("Gimmick!",1,
-		    "[Gimmick!] Opening local database '"
-		    <<GetLocalDatabasePath()<<"' "
-		    <<std::endl);
-	if ( ! mLocalDatabase->Open(true) )
-	  {
-	    creaMessage("Gimmick!",1,
-			"[Gimmick!] !! ERROR OPENING '"<<GetLocalDatabasePath()<<"'");
-	    return false;
-	  }
-	
-      }
-    return true;
+   ///////////////////////////////////////////////////////////////////////
+   // 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
+		
+		GimmickDebugMessage(1,"Opening local database '" <<i_locDB<< "' " << std::endl);
+        	if ( !sqlTreeH->Open(true) )
+		{
+			GimmickError("ERROR OPENING '"<<i_locDB<<"'");
+		}
+	}
+	return sqlTreeH;
   }
-  //================================================================
-
 
   //==============================================================
-  bool Gimmick::Finalize()
+  void Gimmick::Finalize()
   {
-    delete mLocalDatabase;
+	  if(mTreeHandlerMap.size() >0)
+	  {
+		// delete SQLiteTreeHandler Object
+		for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
+													it!= mTreeHandlerMap.end(); 
+							++it)
+		{
+			delete it->second;
+		}
+	  }
   }
   //==============================================================
 
@@ -122,38 +192,101 @@ namespace creaImageIO
   }
   //================================================================
 
+
   //================================================================
   const std::string& Gimmick::GetLocalDatabasePath()
   {
     if (mLocalDatabasePath.size()==0) 
       {
 	mLocalDatabasePath = GetUserSettingsDirectory();
-	mLocalDatabasePath += "local_database.sqlite3";
+	mLocalDatabasePath += "Shared/gimmick/";
+	mLocalDatabasePath += mLocalDBName;
+	mLocalDatabasePath +=".sqlite3";
 	boost::algorithm::replace_all( mLocalDatabasePath,
 				       INVALID_FILE_SEPARATOR , 
 				       VALID_FILE_SEPARATOR);
       }
     return mLocalDatabasePath;    
   }
+
   //========================================================================
 
   //========================================================================
-  bool Gimmick::CreateUserSettingsDirectory()
+  void Gimmick::CreateUserSettingsDirectory()
   {
-    if (! boost::filesystem::is_directory( GetUserSettingsDirectory() ) )
+	 // std::string st("C:/Documents and Settings/cervenansky/.gimmick/");
+	 // 	boost::algorithm::replace_all( st, 
+		//		       INVALID_FILE_SEPARATOR , 
+		//		       VALID_FILE_SEPARATOR);
+		//const boost::filesystem::path mpath(st);
+//C:\Documents and Settings\cervenansky\.gimmick");
+	  //if ( !boost::filesystem::exists( path ) ) 		  return ;
+	 //  boost::filesystem::directory_iterator end_itr; // default construction yields past-the-end
+	 //  for ( boost::filesystem::directory_iterator itr( path );  itr != end_itr;  ++itr )
+	 // {
+		//// If is directory & recurse : do recurse
+		//	if ( boost::filesystem::is_directory(itr->status()) )
+		//	return;
+	 //  }
+
+	  //JCP settings dir 02/10/2009
+	  const std::string settingsdirectory = GetUserSettingsDirectory();
+	  	//boost::algorithm::replace_all( mUserSettingsDirectory, 
+				//       INVALID_FILE_SEPARATOR , 
+				//       VALID_FILE_SEPARATOR);
+;//("E:\frederic");
+		  //("C:\\Documents and Settings\\cervenansky\\.gimmick\\"); // settingsdirectory );
+		bool isdir = false;
+	   isdir = boost::filesystem::is_directory(settingsdirectory); // settingsdirectory );
+    if (! isdir )
       {
-	creaMessage("Gimmick!",1,
-		    "[Gimmick!] Directory '"<<GetUserSettingsDirectory()<<"' "
-		    << "does not exist : creating it"<<std::endl);
+	GimmickMessage(1,"Directory '"<<GetUserSettingsDirectory()<<"' "
+ 		       << "does not exist : creating it"<<std::endl);
 	
 	if ( ! boost::filesystem::create_directory( GetUserSettingsDirectory() ) )
 	  {
-	    creaMessage("Gimmick!",1,
-			"[Gimmick!] !! ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
-	    return false;
+	    GimmickError("ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
 	  }
       }
-    return true;
+
+	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+=mLocalDescpName;
+
+	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
+
+// The following stuff works on Linux, NOT CHECKED on Windows // JPR
+		
+#if defined(_WIN32)		
+		path+="/bin/Shared/gimmick/";
+		path+=mLocalDescpName;
+#else		
+		path+="/Shared/gimmick/";
+		path+= mLocalDescpName;
+#endif		
+		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);
+	}
   }
   //========================================================================
 
@@ -162,38 +295,185 @@ namespace creaImageIO
   /// Sets message level
   void Gimmick::SetMessageLevel(int l)
   {
-    crea::MessageManager::SetMessageLevel("Gimmick!",l);
+    SetGimmickMessageLevel(l);
   }
   //========================================================================
 
   //========================================================================
-  /// Add a file to the local database
-  void Gimmick::AddFileToLocalDatabase(const std::string& f)
+  /// Sets message level
+  void Gimmick::SetDebugMessageLevel(int l)
   {
-    ImageFinder finder(mLocalDatabase);
-    if (finder.IsHandledFile(f)) finder.AddFile(f);
-    
+    SetGimmickDebugMessageLevel(l);
+  }
+  //========================================================================
+
+  //========================================================================
+  /// 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() )
+      {
+	GimmickError("TreeHandler '"<<name<<"' does not exist");
+      }
+    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)
   {
-    mLocalDatabase->GetTree().Print();
+    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)
+  {
+	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());
+	 }
+  }	
 }