In This Package:

RunDataWriterAlg.cc

Go to the documentation of this file.

#include "RunDataWriterAlg.h"

#include "Event/RunHeader.h"
#include "Event/RunData.h"
#include "DybKernel/IDybStorageSvc.h"
#include "DataSvc/IRunDataSvc.h"
#include "GaudiKernel/RegistryEntry.h"

RunDataWriterAlg::RunDataWriterAlg(const std::string& name, 
                                   ISvcLocator* pSvcLocator)
  : GaudiAlgorithm(name,pSvcLocator),
    m_storageSvc(0),
    m_runDataSvc(0)
{
  declareProperty("RunDataLocation",
                  m_runDataLocation=DayaBay::RunHeaderLocation::Default,
                  "Location in file where Run Data is to be found.");
  declareProperty("WriteToFile",m_writeToFile=false,
                  "Write the run data to the current output file?");
  declareProperty("WriteToDatabase",m_writeToDatabase=false,
                  "Write the run data to the offline database?");
}

RunDataWriterAlg::~RunDataWriterAlg()
{
}

StatusCode RunDataWriterAlg::initialize()
{
  // Initialize the necessary services
  if( m_writeToFile ){
    StatusCode sc = this->service("DybStorageSvc",m_storageSvc,true);
    if(sc.isFailure()){
      error() << "Failed to get DybStorageSvc" << endreq;
      return sc;
    }
  }
  StatusCode sc = this->service("RunDataSvc",m_runDataSvc,true);
  if(sc.isFailure()){
    error() << "Failed to get RunDataSvc" << endreq;
    return sc;
  }
  return StatusCode::SUCCESS;
}

StatusCode RunDataWriterAlg::execute()
{
  return StatusCode::SUCCESS;
}

StatusCode RunDataWriterAlg::finalize()
{
  if( m_writeToFile ) m_storageSvc->release();
  m_runDataSvc->release();
  return StatusCode::SUCCESS;
}


StatusCode RunDataWriterAlg::endRun()
{
  // Write the run data to file/database at the end of the run
  StatusCode sc = StatusCode::SUCCESS;
  if( m_writeToFile ) sc = this->writeDataToFile();
  if( !sc.isSuccess() ) return sc;
  if( m_writeToDatabase ) sc = this->writeDataToDatabase();
  return sc;
}

StatusCode RunDataWriterAlg::writeDataToFile(){
  // Write Run Data to current output file

  debug() << "Writing RunData" << endreq;
  // Create Run Header
  DayaBay::RunHeader runHeader;
  runHeader.setRunDataList( m_runDataSvc->cachedRunData() );
  // Have to put RunHeader in a registry for DybStorageSvc to work correctly
  DataSvcHelpers::RegistryEntry* regEntry 
    = new DataSvcHelpers::RegistryEntry(m_runDataLocation);
  regEntry->setObject(&runHeader);
  debug() << "Storing RunHeader" << endreq;
  StatusCode sc = m_storageSvc->store(&runHeader, m_runDataLocation);
  debug() << "Stored RunHeader" << endreq;
  if(sc.isFailure()){
    error() << "Failed to store Run Header to output file" << endreq;
  }
  // Clear run data to prevent deleting
  debug() << "Clearing RunData in RunHeader" << endreq;
  std::vector<DayaBay::RunData*> emptyList;
  runHeader.setRunDataList( emptyList );
  debug() << "Cleared RunData in RunHeader" << endreq;
  return sc;
}

StatusCode RunDataWriterAlg::writeDataToDatabase(){
  // Write Run Data to offline database
  warning() << "Writing Run Data to offline database is not yet implemented."
            << endreq;
  return StatusCode::SUCCESS;
}

---