// reader for single-top n-tuples, LHC run 2
#ifndef ATLAS_AtlEvtReaderD3PDSgTopR2
#define ATLAS_AtlEvtReaderD3PDSgTopR2

#include <vector>
#include <Rtypes.h>
#include "AtlEvtReaderD3PDBase.h"

class AtlSelector;
class TBranch;
class TFile;
class TTree;
class AtlTriggerConf;

class AtlEvtReaderD3PDSgTopR2 : public AtlEvtReaderD3PDBase {

  public:
    typedef AtlEvtReaderD3PDBase super;
    
    enum EDataType {
        kUndefinedDataType,
        kRealData,
        kMC,
    };
    
    static size_t const PeriodLimit = 2;
    
  protected:
    Float_t v_weight_mc;
    Float_t v_weight_pileup;
    Float_t v_weight_sherpa_22_vjets;
    Float_t v_weight_jvt;
    Float_t v_weight_leptonSF;
    Float_t v_weight_bTagSF_70;
    Float_t v_weight_bTagSF_77;
    Float_t v_weight_bTagSF_85;
    std::vector<float> * v_weight_bTagSF_70_vec;
    std::vector<float> * v_weight_bTagSF_77_vec;
    std::vector<float> * v_weight_bTagSF_85_vec;
    Float_t v_weight_leptonSF_EL_SF_Trigger_UP;
    Float_t v_weight_leptonSF_EL_SF_Trigger_DOWN;
    Float_t v_weight_leptonSF_EL_SF_Reco_UP;
    Float_t v_weight_leptonSF_EL_SF_Reco_DOWN;
    Float_t v_weight_leptonSF_EL_SF_ID_UP;
    Float_t v_weight_leptonSF_EL_SF_ID_DOWN;
    Float_t v_weight_leptonSF_EL_SF_Isol_UP;
    Float_t v_weight_leptonSF_EL_SF_Isol_DOWN;
    Float_t v_weight_leptonSF_MU_SF_Trigger_STAT_UP;
    Float_t v_weight_leptonSF_MU_SF_Trigger_STAT_DOWN;
    Float_t v_weight_leptonSF_MU_SF_Trigger_SYST_UP;
    Float_t v_weight_leptonSF_MU_SF_Trigger_SYST_DOWN;
    Float_t v_weight_leptonSF_MU_SF_ID_STAT_UP;
    Float_t v_weight_leptonSF_MU_SF_ID_STAT_DOWN;
    Float_t v_weight_leptonSF_MU_SF_ID_SYST_UP;
    Float_t v_weight_leptonSF_MU_SF_ID_SYST_DOWN;
    Float_t v_weight_leptonSF_MU_SF_Isol_STAT_UP;
    Float_t v_weight_leptonSF_MU_SF_Isol_STAT_DOWN;
    Float_t v_weight_leptonSF_MU_SF_Isol_SYST_UP;
    Float_t v_weight_leptonSF_MU_SF_Isol_SYST_DOWN;
    ULong64_t v_eventNumber;
    UInt_t v_runNumber;
    UInt_t v_mcChannelNumber;
    Float_t v_mu;
    std::vector<float> * v_el_pt;
    std::vector<float> * v_el_eta;
    std::vector<float> * v_el_phi;
    std::vector<float> * v_el_e;
    std::vector<float> * v_el_charge;
    std::vector<float> * v_el_topoetcone20;
    std::vector<float> * v_el_ptvarcone20;
    std::vector<float> * v_mu_pt;
    std::vector<float> * v_mu_eta;
    std::vector<float> * v_mu_phi;
    std::vector<float> * v_mu_e;
    std::vector<float> * v_mu_charge;
    std::vector<float> * v_mu_topoetcone20;
    std::vector<float> * v_mu_ptvarcone30;
    std::vector<float> * v_jet_pt;
    std::vector<float> * v_jet_eta;
    std::vector<float> * v_jet_phi;
    std::vector<float> * v_jet_e;
    std::vector<char> * v_jet_isbtagged_70;
    std::vector<char> * v_jet_isbtagged_77;
    std::vector<char> * v_jet_isbtagged_85;
    std::vector<float> * v_jet_mv2c00;
    std::vector<float> * v_jet_mv2c10;
    std::vector<float> * v_jet_mv2c20;
    std::vector<float> * v_jet_ip3dsv1;
    std::vector<float> * v_jet_jvt;
    std::vector<int> * v_jet_truthflav;
    Float_t v_met_met;
    Float_t v_met_phi;
    Int_t v_leptonic[PeriodLimit];
    Int_t v_ejets[PeriodLimit];
    Int_t v_mujets[PeriodLimit];
    Int_t v_ee[PeriodLimit];
    Int_t v_mumu[PeriodLimit];
    Int_t v_emu[PeriodLimit];
    Int_t v_eee[PeriodLimit];
    Int_t v_eemu[PeriodLimit];
    Int_t v_emumu[PeriodLimit];
    Int_t v_mumumu[PeriodLimit];
    Int_t v_efatJet[PeriodLimit];
    Int_t v_mufatJet[PeriodLimit];
    Char_t v_HLT_e24_lhmedium_L1EM20VH;
    Char_t v_HLT_e24_lhmedium_iloose_L1EM20VH;
    Char_t v_HLT_e24_lhmedium_L1EM18VH;
    Char_t v_HLT_e26_lhtight_nod0_ivarloose;
    Char_t v_HLT_e60_lhmedium_nod0;
    Char_t v_HLT_e60_lhmedium;
    Char_t v_HLT_e120_lhloose;
    Char_t v_HLT_e140_lhloose_nod0;
    Char_t v_HLT_mu20_iloose_L1MU15;
    Char_t v_HLT_mu20_iloose_mu8noL1;
    Char_t v_HLT_mu26_ivarmedium;
    Char_t v_HLT_mu50;
    std::vector<bool> * v_el_trigMatch;
    std::vector<bool> * v_mu_trigMatch;
    UInt_t v_lbn;
    Float_t v_Vtxz;
    UInt_t v_npVtx;
    UInt_t v_el_n;
    std::vector<float> * v_el_d0;
    std::vector<float> * v_el_z0;
    std::vector<float> * v_el_d0sig;
    std::vector<float> * v_el_z0sig;
    std::vector<float> * v_el_topoetcone30;
    std::vector<float> * v_el_topoetcone40;
    std::vector<float> * v_el_ptcone30;
    std::vector<float> * v_el_ptcone40;
    std::vector<bool> * v_el_tight;
    std::vector<float> * v_el_MT;
    std::vector<float> * v_el_nu_pT;
    std::vector<float> * v_el_nu_eta;
    std::vector<float> * v_el_nu_phi;
    std::vector<float> * v_el_W_pT;
    std::vector<float> * v_el_W_eta;
    std::vector<float> * v_el_W_phi;
    std::vector<int> * v_el_true_type;
    std::vector<int> * v_el_true_origin;
    std::vector<int> * v_el_true_typebkg;
    std::vector<int> * v_el_true_originbkg;
    std::vector<int> * v_el_true_pdg;
    std::vector<float> * v_el_true_pt;
    std::vector<float> * v_el_true_eta;
    UInt_t v_mu_n;
    std::vector<float> * v_mu_d0;
    std::vector<float> * v_mu_z0;
    std::vector<float> * v_mu_d0sig;
    std::vector<float> * v_mu_z0sig;
    std::vector<float> * v_mu_topoetcone30;
    std::vector<float> * v_mu_topoetcone40;
    std::vector<float> * v_mu_ptcone30;
    std::vector<float> * v_mu_ptcone40;
    std::vector<float> * v_mu_MT;
    std::vector<float> * v_mu_nu_pT;
    std::vector<float> * v_mu_nu_eta;
    std::vector<float> * v_mu_nu_phi;
    std::vector<float> * v_mu_W_pT;
    std::vector<float> * v_mu_W_eta;
    std::vector<float> * v_mu_W_phi;
    std::vector<bool> * v_mu_tight;
    std::vector<int> * v_mu_true_type;
    std::vector<int> * v_mu_true_origin;
    std::vector<int> * v_mu_true_typebkg;
    std::vector<int> * v_mu_true_originbkg;
    std::vector<int> * v_mu_true_pdg;
    std::vector<float> * v_mu_true_pt;
    std::vector<float> * v_mu_true_eta;
    UInt_t v_jet_n;
    std::vector<float> * v_jet_m;
    Float_t v_met_px;
    Float_t v_met_py;
    Float_t v_met_sumet;
    Float_t v_MT2;
    Float_t v_pTsys;
    Float_t v_sigma_pTsys;
    Float_t v_Ht;
    bool v_genfilter_BHadron;
    bool v_genfilter_CHadronPt4Eta3;

