/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/MuonEfficiencyCorrections/MuonEfficiencyCorrections/EffiCollection.h

/*
 * EffiCollection.h
 *
 *  Created on: Jun 21, 2014
 *      Author: goblirsc
 */

#ifndef EFFICOLLECTION_H_
#define EFFICOLLECTION_H_

// @class EffiCollection

#include <MuonEfficiencyCorrections/EfficiencyScaleFactor.h>
#include <MuonEfficiencyCorrections/MuonEfficiencyType.h>

#include "PATInterfaces/ISystematicsTool.h"
#include "xAODMuon/Muon.h"
#include <boost/unordered_map.hpp>
#include <map>
#include <string>
#include <iostream>

namespace CP {
    class EffiCollection {
        public:
            EffiCollection();
            EffiCollection(const std::string &file_central, const std::string &file_calo, const std::string &file_forward, const std::string &file_lowpt_central, const std::string &file_lowpt_caloo, SystematicSet sys, CP::MuonEfficiencyType effType, double lowPtTransition = 20000.);
            //Constructor with nominal as fallback..
            EffiCollection(const EffiCollection* Nominal, const std::string &file_central, const std::string &file_calo, const std::string &file_forward, const std::string &file_lowpt_central, const std::string &file_lowpt_caloo, SystematicSet sys, CP::MuonEfficiencyType effType, double lowPtTransition = 20000.);

            EffiCollection(const EffiCollection & other);
            EffiCollection & operator =(const EffiCollection & other);

            EfficiencyScaleFactor* retrieveSF(const xAOD::Muon & mu, unsigned int RunNumber);
            enum CollectionType {
                Central, Calo, Forward, CentralLowPt, CaloLowPt
            };

            // return the name of the systematic variation being run
            std::string sysname() const;

            // a consistency check
            bool CheckConsistency();

            virtual ~EffiCollection();
            unsigned int nBins() const;
            bool SeySystematicBin(unsigned int Bin);
        protected:
            //Create a subclass to handle the periods
            class CollectionContainer {
                public:
                    CollectionContainer(const std::string &FileName, SystematicSet sys, CP::MuonEfficiencyType effType, EffiCollection::CollectionType FileType, bool isLowPt = false, bool hasPtDepSys = false);
                    CollectionContainer(CollectionContainer* Nominal, const std::string &FileName, SystematicSet sys, CP::MuonEfficiencyType effType, EffiCollection::CollectionType FileType, bool isLowPt = false, bool hasPtDepSys = false);

                    CollectionContainer & operator =(const CollectionContainer & other);
                    CollectionContainer(const CollectionContainer & other);
                    virtual ~CollectionContainer();
                    EfficiencyScaleFactor* retrieve(unsigned int RunNumer);
                    bool CheckConsistency();
                    std::string sysname();
                    bool SeySystematicBin(unsigned int Bin);
                protected:
                    bool LoadPeriod(unsigned int RunNumber);
                    typedef std::pair<unsigned int, unsigned int> RunRanges;
                    std::map<RunRanges, EfficiencyScaleFactor*> m_SF;
                    std::map<RunRanges, EfficiencyScaleFactor*>::const_iterator m_currentSF;
                    EffiCollection::CollectionType m_FileType;
            };
            EffiCollection::CollectionContainer* retrieveContainer(CollectionType Type) const;

        private:

            CollectionContainer* m_central_eff;
            CollectionContainer* m_calo_eff;
            CollectionContainer* m_forward_eff;
            CollectionContainer* m_lowpt_central_eff;
            CollectionContainer* m_lowpt_calo_eff;

            double m_lowpt_transition;
            CP::MuonEfficiencyType m_sfType;
    };
}
#endif /* EFFICOLLECTION_H_ */

---