In This Package:

EsFrontEndAlg.cc

Go to the documentation of this file.

#include "EsFrontEndAlg.h"

#include "Event/SimHeader.h"
#include "Event/ElecHeader.h"
#include "Event/ElecFeeCrate.h"
#include "Event/ElecFecCrate.h"

#include "DataSvc/IRunDataSvc.h"
#include "Event/RunData.h"

#include "ElecSim/IEsPulseTool.h"
#include "ElecSim/IEsFrontEndTool.h"

#include "CLHEP/Units/SystemOfUnits.h"

EsFrontEndAlg::EsFrontEndAlg(const std::string& name, ISvcLocator* pSvcLocator)
    : DybAlgorithm<DayaBay::ElecHeader>(name,pSvcLocator)
    , m_pmtTool(0)
    , m_rpcTool(0)
    , m_feeTool(0)
    , m_fecTool(0)
{
    declareProperty("SimLocation",m_simLocation=DayaBay::SimHeaderLocation::Default,
                    "Location in the TES where the input SimHeader is to be found.");
    declareProperty("Detectors",m_detectorNames,"List of active detectors");
    declareProperty("PmtTool",m_pmtToolName="EsPmtEffectPulseTool",
                    "Name of the PMT simulation tool");
    declareProperty("RpcTool",m_rpcToolName="EsIdealPulseTool",
                    "Name of the RPC simulation tool");
    declareProperty("FeeTool",m_feeToolName="EsIdealFeeTool",
                    "Name of the PMT Front-end electronics simulation tool");
    declareProperty("FecTool",m_fecToolName="EsIdealFecTool",
                    "Name of the RPC Front-end electronics simulation tool");
    declareProperty("MaxSimulationTime",
                    m_maxSimulationTime=50*CLHEP::microsecond,
                    "Maximum time window length for electronics simulation");
    // Default configuration includes all PMT detectors
    m_detectorNames.push_back("DayaBayAD1");
    m_detectorNames.push_back("DayaBayAD2");
    m_detectorNames.push_back("DayaBayIWS");
    m_detectorNames.push_back("DayaBayOWS");
    m_detectorNames.push_back("LingAoAD1");
    m_detectorNames.push_back("LingAoAD2");
    m_detectorNames.push_back("LingAoIWS");
    m_detectorNames.push_back("LingAoOWS");
    m_detectorNames.push_back("FarAD1");
    m_detectorNames.push_back("FarAD2");
    m_detectorNames.push_back("FarAD3");
    m_detectorNames.push_back("FarAD4");
    m_detectorNames.push_back("FarIWS");
    m_detectorNames.push_back("FarOWS");
    m_detectorNames.push_back("SABAD1");
    m_detectorNames.push_back("SABAD2");
}

EsFrontEndAlg::~EsFrontEndAlg()
{
}

StatusCode EsFrontEndAlg::initialize()
{

  // Convert detector names to Detector IDs
  int nDetNames = m_detectorNames.size();
  for(int detIdx=0; detIdx<nDetNames; detIdx++){
    DayaBay::Detector det(m_detectorNames[detIdx]);
    if(det.site() == Site::kUnknown 
       || det.detectorId() == DetectorId::kUnknown){
      error() << "Invalid detector name: " << m_detectorNames[detIdx] << endreq;
      return StatusCode::FAILURE;
    }
    m_detectors.push_back(det);
  }
  
  // Get PMT Pulse tool
  try {
    m_pmtTool = tool<IEsPulseTool>(m_pmtToolName);
  }
  catch(const GaudiException& exg) {
    fatal() << "Failed to get pmt tool: \"" << m_pmtToolName << "\"" << endreq;
            return StatusCode::FAILURE;
  }
  info () << "Added tool " << m_pmtToolName << endreq;
  
  // Get RPC Pulse tool
  if(m_pmtToolName == m_rpcToolName){
    m_rpcTool = m_pmtTool;
  }else{
    try {
      m_rpcTool = tool<IEsPulseTool>(m_rpcToolName);
    }
    catch(const GaudiException& exg) {
      fatal() << "Failed to get rpc tool: \"" << m_rpcToolName << "\"" << endreq;
      return StatusCode::FAILURE;
    }
    info () << "Added tool " << m_rpcToolName << endreq;
  }

  // Get PMT Front-end Electronics tool
  try {
    m_feeTool = tool<IEsFrontEndTool>(m_feeToolName);
  }
  catch(const GaudiException& exg) {
    fatal() << "Failed to get fee tool: \"" << m_feeToolName << "\"" << endreq;
            return StatusCode::FAILURE;
  }
  info () << "Added tool " << m_feeToolName << endreq;

  // Get RPC Front-end Electronics tool
  try {
    m_fecTool = tool<IEsFrontEndTool>(m_fecToolName);
  }
  catch(const GaudiException& exg) {
    fatal() << "Failed to get fec tool: \"" << m_fecToolName << "\"" << endreq;
            return StatusCode::FAILURE;
  }
  info () << "Added tool " << m_fecToolName << endreq;

  // Set the 'partition' for this run in RunDataSvc
  IRunDataSvc* runDataSvc = svc<IRunDataSvc>("RunDataSvc",true);
  if(!runDataSvc){
    error() << "Failed to get RunDataSvc" << endreq;
    return StatusCode::FAILURE;
  }
  int task = 1;  // Tells RunDataSvc to use new simulation run data.
  Context emptyContext; // Need a dummy context for this service call
  ServiceMode svcMode(emptyContext, task);
  const DayaBay::RunData* runData = runDataSvc->runData(svcMode);
  if(!runData){
    error() << "Failed to get RunData for simulation" << endreq;
    return StatusCode::FAILURE;
  }
  DayaBay::RunData modifiedRunData(*runData);
  modifiedRunData.setDetectors(m_detectors);
  StatusCode sc = runDataSvc->setRunData(modifiedRunData);
  if( !sc.isSuccess() ){
    error() << "Failed to set detectors in run" << endreq;
  }
  return sc;
}

