CalibrationInterface.h

Go to the documentation of this file.

#ifndef CALIBRATION_INTERFACE_H
#define CALIBRATION_INTERFACE_H
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include <vector>
#include <utility>
#include <iostream>

template <class CategoryT,class CalibDataT> class CalibrationInterface
{
  public:

    CalibrationInterface();
    ~CalibrationInterface();
    
    const CalibDataT* getCalibData(const typename CategoryT::Input & calibrationInput) const 
     {
      return(getCalibData(getIndex(calibrationInput)));
     }

    CalibDataT* getCalibData(const typename CategoryT::Input & calibrationInput) 
     {
      return(getCalibData(getIndex(calibrationInput)));
     }

    const CalibDataT* getCalibData(int index) const ; 
    CalibDataT* getCalibData(int index);

    const CategoryT* getCategoryDefinition(int index) const ;

    int getIndex(const typename CategoryT::Input & calibrationInput) const ; 
    
  
    int addCategoryDefinition(const CategoryT & categoryDefinition);
    int addEntry(const CategoryT & categoryDefinition,const CalibDataT & data);
    
    void setCalibData(int index,const CalibDataT & data); 
   
    const std::vector<std::pair<CategoryT, CalibDataT> > & categoriesWithData() const  {return m_categoriesWithData; }
    int size() const {return m_categoriesWithData.size(); }

  private:
    std::vector<std::pair<CategoryT, CalibDataT> > m_categoriesWithData;
};


template <class CategoryT,class CalibDataT> 
CalibrationInterface<CategoryT,CalibDataT>::CalibrationInterface() 
{
}

template <class CategoryT,class CalibDataT> 
CalibrationInterface<CategoryT,CalibDataT>::~CalibrationInterface()
{
}
    
template <class CategoryT,class CalibDataT> 
int  CalibrationInterface<CategoryT,CalibDataT>::getIndex(const typename CategoryT::Input & calibrationInput) const
{
   int i=0;
   int found=-1;
   for(typename std::vector<std::pair<CategoryT,CalibDataT> >::const_iterator it = m_categoriesWithData.begin();it!=m_categoriesWithData.end();it++)
   {
    
      if((*it).first.match(calibrationInput))
      {
        if(found >=0 ) 
          {
           edm::LogWarning("BTagCalibration") << "WARNING: OVERLAP in categories, using latest one" ;
          }
        
        found=i;
      }
   i++;
  }
  return found;
}
template <class CategoryT,class CalibDataT>
const CalibDataT * CalibrationInterface<CategoryT,CalibDataT>::getCalibData(int i) const
{
 size_t ii=i;
 if(i>=0 && ii < m_categoriesWithData.size() )
  return &m_categoriesWithData[i].second;
 else
  return 0;
}


template <class CategoryT,class CalibDataT> 
CalibDataT * CalibrationInterface<CategoryT,CalibDataT>::getCalibData(int i) 
{
 size_t ii=i;
 if(i>=0 && ii < m_categoriesWithData.size() )
  return &m_categoriesWithData[i].second;
 else
  return 0;
}

template <class CategoryT,class CalibDataT>
int CalibrationInterface<CategoryT,CalibDataT>::addCategoryDefinition(const CategoryT & categoryDefinition) 
{
 CalibDataT emptyData;
 return addEntry(categoryDefinition,emptyData);
}

template <class CategoryT,class CalibDataT>
int CalibrationInterface<CategoryT,CalibDataT>::addEntry(const CategoryT & categoryDefinition,const CalibDataT & data)
{
 std::pair<CategoryT,CalibDataT> newEntry(categoryDefinition,data);
 m_categoriesWithData.push_back(newEntry);
 return  m_categoriesWithData.size();
}


template <class CategoryT,class CalibDataT>
void CalibrationInterface<CategoryT,CalibDataT>::setCalibData(int index,const CalibDataT & data)
{
 m_categoriesWithData[index].second=data;
}
#endif