CMSSW_4_4_3_patch1/src/RecoBTag/XMLCalibration/interface/AlgorithmCalibration.h

Go to the documentation of this file.

#ifndef ALGORITHM_CALIBRATION_H
#define ALGORITHM_CALIBRATION_H

#include <map>
#include <string>
#include <vector>
#include <xercesc/dom/DOM.hpp>
#include <xercesc/dom/DOMElement.hpp>
#include <xercesc/util/XMLString.hpp>
#include <string>
#include <list>
#include <iostream>

#include "CalibrationXML.h"
//#include "CondFormats/BTagObjects/interface/CalibrationInterface.h"
//#include "RecoBTag/TrackProbability/interface/CalibrationInterface.h"
#include "RecoBTag/XMLCalibration/interface/CalibrationInterface.h"

namespace XERCES_CPP_NAMESPACE { class DOMNode; }



//template <class T,class CO> class   CalibrationInterface;
template <class T,class CO> class AlgorithmCalibration : public CalibrationInterface<T,CO>
{
// friend class CalibrationInterface<T,CO>; 
  public:
    typedef XERCES_CPP_NAMESPACE::DOMElement DOMElement;
    typedef XERCES_CPP_NAMESPACE::DOMNode DOMNode;

    AlgorithmCalibration(const std::string & fileName);
    
    ~AlgorithmCalibration();
    
    void startCalibration();

    void updateCalibration(const typename T::Input & calibrationInput);
    template <class CI> void updateCalibration(const typename T::Input & calibrationInputForCategory,const CI & inputForCalibration);

    void saveCalibration(const std::string & fileName);

 protected:

    CO* readObject(DOMNode *);
    bool readCategories();
    
 protected:

    DOMElement * dom() 
    {
      if(m_xml == 0)
       {
        m_xml=new CalibrationXML();
        m_xml->openFile(m_filename);
        
       } 
    return m_xml->calibrationDOM();
    }

  private:
    std::string m_filename;
    CalibrationXML *m_xml;
};


template <class T,class CO> 
AlgorithmCalibration<T,CO>::AlgorithmCalibration(const std::string & filename) : m_filename(filename), m_xml(0)
{
 readCategories();
 if(m_xml) {
  m_xml->closeFile();
 }
}

template <class T,class CO> 
AlgorithmCalibration<T,CO>::~AlgorithmCalibration()
{
   if(m_xml) delete m_xml;
}
    
template <class T,class CO> 
bool AlgorithmCalibration<T,CO>::readCategories()
{
   if(dom()==0) return false;
   
   DOMNode* n1 = dom()->getFirstChild();
   while(n1)
    {
      if (n1->getNodeType() == DOMNode::ELEMENT_NODE   )
      {
          T *cat = new T();
          cat->readFromDOM((DOMElement *)n1);
          CO * obj =readObject(n1->getFirstChild());
          if(obj) 
           {
                this->addEntry(*cat,*obj);  
                delete obj;  
           }
          delete cat;
      }
      n1 = n1->getNextSibling();
    }
    
 return true;
}
template <class T,class CO>
CO * AlgorithmCalibration<T,CO>::readObject(DOMNode * dom)
{
  DOMNode* n1 = dom;
  while(n1)
    {
      if (n1->getNodeType() == DOMNode::ELEMENT_NODE   )
           break;
      n1 = n1->getNextSibling();
    }
     
  if(n1==0) return 0; //Cannot find any calibrated objects
  
  CO * co = new CO();
  co->read((DOMElement *)n1);
  return co;
}

#endif