ZdcRecRun3Decode.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
/*
 * ZdcRecRun3Decode.cxx
 *
 *  Created on: June 23, 2022 (never forget)
 *      Author: steinberg@bnl.gov
 */
 
 
#include <memory>
 
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/StatusCode.h"
#include "StoreGate/StoreGateSvc.h"
#include "StoreGate/WriteHandle.h"
#include "StoreGate/ReadHandle.h"
 
#include "ZdcRec/ZdcRecRun3Decode.h"
#include "ZdcRec/ZdcRecChannelToolLucrod.h"
#include "xAODForward/ZdcModuleToString.h"
#include "ZdcByteStream/ZdcToString.h"
#include "ZdcUtils/ZdcEventInfo.h"
 
 
//==================================================================================================
ZdcRecRun3Decode::ZdcRecRun3Decode(const std::string& name, ISvcLocator* pSvcLocator) :
 
    AthAlgorithm(name, pSvcLocator),
    m_ownPolicy(static_cast<int> (SG::OWN_ELEMENTS)),
    m_DecodeRunMode(0) // 0 = ZDC+RPD (default), 1 = LIS only, 2 = ZDC+RPD + LIS
{
  declareProperty("OwnPolicy",m_ownPolicy) ;
  declareProperty("DecodeRunMode",m_DecodeRunMode); // 0 = ZDC+RPD (default), 1 = LIS only, 2 = ZDC+RPD + LIS 
}
 
//==================================================================================================
 
//==================================================================================================
ZdcRecRun3Decode::~ZdcRecRun3Decode() {}
//==================================================================================================
 
//==================================================================================================
StatusCode ZdcRecRun3Decode::initialize()
{
    MsgStream mLog(msgSvc(), name());
 
    // Reconstruction Tool
    ATH_CHECK( m_ChannelTool.retrieve() );
 
 
    // Reconstruction Tool
 
    ATH_CHECK( m_zdcModuleContainerName.initialize() );
    ATH_CHECK( m_zdcSumContainerName.initialize() );
    ATH_CHECK( m_zldContainerName.initialize(SG::AllowEmpty) );
    ATH_CHECK( m_eventInfoKey.initialize() );
    ATH_CHECK( m_eventInfoDecorKey.initialize() );
 
    if (m_ownPolicy == SG::OWN_ELEMENTS)
        mLog << MSG::DEBUG << "...will OWN its cells." << endmsg;
    else
        mLog << MSG::DEBUG << "...will VIEW its cells." << endmsg;
 
    if(m_DecodeRunMode == 0){
        mLog << MSG::DEBUG << "--> ZDC: Running in ZDC+RPD mode" << endmsg;
        m_nFragments.set(6); // default 
    }
 
    else if(m_DecodeRunMode == 1){
        mLog << MSG::DEBUG << "--> ZDC: Running in LIS only mode" << endmsg;
        m_nFragments.set(1);
        }   
    else if(m_DecodeRunMode == 2){
        mLog << MSG::DEBUG << "--> ZDC: Running in ZDC+RPD + LIS mode" << endmsg;
        m_nFragments.set(7);
    }   
    else {
        mLog << MSG::ERROR << "--> ZDC: Unknown DecodeRunMode " << m_DecodeRunMode << ", should be 0, 1, or 2. Exiting." << endmsg;
        return StatusCode::FAILURE;
    }
 
 
    mLog << MSG::DEBUG << "--> ZDC: ZdcRecRun3Decode initialization complete" << endmsg;
 
    return StatusCode::SUCCESS;
}
//==================================================================================================
 
