X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOSQLiteTreeHandler.cpp;h=4363e7c0a99de19d39fc44b4e0570f0598a9496d;hb=8d73d5f58d80abb928a7f60037837c2933b16bbd;hp=509aa90429e4c1f3e9e4447bc7f5f522a88f9e2d;hpb=cbf693fa62cd51f4ca5c881838bbb609edc447b0;p=creaImageIO.git

diff --git a/src2/creaImageIOSQLiteTreeHandler.cpp b/src2/creaImageIOSQLiteTreeHandler.cpp
index 509aa90..4363e7c 100644
--- a/src2/creaImageIOSQLiteTreeHandler.cpp
+++ b/src2/creaImageIOSQLiteTreeHandler.cpp
@@ -1,4 +1,5 @@
 #include <creaImageIOSQLiteTreeHandler.h>
+#include <creaImageIOSystem.h>
 
 #include "CppSQLite3.h"
 
@@ -20,7 +21,6 @@
 //#include <icons/close.xpm>
 
 #include <creaWx.h>
-#include <creaMessageManager.h>
 using namespace crea;
 
 #include <boost/filesystem.hpp>
@@ -35,28 +35,9 @@ namespace creaImageIO
   SQLiteTreeHandler::SQLiteTreeHandler(const std::string& filename)
     : mFileName(filename)
   {
-    /*
-    mSQLiteTreeHandler = this;
-    NodeTypeDescription::FieldDescriptionMapType& M = 
-      mNodeTypeDescription[Node::Database].GetFieldDescriptionMap();
-
-    boost::filesystem::path full_path(location);
-    mName = full_path.leaf();
-    Field::Description fname("Name",0,0,"Name",0);
-    M[fname.key] = fname;
-    UnsafeSetFieldValue(fname.key,mName);
-    Field::Description flocation("File name",0,0,"File name",0);
-    M[flocation.key] = flocation;
-    UnsafeSetFieldValue(flocation.key,location);
-    */
-    //    Field::Description ftype("Type",0,0,"Type",0);
-    //    M[ftype.key] = ftype;
-    //    UnsafeSetFieldValue(ftype.key,"Invalid location");
-
     mDB = new CppSQLite3DB;
-    //    std::cout << "** SQLite Version: " << mDB->SQLiteVersion() << std::endl;
-
-
+    GimmickMessage(1,"SQLite version : "
+		   <<std::string(mDB->SQLiteVersion())<< std::endl);
   }
   //=============================================================
 
@@ -64,34 +45,10 @@ namespace creaImageIO
   SQLiteTreeHandler::~SQLiteTreeHandler()
   {
     delete mDB;
-    /*
-      Already done in Node now that SQLiteTreeHandler inherits from it
-    ChildrenListType::iterator i;
-    for (i=GetChildrenList().begin(); i!=GetChildrenList().end(); i++)
-      {
-	delete *i;
-      }
-    */
   }
   //=============================================================
   
 
-  //=============================================================
-  void SQLiteTreeHandler::BuildDefaultTreeDescription()
-  {
-    /*
-    for (int i = Node::Patient;
-	 i <= Node::Image; 
-	 ++i)
-      {
-	mNodeTypeDescription[i].BuildDefault(i);
-      }
-    */
-
-    // TODO : GetTree().GetDescription().LoadXML(FILE);
-  }
-  //=============================================================
-
   //=============================================================
   //  void SQLiteTreeHandler::Print() const 
   //  {
@@ -117,48 +74,6 @@ namespace creaImageIO
   */
   //=====================================================================
 
-  //=====================================================================
-  char* format_sql(const std::string& s)
-  { 
-    return sqlite3_mprintf("%q",s.c_str());
-  }
-  //=====================================================================
-
-  //  sqlite3_exec(db, zSQL, 0, 0, 0);
-  //  sqlite3_free(zSQL);
-  //	char* CHAIN = format_sql(QUER);		\
-//  sqlite3_free(CHAIN);				\
-
-  //=====================================================================
-#define QUERYDB(QUER,RES)						\
-    try									\
-      {									\
-	RES = mDB->execQuery(QUER.c_str());				\
-      }									\
-    catch (CppSQLite3Exception& e)					\
-      {									\
-	std::cout << "SQLite query '"<<QUER<<"' : "<< e.errorCode() << ":" \
-		  << e.errorMessage()<<std::endl;			\
-	creaError("SQLite query '"<<QUER<<"' : "<< e.errorCode() << ":" \
-		   << e.errorMessage() );				\
-      }									\
-    
-  //=====================================================================
-  
-  //=====================================================================
-#define UPDATEDB(UP)							\
-  try									\
-    {									\
-      mDB->execDML(UP.c_str());						\
-    }									\
-  catch (CppSQLite3Exception& e)					\
-    {									\
-      std::cout << "SQLite update '"<<UP<<"' Error : "<< e.errorCode() << ":" \
-		<< e.errorMessage()<<std::endl;				\
-      creaError("SQLite update '"<<UP<<"' Error : "<< e.errorCode() << ":" \
-		 << e.errorMessage() );					\
-    }									
-  //=====================================================================
 
   //=====================================================================
   bool SQLiteTreeHandler::Open(bool writable)
