WriteRawData.cxx

Go to the documentation of this file.

// Include files.
#include "WriteRawData.h"
#include "GaudiKernel/IDataManagerSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "RawDataCnv/EventManagement/RAWEVENT.h"
#include <vector>
 
// Event Model related classes
#include "EmcRawEvent/EmcDigi.h"
#include "EventModel/Event.h"
#include "MdcRawEvent/MdcDigi.h"
#include "MucRawEvent/MucDigi.h"
#include "RawEvent/DigiEvent.h"
#include "TofRawEvent/TofDigi.h"
 
#include "EventModel/EventModel.h"
 
using namespace std;
 
// Constructor.
WriteRawData::WriteRawData( const string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {}
 
// WriteRawData::initialize().
StatusCode WriteRawData::initialize() {
 
  MsgStream log( msgSvc(), name() );
  m_count = 0;
 
  log << MSG::INFO << "in initialize()" << endmsg;
 
  return StatusCode::SUCCESS;
}
 
// WriteRawData::execute().
StatusCode WriteRawData::execute() {
  StatusCode sc;
  MsgStream  log( msgSvc(), name() );
 
  log << MSG::INFO << "================================================" << endmsg;
  log << MSG::INFO << "In execute()" << endmsg;
  log << MSG::INFO << "++++++++++++++++++++++++++++++++++++++++++++++++" << endmsg;
 
  static int evtnum = 0;
  static int runnum = 999;
 
  // Create the Event header and set it as "root" of the event store
  Event::EventHeader* event = new Event::EventHeader;
  event->setEventNumber( evtnum++ );
  event->setRunNumber( runnum );
 
  IDataManagerSvc* evtmgr = dynamic_cast<IDataManagerSvc*>( eventSvc() );
  sc                      = evtmgr->setRoot( EventModel::EventHeader, event );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event object" << endmsg;
    return sc;
  }
 
  DigiEvent* digi = new DigiEvent;
  log << MSG::INFO << " DigiEvent clID :: " << digi->clID() << endmsg;
  sc = eventSvc()->registerObject( EventModel::Digi::Event, digi );
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event/Digi object" << endmsg;
    return sc;
  }
 
  sc = writeMdcDigiData();
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event/Digi/MdcDigiCol object" << endmsg;
    return sc;
  }
 
  sc = writeEmcDigiData();
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event/Digi/EmcDigiCol object" << endmsg;
    return sc;
  }
 
  sc = writeTofDigiData();
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event/Digi/EmcDigiCol object" << endmsg;
    return sc;
  }
 
  sc = writeMucDigiData();
  if ( sc.isFailure() )
  {
    log << MSG::ERROR << "Unable to register /Event/Digi/EmcDigiCol object" << endmsg;
    return sc;
  }
 
  m_count++;
 
  return StatusCode::SUCCESS;
}
 
StatusCode WriteRawData::writeMdcDigiData() {
  int         n      = 10;
  MdcDigiCol* mdcCol = new MdcDigiCol;
  for ( int i = 0; i < n; i++ )
  {
    // Create new MdcDigi
    MdcDigi* mdcDigi = new MdcDigi( m_count );
    mdcDigi->setTimeChannel( 8 );
    mdcDigi->setChargeChannel( 10 );
    mdcDigi->setOverflow( 0 );
    // And add the stuff to the container
    mdcCol->push_back( mdcDigi );
  }
 
  StatusCode sc = StatusCode::SUCCESS;
  sc            = eventSvc()->registerObject( EventModel::Digi::MdcDigiCol, mdcCol );
  return sc;
}
 
StatusCode WriteRawData::writeEmcDigiData() {
  int         n      = 10;
  EmcDigiCol* emcCol = new EmcDigiCol;
  for ( int i = 0; i < n; i++ )
  {
    // Create new EmcDigi
    EmcDigi* emcDigi = new EmcDigi( m_count );
    emcDigi->setTimeChannel( 9 );
    emcDigi->setChargeChannel( 9 );
    emcDigi->setMeasure( 0 );
    // And add the stuff to the container
    emcCol->push_back( emcDigi );
  }
 
  StatusCode sc = StatusCode::SUCCESS;
  sc            = eventSvc()->registerObject( EventModel::Digi::EmcDigiCol, emcCol );
  return sc;
}
 
StatusCode WriteRawData::writeTofDigiData() {
  int         n      = 10;
  TofDigiCol* tofCol = new TofDigiCol;
  for ( int i = 0; i < n; i++ )
  {
    // Create new TofDigi
    TofDigi* tofDigi = new TofDigi( m_count );
    tofDigi->setTimeChannel( 10 );
    tofDigi->setChargeChannel( 8 );
    tofDigi->setOverflow( 0 );
    // And add the stuff to the container
    tofCol->push_back( tofDigi );
  }
 
  StatusCode sc = StatusCode::SUCCESS;
  sc            = eventSvc()->registerObject( EventModel::Digi::TofDigiCol, tofCol );
  return sc;
}
 
StatusCode WriteRawData::writeMucDigiData() {
  int         n      = 10;
  MucDigiCol* mucCol = new MucDigiCol;
  for ( int i = 0; i < n; i++ )
  {
    // Create new MucDigi
    MucDigi* mucDigi = new MucDigi( m_count );
    // And add the stuff to the container
    mucCol->push_back( mucDigi );
  }
 
  StatusCode sc = StatusCode::SUCCESS;
  sc            = eventSvc()->registerObject( EventModel::Digi::MucDigiCol, mucCol );
 
  return sc;
}
 
// WriteRawData::finalize().
StatusCode WriteRawData::finalize() {
 
  MsgStream log( msgSvc(), name() );
 
  log << MSG::INFO << "In finalize()" << endmsg;
 
  return StatusCode::SUCCESS;
}