2 #include <creaImageIOGimmick.h>
4 #include <creaImageIOSystem.h>
5 #include <boost/filesystem.hpp>
6 #include <boost/algorithm/string.hpp>
8 #ifndef PATH_MAX // If not defined yet : do it
18 //==============================================================
22 RegisterGimmickMessageTypes();
26 //==============================================================
29 //==============================================================
34 mSettings->writeSettingsFile();
42 //==============================================================
44 //==============================================================
45 void Gimmick::Initialize(const std::string& path)
50 //==============================================================
51 void Gimmick::Initialize()
53 std::string i_nameDB = "Local database";
54 // Create the UserSettings dir if does not exist
55 CreateUserSettingsDirectory();
56 // Sets the current directory to the home dir
57 mCurrentDirectory = GetHomeDirectory();
58 mSynchronizer= new Synchronizer(GetUserSettingsDirectory()+"Shared/gimmick/");
60 mSettings = new Settings(mCurrentDirectory);
62 std::string dbpath = GetLocalDatabasePath();
63 // Create or open local database
64 std::string dpath= mCurrentDirectory + "/.gimmick/Shared/gimmick/localdatabase_Descriptor.dscp";
65 boost::algorithm::replace_all( dpath,
66 INVALID_FILE_SEPARATOR ,
67 VALID_FILE_SEPARATOR);
68 mLocalDatabase = createDB(i_nameDB, dpath, dbpath);
69 // Add it to the TreeHandlerMap
70 mTreeHandlerMap[i_nameDB] = mLocalDatabase;
72 //Add additional DB from user Settings
76 ///////////////////////////////////////////////////////////////////////
77 // add DB to TreeHandler Map //
78 // @param i_name : DB name //
79 // @param i_location : DB location //
81 ////////////////////////////////////////////////////////////////////////
82 void Gimmick::addDB(const std::string &i_name,
83 const std::string &i_location)
85 if(mTreeHandlerMap.find(i_name) == mTreeHandlerMap.end())
87 mTreeHandlerMap[i_name] = new SQLiteTreeHandler(i_location);
88 mTreeHandlerMap[i_name]->Open(true);
89 mSettings->addDB(i_location);
93 ///////////////////////////////////////////////////////////////////////////
94 // create a DB from a attributes descriptor file for medical images //
95 // @param i_name : DB name //
96 // @param i_locDesc : location of descriptor file //
97 // @param i_locDB : location of DB //
98 // return : the SQLiteTreeHandler object on DB //
99 /////////////////////////////////////////////////////////////////////////
100 SQLiteTreeHandler* Gimmick::createDB(const std::string &i_name,
101 const std::string &i_locDesc,
102 const std::string &i_locDB)
104 SQLiteTreeHandler* sqlTreeH( new SQLiteTreeHandler(i_locDB) );
105 // Create or open local database
106 if (! boost::filesystem::exists(i_locDB) )
108 std::string mess = "Local database '";
110 mess += "' does not exist : creating it";
111 GimmickMessage(1,mess<<std::endl);
113 // CREATING DB STRUCTURE
114 sqlTreeH->GetTree().GetDescriptor().createDescriptorfromFile(i_locDesc);
115 if ( ! sqlTreeH->Create(true) )
117 GimmickError("ERROR CREATING '"<<i_locDB<<"'");
119 sqlTreeH->SetAttribute(0,"Name",i_name);
124 GimmickMessage(1,"Opening local database '" <<i_locDB<< "' " << std::endl);
125 if ( !sqlTreeH->Open(true) )
127 GimmickError("ERROR OPENING '"<<i_locDB<<"'");
133 //==============================================================
134 void Gimmick::Finalize()
136 if(mTreeHandlerMap.size() >0)
138 // delete SQLiteTreeHandler Object
139 for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
140 it!= mTreeHandlerMap.end();
147 //==============================================================
149 //================================================================
152 #define VALID_FILE_SEPARATOR "\\"
153 #define INVALID_FILE_SEPARATOR "/"
155 #define INVALID_FILE_SEPARATOR "\\"
156 #define VALID_FILE_SEPARATOR "/"
158 //================================================================
160 //================================================================
161 const std::string& Gimmick::GetHomeDirectory()
163 if (mHomeDirectory.size()==0)
165 #if defined(__GNUC__)
166 mHomeDirectory = getenv("HOME");
167 #elif defined(_WIN32)
168 mHomeDirectory = getenv("USERPROFILE");
171 return mHomeDirectory;
173 //================================================================
174 const std::string& Gimmick::GetUserSettingsDirectory()
176 if (mUserSettingsDirectory.size()==0)
178 mUserSettingsDirectory = GetHomeDirectory();
179 mUserSettingsDirectory += "/.gimmick/";
180 boost::algorithm::replace_all( mUserSettingsDirectory,
181 INVALID_FILE_SEPARATOR ,
182 VALID_FILE_SEPARATOR);
184 return mUserSettingsDirectory;
186 //================================================================
189 //================================================================
190 const std::string& Gimmick::GetLocalDatabasePath()
192 if (mLocalDatabasePath.size()==0)
194 mLocalDatabasePath = GetUserSettingsDirectory();
195 mLocalDatabasePath += "Shared/gimmick/local_database.sqlite3";
196 boost::algorithm::replace_all( mLocalDatabasePath,
197 INVALID_FILE_SEPARATOR ,
198 VALID_FILE_SEPARATOR);
200 return mLocalDatabasePath;
203 //========================================================================
205 //========================================================================
206 void Gimmick::CreateUserSettingsDirectory()
208 //JCP settings dir 02/10/2009
209 const std::string settingsdirectory = GetUserSettingsDirectory();
210 bool isdir = boost::filesystem::is_directory( settingsdirectory );
213 GimmickMessage(1,"Directory '"<<GetUserSettingsDirectory()<<"' "
214 << "does not exist : creating it"<<std::endl);
216 if ( ! boost::filesystem::create_directory( GetUserSettingsDirectory() ) )
218 GimmickError("ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
222 std::string setDir=GetUserSettingsDirectory();
223 boost::algorithm::replace_all( setDir,
224 INVALID_FILE_SEPARATOR ,
225 VALID_FILE_SEPARATOR);
227 boost::filesystem::create_directory( setDir );
229 boost::filesystem::create_directory( setDir );
230 setDir+="localdatabase_Descriptor.dscp";
232 if(!boost::filesystem::is_regular(setDir))
235 //EED int err = GetBinaryDirectory(name, PATH_MAX);
236 crea::System::GetAppPath(name,PATH_MAX);
237 std::cout<<name<<std::endl;
239 std::string path=name;
240 path=path.substr(0,path.size()-1);
241 path=path.substr(0,path.find_last_of("/"));
242 //Creating directories
244 // The following stuff works on Linux, NOT CHECKED on Windows // JPR
247 path+="/bin/Shared/gimmick/localdatabase_Descriptor.dscp";
249 path+="/Shared/gimmick/localdatabase_Descriptor.dscp";
251 std::cout <<"From: " << path <<std::endl;
252 std::cout <<"To: " << setDir <<std::endl;
253 boost::algorithm::replace_all( path,
254 INVALID_FILE_SEPARATOR ,
255 VALID_FILE_SEPARATOR);
256 boost::filesystem::copy_file(path,setDir);
259 //========================================================================
262 //========================================================================
263 /// Sets message level
264 void Gimmick::SetMessageLevel(int l)
266 SetGimmickMessageLevel(l);
268 //========================================================================
270 //========================================================================
271 /// Sets message level
272 void Gimmick::SetDebugMessageLevel(int l)
274 SetGimmickDebugMessageLevel(l);
276 //========================================================================
278 //========================================================================
279 /// Returns the tree handler with the given name
280 TreeHandler* Gimmick::GetTreeHandler(const std::string& name) const
282 TreeHandlerMapType::const_iterator i;
283 i = GetTreeHandlerMap().find(name);
284 if ( i == GetTreeHandlerMap().end() )
286 GimmickError("TreeHandler '"<<name<<"' does not exist");
291 //========================================================================
292 /// Add the files to the tree handler
293 void Gimmick::AddFiles(const std::string& d,
294 const std::vector<std::string>& filenames)
296 GimmickMessage(2,"Adding files to '"<<d<<"'"<<std::endl);
298 mImageAdder.SetCurrentDatabase(d);
299 mImageAdder.SetTreeHandler(GetTreeHandler(d));
300 mImageAdder.SetSynchronizer(mSynchronizer);
301 mImageAdder.AddFiles(filenames);
303 //========================================================================
305 //========================================================================
306 /// Add a dir to the local database
307 void Gimmick::AddDir(const std::string& d, const std::string& f,
310 GimmickMessage(2,"Adding dir '"<<f<<"' to '"<<d<<"' recurse:"
311 <<recurse<<std::endl);
313 TreeHandler * handler=GetTreeHandler(d);
314 mImageAdder.SetCurrentDatabase(d);
315 mImageAdder.SetTreeHandler(handler);
316 mImageAdder.SetSynchronizer(mSynchronizer);
317 mImageAdder.AddDirectory(f,recurse);
320 //========================================================================
322 //========================================================================
323 void Gimmick::RemoveFile(const std::string& d,
326 mImageAdder.SetCurrentDatabase(d);
327 mImageAdder.SetSynchronizer(mSynchronizer);
328 mImageAdder.RemoveFile(node);
330 //========================================================================
332 //========================================================================
334 void Gimmick::CopyFiles(const std::vector<std::string>& filenames, const std::string& d )
336 TreeHandler * handler=GetTreeHandler(d);
337 mImageAdder.SetCurrentDatabase(d);
338 mImageAdder.SetTreeHandler(handler);
339 mImageAdder.SetSynchronizer(mSynchronizer);
340 mImageAdder.CopyFiles(filenames, mSettings->getValue(SETTINGS_COPY_PATH));
343 //========================================================================
345 std::string Gimmick::Synchronize(const std::string& d, bool repair, bool checkAttributes)
347 TreeHandler * handler=GetTreeHandler(d);
348 mImageAdder.SetCurrentDatabase(d);
349 mImageAdder.SetTreeHandler(handler);
350 mImageAdder.SetSynchronizer(mSynchronizer);
351 return mImageAdder.Synchronize(repair, checkAttributes);
354 //========================================================================
356 void Gimmick::Print(const std::string& d)
358 GetTreeHandler(d)->GetTree().Print();
360 //========================================================================
362 void Gimmick::GetSetting(const std::string& name, std::string& value)
364 value = mSettings->getValue(name);
366 //========================================================================
368 //========================================================================
370 void Gimmick::GetAttributes(const std::string& d,
371 const std::string& filename,
372 const std::vector<std::string>& params,
373 std::vector<std::string>& results)
375 TreeHandler * handler=GetTreeHandler(d);
376 mImageAdder.SetCurrentDatabase(d);
377 mImageAdder.SetTreeHandler(handler);
378 mImageAdder.SetSynchronizer(mSynchronizer);
379 mImageAdder.GetAttributes(params, filename, results);
381 //========================================================================
383 //========================================================================
385 void Gimmick::UpdateSetting(const std::string& name, const std::string& value)
387 mSettings->updateSetting(name,value);
388 mSettings->writeSettingsFile();
390 //========================================================================
392 void Gimmick::DeleteDrive(const std::string& drive)
394 for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
395 it!= mTreeHandlerMap.end();
398 mImageAdder.SetTreeHandler(it->second);
399 mImageAdder.DeleteDriveFromMainDB(drive);
401 mImageAdder.SetSynchronizer(mSynchronizer);
402 mImageAdder.DeleteDriveFromOtherDB(drive);
405 //========================================================================
406 void Gimmick::EditField(tree::Node* node, const std::string& d, const std::string& name, const std::string& key, const std::string& val)
408 TreeHandler * handler=GetTreeHandler(d);
409 mImageAdder.SetCurrentDatabase(d);
410 mImageAdder.SetTreeHandler(handler);
411 mImageAdder.EditField(node,name,key,val);
413 //========================================================================
415 ////////////////////////////////////////////////////////////////////////
416 // add DB from Settings file //
419 ////////////////////////////////////////////////////////////////////////
420 void Gimmick::addDBSettings()
423 std::string pathSettings = mSettings->getValue(SETTINGS_DBPATH);
425 // split to find all paths
426 std::vector<std::string> paths;
427 std::string separator = ";";
428 std::string::size_type last_pos = pathSettings.find_first_not_of(separator);
429 //find first separator
430 std::string::size_type pos = pathSettings.find_first_of(separator, last_pos);
431 while(std::string::npos != pos || std::string::npos != last_pos)
433 paths.push_back(pathSettings.substr(last_pos, pos - last_pos));
434 last_pos = pathSettings.find_first_not_of(separator, pos);
435 pos = pathSettings.find_first_of(separator, last_pos);
438 std::vector<std::string>::iterator it_path = paths.begin();
439 for(; it_path != paths.end(); ++it_path)
441 pos = it_path->find_last_of("\\");
442 last_pos = it_path->find_last_of(".");
443 std::string name = it_path->substr(pos +1, last_pos -pos-1 );
444 addDB(name, it_path->c_str());