@@ -181,7 +96,7 @@ namespace creaImageIO
   //=====================================================================
   bool SQLiteTreeHandler::Close()
   {
-    return false;
+    return true;
   }
   //=====================================================================
 
@@ -191,19 +106,14 @@ namespace creaImageIO
   {
     return false;
   }
-  //=====================================================================
-
-  //===================================================================== 
-  unsigned int SQLiteTreeHandler::GetNumberOfChildren(tree::Node* n) 
-  { 
-    return 0; 
-  }
+  
   //===================================================================== 
 
   //===================================================================== 
   int SQLiteTreeHandler::LoadChildren(tree::Node* parent, int maxlevel)
   {
-    return 0;
+    if (parent==0) parent = GetTree().GetTree();
+    return DBLoadChildren(parent,maxlevel);
   }
   //===================================================================== 
 
@@ -216,32 +126,117 @@ namespace creaImageIO
   }
   //===================================================================== 
 
-
   //===================================================================== 
-  int SQLiteTreeHandler::AddBranch( const std::map<std::string,std::string>& attr )
+  int SQLiteTreeHandler::AddBranch( const AttributeMapType& attr )
   {
-    return -1;
+    tree::Node* parent = DBGetParent(attr);
+    DBGraftToParent(parent,attr);
+    return (parent->GetLevel()+1);
+	  
   }
   //===================================================================== 
  
 
   //===================================================================== 
-   bool SQLiteTreeHandler::Remove(tree::Node*)
+   bool SQLiteTreeHandler::Remove(tree::Node* node)
    {
-     return false;
+   DBRecursiveRemoveNode(node);
+ 
+    //    std::cout << "DELETE"<<std::endl;
+    if (node->GetParent())
+      {
+	node->GetParent()->RemoveChildrenFromList(node);
+      }
+    delete node;
+    //    std::cout << "DELETE OK"<<std::endl;
+    return true;
    }
+
+    //========================================================================
+
+  
+  
   //===================================================================== 
 
+  //===================================================================== 
+  /// Sets an attribute of a Node
+  bool SQLiteTreeHandler::SetAttribute(tree::Node* n, 
+				       const std::string& key,
+				       const std::string& value)
+  {
+    if (n==0) n=GetTree().GetTree();
+    return DBSetAttribute(n,key,value);
+  }
+  //===================================================================== 
+
+
+
+
+
+
+
+
+
+
+
+
+
+  //=====================================================================
+  // SQLite DB specific methods
+  //=====================================================================
 
 
 
 
+  //=====================================================================
+  char* format_sql(const std::string& s)
+  { 
+    return sqlite3_mprintf("%q",s.c_str());
+  }
+  //=====================================================================
+
+  //  sqlite3_exec(db, zSQL, 0, 0, 0);
+  //  sqlite3_free(zSQL);
+  //	char* CHAIN = format_sql(QUER);		\
+//  sqlite3_free(CHAIN);				\
+
+  //=====================================================================
+#define QUERYDB(QUER,RES)						\
+    try									\
+      {									\
+	GimmickMessage(2,"SQL query: '"<<QUER<<"'"<<std::endl);		\
+	RES = mDB->execQuery(QUER.c_str());				\
+      }									\
+    catch (CppSQLite3Exception& e)					\
+      {									\
+	GimmickError("SQLite query '"<<QUER<<"' : "			\
+		     << e.errorCode() << ":"				\
+		     << e.errorMessage() );				\
+      }									\
+    
+  //=====================================================================
+  
+  //=====================================================================
+#define UPDATEDB(UP)							\
+  try									\
+    {									\
+      GimmickMessage(2,"SQL update: '"<<UP<<"'"<<std::endl);		\
+      mDB->execDML(UP.c_str());						\
+    }									\
+  catch (CppSQLite3Exception& e)					\
+    {									\
+      GimmickError("SQLite update '"<<UP<<"' Error : "			\
+		   << e.errorCode() << ":"				\
+		   << e.errorMessage() );				\
+    }									
+  //=====================================================================
 
 
   //=====================================================================
   bool SQLiteTreeHandler::DBOpen()
   {
-    //    std::cout << "### Opening SQLite database '"<<GetFileName()<<std::endl;
+    GimmickMessage(1,"Opening SQLite database '"<<GetFileName()
+		   <<"' ... "<<std::endl);
     // OPENING FILE
     if (!boost::filesystem::exists(GetFileName())) 
       {
@@ -254,20 +249,19 @@ namespace creaImageIO
       }
     catch (CppSQLite3Exception& e)
       {
-	std::cerr << "Opening '"<<GetFileName()<<"' : "
-		  << e.errorCode() << ":" 
-		  << e.errorMessage() << std::endl;
+	GimmickError("Opening '"<<GetFileName()<<"' : "
+		     << e.errorCode() << ":" 
+		     << e.errorMessage());
 	return false;
       }
-    // TESTING STRUCTURE VALIDITY
-    if (!DBStructureIsValid())
+    // IMPORT TREE DESCRIPTION (AND TEST DB VALIDITY)
+    if (!DBImportTreeDescription())
       {
- 	std::cerr << "Opening '"<<GetFileName()<<"' : "
-		  << " invalid database structure" << std::endl;
 	return false;
       }
-    // IMPORTING NODE TYPE DESCRIPTIONS
-    DBImportTreeDescription();
+
+    GimmickDebugMessage(1,"Opening SQLite database '"<<GetFileName()
+		   <<"' ... OK"<<std::endl);
     return true;
   }
   //=====================================================================
