/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/PMGTools/PMGTools/PMGSherpaVjetsSysTool.h

// Dear emacs, this is -*- c++ -*-
// $Id: PMGSherpaVjetsSysTool.h tripiana $
//
// Tool calculating event weights for systematic variations of Sherpa V+jets (V=Z,W,gamma) samples
//
// @author Martin Tripiana <martin.tripiana@cern.ch>
//         John Anders     <john.kenneth.anders@cern.ch>
//
// $Date: 2016-07-12 $
//

#ifndef PMGTOOLS_PMGSHERPAVJETSSYSTOOL_H
#define PMGTOOLS_PMGSHERPAVJETSSYSTOOL_H

// Infrastructure include(s):
#include "AsgTools/AsgTool.h"

// Interface include(s):
#include "PMGTools/IPMGSherpaVjetsSysTool.h"

// System include(s):
#include <map>
#include <vector>
#include <memory>

// ROOT include(s):
#include <TH2F.h>
#include <TFile.h>

namespace PMGTools {

  class PMGSherpaVjetsSysTool : public virtual IPMGSherpaVjetsSysTool,
                                public asg::AsgTool { 

    //proper constructor for athena 
    ASG_TOOL_CLASS( PMGSherpaVjetsSysTool, PMGTools::IPMGSherpaVjetsSysTool )

  public:
    PMGSherpaVjetsSysTool( const std::string& name );

    StatusCode initialize() override;

    double getWeight( const std::string& sysVariation = "nom") override;

  protected:
    void setNjets(int njets) override;
    void cacheDSID(bool do_it=true) override;

    std::vector<std::string> getSupportedVariations(SysParType ptype) override;

    unsigned int getNtruthJets() override;

  private:
    int getPtMapping(SysParType& pType);

    int m_current_njets;

    bool m_use_cache;
    SysParType m_cache_ptype;
    unsigned int m_cache_ptidx;

    std::unique_ptr< TFile > m_file;

    std::map<std::string, TH2F*> m_map_w;
    std::map<std::string, TH2F*> m_map_zll;
    std::map<std::string, TH2F*> m_map_znunu;
    std::map<std::string, TH2F*> m_map_gamma;

    std::vector<std::string> m_supported_w;
    std::vector<std::string> m_supported_zll;
    std::vector<std::string> m_supported_znunu;
    std::vector<std::string> m_supported_gamma;


    std::string m_weightsFile;

    std::string m_truthJetContainer;
    std::string m_truthParticleContainer;

  }; // class PMGSherpaVjetsSysTool

} //namespace PMGTools

#endif //> !PMGTOOLS_PMGSHERPAVJETSSYSTOOL_H

---