//#include <icons/close.xpm>
#include <creaWx.h>
-#include <creaMessageManager.h>
using namespace crea;
#include <boost/filesystem.hpp>
: mFileName(filename)
{
mDB = new CppSQLite3DB;
+ mIsAdding=false;
GimmickMessage(1,"SQLite version : "
<<std::string(mDB->SQLiteVersion())<< std::endl);
}
{
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);
}
//=====================================================================
}
//=====================================================================
-
//=====================================================================
int SQLiteTreeHandler::AddBranch( const AttributeMapType& attr )
{
tree::Node* parent = DBGetParent(attr);
- for (int level = parent->GetLevel()+1;
- level < GetTree().GetNumberOfLevels();
- level++)
- {
- // Create Node
- parent = new tree::Node(parent,attr);
-
- // Insert into DB
- if (node->GetType()!=Node::Patient)
- node->SetFieldValue("PARENT_ID",parent_id);
-
- // Insert in DB
- std::string val;
- BuildSQLFieldsValues(node,val);
- std::string insert("INSERT INTO ");
- insert += std::string(SQLiteTreeHandlerStructure::Table(node->GetType()))
- + " " + val + ";";
- // std::cout << "** SQL = '"<<insert<<"'"<<std::endl;
- UPDATEDB(insert);
- // std::cout << "** SQL OK"<<std::endl;
-
- // Store DB id of newly created node;
- long lastrow = mDB->lastRowId();
- std::stringstream ri;
- ri << mDB->lastRowId();
- node_id = ri.str();
- // std::cout << "LastRowId='"<<mDB->lastRowId()<<"' vs '"<<created_id<<"'"<<std::endl;
-
- node->SetFieldValue("ID",node_id);
- /*
- // 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++;
- */
- }
- return -1;
+ 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;
+ bool remove=false;
+ tree::Node* parent=node->GetParent();
+ if (parent)
+ {
+ int nC = parent->RemoveChildrenFromList(node);
+ if(nC>0)
+ {
+ std::stringstream out;
+ out <<nC;
+ SetAttribute(parent,"NumberOfChildren",out.str());
+ }
+ else
+ {
+ remove=true;
+ }
+
+ }
+ delete node;
+ if(remove&&parent->GetLevel()>0)
+ {
+ Remove(parent);
+ }
+ // 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);
+ }
+ //=====================================================================
+ //=====================================================================
+ /// Sets an attribute
+ void SQLiteTreeHandler::SetAttribute(const std::string& levelDescriptor,
+ const std::string& key,
+ const std::string& value,
+ const std::string& searchParam,
+ const std::string& searchVal)
+ {
+ DBSetAttribute(levelDescriptor,key,value,searchParam, searchVal);
+ }
+ //=====================================================================
+ /// Deletes a tuple
+ void SQLiteTreeHandler::DeleteTuple(std::string levelDescriptor,
+ std::string key, std::string value)
+ {
+ DBDelete(levelDescriptor,key,value);
+ }
+ //=====================================================================
#define QUERYDB(QUER,RES) \
try \
{ \
- GimmickMessage(2,"SQL: '"<<QUER<<"'"<<std::endl); \
+ GimmickMessage(2,"SQL query: '"<<QUER<<"'"<<std::endl); \
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() ); \
+ GimmickError("SQLite query '"<<QUER<<"' : " \
+ << e.errorCode() << ":" \
+ << e.errorMessage() ); \
} \
//=====================================================================
#define UPDATEDB(UP) \
try \
{ \
- GimmickMessage(2,"SQL: '"<<UP<<"'"<<std::endl); \
+ GimmickMessage(2,"SQL update: '"<<UP<<"'"<<std::endl); \
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() ); \
+ GimmickError("SQLite update '"<<UP<<"' Error : " \
+ << e.errorCode() << ":" \
+ << e.errorMessage() ); \
}
//=====================================================================
}
catch (CppSQLite3Exception& e)
{
- std::cerr << "Opening '"<<GetFileName()<<"' : "
- << e.errorCode() << ":"
- << e.errorMessage() << std::endl;
+ GimmickError("Opening '"<<GetFileName()<<"' : "
+ << e.errorCode() << ":"
+ << e.errorMessage());
return false;
}
// IMPORT TREE DESCRIPTION (AND TEST DB VALIDITY)
return false;
}
- GimmickMessage(1,"Opening SQLite database '"<<GetFileName()
+ GimmickDebugMessage(1,"Opening SQLite database '"<<GetFileName()
<<"' ... OK"<<std::endl);
return true;
}
if (boost::filesystem::exists(GetFileName()))
{
- creaMessage("Gimmick!",1,
- "[Gimmick!] !! ERROR '"<<GetFileName()<<"' : "
- << "file already exists"<<std::endl);
+ GimmickError(GetFileName()<<"' : "
+ << "file already exists");
return false;
}
}
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;
- // Create LEVELS table
- command = "create table LEVELS\n";
- command += "( Name text )\n";
+ command = "INSERT INTO LEVELS (Name) VALUES ('";
+ command += GetTree().GetLevelDescriptor(l).GetName();
+ command += "')";
UPDATEDB(command);
-
- // Iterate the Levels
- for (int l=0; l<GetTree().GetNumberOfLevels(); ++l)
+
+ // Create table of level (for level>0, i.e. not Root)
+ if (l>=0)
{
- command = "INSERT INTO LEVELS (Name) VALUES ('";
+ command = "CREATE TABLE ";
command += GetTree().GetLevelDescriptor(l).GetName();
- command += "')";
+ 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);
- // Create table of level (for level>0, i.e. not Root)
- if (l>0)
+
+ // Add Attribute 'ID' to Description
+ GetTree().GetDescriptor().Add
+ (AttributeDescriptor( "ID",
+ "Database Identifier",
+ 0,0,
+ AttributeDescriptor::PRIVATE
+ ),l);
+
+ if (l>1)
{
- command = "CREATE TABLE ";
- command += GetTree().GetLevelDescriptor(l).GetName();
- command += "\n(\nID INTEGER PRIMARY KEY";
- if (l>1)
- {
- command += ",\nPARENT_ID int not null";
- }
- AppendAttributesSQLDefinition(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().GetAttributeDescriptorList(l).push_back
- (AttributeDescriptor( "ID",
- "Database Identifier",
+ // Add Attribute 'PARENT_ID' to Description
+ GetTree().GetDescriptor().Add
+ (AttributeDescriptor( "PARENT_ID",
+ "Database Parent Identifier",
0,0,
AttributeDescriptor::PRIVATE
- ));
-
- if (l>1)
- {
- // Add Attribute 'PARENT_ID' to Description
- GetTree().GetAttributeDescriptorList(l).push_back
- (AttributeDescriptor( "PARENT_ID",
- "Database Parent Identifier",
- 0,0,
- AttributeDescriptor::PRIVATE
- ));
- }
-
+ ),l);
}
- // Create table *_ATTRIBUTES
+ }
+
+ // Create table *_ATTRIBUTES
+
+ 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);
+
+
+ // Fill the table *_ATTRIBUTES
+ LevelDescriptor::AttributeDescriptorListType::const_iterator i;
+ for (i = GetTree().GetAttributeDescriptorList(l).begin();
+ i != GetTree().GetAttributeDescriptorList(l).end();
+ ++i)
+ {
- 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);
-
-
- // 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());
- }
+ 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());
}
- }
- catch (std::exception)
- {
- return false;
- }
+ } // 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);
//=====================================================================
//=====================================================================
- void SQLiteTreeHandler::AppendAttributesSQLDefinition(int level,
+ void SQLiteTreeHandler::SQLAppendAttributesDefinition(int level,
std::string& s)
{
LevelDescriptor::AttributeDescriptorListType::const_iterator i;
tree::Descriptor& desc = GetTree().GetDescriptor();
// clears the existing one
- desc.GetLevelDescriptorList().clear();
+ desc.Clear();
int nblevel = 0;
std::string query = "SELECT * FROM LEVELS";
{
std::string name = q.getStringField(0);
GimmickMessage(2," * Importing level '"<<name<<"'"<<std::endl);
- desc.GetLevelDescriptorList().push_back(LevelDescriptor(name));
+ desc.Add(LevelDescriptor(name));
nblevel++;
q.nextRow();
}
std::string name(q.getStringField(1));
GimmickMessage(2," - Importing attribute '"<<key<<"' '"<<name
<<"'"<<std::endl);
- desc.GetAttributeDescriptorList(level).push_back
+ 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;
}
}
+
+ // 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;
+
+
+ //========================================================================
+ ///
+ 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)
+ {
+ if (str[i]==0)
+ {
+ str = str.substr(0,i);
+ break;
+ }
+ }
+ // 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")
+ {
+ 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);
+ }
+ atts[atts.size()-1]=' ';
+ values[values.size()-1]=' ';
+
+ str = "("+atts+") VALUES ("+values+")";
+ GimmickMessage(4,"Result = '"<<str<<"'"<<std::endl);
+ }
+ //=====================================================================
+
//=====================================================================
tree::Node* SQLiteTreeHandler::DBGetParent( const AttributeMapType& attr)
{
{
go_down = true;
parent = *i;
+ break;
}
}
}
//=====================================================================
- int SQLiteTreeHandler::DBLoadChildren(tree::Node* 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 (parent->GetChildrenLoaded()) return nbloaded;
+ // 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);
+ }
+ node->SetChildrenLoaded(true);
+ return nbloaded;
+
+ }
+ else
+ {
+ /// If children not loaded : do it and recurse
// Query DB
- int level = parent->GetLevel();
+ int level = node->GetLevel();
std::string query = "SELECT * FROM ";
+
query += GetTree().GetLevelDescriptor(level+1).GetName();
- if (level>1)
+ if (level>0)
{
- query += " WHERE PARENT_ID='" + parent->UnsafeGetAttribute("ID")
+ query += " WHERE PARENT_ID='" + node->UnsafeGetAttribute("ID")
+ "'";
}
CppSQLite3Query q;
QUERYDB(query,q);
+ int p=0;
while (!q.eof())
{
+
+ // std::cout << "DBLoadCh : creating node level "<<level+1<<std::endl;
+
nbloaded++;
- Node* n = new Node(parent);
+ Node* n = new Node(node);
for (int fld = 0; fld < q.numFields(); fld++)
{
n->UnsafeSetAttribute(q.fieldName(fld),q.getStringField(fld));
mTypeIdToNodeMap[ti] = n;
*/
// recurse
- if ( numberoflevels > 1 )
+ if ( numberoflevels != 1 )
{
// msw[2].Pause();
nbloaded += DBLoadChildren(n, numberoflevels-1);
q.nextRow();
}
- parent->SetChildrenLoaded(true);
+ node->SetChildrenLoaded(true);
+
// msw[2].Pause();
return nbloaded;
+ }
+ }
+ //=====================================================================
- // 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() )
- {
- // std::cout << "--> Children already loaded"<<std::endl;
- return nbloaded;
- }
- if ( xparent->GetType() == Node::Image )
- {
- return nbloaded;
- }
- if ( xparent->GetType() >= maxlevel )
- {
- return nbloaded;
- }
-
- // msw[2].Pause();
- // msw[2].Resume();
+ //======================================================================
+ 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 + ";";
- Node::Type type = xparent->GetType()+1;
+ UPDATEDB(insert);
+
+ // Store DB id of newly created node;
+ long lastrow = mDB->lastRowId();
+ std::stringstream ri;
+ ri << mDB->lastRowId();
+ n->SetAttribute("ID",ri.str());
+ }
+ //======================================================================
- // Query DB
+ //======================================================================
+ /// Graft the branch defined by the attributes to the parent
+ void SQLiteTreeHandler::DBGraftToParent( tree::Node* parent,
+ const AttributeMapType& attr)
+ {
+ // std::cout <<"Grafting to parent '"<<parent->GetLabel()
+ // <<"'"<<std::endl;
- std::string query = "SELECT * FROM ";
- query += GetTree().GetDescriptor().GetLevelDescriptor(level).GetName();
- //SQLiteTreeHandlerStructure::Table(type);
- if (parent!=0)
+ for (int level = parent->GetLevel()+1;
+ level < GetTree().GetNumberOfLevels();
+ level++)
{
- query += " WHERE PARENT_ID='" + parent->GetFieldValue("ID") + "'";
+ // Create Node
+ tree::Node* child = new tree::Node(parent,attr);
+ child->SetChildrenLoaded(true);
+ if (level>1)
+ {
+ int nc = GetNumberOfChildren(parent)+1;
+
+ // std::cout<<"Number of children "<<parent->GetNumberOfChildren()<<std::endl;
+ std::stringstream out;
+ out << nc;
+ SetAttribute(parent,"NumberOfChildren",out.str());
+ }
+
+ // 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++;
+ */
}
+ }
+ //======================================================================
- // std::cout << "** SQL = '"<<query<<"'"<<std::endl;
- CppSQLite3Query q;
- QUERYDB(query,q);
+ //=====================================================================
+ /// 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 +="'";
+ // sql += " LIMIT 1";
+ UPDATEDB(sql);
+ }
- while (!q.eof())
- {
- nbloaded++;
- Node* n = new Node(type,
- this,xparent);
- for (int fld = 0; fld < q.numFields(); fld++)
+ //=====================================================================
+ /// Sets an attribute of a Node
+ void SQLiteTreeHandler::DBSetAttribute(const std::string& levelDescriptor,
+ const std::string& key,
+ const std::string& value,
+ const std::string& searchParam,
+ const std::string& searchVal)
+ {
+
+ std::string sql = "UPDATE ";
+ sql += levelDescriptor;
+ sql += " SET ";
+ sql += key;
+ sql += " = '";
+ sql += value;
+ sql += "' WHERE ";
+ sql += searchParam;
+ sql += " = '";
+ sql += searchVal;
+ sql += "'";
+ 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);
+
+
+ if(node->GetNumberOfChildren()!=0)
+ {
+ Node::ChildrenListType::iterator i;
+ for (i = node->GetChildrenList().begin();
+ i != node->GetChildrenList().end();
+ i++)
+ {
+ DBRecursiveRemoveNode((*i));
+ }
+ }
+ else if(node->GetLevel()<GetTree().GetNumberOfLevels()-1)
+ {
+ DBRecursiveRemoveNode(node->GetLevel()+1,node->GetAttribute("ID"));
+ }
+ }
+
+ //=====================================================================
+ void SQLiteTreeHandler::DBRecursiveRemoveNode(int level, std::string parentId)
+ {
+ std::stringstream out;
+ std::stringstream result;
+ out<<"SELECT ID FROM "<<GetTree().GetLevelDescriptor(level).GetName()<<" WHERE PARENT_ID='"<<parentId<<"'";
+
+ CppSQLite3Query q;
+ QUERYDB(out.str(),q);
+
+ while (!q.eof())
{
- n->SetFieldValue(q.fieldName(fld),q.getStringField(fld));
+ for (int fld = 0; fld < q.numFields(); fld++)
+ {
+ result<<q.getStringField(fld)<<"#";
+ }
+ q.nextRow();
}
- // Index
- TypeId ti;
- ti.type = type;
- ti.id = n->GetFieldValue("ID");
- mTypeIdToNodeMap[ti] = n;
- // recurse
- if ( type < maxlevel )
+ std::string res=result.str();
+ size_t ini=0;
+ size_t fin=0;
+ while(fin<res.size()-1)
+ {
+ fin=res.find('#',ini);
+ DBDelete(GetTree().GetLevelDescriptor(level).GetName(),"ID",res.substr(ini,fin-ini));
+ if(level<GetTree().GetNumberOfLevels()-1)
{
- msw[2].Pause();
- nbloaded += DBLoadChildren(n,maxlevel);
- msw[2].Resume();
+ DBRecursiveRemoveNode(level+1,res.substr(ini,fin-ini));
+ }
+ ini=fin+1;
}
- // next entry in db
- q.nextRow();
- }
-
- xparent->SetChildrenLoaded(true);
+
- // msw[2].Pause();
- return nbloaded;
- */
}
+
//=====================================================================
+ void SQLiteTreeHandler::DBDelete(std::string levelDescriptor, std::string key, std::string value)
+ {
+
+ std::stringstream query;
+ query<<"DELETE FROM "<<levelDescriptor<<" WHERE "<<key<<"='"<<value<<"';";
+
+ UPDATEDB(query.str());
+ GimmickDebugMessage(2," Deleting: Query: "<<query.str()<<std::endl);
+ }
+
+
+ //=====================================================================
+ void SQLiteTreeHandler::GetAttribute(std::string levelDescriptor,
+ std::string searchParam,
+ std::string searchVal,
+ std::string key,
+ std::string& result)
+ {
+ std::stringstream out;
+ std::stringstream results;
+ out<<"SELECT "<<key<<" FROM "<<levelDescriptor;
+ if(searchParam!="")
+ {
+ out<<" WHERE "<<searchParam<<"='"<<searchVal<<"'";
+ }
+
+ CppSQLite3Query q;
+ QUERYDB(out.str(),q);
+
+
+ while (!q.eof())
+ {
+ for (int fld = 0; fld < q.numFields(); fld++)
+ {
+ results<<q.getStringField(fld);
+ if(searchParam=="")
+ {
+ results<<"#";
+ }
+ }
+ q.nextRow();
+ }
+ result=results.str();
+
+ }
+ //=====================================================================
+ unsigned int SQLiteTreeHandler::GetNumberOfChildren(tree::Node* n)
+ {
+ // Query DB
+ int nb=0;
+ int level = n->GetLevel();
+ if(level<GetTree().GetNumberOfLevels()&& level>0)
+ {
+ std::string query = "SELECT NumberOfChildren FROM ";
+ query += GetTree().GetLevelDescriptor(level).GetName();
+ if (level>0)
+ {
+ query += " WHERE ID='" + n->UnsafeGetAttribute("ID")
+ + "'";
+ }
+ CppSQLite3Query q;
+ QUERYDB(query,q);
+
+
+ while (!q.eof())
+ {
+ for (int fld = 0; fld < q.numFields(); fld++)
+ {
+ nb=q.getIntField(fld);
+ }
+ q.nextRow();
+ }
+ }
+ /*
+ if(nb==0)
+ {
+ nb=1;
+ }
+ */
+ return nb;
+ }
+ //=====================================================================
+ void SQLiteTreeHandler::GetTopLevelNodeId(const std::string& searchParam, const std::string& searchValue, std::string& parent_id)
+ {
+ int level=GetTree().GetNumberOfLevels()-1;
+ std::string sp=searchParam.c_str();
+ std::string sv=searchValue.c_str();
+ while(level>1)
+ {
+ std::stringstream out;
+ std::stringstream results;
+ out<<"SELECT PARENT_ID FROM "<<GetTree().GetLevelDescriptor(level).GetName();
+ out<<" WHERE "<<sp<<"='"<<sv<<"'";
+ CppSQLite3Query q;
+ QUERYDB(out.str(),q);
+
+
+ while (!q.eof())
+ {
+ for (int fld = 0; fld < q.numFields(); fld++)
+ {
+ results<<q.getStringField(fld);
+ }
+ q.nextRow();
+ }
+ level=level-1;
+ sp="ID";
+ sv=results.str();
+ }
+ parent_id=sv;
+ }
/*
//=====================================================================
// 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())
// std::cout << "DELETE OK"<<std::endl;
return true;
}
+
+
//========================================================================
//=====================================================================
DBRecursiveRemoveNode((*i));
}
}
+
//=====================================================================
//=====================================================================
}
//=====================================================================
- //========================================================================
- 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+")";
+
- }
- //=====================================================================
*/