@@ -275,13 +269,13 @@ namespace creaImageIO
   //=====================================================================
   bool SQLiteTreeHandler::DBCreate()
   {
-    //    std::cout << "### Creating SQLite database '"<<GetFileName()<<std::endl;
+    GimmickMessage(1,"Creating SQLite database '"<<GetFileName()
+		   <<"' ... "<<std::endl);
 
     if (boost::filesystem::exists(GetFileName())) 
       {
-	creaMessage("Gimmick!",1,
-		    "[Gimmick!] !! ERROR '"<<GetFileName()<<"' : "
-		    << "file already exists"<<std::endl);
+	GimmickError(GetFileName()<<"' : "
+		     << "file already exists");
 	return false;
       }
     
@@ -292,288 +286,600 @@ namespace creaImageIO
       }
     catch (CppSQLite3Exception& e)
       {
-	creaMessage("Gimmick!",1,
-		    "[Gimmick!] !! ERROR '"
-		    << e.errorCode() << ":" 
-		    << e.errorMessage() <<std::endl);
+	GimmickError(e.errorCode() << ":" 
+		     << e.errorMessage() <<std::endl);
 	return false;
       }
     
+     
     // CREATING TABLES
-    try
+    
+    std::string command;
+    // Create LEVELS table
+    command = "create table LEVELS\n";
+    command += "( Name text )\n";
+    UPDATEDB(command);
+    
+    // Iterate the Levels
+    for (int l=0; l<GetTree().GetNumberOfLevels(); ++l)
       {
-	std::string command;
-	// Level 0
-	command = "create table ";
-	command += GetTree().GetLevelDescriptor(0).GetName();
-	command += "\n(\nID INTEGER PRIMARY KEY,\n";
-	AppendAttributesSQLDefinition(0,command);
-	command += "\n)";
+	command = "INSERT INTO LEVELS (Name) VALUES ('";
+	command += GetTree().GetLevelDescriptor(l).GetName();
+	command += "')";
 	UPDATEDB(command);
 	
-	// Iterate the other Levels
-	for (int l=1; l<GetTree().GetNumberOfLevels(); ++l)
-	  {
-	    command = "create table ";
-	    command += GetTree().GetLevelDescriptor(l).GetName();
-	    command += "\n(\nID INTEGER PRIMARY KEY,\nPARENT_ID int not null,\n";	
-	    AppendAttributesSQLDefinition(l,command);
-	    command +=",\n";
-	    command +="constraint FK_PARENT foreign key (PARENT_ID) references PATIENT(ID) on delete restrict on update restrict\n)";
-	    UPDATEDB(command);
-	  }
-	
-	// Create tables *_ATTRIBUTES
-	for (int l=1; l<GetTree().GetNumberOfLevels(); ++l)
+	// Create table of level (for level>0, i.e. not Root)
+	if (l>=0)
 	  {
-	    command = "create table ";
+	    command = "CREATE TABLE ";
 	    command += GetTree().GetLevelDescriptor(l).GetName();
-	    command += "_ATTRIBUTES\n(\n";
-	    command += "Key text,\n";
-	    command += "Group int,\n";
-	    command += "Element int,\n";	    
-	    command += "Name text,\n";	    
-	    command += "Flags int\n";	    
+	    command += "\n(\nID INTEGER PRIMARY KEY";
+	    if (l>1) 
+	      {
+		command += ",\nPARENT_ID int not null";	
+	      }
+	    SQLAppendAttributesDefinition(l,command);
+	    if (l>1) 
+	      {
+		command += ",\nconstraint FK_PARENT foreign key (PARENT_ID) references ";
+		command += GetTree().GetLevelDescriptor(l-1).GetName();
+		command += "(ID) on delete restrict on update restrict";
+	      }
 	    command += "\n)";
 	    UPDATEDB(command);
+	    
+	    
+	    // Add Attribute 'ID' to Description
+	    GetTree().GetDescriptor().Add
+	      (AttributeDescriptor( "ID",
+				    "Database Identifier",
+				    0,0,
+				    AttributeDescriptor::PRIVATE
+				    ),l);
+	    
+	    if (l>1) 
+	      {
+		// Add Attribute 'PARENT_ID' to Description
+		GetTree().GetDescriptor().Add
+		  (AttributeDescriptor( "PARENT_ID",
+					"Database Parent Identifier",
+					0,0,
+					AttributeDescriptor::PRIVATE
+					),l);
+	      }
+	    
 	  }
-	// Fill the tables *_ATTRIBUTES
-	DBExportTreeDescription();
 	
-      }
-    catch (std::exception)
-      {
-	return false;
-      }
-    // LG : TEST
-    if (DBStructureIsValid())
-      {
-	//	std::cout << "*** DONE ***"<<std::endl;
-	//	mDBLoaded = true;
-	GetTree().SetChildrenLoaded(true);
-	return true;
+	// Create table *_ATTRIBUTES
 	
-      }
-    else 
-      {
-	//	std::cout << "*** AAAARRRRGGG ***"<<std::endl;
+	command = "CREATE TABLE ";
+	command += GetTree().GetLevelDescriptor(l).GetName();
+	command += "_Attributes\n(\n";
+	command += "Key text,\n";
+	command += "Name text,\n";	    
+	command += "DicomGroup int,\n";
+	command += "DicomElement int,\n";	    
+	command += "Flags int\n";	    
+	command += "\n)";
+	UPDATEDB(command);
 	
-	return false;
-      }
+	
+	// Fill the table *_ATTRIBUTES
+	LevelDescriptor::AttributeDescriptorListType::const_iterator i;
+	for (i  = GetTree().GetAttributeDescriptorList(l).begin();
+	     i != GetTree().GetAttributeDescriptorList(l).end();
+	     ++i)
+	  {
+	    
+	    std::stringstream insert;
+	    insert << "INSERT INTO "
+		   << GetTree().GetLevelDescriptor(l).GetName()
+		   << "_Attributes (Key,Name,DicomGroup,DicomElement,Flags) "
+		   << "VALUES ('"
+		   << i->GetKey() << "','"
+		   << i->GetName() << "',"
+		   << i->GetGroup() << ","
+		   << i->GetElement() << ","
+		   << i->GetFlags() << ");";
+	    UPDATEDB(insert.str());
+	  }
+
+      } // For l=0...
+
+    // Initialize the root attributes
+    GetTree().InitializeAttributeMap();
+    // Insert the root in the level 0 table 
+    DBInsert(GetTree().GetTree());
+    
+    
+    GetTree().SetChildrenLoaded(true);
+    GimmickMessage(1,"Creating SQLite database '"<<GetFileName()
+		   <<"' ... OK"<<std::endl);
+    return true;
   }
   //=====================================================================
   
   //=====================================================================
