MpdXMLNode.cxx

Go to the documentation of this file.

/*
 * MpdXMLNode.cxx
 *
 *  Created on: 01-10-2013
 *      Author: Daniel Wielanek
 *    E-mail: daniel.wielanek@gmail.com
 *    Warsaw University of Technology, Faculty of Physics
 */
 
#include "MpdXMLNode.h"
#include "TDOMParser.h"
#include "TXMLEngine.h"
#include <iostream>
 
 
MpdXMLNode::MpdXMLNode(TString name,TString value) : TNamed(name, value){
  fChildren.SetOwner(kTRUE);
  fAttrib.SetOwner(kTRUE);
}
 
MpdXMLNode::MpdXMLNode(const MpdXMLNode& other):
        MpdXMLNode(other.GetName(),other.GetValue()){
  for(int i=0;i<other.fChildren.GetEntries();i++){
        fChildren.Add(new MpdXMLNode(*other.GetChild(i)));
  }
  for(int i=0;i<other.fAttrib.GetEntries();i++){
        fAttrib.Add(new MpdXMLAttrib(*other.GetAttrib(i)));
  }
}
 
MpdXMLNode& MpdXMLNode::operator=(const MpdXMLNode& other) {
  if(&other == this)
     return *this;
  SetName(other.GetName());
  SetValue(other.GetValue());
  fChildren.Clear();
  fAttrib.Clear();
  for(int i=0;i<other.fChildren.GetEntries();i++){
        fChildren.Add(new MpdXMLNode(*other.GetChild(i)));
  }
  for(int i=0;i<other.fAttrib.GetEntries();i++){
        fAttrib.Add(new MpdXMLAttrib(*other.GetAttrib(i)));
  }
  return *this;
}
 
void MpdXMLNode::Copy(TXMLNode* node){
  fChildren.Clear();
  fAttrib.Clear();
  SetName(node->GetNodeName());
  SetTitle(node->GetText());
  if(node->HasChildren()){
    TXMLNode *child = node->GetChildren();
    do{
      if(child==NULL) break;
      TString name = child->GetNodeName();
      if(name!="text"){//skip "text" nodes
                MpdXMLNode *tempnode = new MpdXMLNode();
        tempnode->Copy(child);
        fChildren.Add(tempnode);
      }
      if(child->HasNextNode())
        child=child->GetNextNode();
    }while(child->HasNextNode());
  }
  if(node->HasAttributes()){
    TList *atr_list = node->GetAttributes();
    for(int i=0;i<atr_list->GetEntries();i++){
      TXMLAttr *atrib = (TXMLAttr*)atr_list->At(i);
            fAttrib.Add(new MpdXMLAttrib(atrib->GetName(),atrib->GetValue()));
    }
  }
}
 
void MpdXMLNode::AddAttrib(MpdXMLAttrib* attrib) {
  TString new_atr = attrib->GetName();
  if(GetAttrib(new_atr)!=nullptr){
        std::cout<<"MpdXMLNode::AddAttrib Can't have two attributes with the same name!"<<std::endl;
    return;
  }
  fAttrib.AddLast(attrib);
}
 
Int_t MpdXMLNode::GetNChildren(TString name) const{
  Int_t counter =0;
  for(int i=0;i<GetNChildren();i++){
    TString name_temp = GetChild(i)->GetName();
    if(name_temp==name){
      counter++;
    }
  }
  return counter;
}
 
MpdXMLNode* MpdXMLNode::GetChild(TString name, Int_t count) const{
  Int_t control_index = 0;
  for(int i=0;i<fChildren.GetEntries();i++){
        MpdXMLNode *node = GetChild(i);
    TString temp = node->GetName();
    if(temp==name){
      control_index++;
    }
    if(control_index>count)
      return node;
  }
  return nullptr;
}
 
MpdXMLAttrib* MpdXMLNode::GetAttrib(TString name) const{
    return static_cast<MpdXMLAttrib*>(fAttrib.FindObject(name));
}
 
MpdXMLNode* MpdXMLNode::GetChild(Int_t index) const{
    return static_cast<MpdXMLNode*>(fChildren.At(index));
}
 
MpdXMLAttrib* MpdXMLNode::GetAttrib(Int_t index) const{
    return static_cast<MpdXMLAttrib*>(fAttrib.At(index));
}
 
MpdXMLNode::~MpdXMLNode() {
}
 
//---------- MpdXMLFile ------------------------------------------------------------------------------------
 
MpdXMLFile::MpdXMLFile(TString name, TString mode) :fName(name){
  if(mode=="read"||mode=="READ"){
    fOverwrite  = kFALSE;
    TDOMParser Parser;
    Parser.SetValidate(kFALSE);
    Parser.ParseFile(name);
    TXMLNode *MainNode = Parser.GetXMLDocument()->GetRootNode();
        fRootNode.reset(new MpdXMLNode());
    fRootNode->Copy(MainNode);
  }else{
    fOverwrite  = kTRUE;
  }
}
 
void MpdXMLFile::CreateRootNode(TString name) {
    fRootNode.reset(new MpdXMLNode(name));
}
 
void MpdXMLFile::SetRootNode(MpdXMLNode *node) {
  fRootNode.reset(node);
}
 
void MpdXMLFile::Close() {
  if(fOverwrite){
    if(!fRootNode){
            std::cout<<"MpdXMLFile::Close() No root node!"<<std::endl;
      return;
    }
    TXMLEngine engine;
    XMLNodePointer_t mainnode = engine.NewChild(0, 0, fRootNode->GetName());
    ExportNode(mainnode,engine,*fRootNode.get());
    XMLDocPointer_t xmldoc =  engine.NewDoc();
    engine.DocSetRootElement(xmldoc,mainnode);
    engine.SaveDoc(xmldoc,fName);
    engine.FreeDoc(xmldoc);
    fRootNode.reset(nullptr);
  }
}
 
void MpdXMLFile::ExportNode(XMLNodePointer_t& nodePointer,
        TXMLEngine &engine,const MpdXMLNode &node) const {
  for(int i=0;i<node.GetNChildren();i++){
    XMLNodePointer_t child = engine.NewChild(nodePointer,0,node.GetChild(i)->GetName(),node.GetChild(i)->GetValue());
    for(int j=0;j<node.GetChild(i)->GetNAttributes();j++){
      engine.NewAttr(child,0,node.GetChild(i)->GetAttrib(j)->GetName(),node.GetChild(i)->GetAttrib(j)->GetValue());
    }
    ExportNode(child,engine,*node.GetChild(i));
  }
}
 
MpdXMLFile::~MpdXMLFile() {
  if(fRootNode&&fOverwrite)
    Close();
}