In This Package:

SimHeaderCnv.cc

Go to the documentation of this file.

#include "SimHeaderCnv.h"
#include "PerBaseEvent/HeaderObjectCnv.h"

#include "PerSimEvent/PerSimHitHeader.h"
#include "Event/SimHitHeader.h"

#include "PerSimEvent/PerSimUnobservableStatistics.h"
#include "Event/SimUnobservableStatisticsHeader.h"

using namespace DayaBay;
using namespace std;

SimHeaderCnv::SimHeaderCnv(ISvcLocator* svc)
    : RootIOTypedCnv<PerSimHeader,SimHeader>("PerSimHeader",
                                             classID(),svc)
{
    // set maximum number of warnings of "no particle history to save"
    m_maxNoPHtoS = 10;
    m_NoPHtoSmsgs=  0;

    m_pruneSimHits = false;
    char* foo = getenv("NUWA_PRUNESIMHITS");
    if (foo) {
        m_pruneSimHits = true;
    }
}

SimHeaderCnv::~SimHeaderCnv()
{
}


StatusCode SimHeaderCnv::PerToTran(const PerSimHeader& perobj,
                                   DayaBay::SimHeader& tranobj)
{
    StatusCode sc = HeaderObjectCnv::toTran(perobj,tranobj);
    if (sc.isFailure()) return sc;

    MsgStream log(msgSvc(), "SimHeaderCnv::PerToTran");

    // Convert the historian part
    vector<SimTrack*> timap;
    const PerSimParticleHistory* pph = perobj.history;
    if (!pph) {
        log << MSG::DEBUG
            << "No particle history in persistent object." << endreq;
        tranobj.setParticleHistory(0);
    }
    else {
        SimParticleHistory* ph = convert(timap,tranobj,*pph);
        if (ph) {
            tranobj.setParticleHistory(ph);
        }
        else {
            log << MSG::ERROR
                << "Failed to convert the SimParticleHistory" << endreq;
        }
    }

    if (perobj.hits) { // Convert the Hits part
        const PerSimHitHeader::phc_map& pcolmap = perobj.hits->hitCollections;

        log << MSG::DEBUG << "Reading " << pcolmap.size() << " hit collections" << endreq;

        SimHitHeader* shit_header = new SimHitHeader;
        shit_header->setHeader(&tranobj);
        tranobj.setHits(shit_header);

        PerSimHitHeader::phc_map::const_iterator it, done = pcolmap.end();
        for (it = pcolmap.begin(); it != done; ++it) {
            int detectorID = it->first;
            const PerSimHitCollection* phc = it->second;
            SimHitCollection* hc = convert(timap,*phc);
            hc->setHeader(shit_header);
            shit_header->addHitCollection(hc);

            DayaBay::Detector det((short)detectorID);
            log << MSG::DEBUG << "Reading " << phc->hits.size() 
                << " hits for detector " << det.detName()
                << " ("
                << detectorID 
                << ")"
                << endreq;
        }
    }
    else {
        log << MSG::DEBUG << "No persistent hits to convert" << endreq;
    }

    { // Convert the stats part
        const PerSimUnobservableStatistics* pstat = perobj.stats;
        if (!pstat) {
            log << MSG::DEBUG
                << "No unobservable statistics in persistent object." << endreq;
            tranobj.setUnobservableStatistics(0);
        }
        else {
            SimUnobservableStatisticsHeader* stat = convert(*pstat);
            tranobj.setUnobservableStatistics(stat);
        }
    }

    return StatusCode::SUCCESS;
}

