PndSttMatchTracks.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

// -------------------------------------------------------------------------
// -----                  CbmStsMatchTracks header file                -----
// -----                  Created 22/11/05  by V. Friese               -----
// -------------------------------------------------------------------------

#ifndef PNDSTTMATCHTRACKS_H
#define PNDSTTMATCHTRACKS_H 1

#include <PndPersistencyTask.h>
#include <vector>
#include <map>
#include "PndSttHit.h"
#include "FairMCPoint.h"

class TClonesArray;

class PndSttMatchTracks : public PndPersistencyTask {

 public:
  PndSttMatchTracks();

  PndSttMatchTracks(Int_t verbose);

  PndSttMatchTracks(const char *name, const char *title = "Pnd Stt Match Tracks Task", Int_t verbose = 1);

  virtual ~PndSttMatchTracks();

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  void AddHitCollectionName(char *hitCollectionName, char *pointCollectionName);
  PndSttHit *GetHitFromCollections(Int_t hitCounter);
  FairMCPoint *GetPointFromCollections(Int_t hitCounter);

  void SetPersistence(Bool_t persistence) { SetPersistency(persistence); }

 private:
  void AddAllCollections();
  void AddHitCollection(char const *collectionName, char const *pointCollectionName);

  TClonesArray *fTrackCandidates; // Array of PndTrackCand
  TClonesArray *fMatches;         // Array of PndSttTrackMatch

  std::map<Int_t, Int_t> fMatchMap;

  Int_t fVerbose;

  Bool_t fCollectionsComplete;

  std::vector<std::string> fHitCollectionNames;
  std::vector<std::string> fPointCollectionNames;
  TList fHitCollectionList;
  TList fPointCollectionList;

  PndSttMatchTracks(const PndSttMatchTracks &L);
  PndSttMatchTracks &operator=(const PndSttMatchTracks &) { return *this; };

  ClassDef(PndSttMatchTracks, 1);
};

#endif