StatusCode EsFrontEndAlg::execute()
{

    // Get input header
    DayaBay::SimHeader* simHeader = getTES<DayaBay::SimHeader>(m_simLocation);

    // Create output header 
    DayaBay::ElecHeader* elecHeader = MakeHeaderObject();

    // Set the simulation time range
    // Add padding to ElecHeader time window
    TimeStamp earliest = simHeader->earliest();
    TimeStamp latest = simHeader->latest();

    earliest.Add(-DayaBay::preTimeTolerance/CLHEP::second);
    latest.Add(DayaBay::postTimeTolerance/CLHEP::second);

    elecHeader->setContext(simHeader->context());
    elecHeader->setEarliest(earliest);
    elecHeader->setLatest(latest);
    elecHeader->setTimeStamp(earliest);

    // Add the pulse collection header
    DayaBay::ElecPulseHeader* pulseHeader = 
      new DayaBay::ElecPulseHeader(elecHeader);
    elecHeader->setPulseHeader(pulseHeader);

    // Add the crate header
    DayaBay::ElecCrateHeader* crateHeader = 
      new DayaBay::ElecCrateHeader(elecHeader);
    elecHeader->setCrateHeader(crateHeader);

    double simDeltaT = (latest-earliest).GetSeconds()*CLHEP::second;
    if( simDeltaT > m_maxSimulationTime ){
      //error() << "SimHits span to much time (" << simDeltaT/CLHEP::microsecond 
            //  << " us).  Simulate kinematics individually," 
            //  << " or use the Fifteen algorithm." << endreq;
      //return StatusCode::FAILURE;
      info() << "SimHits span too much time (" << simDeltaT/CLHEP::second 
              << " us).  Simulate kinematics individually." << endreq;
      return StatusCode::SUCCESS;
    }

    // Get the SimHit Header
    const DayaBay::SimHitHeader* simHitHeader = simHeader->hits();
    if ( !simHitHeader ) {
      info() << "Empty simhitheader. Do not try to process non-existent hit collections." <<endreq;
    }
    else {
    const DayaBay::SimHitHeader::hc_map& hcMap = simHitHeader->hitCollection();

    info() << "Processing hit collections" <<endreq;

    DayaBay::SimHitHeader::hc_map::const_iterator hcIter;
    for (hcIter=hcMap.begin(); hcIter != hcMap.end(); ++hcIter) {

      DayaBay::Detector det(hcIter->first);
      debug() << "Checking " << det.detName() << " for hits." << endreq;
      DayaBay::SimHitCollection* hits = hcIter->second;
      if(!hits) return StatusCode::FAILURE;

      debug() << "Got hit collection from " << det.detName()<<" (id= "
              << det.siteDetPackedData() << ") " << " with " 
              << hits->collection().size() << " hits." << endreq;

      StatusCode sc;
      // Check if this detector should be simulated.
      if( std::find(m_detectors.begin(),m_detectors.end(),det) 
          == m_detectors.end()){
        debug() << "Detector " << det.detName() << " not simulated." << endreq;
        continue;
      }
      if( det.isAD() || det.isWaterShield() ){
        // Process PMT hits
        debug() << "Processing PMT hits" <<endreq;
        DayaBay::ElecPulseCollection* pulses = 
          new DayaBay::ElecPulseCollection(pulseHeader, det);
        sc = m_pmtTool->generatePulses(hits, pulses);
        if( sc != StatusCode::SUCCESS ) return sc;
        pulseHeader->addPulseCollection(pulses);
        
        DayaBay::ElecFeeCrate* crate = new DayaBay::ElecFeeCrate(det,
                                                                 crateHeader);
        sc = m_feeTool->generateSignals(pulses, crate);
        if( sc != StatusCode::SUCCESS ) return sc;
        crateHeader->addCrate(crate);

      }else if(det.detectorId() == DetectorId::kRPC){
        // Process RPC hits
        debug() << "Processing RPC hits" <<endreq;
        DayaBay::ElecPulseCollection* pulses = 
          new DayaBay::ElecPulseCollection(pulseHeader, det);
        sc = m_rpcTool->generatePulses(hits, pulses);
        if( sc != StatusCode::SUCCESS ) return sc;
        pulseHeader->addPulseCollection(pulses);
        
        DayaBay::ElecFecCrate* crate = new DayaBay::ElecFecCrate(det, 
                                                                 crateHeader);
        sc = m_fecTool->generateSignals(pulses, crate);
        if( sc != StatusCode::SUCCESS ) return sc;
        crateHeader->addCrate(crate);
      }else{
        error() << "Unknown detector " << det << endreq; 
        return StatusCode::FAILURE;
      }
    }

    }
    return StatusCode::SUCCESS;
}

StatusCode EsFrontEndAlg::finalize()
{
    return StatusCode::SUCCESS;
}

---