    TBranch * b_weight_mc; //!
    TBranch * b_weight_pileup; //!
    TBranch * b_weight_sherpa_22_vjets; //!
    TBranch * b_weight_jvt; //!
    TBranch * b_weight_leptonSF; //!
    TBranch * b_weight_bTagSF_70; //!
    TBranch * b_weight_bTagSF_77; //!
    TBranch * b_weight_bTagSF_85; //!
    TBranch * b_eventNumber; //!
    TBranch * b_runNumber; //!
    TBranch * b_mcChannelNumber; //!
    TBranch * b_mu; //!
    TBranch * b_el_pt; //!
    TBranch * b_el_eta; //!
    TBranch * b_el_phi; //!
    TBranch * b_el_e; //!
    TBranch * b_el_charge; //!
    TBranch * b_el_topoetcone20; //!
    TBranch * b_el_ptvarcone20; //!
    TBranch * b_mu_pt; //!
    TBranch * b_mu_eta; //!
    TBranch * b_mu_phi; //!
    TBranch * b_mu_e; //!
    TBranch * b_mu_charge; //!
    TBranch * b_mu_topoetcone20; //!
    TBranch * b_mu_ptvarcone30; //!
    TBranch * b_jet_pt; //!
    TBranch * b_jet_eta; //!
    TBranch * b_jet_phi; //!
    TBranch * b_jet_e; //!
    TBranch * b_jet_isbtagged_70; //!
    TBranch * b_jet_isbtagged_77; //!
    TBranch * b_jet_isbtagged_85; //!
    TBranch * b_jet_mv2c00; //!
    TBranch * b_jet_mv2c10; //!
    TBranch * b_jet_mv2c20; //!
    TBranch * b_jet_ip3dsv1; //!
    TBranch * b_jet_jvt; //!
    TBranch * b_jet_truthflav; //!
    TBranch * b_met_met; //!
    TBranch * b_met_phi; //!
    TBranch * b_leptonic; //!
    TBranch * b_ejets; //!
    TBranch * b_mujets; //!
    TBranch * b_ee; //!
    TBranch * b_mumu; //!
    TBranch * b_emu; //!
    TBranch * b_eee; //!
    TBranch * b_eemu; //!
    TBranch * b_emumu; //!
    TBranch * b_mumumu; //!
    TBranch * b_efatJet; //!
    TBranch * b_mufatJet; //!
    TBranch * b_leptonic_2015; //!
    TBranch * b_ejets_2015; //!
    TBranch * b_mujets_2015; //!
    TBranch * b_ee_2015; //!
    TBranch * b_mumu_2015; //!
    TBranch * b_emu_2015; //!
    TBranch * b_eee_2015; //!
    TBranch * b_eemu_2015; //!
    TBranch * b_emumu_2015; //!
    TBranch * b_mumumu_2015; //!
    TBranch * b_efatJet_2015; //!
    TBranch * b_mufatJet_2015; //!
    TBranch * b_leptonic_2016; //!
    TBranch * b_ejets_2016; //!
    TBranch * b_mujets_2016; //!
    TBranch * b_ee_2016; //!
    TBranch * b_mumu_2016; //!
    TBranch * b_emu_2016; //!
    TBranch * b_eee_2016; //!
    TBranch * b_eemu_2016; //!
    TBranch * b_emumu_2016; //!
    TBranch * b_mumumu_2016; //!
    TBranch * b_efatJet_2016; //!
    TBranch * b_mufatJet_2016; //!
    TBranch * b_HLT_e24_lhmedium_L1EM20VH;
    TBranch * b_HLT_e24_lhmedium_iloose_L1EM20VH; //!
    TBranch * b_HLT_e24_lhmedium_L1EM18VH; //!
    TBranch * b_HLT_e26_lhtight_nod0_ivarloose;
    TBranch * b_HLT_e60_lhmedium_nod0;
    TBranch * b_HLT_e60_lhmedium;
    TBranch * b_HLT_e120_lhloose;
    TBranch * b_HLT_e140_lhloose_nod0;
    TBranch * b_HLT_mu20_iloose_L1MU15;
    TBranch * b_HLT_mu20_iloose_mu8noL1; //!
    TBranch * b_HLT_mu26_ivarmedium;
    TBranch * b_HLT_mu50;
    TBranch * b_el_trigMatch; //!
    TBranch * b_mu_trigMatch; //!
    TBranch * b_lbn; //!
    TBranch * b_Vtxz; //!
    TBranch * b_npVtx; //!
    TBranch * b_el_n; //!
    TBranch * b_el_d0; //!
    TBranch * b_el_z0; //!
    TBranch * b_el_d0sig; //!
    TBranch * b_el_z0sig; //!
    TBranch * b_el_topoetcone30; //!
    TBranch * b_el_topoetcone40; //!
    TBranch * b_el_ptcone30; //!
    TBranch * b_el_ptcone40; //!
    TBranch * b_el_tight; //!
    TBranch * b_el_MT; //!
    TBranch * b_el_nu_pT; //!
    TBranch * b_el_nu_eta; //!
    TBranch * b_el_nu_phi; //!
    TBranch * b_el_W_pT; //!
    TBranch * b_el_W_eta; //!
    TBranch * b_el_W_phi; //!
    TBranch * b_el_true_type; //!
    TBranch * b_el_true_origin; //!
    TBranch * b_el_true_typebkg; //!
    TBranch * b_el_true_originbkg; //!
    TBranch * b_el_true_pdg; //!
    TBranch * b_el_true_pt; //!
    TBranch * b_el_true_eta; //!
    TBranch * b_mu_n; //!
    TBranch * b_mu_d0; //!
    TBranch * b_mu_z0; //!
    TBranch * b_mu_d0sig; //!
    TBranch * b_mu_z0sig; //!
    TBranch * b_mu_topoetcone30; //!
    TBranch * b_mu_topoetcone40; //!
    TBranch * b_mu_ptcone30; //!
    TBranch * b_mu_ptcone40; //!
    TBranch * b_mu_MT; //!
    TBranch * b_mu_nu_pT; //!
    TBranch * b_mu_nu_eta; //!
    TBranch * b_mu_nu_phi; //!
    TBranch * b_mu_W_pT; //!
    TBranch * b_mu_W_eta; //!
    TBranch * b_mu_W_phi; //!
    TBranch * b_mu_tight; //!
    TBranch * b_mu_true_type; //!
    TBranch * b_mu_true_origin; //!
    TBranch * b_mu_true_typebkg; //!
    TBranch * b_mu_true_originbkg; //!
    TBranch * b_mu_true_pdg; //!
    TBranch * b_mu_true_pt; //!
    TBranch * b_mu_true_eta; //!
    TBranch * b_jet_n; //!
    TBranch * b_jet_m; //!
    TBranch * b_met_px; //!
    TBranch * b_met_py; //!
    TBranch * b_met_sumet; //!
    TBranch * b_MT2; //!
    TBranch * b_pTsys; //!
    TBranch * b_sigma_pTsys; //!
    TBranch * b_Ht; //!
    TBranch * b_genfilter_BHadron; //!
    TBranch * b_genfilter_CHadronPt4Eta3; //!

