//=====================================================================
int SQLiteTreeHandler::AddBranch( const AttributeMapType& attr )
{
- //std::cout << "AddBranch" << std::endl;
tree::Node* parent = DBGetParent(attr);
- //std::cout << " parent = "<<parent->GetLabel()<<" level "<<parent->GetLevel()<<std::endl;
DBGraftToParent(parent,attr);
-
+ if(parent->GetLevel()<GetTree().GetNumberOfLevels()-1&&parent->GetLevel()>0)
+ {
+ int nC=GetNumberOfChildren(parent);
+ nC=nC+1;
+ std::stringstream out;
+ out <<nC;
+ SetAttribute(parent,"NumberOfChildren",out.str());
+ }
return (parent->GetLevel()+1);
}
DBRecursiveRemoveNode(node);
// std::cout << "DELETE"<<std::endl;
- if (node->GetParent())
+ bool remove=false;
+ tree::Node* parent=node->GetParent();
+ if (parent)
{
- node->GetParent()->RemoveChildrenFromList(node);
+ parent->RemoveChildrenFromList(node);
+ int nC=GetNumberOfChildren(parent);
+ nC=nC-1;
+ 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;
}
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()||mIsAdding)
+ if (node->GetChildrenLoaded())
{
// Iterate the children
{
// Create Node
tree::Node* child = new tree::Node(parent,attr);
+ GetNumberOfChildren(parent);
child->SetChildrenLoaded(true);
// std::cout<<"Number of children "<<parent->GetNumberOfChildren()<<std::endl;
+
// Set PARENT_ID if necessary
if ( parent->GetLevel()>0 )
child->SetAttribute("PARENT_ID",parent->GetAttribute("ID"));
// Query DB
int nb=0;
int level = n->GetLevel();
- if(GetTree().GetNumberOfLevels()!=(level+1))
+
+ if(level<GetTree().GetNumberOfLevels()&& level>0)
{
- std::string query = "SELECT * FROM ";
- query += GetTree().GetLevelDescriptor(level+1).GetName();
+ std::string query = "SELECT NumberOfChildren FROM ";
+ query += GetTree().GetLevelDescriptor(level).GetName();
if (level>0)
{
- query += " WHERE PARENT_ID='" + n->UnsafeGetAttribute("ID")
+ query += " WHERE ID='" + n->UnsafeGetAttribute("ID")
+ "'";
}
-
CppSQLite3Query q;
QUERYDB(query,q);
while (!q.eof())
{
- nb++;
+ for (int fld = 0; fld < q.numFields(); fld++)
+ {
+ nb=q.getIntField(fld);
+ }
q.nextRow();
}
}
+ if(nb==0){nb=1;}
return nb;
}