LCDG4LcioHelper.hh

Go to the documentation of this file.

//\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/
//
// File: LCDG4LcioHelper.hh
// Module: lcdg4
//
// Purpose: Prepare LCIO data objects for persistency, and
//      save them into an LCIO output file
//
// 20040805 - G.Lima - Created
//
//\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/
#ifndef _LCDG4LCIOHELPER_HH_
#define _LCDG4LCIOHELPER_HH_

#include "LCDG4CalHit.hh"
#include "LCDG4TrackerHit.hh"
#include "LCDMcPart.hh"

#include "IO/LCWriter.h"
#include "IMPL/SimCalorimeterHitImpl.h"
#include "IMPL/SimTrackerHitImpl.h"
#include "IMPL/MCParticleImpl.h"
#include "IMPL/LCRunHeaderImpl.h"

#include <fstream>
#include <vector>

// forward declarations
class G4Run;
class G4Event;
class IMPL::LCCollectionVec;

class LCDG4LcioHelper {
public:  // typedefs
  typedef std::map<int,IMPL::MCParticleImpl*> TagToMCParticleMap;
  typedef std::map<IMPL::MCParticleImpl*,int> MCParticleToTagMap;

  // object accessor
  static LCDG4LcioHelper* getInstance();

  // destructor
  ~LCDG4LcioHelper();

  IO::LCWriter* getWriter() const { return _writer; }
  void openLcioFile(const std::string& filename);
  void beginRun(const G4Run*, const std::string& detname);
  void endRun();
  void writeEvent(const G4Event*);
  void reset();

  void saveCollection(const std::string& collName,
                      IMPL::LCCollectionVec* coll);

  // Create calorimeter-like hits
  IMPL::SimCalorimeterHitImpl* getCalHitLcio(const LCDG4CalHit& hit);
  // Create tracker-like hits
  IMPL::SimTrackerHitImpl* getTrkHitLcio(const LCDG4TrackerHit& hit);

  // Create MCParticle table
  void buildMCParticleColl(const std::vector<LCDMcPart*>& lcdMcpColl);
  void buildParentageLinks(const std::vector<LCDMcPart*>& lcdMcpColl);
  IMPL::MCParticleImpl* getMCParticleLcio(LCDMcPart& part);
  void prepareMCParticleCollection(IMPL::LCCollectionVec* newColl);
  void insertAllFamily(IMPL::LCCollectionVec* newColl,
                       TagToMCParticleMap::iterator head);

  void dumpParticleLists(const std::vector<LCDMcPart*>& lcdMcpColl) const;

private:
  // constructor
  LCDG4LcioHelper();

private: // attributes
  static LCDG4LcioHelper* _me;

  IO::LCWriter* _writer;
  TagToMCParticleMap* _mcPartColl;
  MCParticleToTagMap _mcp2tag;
  IMPL::LCRunHeaderImpl* _runHdr;
  std::map<std::string, IMPL::LCCollectionVec*> _savedColls;
};

#endif // _LCDG4LCIOHELPER_HH_

---