    // Truth tree
    ULong64_t v_tt_eventNumber;
    UInt_t    v_tt_runNumber;
    Float_t   v_tt_MC_b_from_t_pt;
    Float_t   v_tt_MC_b_from_t_eta;
    Float_t   v_tt_MC_b_from_t_phi;
    Float_t   v_tt_MC_b_from_t_m;
    Float_t   v_tt_MC_b_from_tbar_pt;
    Float_t   v_tt_MC_b_from_tbar_eta;
    Float_t   v_tt_MC_b_from_tbar_phi;
    Float_t   v_tt_MC_b_from_tbar_m;
    TBranch  *b_tt_eventNumber; //!
    TBranch  *b_tt_runNumber; //!
    TBranch  *b_tt_MC_b_from_t_pt; //!
    TBranch  *b_tt_MC_b_from_t_eta; //!
    TBranch  *b_tt_MC_b_from_t_phi; //!
    TBranch  *b_tt_MC_b_from_t_m; //!
    TBranch  *b_tt_MC_b_from_tbar_pt; //!
    TBranch  *b_tt_MC_b_from_tbar_eta; //!
    TBranch  *b_tt_MC_b_from_tbar_phi; //!
    TBranch  *b_tt_MC_b_from_tbar_m; //!
    
    EDataType fDataType;
    Int_t     fD3PDversion;
    Int_t     fPeriodCount;
    Int_t     fIsSherpa22;