-  void SQLiteTreeHandler::AppendAttributesSQLDefinition(int level,
+  void SQLiteTreeHandler::SQLAppendAttributesDefinition(int level,
 							std::string& s)
   {
-    /*
-    std::vector<std::string*> keys;
-    NodeTypeDescription::FieldDescriptionMapType::iterator i;
-    for (i  = GetNodeTypeDescription(c).GetFieldDescriptionMap().begin();
-	 i != GetNodeTypeDescription(c).GetFieldDescriptionMap().end();
+    LevelDescriptor::AttributeDescriptorListType::const_iterator i;
+    for (i  = GetTree().GetAttributeDescriptorList(level).begin();
+	 i != GetTree().GetAttributeDescriptorList(level).end();
 	 ++i)
       {
-	if (i->second.flags==1) continue;
-	keys.push_back(&(i->second.key));
+	//	if (i->second.flags==1) continue;
+	s += ",\n";
+	s += i->GetKey();
+	s += " text";
       }
-    std::vector<std::string*>::iterator j;
-    for (j=keys.begin();j!=keys.end();)
-      {
-	s += **j + " text";
-	++j;
-	if (j!=keys.end())
-	  s += ",\n";
-      }
-    */
   }
   //=====================================================================
   
-  //=====================================================================
-  void SQLiteTreeHandler::DBExportTreeDescription()
-  {
-    /*
-    //    std::cout<<"ExportNodeTypeDescriptionsToDB()"<<std::endl;
-    for (Node::Type type=Node::Patient; 
-	 type<=Node::Image; 
-	 type++)
-      {
-	NodeTypeDescription::FieldDescriptionMapType::iterator i;
-	for (i = GetNodeTypeDescription(type).GetFieldDescriptionMap().begin();
-	     i!= GetNodeTypeDescription(type).GetFieldDescriptionMap().end();
-	     i++)
-	  {
-	    
-	    std::stringstream insert;
-	    insert << "INSERT INTO "
-		   << std::string(SQLiteTreeHandlerStructure::Table(type))
-		   << "_FIELDS (Key,DicomGroup,DicomElement,Name,Flags) "
-		   << "VALUES ('"
-		   << (*i).second.GetKey() << "',"
-		   << (*i).second.GetGroup() << ","
-		   << (*i).second.GetElement() << ",'"
-		   << (*i).second.GetName() << "',"
-		   << (*i).second.GetFlags() << ");";
-
-	    //	    std::cout << "** SQL = '"<<insert.str()<<"'"<<std::endl;
-	    
-	    UPDATEDB(insert.str());
-	  }
-      }
-    */
-  }
-  //=====================================================================
   
   //=====================================================================
-  void SQLiteTreeHandler::DBImportTreeDescription()
+  bool SQLiteTreeHandler::DBImportTreeDescription()
   {
-    //    std::cout<<"ImportNodeTypeDescriptionsFromDB()"<<std::endl;
-    /*
-    for (Node::Type type=Node::Patient; 
-	 type<=Node::Image; 
-	 type++)
+    GimmickMessage(1,"Importing tree description from database ..."
+		   <<std::endl);
+
+    // Test table 'LEVELS' existence
+    if ( ! mDB->tableExists("LEVELS") )
       {
-	std::string query = "SELECT * FROM ";
-	query += SQLiteTreeHandlerStructure::Table(type);
-	query += "_FIELDS";
+	GimmickMessage(1,"!! ERROR : Table 'LEVELS' does not exist"
+		       <<std::endl);
+	return false;
+      }
+
+    tree::Descriptor& desc = GetTree().GetDescriptor();
+    // clears the existing one
+    desc.Clear();
+     
+    int nblevel = 0;
+    std::string query = "SELECT * FROM LEVELS";
+    CppSQLite3Query q;
+    QUERYDB(query,q);
 
-	//	std::cout << "** SQL = '"<<query<<"'"<<std::endl;
+    while (!q.eof())
+      {
+	std::string name = q.getStringField(0);
+	GimmickMessage(2," * Importing level '"<<name<<"'"<<std::endl);
+	desc.Add(LevelDescriptor(name));
+	nblevel++;
+	q.nextRow();
+      }   
+    
+    for (int level = 0; level < nblevel; ++level )
+      {
+	std::string table = GetTree().GetLevelDescriptor(level).GetName();
+	table += "_Attributes";
+	// Test table 'LEVELS' existence
+	if ( ! mDB->tableExists(table.c_str()) )
+	  {
+	    GimmickMessage(1,"!! ERROR : Table '"<<table<<"' does not exist"
+			   <<std::endl);
+	    return false;
+	  }
 	
+	std::string query = "SELECT * FROM ";
+	query += table;
 	CppSQLite3Query q;
 	QUERYDB(query,q);
 	
+	GimmickMessage(2," * Level '"
+		       <<GetTree().GetLevelDescriptor(level).GetName()
+		       <<"'"<<std::endl);
+
+	// Test that ID and PARENT_ID mandatory attributes exist
+	bool ID_found = false;
+	bool PARENT_ID_found = false;
+	if (level==0) ID_found = true;
+	if (level<=1) PARENT_ID_found = true;
 
 	while (!q.eof())
 	  {
-	    Field::Description d(q.getStringField(0), // Key
-				 q.getIntField(1), // Group
-				 q.getIntField(2), // Element 
-				 q.getStringField(3), // Name
-				 q.getIntField(4) // Flags
-				 );
-	    GetNodeTypeDescription(type).GetFieldDescriptionMap()[d.key] = d;
-	    //	    std::cout << d << std::endl;
+	    std::string key(q.getStringField(0));
+	    std::string name(q.getStringField(1));
+	    GimmickMessage(2,"  - Importing attribute '"<<key<<"' '"<<name
+			   <<"'"<<std::endl);
+	    desc.Add
+	      (AttributeDescriptor( key, // Key
+				    name, // Name
+				    q.getIntField(2), // Group
+				    q.getIntField(3), // Element 
+				    q.getIntField(4) // Flags
+				    ),level);
+	    if ( key == "ID" ) 
+	      {
+		ID_found = true;
+	      }
+	    if ( key == "PARENT_ID" ) 
+	      {
+		PARENT_ID_found = true;
+	      }
 	    q.nextRow();
 	  }
+	
+	if ( ! (ID_found || PARENT_ID_found ) )
+	  {
+	    GimmickMessage(1,"!! ERROR : Table '"<<table
+			   <<"' does not contain mandatory attribute ID or PARENT_ID"
+			   <<std::endl);
+	    return false;
+ 
+	  }
       }
-    */
+
+
+    // Create the attributes table for Root (i.e. Tree)
+    LevelDescriptor::AttributeDescriptorListType::const_iterator a;
+    for (a = GetTree().GetAttributeDescriptorList(0).begin();
+	 a!= GetTree().GetAttributeDescriptorList(0).end();
+	 ++a)
+      {
+	/*
+	std::string v;
+	AttributeMapType::const_iterator i = attr.find(a->GetKey());
+	if ( i != attr.end() )  
+	  {
+	    v = i->second;
+	  }
+	*/
+	GetTree().UnsafeSetAttribute( a->GetKey(), "" );
+      }
+
+    // Reading Root attributes
+    // Query DB
+    query = "SELECT * FROM ";
+    query += GetTree().GetLevelDescriptor(0).GetName();
+    QUERYDB(query,q);
+
+    for (int fld = 0; fld < q.numFields(); fld++)
+      {
+	GetTree().UnsafeSetAttribute(q.fieldName(fld),
+				     q.getStringField(fld));	    
+      }
+
+    GimmickMessage(1,"Importing tree description from database ... OK"
+		   <<std::endl);
+    return true;
   }
   //=====================================================================
 
 
-    //=====================================================================
-  bool SQLiteTreeHandler::DBStructureIsValid()
-  {
-    bool success = false; //true;
 
-    // TO DO : TABLE WHICH STORES THE LEVELS
-    /*
-    
-    for (int i = SQLiteTreeHandlerStructure::TableBegin();
-	 i    != SQLiteTreeHandlerStructure::TableEnd();++i)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+  //========================================================================
+  /// 
+  std::string& SQLformat(std::string& str)
+  {
+    // quote must be doubled
+    boost::algorithm::replace_all(str,"'","''");
+    // Found strange strings which contained NULL char INSIDE string 
+    int i,size=str.size();
+    for (i=0;i<size;++i) 
       {
-	bool ok = mDB->tableExists(SQLiteTreeHandlerStructure::Table(i));
-	if (ok) 
+	if (str[i]==0) 
 	  {
-	    //	    std::cout << "** Table "<<SQLiteTreeHandlerStructure::Table(i)
-	    //		      <<" exists"<<std::endl;
-	    // TO DO : TEST MANDATORY FIELDS EXIST
+	    str = str.substr(0,i);
+	    break;
 	  }
-	else 
+      }
+    //    if (i<str.size())
+    return str;
+  }
+  //========================================================================
+  
+  //=====================================================================
+  void SQLiteTreeHandler::SQLAppendAttributesValues(tree::Node* n, 
+						    std::string& str)
+  {
+    GimmickMessage(4,"SQLAppendAttributesValues"<<std::endl);
+    std::string atts="";
+    std::string values="";
+    tree::Node::AttributeMapType::iterator i;
+    for (i =  n->GetAttributeMap().begin();
+	 i != n->GetAttributeMap().end();
+	 i++)
+      {
+	if (i->first=="ID") 
 	  {
-	    //	    std::cout << "** Table "<<SQLiteTreeHandlerStructure::Table(i)
-	    //		      <<" does *NOT* exist"<<std::endl;
-	    success = false;
+	    continue;
 	  }
+	//	std::cout << "("<<i->first<<","<<i->second<<")"<<std::endl;
+	atts += "'" + i->first + "'";
+	values += "'" + SQLformat(i->second) + "'"; 
+	atts += ",";
+	values += ",";
+	GimmickMessage(4,"'"<<i->first<<"' = '"<<i->second<<"'"<<std::endl);
       }
-    */
-    return success;
-  }  
+    atts[atts.size()-1]=' ';
+    values[values.size()-1]=' ';
+
+    str = "("+atts+") VALUES ("+values+")";
+    GimmickMessage(4,"Result = '"<<str<<"'"<<std::endl);
+  }
   //=====================================================================
 