//==================================================================================================
StatusCode ZdcRecRun3Decode::execute()
{
 
  ATH_MSG_DEBUG("In ZdRecRun3");
 
  const EventContext& ctx = Gaudi::Hive::currentContext();
 
  ATH_MSG_DEBUG ("--> ZDC: ZdcRecRun3Decode execute starting on "
                 << ctx.evt()
                 << "th event");
 
  //Look for the container presence
  if (m_zldContainerName.empty()) {
    return StatusCode::SUCCESS;
  }
 
  ATH_MSG_DEBUG("Trying to get LUCROD DATA!");
  SG::ReadHandle<ZdcLucrodDataContainer> zldContainer    (m_zldContainerName, ctx);
  ATH_MSG_DEBUG("Did I get LUCROD DATA?");
 
 
  if (zldContainer->size() < m_nFragments)
    {
      int zdcLucrod = 0;
      int rpdLucrod = 0;
      int lisLucrod = 0;    
      for (auto zld : *zldContainer)
    {
      uint32_t lucrod_id =  zld->GetLucrodID();
      if (lucrod_id == ZdcEventInfo::LucrodLowGain || lucrod_id == ZdcEventInfo::LucrodHighGain) // ZDC LUCRODs
        {
          zdcLucrod++;
        }
      else if (lucrod_id == ZdcEventInfo::LucrodRPD1A
           || lucrod_id == ZdcEventInfo::LucrodRPD1C
           || lucrod_id == ZdcEventInfo::LucrodRPD2A
           || lucrod_id == ZdcEventInfo::LucrodRPD2C)
        {
          rpdLucrod++;
        }
        else if (lucrod_id == ZdcEventInfo::LucrodLIS) // LIS LUCROD
        {
          lisLucrod++;
        }
      else
        {
          ATH_MSG_WARNING("Unidentified LUCROD ID = " << lucrod_id);
        }
    }
      SG::ReadHandle<xAOD::EventInfo> eventInfo (m_eventInfoKey);
      if (!eventInfo->updateErrorState(xAOD::EventInfo::ForwardDet,xAOD::EventInfo::Error))
    {
      ATH_MSG_WARNING( " cannot set EventInfo error state for ForwardDet "  );
    }
      if (!eventInfo->updateEventFlagBit(xAOD::EventInfo::ForwardDet,ZdcEventInfo::DECODINGERROR)) 
    {
      ATH_MSG_WARNING( " cannot set flag bit for ForwardDet "  );      
    }
      if (rpdLucrod < ZdcEventInfo::nTotalRpdLucrod && m_DecodeRunMode != 1)
    {
      if (!eventInfo->updateEventFlagBit(xAOD::EventInfo::ForwardDet,ZdcEventInfo::RPDDECODINGERROR)) 
        {
          ATH_MSG_WARNING( " cannot set RPDDECODINGERROR flag bit for ForwardDet "  );      
        }
    }
      if (zdcLucrod < ZdcEventInfo::nTotalZdcLucrod && m_DecodeRunMode != 1)
    {
      if (!eventInfo->updateEventFlagBit(xAOD::EventInfo::ForwardDet,ZdcEventInfo::ZDCDECODINGERROR)) 
        {
          ATH_MSG_WARNING( " cannot set ZDCDECODINGERROR flag bit for ForwardDet "  );      
        }
    }
      if (lisLucrod < ZdcEventInfo::nTotalLisLucrod && (m_DecodeRunMode == 1 || m_DecodeRunMode == 2))
    {
      if (!eventInfo->updateEventFlagBit(xAOD::EventInfo::ForwardDet,ZdcEventInfo::LISDECODINGERROR)) 
        {
          ATH_MSG_WARNING( " cannot set LISDECODINGERROR flag bit for ForwardDet "  );      
        }
    }
    }
 
  //Create the containers to hold the reconstructed information (you just pass the pointer and the converter does the work) 
  std::unique_ptr<xAOD::ZdcModuleContainer> moduleContainer( new xAOD::ZdcModuleContainer());
  std::unique_ptr<xAOD::ZdcModuleAuxContainer> moduleAuxContainer( new xAOD::ZdcModuleAuxContainer() );
  moduleContainer->setStore( moduleAuxContainer.get() );
 
  //Create the containers to hold the reconstructed information (you just pass the pointer and the converter does the work) 
  std::unique_ptr<xAOD::ZdcModuleContainer> moduleSumContainer( new xAOD::ZdcModuleContainer());
  std::unique_ptr<xAOD::ZdcModuleAuxContainer> moduleSumAuxContainer( new xAOD::ZdcModuleAuxContainer() );
  moduleSumContainer->setStore( moduleSumAuxContainer.get() );
 
  ATH_MSG_DEBUG("Trying to convert!");
 
  // rearrange ZDC channels and perform fast reco on all channels (including non-big tubes)
  int ncha = m_ChannelTool->convertLucrod2ZM(zldContainer.get(), moduleContainer.get(), moduleSumContainer.get() );
  ATH_MSG_DEBUG("m_ChannelTool->convertLucrod2ZM returned " << ncha << " channels");
 
  ATH_MSG_DEBUG("Dumping modules");
  ATH_MSG_DEBUG( ZdcModuleToString(*moduleContainer) );
  ATH_MSG_DEBUG("Dumping module sums");
  ATH_MSG_DEBUG( ZdcModuleToString(*moduleSumContainer) );
  
  // eventually reconstruct RPD, using ML libraries
  // ATH_CHECK( m_rpdTool...)
 
  SG::WriteHandle<xAOD::ZdcModuleContainer> moduleContainerH (m_zdcModuleContainerName, ctx);
  ATH_CHECK( moduleContainerH.record (std::move(moduleContainer),
                      std::move(moduleAuxContainer)) );
 
  SG::WriteHandle<xAOD::ZdcModuleContainer> moduleSumContainerH (m_zdcSumContainerName, ctx);
  ATH_CHECK( moduleSumContainerH.record (std::move(moduleSumContainer),
                     std::move(moduleSumAuxContainer)) );
 
  return StatusCode::SUCCESS;
 
}
//==================================================================================================
 
//==================================================================================================
StatusCode ZdcRecRun3Decode::finalize()
{
 
  ATH_MSG_DEBUG( "--> ZDC: ZdcRecRun3Decode finalize complete" );
 
  return StatusCode::SUCCESS;
 
}
//==================================================================================================