TileCellIDCToNtuple.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
//*****************************************************************************
//  Filename : TileCellIDCToNtuple.cxx
//  Author   : Zhifang
//  Created  : Jan, 2003
//
//  DESCRIPTION:
//     Implement the algorithm
//
//  HISTORY:
//
//  BUGS:
//
//*****************************************************************************
 
//Gaudi Includes
#include "GaudiKernel/INTupleSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/SmartDataPtr.h"
 
//Atlas include
#include "AthenaKernel/errorcheck.h"
#include "EventContainers/SelectAllObject.h"
 
// Calo include
#include "CaloIdentifier/TileID.h"
 
//TileCalo include
#include "TileEvent/TileCellIDC.h"
#include "TileRec/TileCellIDCToNtuple.h"
 
const int max_chan=5216;
 
// Constructor & deconstructor
TileCellIDCToNtuple::TileCellIDCToNtuple(const std::string& name, ISvcLocator* pSvcLocator)
   : AthAlgorithm(name, pSvcLocator)
   , m_ntuplePtr(0)
   , m_ntupleID("h90")
   , m_ntupleLoc("/FILE1/TileRec")
   , m_cellContainer("TileCellIDC")
   , m_tileID(0)
{
  declareProperty("TileCellIDC", m_cellContainer);    
  declareProperty("NTupleLoc", m_ntupleLoc);
  declareProperty("NTupleID", m_ntupleID);
}
 
TileCellIDCToNtuple::~TileCellIDCToNtuple()
{
}
 
// Alg standard interface function
StatusCode TileCellIDCToNtuple::initialize() 
{ 
  ATH_MSG_INFO( "Initialization started" );
 
  // retrieve TileID helper from det store
 
  CHECK( detStore()->retrieve(m_tileID) );
 
  m_ntupleLoc="/NTUPLES" + m_ntupleLoc;
 
  SmartDataPtr<NTuple::Directory> DirPtr(ntupleSvc(), m_ntupleLoc);
  if(!DirPtr) DirPtr=ntupleSvc()->createDirectory(m_ntupleLoc);
  if(!DirPtr) {
    ATH_MSG_ERROR( "Invalid Ntuple Directory: " );
    return StatusCode::FAILURE;
  }
  m_ntuplePtr=ntupleSvc()->book(DirPtr.ptr(), m_ntupleID,
                                    CLID_ColumnWiseTuple, "TileCellIDC-Ntuple");
  if(!m_ntuplePtr) {
    ATH_MSG_ERROR( "Failed to book ntuple: TileCellIDCNtuple" );
    return StatusCode::FAILURE;
  }
 
  CHECK( m_ntuplePtr->addItem("TileCell/ncell",m_nchan,0,max_chan) );
  CHECK( m_ntuplePtr->addItem("TileCell/totalE",m_tolE) );
 
  CHECK( m_ntuplePtr->addItem("TileCell/energy",m_nchan,m_energy) );
  CHECK( m_ntuplePtr->addItem("TileCell/enediff",m_nchan,m_enediff) );
  CHECK( m_ntuplePtr->addItem("TileCell/time",m_nchan,m_time) );
  CHECK( m_ntuplePtr->addItem("TileCell/quality",m_nchan,m_quality) );
 
  CHECK( m_ntuplePtr->addItem("TileCell/detector",m_nchan,m_detector,0,3) );
  CHECK( m_ntuplePtr->addItem("TileCell/side",m_nchan,m_side,-1,1) );
  CHECK( m_ntuplePtr->addItem("TileCell/sample",m_nchan,m_sample,0,3) );
  CHECK( m_ntuplePtr->addItem("TileCell/eta",m_nchan,m_eta,0,15) );
  CHECK( m_ntuplePtr->addItem("TileCell/phi",m_nchan,m_phi,0,63) );
 
  ATH_MSG_INFO( "Initialization completed" );
  return StatusCode::SUCCESS;
}
 
StatusCode TileCellIDCToNtuple::execute(const EventContext& /*ctx*/)
{
  // step1: read Cells from TDS
  const TileCellIDC* CellCnt;
  CHECK( evtStore()->retrieve(CellCnt, m_cellContainer) );
 
  // step2: to fill items in ntuple
  SelectAllObject<TileCellIDC> selCells(CellCnt);
  SelectAllObject<TileCellIDC>::const_iterator it=selCells.begin(); 
  SelectAllObject<TileCellIDC>::const_iterator end=selCells.end(); 
  m_nchan=0;
  m_tolE=0.0;
  for(; it != end; ++it) {
    const TileCell *cell = (*it);
    m_tolE+=cell->energy();
    m_energy[m_nchan]=cell->energy();
    m_enediff[m_nchan]=cell->eneDiff();
    m_time[m_nchan]=cell->time();
    m_quality[m_nchan]=cell->quality();
    
    Identifier id=m_tileID->cell_id(cell->subcalo_hash());
    m_detector[m_nchan]=m_tileID->section(id);
    m_side[m_nchan]=m_tileID->side(id);
    m_sample[m_nchan]=m_tileID->sample(id);
    m_eta[m_nchan]=m_tileID->tower(id);
    m_phi[m_nchan]=m_tileID->module(id);
    
    m_nchan++;
 
    if (m_nchan >= max_chan) break;
  }      
 
  // step3: coomit ntuple
  CHECK( ntupleSvc()->writeRecord(m_ntuplePtr) );
 
  // Execution completed.
  ATH_MSG_DEBUG( "execute() completed successfully" );
  return StatusCode::SUCCESS;
}
 
StatusCode TileCellIDCToNtuple::finalize()
{
  ATH_MSG_INFO( "finalize() successfully" );
  return StatusCode::SUCCESS;
}