-  /*
   //=====================================================================
-  int SQLiteTreeHandler::DBLoadChildren(Node* parent, 
-					int maxlevel)
+  tree::Node* SQLiteTreeHandler::DBGetParent( const AttributeMapType& attr) 
   {
-
-    //    std::cout << "SQLiteTreeHandler::DBLoadChildren("<<parent<<","<<maxlevel
-    //	      << ")"<<std::endl;
-    int nbloaded = 0;
-    if (parent == GetTree()) { parent = 0; }
-    Node* xparent = parent;
-    if ( xparent==0 ) xparent = GetTree();
-    if ( xparent->ChildrenLoaded() ) 
+    Node* parent = GetTree().GetTree();
+    bool go_down;
+    do 
       {
-	//	std::cout << "--> Children already loaded"<<std::endl;
-	return nbloaded;
-      }
-    if ( xparent->GetType() == Node::Image ) 
-      {
-	return nbloaded;
+	go_down = false;
+	// Load the children of the current parent
+	DBLoadChildren(parent);
+	// Iterate the children 
+	tree::Node::ChildrenListType::const_iterator i;
+	for (i = parent->GetChildrenList().begin();
+	     i!= parent->GetChildrenList().end();
+	     ++i)
+	  {
+	    if ( (*i)->Matches( attr ) ) 
+	      {
+		go_down = true;
+		parent = *i;
+		break;
+	      }
+	  }	
       }
-    if ( xparent->GetType() >= maxlevel ) 
+    while (go_down);
+    return parent;
+  }
+  //=====================================================================
+
+
+  //=====================================================================
+  int SQLiteTreeHandler::DBLoadChildren(tree::Node* node, 
+					int numberoflevels)
+  {
+    if (node->GetLevel()+1 >= node->GetTree()->GetNumberOfLevels() ) 
+      return 0;
+
+    GimmickMessage(2,"Loading children of '"<<node->GetLabel()
+		   <<"'"<<std::endl);
+
+    int nbloaded = 0;
+    // If children loaded we do not have to do it but we need to recurse
+    // in order to load the children's children if necessary, and so on...
+    if (node->GetChildrenLoaded()) 
       {
+	// Iterate the children 
+
+	/*tree::Node::ChildrenListType::iterator i;
+	for (i = node->GetChildrenList().begin();
+	     i!= node->GetChildrenList().end();
+	     ++i)
+	  {
+	    nbloaded += DBLoadChildren(*i,numberoflevels-1);
+	  }*/
 	return nbloaded;