    TString   fBtagSFVariationName;
    Int_t     fBtagSFVariationComponent;
    TString   fLeptonSFVariationName;
    TString   fPileupSFVariationName;
    TString   fJvtSFVariationName;
    TString   fTreeName;
    
    TTree    *fTruthTree;   // MC truth tree

    AtlTriggerConf *fTriggerConfDbase; // Trigger configuration dbase
    TTree          *fTriggerConfTree;  // Tree to store the config dbase
    
  public:
    AtlEvtReaderD3PDSgTopR2(AtlSelector * parent, EDataType dataType,
			    const char* D3PDversion, char const * systematicName,
			    char const * treeNameSuffix);
    ~AtlEvtReaderD3PDSgTopR2() /*override*/;
    
    void SetBranches(TTree * tree) /*override*/;
    InitialSumOfWeights_t GetInitialSumOfWeights(TFile * inputFile) const /*override*/;
    virtual void Print(Option_t *option = "");
    char const * GetTreeName() const { return fTreeName.Data(); }
    virtual Bool_t Notify();
    
  protected:
    virtual void BuildEvent();
    virtual void BuildEventHeader();
    virtual void BuildEnergySum();
    virtual void BuildElectrons();
    virtual void BuildMuons();
    virtual void BuildJets();
    virtual void BuildFromTruthTree();
    virtual void BuildTriggerDecision();
    virtual void BuildTriggerConfig(UInt_t RunStart, UInt_t RunEnd);
    Int_t GetEntry(TTree * t, Long64_t entry) /*override*/;
    virtual void ClearBranches();
    void LoadTruthTree();
    
    inline Bool_t IsMC() const {
	//
	// Is DATA or MC ?
	//
	return (fDataType == kMC);
    }
    
    ClassDef(AtlEvtReaderD3PDSgTopR2, 0)
};

#endif