In This Package:

SimPruneTool.cc

Go to the documentation of this file.

#include "SimPruneTool.h"

#include "Event/SimHeader.h"

#include "DybKernel/ObjectReg.h"

#include <string>
using namespace std;
using namespace DayaBay;

SimPruneTool::SimPruneTool(const std::string& type,
                           const std::string& name,
                           const IInterface* parent)
    : GaudiTool(type,name,parent)
{
    declareInterface<IRegSeqVisitor>(this);

    declareProperty("Location", m_location = SimHeader::defaultLocation(),
                    "Location of SimHeader that gets its hits pruned.");
}

SimPruneTool::~SimPruneTool()
{
}
 
StatusCode SimPruneTool::visit(DayaBay::RegistrationSequence& rs)
{
    debug () << "visit() with " << rs.size() << " registrations" << endreq;

    // vector<ObjectReg>
    IRegistrationSequence::Registrations regs = rs.registrations();
    for (size_t ind=0; ind<regs.size(); ++ind) {
        ObjectReg& objreg = regs[ind];

        const string& path = objreg.path();
        if (path != m_location) continue;

        DataObject* dobj = objreg.object();
        if (!dobj) continue;

        SimHeader* sh = dynamic_cast<SimHeader*>(dobj);
        if (!sh) {
            error() << "Data object at \"" << path << "\" not a SimHeader" << endreq;
            return StatusCode::FAILURE;
        }

        const SimHitHeader* chits = sh->hits();
        if (!chits) {
            debug() << "No SimHitHeader, my job is already done" << endreq;
            return StatusCode::SUCCESS;
        }
        SimHitHeader* shits = const_cast<SimHitHeader*>(chits);
        debug() << "Pruning " << shits->hitCollection().size() << " hit collections" << endreq;
        sh->setHits(0);
        delete shits;       // One usually uses TP for this operation.

        break;
    }

    return StatusCode::SUCCESS;
}

---