-      }
-   
-    //    msw[2].Pause();
-    //    msw[2].Resume();
 
-    Node::Type type = xparent->GetType()+1;
+      }
+    
+    /// If children not loaded : do it and recurse
 
     // Query DB
-
+    int level = node->GetLevel();
     std::string query = "SELECT * FROM ";
-    query += GetTree().GetDescriptor().GetLevelDescriptor(level).GetName();
-      //SQLiteTreeHandlerStructure::Table(type);
-    if (parent!=0)
+	
+    query += GetTree().GetLevelDescriptor(level+1).GetName();
+    if (level>0)
       {
-	query += " WHERE PARENT_ID='" + parent->GetFieldValue("ID") + "'";
+	query += " WHERE PARENT_ID='" + node->UnsafeGetAttribute("ID") 
+	  + "'";
       }
-
-    //    std::cout << "** SQL = '"<<query<<"'"<<std::endl;
-
     CppSQLite3Query q;
     QUERYDB(query,q);
 
     while (!q.eof())
       {
+	//	  std::cout<<"Name in q: "<<q.fieldName(3)<<" Value: "<<q.getStringField(3)<<std::endl;
 	nbloaded++;
-	Node* n = new Node(type,
-			   this,xparent);
+	Node* n = new Node(node);
 	for (int fld = 0; fld < q.numFields(); fld++)
 	  {
-	    n->SetFieldValue(q.fieldName(fld),q.getStringField(fld));	    
+	    n->UnsafeSetAttribute(q.fieldName(fld),q.getStringField(fld));	    
 	  }
+	/*
 	// Index 
 	TypeId ti;
 	ti.type = type;
 	ti.id = n->GetFieldValue("ID");    
 	mTypeIdToNodeMap[ti] = n;
+	*/
 	// recurse 
-	if ( type < maxlevel ) 
+	if ( numberoflevels != 1 ) 
 	  {
-	    msw[2].Pause();
-	    nbloaded += DBLoadChildren(n,maxlevel);
-	    msw[2].Resume();
+	    //  msw[2].Pause();
+	//    nbloaded += DBLoadChildren(n, numberoflevels-1);
+	    //	    msw[2].Resume();
 	  }
 	// next entry in db
 	q.nextRow();
       }
 
