EvtCollectionWrite.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
//      ====================================================================
//  EvtCollection.Write.cpp
//      --------------------------------------------------------------------
//
//      Author    : Markus Frank
//
//      ====================================================================
#define EVTCOLLECTION_WRITE_CPP
 
// Framework include files
#include <GaudiKernel/INTupleSvc.h>
#include <GaudiKernel/IRegistry.h>
#include <GaudiKernel/MsgStream.h>
#include <GaudiKernel/SmartDataPtr.h>
 
// Example related include files
#include "EvtCollectionWrite.h"
#include <GaudiTestSuite/Event.h>
#include <GaudiTestSuite/MyTrack.h>
 
#include <cmath>
 
using namespace Gaudi::TestSuite;
 
DECLARE_COMPONENT( EvtCollectionWrite )
 
StatusCode EvtCollectionWrite::initialize() {
  StatusCode status = m_evtTupleSvc.retrieve();
  if ( status.isSuccess() ) {
    NTuplePtr nt( m_evtTupleSvc.get(), "/NTUPLES/EvtColl/Dir1/Dir2/Dir3/Collection" );
    if ( !nt ) { // Check if already booked
      nt = m_evtTupleSvc->book( "/NTUPLES/EvtColl/Dir1/Dir2/Dir3/Collection", CLID_ColumnWiseTuple, "Hello World" );
      if ( nt ) {
        // Add an index column
        status = nt->addItem( "Ntrack", m_ntrkColl, 0, 5000 );
        status = nt->addItem( "Energy", m_eneColl );
        status = nt->addItem( "Track", m_trackItem );
        status = nt->addItem( "Addr", m_evtAddrColl );
        status = nt->addItem( "TrkMom", m_ntrkColl, m_trkMom );
        status = nt->addItem( "TrkMomFix", 100, m_trkMomFixed );
        status = nt->addItem( "Address", m_evtAddrCollEx );
      } else { // did not manage to book the N tuple....
        return StatusCode::FAILURE;
      }
    } else { // Just reconnect to existing items
      status = nt->item( "Ntrack", m_ntrkColl );
      status = nt->item( "Energy", m_eneColl );
      status = nt->item( "TrkMom", m_trkMom );
      status = nt->item( "Track", m_trackItem );
      status = nt->item( "TrkMomFix", m_trkMomFixed );
      status = nt->item( "Addr", m_evtAddrColl );
      status = nt->item( "Address", m_evtAddrCollEx );
    }
  }
  return status;
}
 
// Event callback
StatusCode EvtCollectionWrite::execute() {
  auto&                    log = msgStream();
  SmartDataPtr<DataObject> evtRoot( eventSvc(), "/Event" );
  SmartDataPtr<Event>      evt( eventSvc(), "/Event/Header" );
  if ( evt ) {
    int                         evt_num = evt->event();
    SmartDataPtr<MyTrackVector> trkCont( eventSvc(), "/Event/MyTracks" );
    if ( trkCont != 0 ) {
      // Force an object update since now the original tracks should be
      // present and the local pointers can be updated!
      m_evtAddrCollEx = evtRoot->registry()->address();
      m_evtAddrColl   = evtRoot->registry()->address();
      m_ntrkColl      = trkCont->size();
      m_eneColl       = 0.f;
      log << MSG::DEBUG << " ->Track:";
      for ( size_t j = 0; j < 100; ++j ) { m_trkMomFixed[j] = 0; }
      int cnt = 0;
      for ( const auto& i : *trkCont ) {
        float p = sqrt( i->px() * i->px() + i->py() * i->py() + i->pz() * i->pz() );
        if ( cnt < 5000 ) m_trkMom[cnt] = p;
        if ( cnt < 5 ) m_trkMomFixed[cnt] = p;
        m_eneColl += p;
        ++cnt;
      }
      m_trackItem = ( 0 == m_ntrkColl ) ? 0 : ( *trkCont->begin() );
      if ( evt_num < 10 || evt_num % 500 == 0 ) {
        log << endmsg;
        log << MSG::INFO << "================ EVENT:" << evt->event() << " RUN:" << evt->run()
            << " ====== N(Track)=" << m_ntrkColl;
        if ( m_ntrkColl < m_nMCcut.value() ) {
          log << " FAILED selection (<" << m_nMCcut.value() << ") ============" << endmsg;
        } else {
          log << " PASSED selection (>=" << m_nMCcut.value() << ") ============" << endmsg;
          return m_evtTupleSvc->writeRecord( "/NTUPLES/EvtColl/Dir1/Dir2/Dir3/Collection" );
        }
      }
      return StatusCode::SUCCESS;
    }
  }
  error() << "Unable to retrieve Event Header object" << endmsg;
  return StatusCode::FAILURE;
}