EGInvariantMassTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

// EGInvariantMassTool.h
// author: giovanni.marchiori@cern.ch
 
#ifndef DERIVATIONFRAMEWORK_EGINVARIANTMASSTOOL_H
#define DERIVATIONFRAMEWORK_EGINVARIANTMASSTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
//
#include "ExpressionEvaluation/ExpressionParserUser.h"
#include "GaudiKernel/EventContext.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODBase/IParticleContainer.h"
//
#include <string>
#include <vector>
namespace DerivationFramework {
 
enum EEGInvariantMassToolParser
{
  kParser1,
  kParser2,
  kNumEGInvariantMassToolParser
};
class EGInvariantMassTool : public extends<ExpressionParserUser<AthAlgTool, kNumEGInvariantMassToolParser>, IAugmentationTool>
{
public:
 
  using base_class::base_class;
 
  virtual StatusCode initialize() override final;
  virtual StatusCode addBranches(const EventContext& ctx) const override final;
 
private:
  StatusCode getInvariantMasses(const EventContext& ctx,
                                std::vector<float>&) const;
 
  Gaudi::Property<std::string> m_expression1{this, "Object1Requirements", "true"};
  Gaudi::Property<std::string> m_expression2{this, "Object2Requirements", "true"};
  SG::WriteHandleKey<std::vector<float>> m_sgName{ this,
                                                   "StoreGateEntryName",
                                                   "",
                                                   "SG key of output object" };
 
  SG::ReadHandleKey<xAOD::IParticleContainer> m_container1Name{
    this,
    "Container1Name",
    "",
    "SG key of first  container"
  };
  SG::ReadHandleKey<xAOD::IParticleContainer> m_container2Name{
    this,
    "Container2Name",
    "",
    "SG key of second container"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_pt1BranchName{
    this,
    "Pt1BranchName",
    "",
    "Pt1 if different than default"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_eta1BranchName{
    this,
    "Eta1BranchName",
    "",
    "Eta1 if different than default"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_phi1BranchName{
    this,
    "Phi1BranchName",
    "",
    "Phi1 if different than default"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_pt2BranchName{
    this,
    "Pt2BranchName",
    "",
    "Pt2 if different than default"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_eta2BranchName{
    this,
    "Eta2BranchName",
    "",
    "Eta2 if different than default"
  };
 
  SG::ReadHandleKey<std::vector<float>> m_phi2BranchName{
    this,
    "Phi2BranchName",
    "",
    "Phi2 if different than default"
  };
 
  Gaudi::Property<float> m_mass1Hypothesis{this, "Mass1Hypothesis", 0.f};
  Gaudi::Property<float> m_mass2Hypothesis{this, "Mass2Hypothesis", 0.f};
  Gaudi::Property<float> m_mindR{this, "MinDeltaR", 0.f};
  Gaudi::Property<bool> m_checkCharge{this, "CheckCharge", true};
  Gaudi::Property<bool> m_doTransverseMass{this,"DoTransverseMass",  true};
};
}
 
#endif // DERIVATIONFRAMEWORK_EGINVARIANTMASSTOOL_H