-    xparent->SetChildrenLoaded(true);
+    node->SetChildrenLoaded(true);
     
     //    msw[2].Pause();
     return nbloaded;
   }
   //=====================================================================
 
+  //======================================================================
+  void SQLiteTreeHandler::DBInsert(tree::Node* n)
+  {
+    GimmickMessage(2,"Inserting in DB '"<<n->GetLabel()
+		   <<"'"<<std::endl);
+    std::string val;
+    SQLAppendAttributesValues(n,val);
+    std::string insert("INSERT INTO ");
+    insert += GetTree().GetLevelDescriptor(n->GetLevel()).GetName();
+    insert += " " + val + ";";
 
+    UPDATEDB(insert);
+	
+    // Store DB id of newly created node;
+    long lastrow = mDB->lastRowId();
+    std::stringstream ri;
+    ri << mDB->lastRowId();
+    n->SetAttribute("ID",ri.str());
+  }
+  //======================================================================
 
+  //======================================================================
+  /// Graft the branch defined by the attributes to the parent
+  void SQLiteTreeHandler::DBGraftToParent( tree::Node* parent, 
+					    const AttributeMapType& attr)
+  {
+    GimmickMessage(2,"Grafting to parent '"<<parent->GetLabel()
+		   <<"'"<<std::endl);
 
+    for (int level = parent->GetLevel()+1;
+	 level < GetTree().GetNumberOfLevels();
+	 level++)
+      {
+	// Create Node
+	tree::Node* child = new tree::Node(parent,attr);
+	
+	// Set PARENT_ID if necessary 
+	if ( parent->GetLevel()>0 )
+	  child->SetAttribute("PARENT_ID",parent->GetAttribute("ID"));
+	
+	// Insert in DB
+	DBInsert(child);
+	/*
+	std::string val;
+	SQLAppendAttributesValues(child,val);
+	std::string insert("INSERT INTO ");
+	insert += GetTree().GetLevelDescriptor(child->GetLevel()).GetName();
+	insert += " " + val + ";";
+	UPDATEDB(insert);
+	
+	// Store DB id of newly created node;
+	long lastrow = mDB->lastRowId();
+	std::stringstream ri;
+	ri << mDB->lastRowId();
+	child->SetAttribute("ID",ri.str());
+	*/
+	// Down one level
+	parent = child;
+	
+	/*
+	// Insert in TypeId map
+	TypeId ti;
+	ti.type = node->GetType();
+	ti.id = node_id;
+	mTypeIdToNodeMap[ti] = node;
+	//    std::cout << "== Insert TypeId ("<<ti.type<<","<<ti.id<<")"<<std::endl; 
+	// 
+	msw[2].Pause();
+	
+	if (node->GetType()==Node::Patient) summary.added_patients++;
+	if (node->GetType()==Node::Study) summary.added_studies++;
+	if (node->GetType()==Node::Series) summary.added_series++;
+	if (node->GetType()==Node::Image) summary.added_images++;
+	*/
+      }
+  }
+  //======================================================================
+
+
+  //===================================================================== 
+  /// Sets an attribute of a Node
+  bool SQLiteTreeHandler::DBSetAttribute(tree::Node* n, 
+					 const std::string& key,
+					 const std::string& value)
+  {
+    GimmickMessage(3,"Setting Attribute of '"<<n->GetLabel()<<
+		   "' "<<key<<"='"<<value<<"'"<<std::endl);
+
+    n->SetAttribute(key,value);
+    std::string sql = "UPDATE ";
+    sql += GetTree().GetLevelDescriptor(n->GetLevel()).GetName();
+    sql += " SET ";
+    sql += key;
+    sql += "='";
+    sql += value;
+    sql += "' WHERE ID=";
+    sql += n->GetAttribute("ID");
+    //    sql += " LIMIT 1";
+    UPDATEDB(sql);
+  }
+  //===================================================================== 
 
 
+  //=====================================================================
+  void SQLiteTreeHandler::DBRecursiveRemoveNode(Node* node)
+  {
+       
+    std::string query = "DELETE FROM ";
+   
+    query += GetTree().GetLevelDescriptor(node->GetLevel()).GetName();
+    query += " WHERE ID='"+ node->GetAttribute("ID") + "';";
+ 
+    UPDATEDB(query);
+	GimmickDebugMessage(2,
+			    " Deleting '"
+				<<node->GetLabel()<<"' with ID '"
+			    <<node->GetAttribute("ID")
+			    <<"' in level "<< GetTree().GetLevelDescriptor(node->GetLevel()).GetName()
+			    <<std::endl);
+
+    Node::ChildrenListType::iterator i;
+    for (i  = node->GetChildrenList().begin();
+	 i != node->GetChildrenList().end();
+	 i++)
+      {
+	DBRecursiveRemoveNode((*i));
+      }
+  }
+
+  //===================================================================== 
+  unsigned int SQLiteTreeHandler::GetNumberOfChildren(tree::Node* n) 
+  { 
+	  // Query DB
+	int nb=0;
+    int level = n->GetLevel();
+	if(GetTree().GetNumberOfLevels()!=(level+1))
+	{
+	std::string query = "SELECT * FROM ";
+    query += GetTree().GetLevelDescriptor(level+1).GetName();
+    if (level>0)
+      {
+	query += " WHERE PARENT_ID='" + n->UnsafeGetAttribute("ID") 
+	  + "'";
+      }
+
+    CppSQLite3Query q;
+    QUERYDB(query,q);
+	
+
+		while (!q.eof())
+		{
+			nb++;
+			q.nextRow();
+		}
+	}
+    return nb; 
+  }
+
+  /*
   //=====================================================================
   bool SQLiteTreeHandler::DBInsert(Node* alien_node,
 				   UpdateSummary& summary)
@@ -859,12 +1165,14 @@ namespace creaImageIO
     //    std::cout << " ** Node = "<<i->second<<std::endl;
     return i->second;
   }
+ 
   //=====================================================================
 
   //=====================================================================
   bool SQLiteTreeHandler::Remove(Node* node)
   {
-    DBRecursiveRemoveNode(node);
+
+    //DBRecursiveRemoveNode(node);
  
     //    std::cout << "DELETE"<<std::endl;
     if (node->GetParent())
@@ -875,6 +1183,8 @@ namespace creaImageIO
     //    std::cout << "DELETE OK"<<std::endl;
     return true;
   }
+  
+  
   //========================================================================
 
   //=====================================================================
@@ -897,6 +1207,7 @@ namespace creaImageIO
 	DBRecursiveRemoveNode((*i));
       }
   }
+  
   //=====================================================================
   
   //=====================================================================
@@ -921,57 +1232,8 @@ namespace creaImageIO
   }
   //=====================================================================
  
-  //========================================================================
-  std::string& format_sql2(std::string& str)
-  {
-    // quote must be doubled
-    //    crea::Utils::Replace( str, "'", "''" );
-    boost::algorithm::replace_all(str,"'","''");
-    // Found strange strings which contained NULL char INSIDE string 
-    int i,size=str.size();
-    for (i=0;i<size;++i) 
-      {
-	if (str[i]==0) 
-	  {
-	    str = str.substr(0,i);
-	    break;
-	  }
-      }
-    //    if (i<str.size())
-    return str;
-  }
-   //========================================================================
-
-  //=====================================================================
-  void SQLiteTreeHandler::BuildSQLFieldsValues(Node* n,
-					std::string& str)
-  {
-    //    std::cout << "BuildSQLFieldsValues('"<<n->GetLabel()<<"')"<<std::endl;
-
-    std::string atts="";
-    std::string values="";
-    Node::FieldValueMapType::iterator i;
-    for (i =  n->GetFieldValueMap().begin();
-	 i != n->GetFieldValueMap().end();
-	 i++)
-      {
-	if (i->first=="ID") 
-	  {
-	    continue;
-	  }
-	//	std::cout << "("<<i->first<<","<<i->second<<")"<<std::endl;
-	atts += "'" + i->first + "'";
-	values += "'" + format_sql2(i->second) + "'"; 
-	atts += ",";
-	values += ",";
-      }
-    atts[atts.size()-1]=' ';
-    values[values.size()-1]=' ';
-
-    str = "("+atts+") VALUES ("+values+")";
+ 
 
-  }
-  //=====================================================================
 */