StatusCode SimHeaderCnv::TranToPer(const DayaBay::SimHeader& tranobj,
                                   PerSimHeader& perobj)
{
    StatusCode sc = HeaderObjectCnv::toPer(tranobj,perobj);
    if (sc.isFailure()) return sc;

    MsgStream log(msgSvc(), "SimHeaderCnv::TranToPer");

    map<const SimTrack*,int> trackMap;

    { // Convert the historian part
        const SimParticleHistory* ph = tranobj.particleHistory();
        delete perobj.history;
        if (!ph) {
          m_NoPHtoSmsgs += 1;
          if ( m_NoPHtoSmsgs < m_maxNoPHtoS ) {
            log << MSG::WARNING
                << "No Particle history to save" << endreq;
          }
          if ( m_NoPHtoSmsgs == m_maxNoPHtoS ) {
            log << MSG::WARNING
                << "No Particle history to save. Additional warnings will be suppressed. " << endreq;
          }
          
            perobj.history = 0;
        }
        else {
            PerSimParticleHistory* pph = convert(trackMap,*ph);
            perobj.history = pph;
        }
    }

    { // Convert the Hits part
        if (perobj.hits) {
            delete perobj.hits;
            perobj.hits = 0;
        }
        perobj.hits = new PerSimHitHeader();
        PerSimHitHeader::phc_map& phcolmap = perobj.hits->hitCollections;
        if (m_pruneSimHits) {
            static int count = 0; 
            ++count;
            if (count<10) {
                log << MSG::INFO << "Not saving SimHits, will say " << 10-count << " more times" << endreq;
            }
        }
        else {
            const SimHitHeader* header = tranobj.hits();
            if (!header) {
                static int count = 0; 
                if (count++ < 10) {
                    log << MSG::WARNING << "No Hit Header (I will say it " 
                        << 10 - count << " times again)" << endreq;
                }
            }
            else {
                const SimHitHeader::hc_map& colmap = header->hitCollection();
                
                log << MSG::DEBUG << "Saving " << colmap.size() << " hit collections" << endreq;
                
                SimHitHeader::hc_map::const_iterator it, done = colmap.end();
                for (it = colmap.begin(); it != done; ++it) {
                    int detectorID = it->first;
                    static int nUnknownDet = 0;
                    if( detectorID == DetectorId::kUnknown ) {
                        nUnknownDet++;
                        if( nUnknownDet<10 ) {
                            log << MSG::WARNING << "No hit should be from unknown detector (I will say it "
                                << 10 - nUnknownDet << " times again)" << endreq;
                        }
                    } 
                    else {
                        const SimHitCollection* hc = it->second;
                        PerSimHitCollection* phc = convert(trackMap,*hc);
                        phcolmap[detectorID] = phc;
                        DayaBay::Detector det((short)detectorID);
                        log << MSG::DEBUG << "Saving " << phc->hits.size() 
                            << " hits for detector " << det.detName()
                            << " ("
                            << detectorID 
                            << ")"
                            << endreq;
                    }
                }
            }
        }
        log << MSG::DEBUG << "Saving " << phcolmap.size() << " hit collections" << endreq;
    }    

    { // Convert the stats part
        const SimUnobservableStatisticsHeader* stat = tranobj.unobservableStatistics();
        if (!stat) {
            log << MSG::DEBUG
                << "No unobservable statistics" << endreq;
        }
        else {
            PerSimUnobservableStatistics* pstat = convert(*stat);
            delete perobj.stats;
            perobj.stats = pstat;
        }
    }

    return StatusCode::SUCCESS;
}


    
StatusCode SimHeaderCnv::fillRepRefs(IOpaqueAddress*, DataObject* dobj)
{
    MsgStream log(msgSvc(), "SimHeaderCnv::fillRepRefs");
    SimHeader* sh = dynamic_cast<SimHeader*>(dobj);

    log << MSG::DEBUG
        << "Saving links to " << sh->inputHeaders().size() 
        << " input headers" << endreq;

    StatusCode sc = HeaderObjectCnv::fillPer(m_rioSvc,*sh,*m_perOutObj);
    if (sc.isFailure()) {
        log << MSG::ERROR << "Failed to fill HeaderObject part" << endreq;
        return sc;
    }

    // ... fill SimHeader part...
    return sc;
}

StatusCode SimHeaderCnv::fillObjRefs(IOpaqueAddress*, DataObject* dobj)
{
    MsgStream log(msgSvc(), "SimHeaderCnv::fillObjRefs");
    HeaderObject* hobj = dynamic_cast<HeaderObject*>(dobj);
    StatusCode sc = HeaderObjectCnv::fillTran(m_rioSvc,*m_perInObj,*hobj);
    if (sc.isFailure()) {
        log << MSG::ERROR << "Failed to fill HeaderObject part" << endreq;
        return sc;
    }

    log << MSG::DEBUG
        << "Restored links to " << hobj->inputHeaders().size() 
        << " input headers" << endreq;

    // ... fill SimHeader part...

    DayaBay::SimHeader* sh = dynamic_cast<DayaBay::SimHeader*>(hobj);
    relate(*sh);

    return sc;
}

---