CalibrationXML.cc

Go to the documentation of this file.

#include "RecoBTag/XMLCalibration/interface/CalibrationXML.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <xercesc/framework/LocalFileFormatTarget.hpp>
#include <xercesc/parsers/XercesDOMParser.hpp>
#include <xercesc/dom/DOM.hpp>
#include <xercesc/sax/HandlerBase.hpp>
#include <xercesc/util/XMLString.hpp>
#include "FWCore/Concurrency/interface/Xerces.h"
#include <iostream>
#include <fstream>
#include <string>

using namespace std;


XERCES_CPP_NAMESPACE_USE

CalibrationXML::CalibrationXML() : errHandler(nullptr), parser(nullptr)
{

}

CalibrationXML::~CalibrationXML()
{
//TODO: delete!!!!  
if(errHandler) delete errHandler;
if(parser)  { 
              delete parser;
              cms::concurrency::xercesTerminate();
            }
}

void CalibrationXML::openFile(const std::string & xmlFileName) 
{
if(errHandler) delete errHandler;
if(parser) { delete parser; cms::concurrency::xercesTerminate(); }

 m_xmlFileName = xmlFileName;
// std::cout << "Opening.." << std::endl;
    // Initialize the XML4C2 system
    try
        {
            cms::concurrency::xercesInitialize();
        }
    catch(const XMLException& toCatch)
    {
        std::cerr << "Error during Xerces-c Initialization.\n"
             << "  Exception message:"
             << XMLString::transcode(toCatch.getMessage()) << std::endl;
   abort();
//FIXME     throw GenTerminate("Error during Xerces-c Initialization.");
    }
    parser = new XercesDOMParser;
    parser->setValidationScheme(XercesDOMParser::Val_Auto);
    parser->setDoNamespaces(false);
    parser->setDoSchema(false);
    parser->setValidationSchemaFullChecking(false);
    errHandler = new HandlerBase;
    parser->setErrorHandler(errHandler);
    parser->setCreateEntityReferenceNodes(false);
    //  Parse the XML file, catching any XML exceptions that might propogate out of it.
    bool errorsOccured = false;
    try
    {
          edm::LogInfo("XMLCalibration") << "Calibration XML: parsing " << m_xmlFileName.c_str() << std::endl;
        parser->parse(m_xmlFileName.c_str());
        int errorCount = parser->getErrorCount();
        if (errorCount > 0) errorsOccured = true;
    }
    catch (const XMLException& e)
    {
        std::cerr << "A DOM error occured during parsing\n   DOMException code: "
             << (long unsigned int)e.getCode() << std::endl;
        errorsOccured = true;
    }
    // If the parse was successful, build the structure we want to have
    if(errorsOccured) { 
        std::cerr << "An error occured during parsing\n"
             << "Please check your input with SAXCount or a similar tool.\n Exiting!\n" << std::endl; 
abort();
//FIXME     throw GenTerminate("An error occured during parsing\n Please check your input with SAXCount or a similar tool.\n Exiting!\n");
    }

    doc = parser->getDocument();
    DOMNode* n1 = doc->getFirstChild();

        while(n1)
        {
                if (n1->getNodeType() == DOMNode::ELEMENT_NODE   ) break;
                n1 = n1->getNextSibling();
        }
    
    if(n1 == nullptr || strcmp("Calibration",XMLString::transcode(n1->getNodeName())))
abort();
//FIXME     throw GenTerminate("The root element in the XML Calibration file is not a Calibration element.\n This should be forbidden at the DTD level.");
    else {   edm::LogInfo("XMLCalibration")  << "Calibration found" ; } 

    m_calibrationDOM = (DOMElement *) n1;
    


}

void CalibrationXML::saveFile(const std::string & xmlFileName)
{
    DOMImplementation * theImpl = DOMImplementationRegistry::getDOMImplementation(XMLString::transcode("Core"));
    DOMLSSerializer *   theSerializer = ((DOMImplementation*)theImpl)->createLSSerializer();
    DOMConfiguration* dc = theSerializer->getDomConfig();
    dc->setParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true);

    XMLFormatTarget* myFormTarget = new LocalFileFormatTarget(XMLString::transcode(xmlFileName.c_str()));
    DOMLSOutput* outputDesc = ((DOMImplementationLS*)theImpl)->createLSOutput();
    outputDesc->setByteStream(myFormTarget);

    theSerializer->write(doc, outputDesc);
     delete myFormTarget;     
}

DOMElement * CalibrationXML::addChild(DOMNode *dom,const std::string & name)
{ 
      DOMNode *n1 = dom;
      int level=0;
      std::string indent="\n";
      while(n1 && level < 100)
      {
       level++;
       indent+="  ";
       n1 = n1->getParentNode();
      }
      assert(dom); 
      if(dom->getFirstChild()==nullptr)
             dom->appendChild(dom->getOwnerDocument()->createTextNode(XMLString::transcode(indent.c_str()))); 
         
      DOMElement * child = (DOMElement *)dom->appendChild(dom->getOwnerDocument()->createElement(XMLString::transcode(name.c_str()))); 
          dom->appendChild(dom->getOwnerDocument()->createTextNode(XMLString::transcode(indent.c_str())));           
      return child;
}