Select_Bmumu.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
//============================================================================
// Select_Bmumu.h
//============================================================================
// 
// Author : Wolfgang Walkowiak <Wolfgang.Walkowiak@cern.ch.>
// Changes:
// 
// Based on Select_onia2mumu.h.
// Original author: Daniel Scheirich <daniel.scheirich@cern.ch>
//
// Select B candidates for the B(s)mumu analysis including for
// the reference channels used.
//
// For an example see BPHY8.py .
//
// Job options provided by this class:
//                           
//============================================================================
//
#ifndef DERIVATIONFRAMEWORK_Select_Bmumu_H
#define DERIVATIONFRAMEWORK_Select_Bmumu_H
 
#include <string>
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkBPhys/CfAthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "MuonAnalysisInterfaces/IMuonSelectionTool.h"
#include "xAODBPhys/BPhysHelper.h"
 
namespace Trk {
  class V0Tools;
}
 
namespace xAOD {
  class BPhysHypoHelper;
}
 
namespace SG {
  class AuxElement;
}
 
namespace DerivationFramework {
 
  class ATLAS_NOT_THREAD_SAFE Select_Bmumu : public CfAthAlgTool, public IAugmentationTool {
    public: 
      Select_Bmumu(const std::string& t, const std::string& n, const IInterface* p);
 
      StatusCode initialize() override;
      StatusCode finalize()   override;
      
      virtual StatusCode addBranches() const override;
      
    private:
      void ProcessVertex(xAOD::BPhysHypoHelper&, xAOD::BPhysHelper::pv_type) const;
      bool massCuts(float mass) const;
      bool massInBlindedRegion(float mass) const;
 
      bool checkAllMuonsTight(const std::vector<const xAOD::Muon*>& muons,
                              int maxMuonsToCheck=-1) const;
      
      bool pass(const SG::AuxElement& em, const std::string& hypo) const;
      bool setPass(const SG::AuxElement& em, const std::string& hypo, bool passVal) const;
      bool setPassIfNotAvailable(SG::AuxElement& em, const std::string& hypo,
                 bool passVal) const;
      // std::vector<xAOD::Vertex*> getPrecedingVertices(const xAOD::Vertex* vtx);
      
      ToolHandle<Trk::V0Tools>           m_v0Tools;
      ToolHandle<CP::IMuonSelectionTool> m_muSelectionTool;
      
      std::string m_hypoName;               
      std::string m_inputVtxContainerName;  
      std::vector<double> m_trkMasses;      
      double m_massHypo;                    
      double m_massMax;                     
      double m_massMin;                     
      double m_chi2Max;                     
      int m_DoVertexType;                   
      bool   m_do3d;                        
      double m_blindMassMin;                
      double m_blindMassMax;                
      bool   m_doBlinding;                  
      bool   m_doCutBlinded;                
      bool   m_blindOnlyAllMuonsTight;      
      bool   m_useMuCalcMass;               
 
      std::vector<std::string> m_subDecVtxContNames; 
      std::vector<std::string> m_subDecVtxHypoCondNames; 
      std::vector<std::string> m_subDecVtxHypoFlagNames; 
  }; 
}
 
#endif // DERIVATIONFRAMEWORK_Select_Bmumu_H