#include <creaImageIOGimmick.h>
-#include <creaMessageManager.h>
+#include <creaImageIOSystem.h>
#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>
namespace creaImageIO
{
-
+
+
//==============================================================
Gimmick::Gimmick()
+ : mImageAdder(0)
{
- crea::MessageManager::RegisterMessageType("Gimmick!",
- "Gimmick",1);
+ RegisterGimmickMessageTypes();
}
//==============================================================
//==============================================================
- bool Gimmick::Initialize()
+ void Gimmick::Initialize()
{
+ std::string i_nameDB = "Local database";
// 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());
- // 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[i_nameDB] = mLocalDatabase;
+
+ // Creates files and directories database
+ mTimestampDatabase = new TimestampDatabaseHandler(GetTimestampDatabasePath());
// Create or open local database
- if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
+ if (! boost::filesystem::exists( GetTimestampDatabasePath() ) )
{
- creaMessage("Gimmick!",1,
- "[Gimmick!] Local database '"<<GetLocalDatabasePath()<<"' "
- << "does not exist : creating it"<<std::endl);
-
- // CREATING DEFAULT DB STRUCTURE
- mLocalDatabase->GetTree().GetDescriptor().CreateDefault();
+ std::string mess = "Timestamp database '";
+ mess += GetTimestampDatabasePath();
+ mess += "' does not exist : creating it";
+ GimmickMessage(1,mess<<std::endl);
- if ( ! mLocalDatabase->Create(true) )
+ if ( ! mTimestampDatabase->Create() )
{
- creaMessage("Gimmick!",1,
- "[Gimmick!] !! ERROR CREATING '"<<GetLocalDatabasePath()<<"'");
- return false;
+ GimmickError("ERROR CREATING '"<<GetTimestampDatabasePath()<<"'");
}
- }
+
+ }
else
{
/// Open and test it
- creaMessage("Gimmick!",1,
- "[Gimmick!] Opening local database '"
- <<GetLocalDatabasePath()<<"' "
- <<std::endl);
- if ( ! mLocalDatabase->Open(true) )
+ GimmickMessage(1,"Opening Timestamp database '"
+ <<GetTimestampDatabasePath()<<"' "
+ <<std::endl);
+ if ( ! mTimestampDatabase->Open() )
{
- creaMessage("Gimmick!",1,
- "[Gimmick!] !! ERROR OPENING '"<<GetLocalDatabasePath()<<"'");
- return false;
+ GimmickError("ERROR OPENING '"<<GetTimestampDatabasePath()<<"'");
}
}
- return true;
}
- //================================================================
+
+ ///////////////////////////////////////////////////////////////////////
+ // add DB to TreeHandler Map //
+ // @param i_name : DB name //
+ // @param i_location : DB location //
+ // return : - //
+ ///////////////////////////////////////////////////////////////////////
+ void Gimmick::addDB(std::string &i_name, std::string &i_location)
+ {
+ if(mTreeHandlerMap.find(i_name) == mTreeHandlerMap.end())
+ {
+ mTreeHandlerMap[i_name] = new SQLiteTreeHandler(i_location);
+ mTreeHandlerMap[i_name]->Open(true);
+ }
+ }
+
+
+ ///////////////////////////////////////////////////////////////////////////
+ // 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(std::string &i_name, std::string &i_locDesc, 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;
+ }
//==============================================================
- bool Gimmick::Finalize()
+ void Gimmick::Finalize()
{
- delete mLocalDatabase;
+
+ // delete SQLiteTreeHandler Object
+ for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
+ it!= mTreeHandlerMap.end(); ++it)
+ {
+ delete it->second;
+ }
+ delete mTimestampDatabase;
}
//==============================================================
}
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;
+ }
//========================================================================
//========================================================================
- bool Gimmick::CreateUserSettingsDirectory()
+ void Gimmick::CreateUserSettingsDirectory()
{
if (! boost::filesystem::is_directory( GetUserSettingsDirectory() ) )
{
- 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;
}
//========================================================================
//========================================================================
- /// Sets debug mode
- void Gimmick::SetDebugMode(bool m)
+ /// Sets message level
+ void Gimmick::SetMessageLevel(int l)
{
- if (m)
- {
- crea::MessageManager::SetMessageLevel("Gimmick!",9);
- }
- else
+ SetGimmickMessageLevel(l);
+ }
+ //========================================================================
+
+ //========================================================================
+ /// Sets message level
+ void Gimmick::SetDebugMessageLevel(int l)
+ {
+ 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() )
{
- crea::MessageManager::SetMessageLevel("Gimmick!",1);
+ GimmickError("TreeHandler '"<<name<<"' does not exist");
}
+ return i->second;
+ }
+
+ //========================================================================
+ ///Returns the timestamp database handler
+ TimestampDatabaseHandler* Gimmick::GetTimestampDatabase() const
+ {
+ return mTimestampDatabase;
+ }
+
+
+ //========================================================================
+ /// 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.SetTimestampHandler(mTimestampDatabase);
+ mImageAdder.SetSynchronizer(mSynchronizer);
+ mImageAdder.AddFiles(filenames);
+
+ }
+ //========================================================================
+
+ //========================================================================
+ /// Add a dir to the local database
+ void Gimmick::AddDir(const std::string& d, const std::string& f,
+ bool recurse)
+ {
+ GimmickMessage(2,"Adding dir '"<<f<<"' to '"<<d<<"' recurse:"
+ <<recurse<<std::endl);
+
+ TreeHandler * handler=GetTreeHandler(d);
+ mImageAdder.SetTreeHandler(handler);
+ mImageAdder.SetTimestampHandler(mTimestampDatabase);
+ mImageAdder.SetSynchronizer(mSynchronizer);
+ mImageAdder.AddDirectory(f,recurse);
}
+
+ //========================================================================
+
+ //========================================================================
+ void Gimmick::RemoveFile(const std::string& d,
+ const tree::Node* & node)
+ {
+ mImageAdder.SetSynchronizer(mSynchronizer);
+ mTimestampDatabase->RemoveNode("PATH",node);
+ mImageAdder.RemoveFile(node);
+ }
+ //========================================================================
+
//========================================================================
+
+ std::string Gimmick::Synchronize(bool repair, bool checkAttributes)
+ {
+ TreeHandler * handler=GetTreeHandler("Local database");
+ mImageAdder.SetTreeHandler(handler);
+ mImageAdder.SetTimestampHandler(mTimestampDatabase);
+ mImageAdder.SetSynchronizer(mSynchronizer);
+ return mImageAdder.Synchronize(repair, checkAttributes);
+ }
+
+ //========================================================================
+ ///
+ void Gimmick::Print(const std::string& d)
+ {
+ GetTreeHandler(d)->GetTree().Print();
+ }
+ //========================================================================
+
}
git://git.creatis.insa-lyon.fr / creaImageIO.git / blobdiff