FileWriter.h

Go to the documentation of this file.

 
#ifndef PANDORA_FILE_WRITER_H
#define PANDORA_FILE_WRITER_H 1
 
#include "Pandora/StatusCodes.h"
 
#include "Persistency/PandoraIO.h"
#include "Persistency/Persistency.h"
 
#include <string>
 
namespace pandora
{
 
class Pandora;
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
class FileWriter : public Persistency
{
public:
    FileWriter(const pandora::Pandora &pandora, const std::string &fileName);
 
    virtual ~FileWriter();
 
    StatusCode WriteGeometry();
 
    StatusCode WriteEvent(const CaloHitList &caloHitList, const TrackList &trackList, const MCParticleList &mcParticleList,
        const bool writeMCRelationships = true, const bool writeTrackRelationships = true);
 
protected:
    virtual StatusCode WriteHeader(const ContainerId containerId) = 0;
 
    virtual StatusCode WriteFooter() = 0;
 
    virtual StatusCode WriteSubDetector(const SubDetector *const pSubDetector) = 0;
 
    virtual StatusCode WriteLArTPC(const LArTPC *const pLArTPC) = 0;
 
    virtual StatusCode WriteDetectorGap(const DetectorGap *const pDetectorGap) = 0;
 
    virtual StatusCode WriteCaloHit(const CaloHit *const pCaloHit) = 0;
 
    virtual StatusCode WriteTrack(const Track *const pTrack) = 0;
 
    virtual StatusCode WriteMCParticle(const MCParticle *const pMCParticle) = 0;
 
    virtual StatusCode WriteRelationship(const RelationshipId relationshipId, const void *address1, const void *address2, const float weight = 1.f) = 0;
 
private:
    StatusCode WriteSubDetectorList();
 
    StatusCode WriteLArTPCList();
 
    StatusCode WriteDetectorGapList();
 
    StatusCode WriteTrackList(const TrackList &trackList);
 
    StatusCode WriteCaloHitList(const CaloHitList &caloHitList);
 
    StatusCode WriteMCParticleList(const MCParticleList &mcParticleList);
 
    StatusCode WriteCaloHitToMCParticleRelationships(const CaloHitList &caloHitList);
 
    StatusCode WriteTrackToMCParticleRelationships(const TrackList &trackList);
 
    StatusCode WriteMCParticleRelationships(const MCParticleList &mcParticleList);
 
    StatusCode WriteTrackRelationships(const TrackList &trackList);
 
    StatusCode WriteCaloHitToMCParticleRelationship(const CaloHit *const pCaloHit);
 
    StatusCode WriteTrackToMCParticleRelationship(const Track *const pTrack);
 
    StatusCode WriteMCParticleRelationships(const MCParticle *const pMCParticle);
 
    StatusCode WriteTrackRelationships(const Track *const pTrack);
};
 
} // namespace pandora
 
#endif // #ifndef PANDORA_FILE_WRITER_H