DybGenPrune.cc

Go to the documentation of this file.

#include "DybGenPrune.h"
#include "Event/GenHeader.h"

#include "HepMC/GenEvent.h"

DybGenPrune::DybGenPrune(const std::string& name, ISvcLocator* pSvcLocator)
    : GaudiAlgorithm(name,pSvcLocator)
    , m_genPrune(0)
{
    declareProperty("MaxVertices",m_maxVertices=0,
                    "Maximum number of vertices to survive the pruning");
    declareProperty("Location",m_location = DayaBay::GenHeader::defaultLocation(),
                    "Location in TES in which to look for the GenHeader");    
    declareProperty("GenPruneName",m_genPruneName = "GenPruneTool", "Name of GenHeader pruner");
}

DybGenPrune::~DybGenPrune()
{
}

StatusCode DybGenPrune::initialize()
{
    this->GaudiAlgorithm::initialize();

    m_genPrune = tool<IGenPruneTool>(m_genPruneName);
    if( !m_genPrune ) {
      error() << "Failed to retrive GenPruneTool" << endreq;
      return StatusCode::FAILURE;
    }

    m_genPrune->setMaxVertices( m_maxVertices );

    return StatusCode::SUCCESS;
}

StatusCode DybGenPrune::execute()
{

    DayaBay::GenHeader* gh = get<DayaBay::GenHeader>(m_location);
    if (!gh) {
        warning() << "No GenHeader found at " << m_location 
                  << ", my work here is already done."
                  << endreq;
        return StatusCode::SUCCESS;
    }
    
    m_genPrune->prune( gh );

    return StatusCode::SUCCESS;
}

StatusCode DybGenPrune::finalize()
{

    return this->GaudiAlgorithm::finalize();
}