Reco_mumu.cxx

Go to the documentation of this file.

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

// Reco_mumu.cxx
// Author: Daniel Scheirich <daniel.scheirich@cern.ch>
// Based on the Integrated Simulation Framework
//
// Basic Jpsi->mu mu derivation example
 
#include "Reco_mumu.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODTracking/VertexAuxContainer.h"
#include "TrkVertexAnalysisUtils/V0Tools.h"
#include "BPhysPVTools.h"
 
 
namespace DerivationFramework {
 
  Reco_mumu::Reco_mumu(const std::string& t,
      const std::string& n,
      const IInterface* p) : 
    base_class(t,n,p) {
  }
 
  // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * 
  
  StatusCode Reco_mumu::initialize()
  {
  
    ATH_MSG_DEBUG("in initialize()");
 
    // retrieve V0 tools
    CHECK( m_v0Tools.retrieve() );
    
    // get the JpsiFinder tool
    CHECK( m_jpsiFinder.retrieve() );
     
    // get the PrimaryVertexRefitter tool
    CHECK( m_pvRefitter.retrieve() );
 
    // Get the beam spot service   
    CHECK( m_eventInfo_key.initialize() );
    
    ATH_CHECK(m_CollectionsToCheck.initialize());
    ATH_CHECK(m_pvContainerKey.initialize());
    ATH_CHECK(m_outContainerKey.initialize());
    ATH_CHECK(m_refContainerKey.initialize(m_refitPV));
    return StatusCode::SUCCESS;
 }
  
 
 
  // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * 
  
  StatusCode Reco_mumu::addBranches(const EventContext& ctx) const
  {
    bool callJpsiFinder = true;
    if(m_checkCollections) {
      for( SG::ReadHandle<xAOD::VertexContainer> vertContainer : m_CollectionsToCheck.makeHandles(ctx)){
         if (!vertContainer.isValid()){
             ATH_MSG_FATAL("Failed to retrieve "<<vertContainer.key());
             return StatusCode::FAILURE;
         }
         if(vertContainer->size() == 0) {
            callJpsiFinder = false;
            ATH_MSG_DEBUG("Container VertexContainer (" << vertContainer.key() << ") is empty");
            break;//No point checking other containers
         }
      }
    }
 
    // Jpsi container and its auxilliary store
    std::unique_ptr<xAOD::VertexContainer>    vtxContainer = std::make_unique<xAOD::VertexContainer>();
    std::unique_ptr<xAOD::VertexAuxContainer> vtxAuxContainer = std::make_unique<xAOD::VertexAuxContainer>();
    vtxContainer->setStore(vtxAuxContainer.get());
    
    std::unique_ptr<xAOD::VertexContainer>    refPvContainer =std::make_unique<xAOD::VertexContainer>();
    std::unique_ptr<xAOD::VertexAuxContainer> refPvAuxContainer = std::make_unique<xAOD::VertexAuxContainer>();
    refPvContainer->setStore(refPvAuxContainer.get());
   
    
    if(callJpsiFinder) {
        //----------------------------------------------------
        // call Jpsi finder
        //----------------------------------------------------
        ATH_CHECK(m_jpsiFinder->performSearch(ctx, *vtxContainer));
 
        //----------------------------------------------------
        // retrieve primary vertices
        //----------------------------------------------------
        SG::ReadHandle<xAOD::VertexContainer>    pvContainer{m_pvContainerKey,ctx};
        if (!pvContainer.isValid()){
            ATH_MSG_FATAL("Failed to retrieve "<<m_pvContainerKey.fullKey());
            return StatusCode::FAILURE;
        }
        //----------------------------------------------------
        // Try to retrieve refitted primary vertices
        //----------------------------------------------------
       
       // Give the helper class the ptr to v0tools and beamSpotsSvc to use
        SG::ReadHandle<xAOD::EventInfo> evt(m_eventInfo_key, ctx);
        if(not evt.isValid()) ATH_MSG_ERROR("Cannot Retrieve " << m_eventInfo_key.key() );
        
        // Give the helper class the ptr to v0tools and beamSpotsSvc to use
        BPhysPVTools helper(&(*m_v0Tools), evt.cptr());
        helper.SetMinNTracksInPV(m_PV_minNTracks);
        helper.SetSave3d(m_do3d);
 
        if(m_refitPV &&  vtxContainer->size()){
            ATH_CHECK(helper.FillCandwithRefittedVertices(vtxContainer.get(),  pvContainer.cptr(), refPvContainer.get(), &(*m_pvRefitter) , m_PV_max, m_DoVertexType));
        }else if (!m_refitPV && vtxContainer->size()){
            ATH_CHECK(helper.FillCandExistingVertices(vtxContainer.get(), pvContainer.cptr(), m_DoVertexType));
        }
    }    
    //----------------------------------------------------
    // save in the StoreGate
    //----------------------------------------------------
    SG::WriteHandle<xAOD::VertexContainer> out_handle{m_outContainerKey, ctx};
    ATH_CHECK(out_handle.record(std::move(vtxContainer), std::move(vtxAuxContainer)));
    if(m_refitPV) {
       SG::WriteHandle<xAOD::VertexContainer> refitHandle{m_refContainerKey,ctx};
       ATH_CHECK(refitHandle.record(std::move(refPvContainer), std::move(refPvAuxContainer)));
    }    
 
    return StatusCode::SUCCESS;
  }  
}