CbmLitMCTrackCreator.h

Go to the documentation of this file.

/* Copyright (C) 2011-2020 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer], Timur Ablyazimov */

#ifndef CBMLITMCTRACKCREATOR_H_
#define CBMLITMCTRACKCREATOR_H_
 
#include "CbmDefs.h"
#include "CbmLitMCTrack.h"
#include "CbmStsAddress.h"
 
#include <cassert>
#include <iostream>
#include <map>
 
class TClonesArray;
class FairMCPoint;
class CbmMvdPoint;
class CbmStsPoint;
class CbmTrdPoint;
class CbmMuchPoint;
class CbmGeoStsPar;
class CbmStsDigiPar;
class CbmRichRingFitterEllipseTau;
class CbmMCDataArray;
class CbmDigiManager;

class CbmLitMCTrackCreator {
 private:
  CbmLitMCTrackCreator();
 
 public:
  virtual ~CbmLitMCTrackCreator();

  static CbmLitMCTrackCreator* Instance();

  void Create(Int_t eventNum);
 

  bool TrackExists(int mcEventId, int mcId) const
  {
    return (fLitMCTracks.count(std::make_pair(mcEventId, mcId)) > 0) ? true : false;
  }

  const CbmLitMCTrack& GetTrack(int mcEventId, int mcId) const
  {
    assert(TrackExists(mcEventId, mcId));
    return fLitMCTracks.find(std::make_pair(mcEventId, mcId))->second;
  }

  Int_t GetNofTracks() const { return fLitMCTracks.size(); }
 
 private:
  void ReadDataBranches();

  void AddPoints(ECbmModuleId detId, CbmMCDataArray* array, Int_t iEvent);

  void AddRichHits(Int_t iEvent);

  void AddRingParameters(Int_t iEvent);

  void FairMCPointToLitMCPoint(const FairMCPoint* fairPoint, CbmLitMCPoint* litPoint, int eventId, int refId,
                               int stationId);

  void FillStationMaps(Int_t iEvent);
 
 
  void FairMCPointCoordinatesAndMomentumToLitMCPoint(const FairMCPoint* fairPoint, CbmLitMCPoint* litPoint);
 
  void MvdPointCoordinatesAndMomentumToLitMCPoint(const CbmMvdPoint* mvdPoint, CbmLitMCPoint* litPoint);
 
  void StsPointCoordinatesAndMomentumToLitMCPoint(const CbmStsPoint* stsPoint, CbmLitMCPoint* litPoint);
 
  void TrdPointCoordinatesAndMomentumToLitMCPoint(const CbmTrdPoint* trdPoint, CbmLitMCPoint* litPoint);
 
  void MuchPointCoordinatesAndMomentumToLitMCPoint(const CbmMuchPoint* muchPoint, CbmLitMCPoint* litPoint);
 
  CbmMCDataArray* fMCTracks;    // CbmMCTrack array
  CbmMCDataArray* fMvdPoints;   // CbmMvdPoint array
  CbmMCDataArray* fStsPoints;   // CbmStsPoint array
  CbmMCDataArray* fTrdPoints;   // CbmTrdPoint array
  CbmMCDataArray* fMuchPoints;  // CbmMuchPoint array
  CbmMCDataArray* fTofPoints;   // CbmTofPoint array
  CbmMCDataArray* fRichPoints;  // CbmRichPoint array
  TClonesArray* fRichHits;      // CbmRichHit array
  CbmDigiManager* fDigiMan;
 
  // Stores created CbmLitMCTrack objects.
  // std::map<MC track index, CbmLitMCTrack object>.
  std::map<std::pair<int, int>, CbmLitMCTrack> fLitMCTracks;
 
  // Map <MC point index, station index>
  std::map<std::pair<int, int>, int> fMvdStationsMap;   // for MVD
  std::map<std::pair<int, int>, int> fStsStationsMap;   // for STS
  std::map<std::pair<int, int>, int> fTrdStationsMap;   // for TRD
  std::map<std::pair<int, int>, int> fMuchStationsMap;  // for MUCH
 
  CbmRichRingFitterEllipseTau* fTauFit;  // Ellipse fitter algorithm
 
  CbmLitMCTrackCreator(const CbmLitMCTrackCreator&);
  CbmLitMCTrackCreator operator=(const CbmLitMCTrackCreator&);
};
 
#endif /* CBMLITMCTRACKCREATOR_H_ */