PMGSherpaVjetsSysTool.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $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 "PMGAnalysisInterfaces/IPMGSherpaVjetsSysTool.h"
 
// System include(s):
#include <map>
#include <vector>
#include <memory>
 
// ROOT include(s):
#include "TFile.h"
 
// Forward declaration
class TH2F;
 
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