2 #include <creaImageIOGimmick.h>
3 #include <creaImageIOSystem.h>
4 #include <boost/filesystem.hpp>
5 #include <boost/algorithm/string.hpp>
8 #ifndef PATH_MAX // If not defined yet : do it
11 #include <creaImageIOGimmick.h>
20 //==============================================================
24 RegisterGimmickMessageTypes();
27 mLocalDescpName = "localdatabase_Descriptor.dscp";
28 mLocalDBName = "Local database";
30 //==============================================================
33 //==============================================================
39 mSettings->writeSettingsFile();
47 //==============================================================
49 //==============================================================
50 void Gimmick::Initialize(const std::string i_namedescp, const std::string i_namedb)
52 mLocalDescpName = i_namedescp;
53 mLocalDBName = i_namedb;
57 //==============================================================
58 void Gimmick::Initialize()
60 std::string i_nameDB = mLocalDBName;
61 // Create the UserSettings dir if does not exist
62 CreateUserSettingsDirectory();
63 // Sets the current directory to the home dir
64 mCurrentDirectory = GetHomeDirectory();
65 mSynchronizer= new Synchronizer(GetUserSettingsDirectory()+"share/creaImageIO/");
67 mSettings = new Settings(mCurrentDirectory);
69 std::string dbpath = GetLocalDatabasePath();
71 // Create or open local database
72 std::string dpath= mCurrentDirectory + "/.creaImageIO/share/creaImageIO/" + mLocalDescpName;
74 boost::algorithm::replace_all( dpath,
75 INVALID_FILE_SEPARATOR ,
76 VALID_FILE_SEPARATOR);
77 mLocalDatabase = createDB(i_nameDB, dpath, dbpath);
78 // Add it to the TreeHandlerMap
79 mTreeHandlerMap[i_nameDB] = mLocalDatabase;
81 //Add additional DB from user Settings
85 ///////////////////////////////////////////////////////////////////////
86 // add DB to TreeHandler Map //
87 // @param i_name : DB name //
88 // @param i_location : DB location //
90 ////////////////////////////////////////////////////////////////////////
91 void Gimmick::addDB(const std::string &i_name,
92 const std::string &i_location)
94 if(mTreeHandlerMap.find(i_name) == mTreeHandlerMap.end())
96 mTreeHandlerMap[i_name] = new SQLiteTreeHandler(i_location);
97 mTreeHandlerMap[i_name]->Open(true);
98 mSettings->addDB(i_location);
102 ///////////////////////////////////////////////////////////////////////////
103 // create a DB from a attributes descriptor file for medical images //
104 // @param i_name : DB name //
105 // @param i_locDesc : location of descriptor file //
106 // @param i_locDB : location of DB //
107 // return : the SQLiteTreeHandler object on DB //
108 /////////////////////////////////////////////////////////////////////////
109 SQLiteTreeHandler* Gimmick::createDB(const std::string &i_name,
110 const std::string &i_locDesc,
111 const std::string &i_locDB)
113 SQLiteTreeHandler* sqlTreeH( new SQLiteTreeHandler(i_locDB) );
114 // Create or open local database
115 if (! boost::filesystem::exists(i_locDB) )
117 std::string mess = "Local database '";
119 mess += "' does not exist : creating it";
120 GimmickMessage(1,mess<<std::endl);
122 // CREATING DB STRUCTURE
123 sqlTreeH->GetTree().GetDescriptor().createDescriptorfromFile(i_locDesc);
124 if ( ! sqlTreeH->Create(true) )
126 GimmickError("ERROR CREATING '"<<i_locDB<<"'");
128 sqlTreeH->SetAttribute(0,"Name",i_name);
134 GimmickDebugMessage(1,"Opening local database '" <<i_locDB<< "' " << std::endl);
135 if ( !sqlTreeH->Open(true) )
137 GimmickError("ERROR OPENING '"<<i_locDB<<"'");
143 //==============================================================
144 void Gimmick::Finalize()
146 if(mTreeHandlerMap.size() >0)
148 // delete SQLiteTreeHandler Object
149 for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
150 it!= mTreeHandlerMap.end();
157 //==============================================================
159 //================================================================
162 #define VALID_FILE_SEPARATOR "\\"
163 #define INVALID_FILE_SEPARATOR "/"
165 #define INVALID_FILE_SEPARATOR "\\"
166 #define VALID_FILE_SEPARATOR "/"
168 //================================================================
170 //================================================================
171 const std::string& Gimmick::GetHomeDirectory()
173 if (mHomeDirectory.size()==0)
175 #if defined(__GNUC__)
176 mHomeDirectory = getenv("HOME");
177 #elif defined(_WIN32)
178 mHomeDirectory = getenv("USERPROFILE");
181 return mHomeDirectory;
183 //================================================================
184 const std::string& Gimmick::GetUserSettingsDirectory()
186 if (mUserSettingsDirectory.size()==0)
188 mUserSettingsDirectory = GetHomeDirectory();
189 mUserSettingsDirectory += "/.creaImageIO/";
190 boost::algorithm::replace_all( mUserSettingsDirectory,
191 INVALID_FILE_SEPARATOR ,
192 VALID_FILE_SEPARATOR);
194 return mUserSettingsDirectory;
196 //================================================================
199 //================================================================
200 const std::string& Gimmick::GetLocalDatabasePath()
202 if (mLocalDatabasePath.size()==0)
204 mLocalDatabasePath = GetUserSettingsDirectory();
205 mLocalDatabasePath += "share/creaImageIO/";
206 mLocalDatabasePath += mLocalDBName;
207 mLocalDatabasePath +=".sqlite3";
208 boost::algorithm::replace_all( mLocalDatabasePath,
209 INVALID_FILE_SEPARATOR ,
210 VALID_FILE_SEPARATOR);
212 return mLocalDatabasePath;
215 //========================================================================
217 //========================================================================
218 void Gimmick::CreateUserSettingsDirectory()
221 // std::string st("C:/Documents and Settings/cervenansky/.gimmick/");
222 // boost::algorithm::replace_all( st,
223 // INVALID_FILE_SEPARATOR ,
224 // VALID_FILE_SEPARATOR);
225 //const boost::filesystem::path mpath(st);
226 //C:\Documents and Settings\cervenansky\.gimmick");
227 //if ( !boost::filesystem::exists( path ) ) return ;
228 // boost::filesystem::directory_iterator end_itr; // default construction yields past-the-end
229 // for ( boost::filesystem::directory_iterator itr( path ); itr != end_itr; ++itr )
231 //// If is directory & recurse : do recurse
232 // if ( boost::filesystem::is_directory(itr->status()) )
236 //JCP settings dir 02/10/2009
237 const std::string settingsdirectory = GetUserSettingsDirectory();
238 //boost::algorithm::replace_all( mUserSettingsDirectory,
239 // INVALID_FILE_SEPARATOR ,
240 // VALID_FILE_SEPARATOR);
242 //("C:\\Documents and Settings\\cervenansky\\.gimmick\\"); // settingsdirectory );
244 isdir = boost::filesystem::is_directory(settingsdirectory); // settingsdirectory );
247 GimmickMessage(1,"Directory '"<<GetUserSettingsDirectory()<<"' "
248 << "does not exist : creating it"<<std::endl);
250 if ( ! boost::filesystem::create_directory( GetUserSettingsDirectory() ) )
252 GimmickError("ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
256 std::string setDir=GetUserSettingsDirectory();
257 boost::algorithm::replace_all( setDir,
258 INVALID_FILE_SEPARATOR ,
259 VALID_FILE_SEPARATOR);
261 boost::filesystem::create_directory( setDir );
262 setDir+="creaImageIO/";
263 boost::filesystem::create_directory( setDir );
264 setDir+=mLocalDescpName;
266 if(!boost::filesystem::is_regular(setDir))
269 crea::System::GetAppPath(name,PATH_MAX);
270 std::cout<<name<<std::endl;
272 std::string path=name;
273 path=path.substr(0,path.size()-1);
274 path=path.substr(0,path.find_last_of("/"));
275 //Creating directories
277 // The following stuff works on Linux, NOT CHECKED on Windows // JPR
280 path+="/bin/share/creaImageIO/";
284 path+="/../share/creaImageIO/";
286 #if defined(__APPLE__)
287 path+="/../../../../share/creaImageIO/";
291 path+= mLocalDescpName;
293 std::cout <<"From: " << path <<std::endl;
294 std::cout <<"To: " << setDir <<std::endl;
295 boost::algorithm::replace_all( path,
296 INVALID_FILE_SEPARATOR ,
297 VALID_FILE_SEPARATOR);
298 boost::filesystem::copy_file(path,setDir);
302 //========================================================================
305 //========================================================================
306 /// Sets message level
307 void Gimmick::SetMessageLevel(int l)
309 SetGimmickMessageLevel(l);
311 //========================================================================
313 //========================================================================
314 /// Sets message level
315 void Gimmick::SetDebugMessageLevel(int l)
317 SetGimmickDebugMessageLevel(l);
319 //========================================================================
321 //========================================================================
322 /// Returns the tree handler with the given name
323 TreeHandler* Gimmick::GetTreeHandler(const std::string& name) const
325 TreeHandlerMapType::const_iterator i;
326 i = GetTreeHandlerMap().find(name);
327 if ( i == GetTreeHandlerMap().end() )
329 GimmickError("TreeHandler '"<<name<<"' does not exist");
334 //========================================================================
335 /// Add the files to the tree handler
336 void Gimmick::AddFiles(const std::string& d,
337 const std::vector<std::string>& filenames)
339 GimmickMessage(2,"Adding files to '"<<d<<"'"<<std::endl);
341 mImageAdder.SetCurrentDatabase(d);
342 mImageAdder.SetTreeHandler(GetTreeHandler(d));
343 mImageAdder.SetSynchronizer(mSynchronizer);
344 mImageAdder.AddFiles(filenames);
346 //========================================================================
348 //========================================================================
349 /// Add a dir to the local database
350 void Gimmick::AddDir(const std::string& d, const std::string& f,
353 GimmickMessage(2,"Adding dir '"<<f<<"' to '"<<d<<"' recurse:"
354 <<recurse<<std::endl);
356 TreeHandler * handler=GetTreeHandler(d);
357 mImageAdder.SetCurrentDatabase(d);
358 mImageAdder.SetTreeHandler(handler);
359 mImageAdder.SetSynchronizer(mSynchronizer);
360 mImageAdder.AddDirectory(f,recurse);
363 //========================================================================
365 //========================================================================
366 void Gimmick::RemoveFile(const std::string& d,
369 mImageAdder.SetCurrentDatabase(d);
370 mImageAdder.SetSynchronizer(mSynchronizer);
371 mImageAdder.RemoveFile(node);
373 //========================================================================
375 //void Gimmick::Anonymize(std::vector<std::string>i_files) { }
376 //========================================================================
378 void Gimmick::CopyFiles(const std::vector<std::string>& filenames, const std::string& d )
380 TreeHandler * handler=GetTreeHandler(d);
381 mImageAdder.SetCurrentDatabase(d);
382 mImageAdder.SetTreeHandler(handler);
383 mImageAdder.SetSynchronizer(mSynchronizer);
384 mImageAdder.CopyFiles(filenames, mSettings->getValue(SETTINGS_COPY_PATH));
387 //========================================================================
389 std::string Gimmick::Synchronize(const std::string& d, bool repair, bool checkAttributes)
391 TreeHandler * handler=GetTreeHandler(d);
392 mImageAdder.SetCurrentDatabase(d);
393 mImageAdder.SetTreeHandler(handler);
394 mImageAdder.SetSynchronizer(mSynchronizer);
395 return mImageAdder.Synchronize(repair, checkAttributes);
398 //========================================================================
400 void Gimmick::Print(const std::string& d)
402 GetTreeHandler(d)->GetTree().Print();
404 //========================================================================
406 void Gimmick::GetSetting(const std::string& name, std::string& value)
408 value = mSettings->getValue(name);
410 //========================================================================
412 //========================================================================
414 void Gimmick::GetAttributes(const std::string& d,
415 const std::string& filename,
416 const std::vector<std::string>& params,
417 std::vector<std::string>& results)
419 TreeHandler * handler=GetTreeHandler(d);
420 mImageAdder.SetCurrentDatabase(d);
421 mImageAdder.SetTreeHandler(handler);
422 mImageAdder.SetSynchronizer(mSynchronizer);
423 mImageAdder.GetAttributes(params, filename, results);
425 //========================================================================
427 //========================================================================
428 // get attributes values from database for a given file from database
429 //========================================================================
430 void Gimmick::GetAttributes(const std::string filename, std::map<std::string, std::string> &i_res, OutputAttr i_attr)
432 if (i_attr.inside.size() > 0)
434 std::map<std::string, std::string> map_attr;
435 TreeHandler * handler=GetTreeHandler(i_attr.db);
436 handler->getAllAttributes(filename, map_attr);
437 if(i_attr.inside.front() == "ALL") // we take all values
439 std::map<std::string, std::string>::iterator it = map_attr.begin();
440 for(; it != map_attr.end(); it++)
441 i_res[it->first] = it->second;
445 std::vector<std::string>::iterator it = i_attr.inside.begin();
446 for(; it != i_attr.inside.end(); it++)
447 i_res[(*it)] = map_attr[(*it)];
453 //========================================================================
455 void Gimmick::UpdateSetting(const std::string& name, const std::string& value)
457 mSettings->updateSetting(name,value);
458 mSettings->writeSettingsFile();
460 //========================================================================
462 void Gimmick::DeleteDrive(const std::string& drive)
464 for( TreeHandlerMapType::const_iterator it = mTreeHandlerMap.begin();
465 it!= mTreeHandlerMap.end();
468 mImageAdder.SetTreeHandler(it->second);
469 mImageAdder.DeleteDriveFromMainDB(drive);
471 mImageAdder.SetSynchronizer(mSynchronizer);
472 mImageAdder.DeleteDriveFromOtherDB(drive);
475 //========================================================================
476 void Gimmick::EditField(tree::Node* node, const std::string& d, const std::string& name, const std::string& key, const std::string& val)
478 TreeHandler * handler=GetTreeHandler(d);
479 mImageAdder.SetCurrentDatabase(d);
480 mImageAdder.SetTreeHandler(handler);
481 mImageAdder.EditField(node,name,key,val);
483 //========================================================================
485 ////////////////////////////////////////////////////////////////////////
486 // add DB from Settings file //
489 ////////////////////////////////////////////////////////////////////////
490 void Gimmick::addDBSettings()
493 std::string pathSettings = mSettings->getValue(SETTINGS_DBPATH);
495 // split to find all paths
496 std::vector<std::string> paths;
497 std::string separator = ";";
498 std::string::size_type last_pos = pathSettings.find_first_not_of(separator);
499 //find first separator
500 std::string::size_type pos = pathSettings.find_first_of(separator, last_pos);
501 while(std::string::npos != pos || std::string::npos != last_pos)
503 paths.push_back(pathSettings.substr(last_pos, pos - last_pos));
504 last_pos = pathSettings.find_first_not_of(separator, pos);
505 pos = pathSettings.find_first_of(separator, last_pos);
508 std::vector<std::string>::iterator it_path = paths.begin();
509 for(; it_path != paths.end(); ++it_path)
511 pos = it_path->find_last_of("\\");
512 last_pos = it_path->find_last_of(".");
513 std::string name = it_path->substr(pos +1, last_pos -pos-1 );
514 addDB(name, it_path->c_str());
519 ///////////////////////////////////////////////////////////////////////////////
520 // Fill attributes structure with attributes present in database (inside vector
521 // and not present (outside)
522 ///////////////////////////////////////////////////////////////////////////////
523 void Gimmick::fillVectInfos(std::vector<std::string> i_attr, OutputAttr &infos)
525 //test if a tag is present in Database descriptor
526 TreeHandler * handler=GetTreeHandler(infos.db);
527 mImageAdder.SetTreeHandler(handler);
528 std::vector<std::string>::const_iterator it = i_attr.begin();
529 for (;it != i_attr.end(); it++)
531 if( mImageAdder.isAttributeExist((*it)) != "" ) // in DB
533 infos.inside.push_back((*it));
537 infos.outside.push_back((*it)); // Need to scan again the files
542 const std::string Gimmick::getSummary()
544 const AddProgress& p = GetAddProgress();
545 std::stringstream mess;
546 mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs() << "\n";
547 mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
548 mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
549 mess << "Files\tadded \t: " << p.GetNumberAddedFiles() << "\n\n";