PndPreselectSttHitsTask.h

Go to the documentation of this file.

//  PndPreselectSttHitsTask
//  Finds Track

#pragma once

#include "FairTask.h"
#include <vector>

#include "PndPreselectSttHits.h"
#include "PndSttStrawMap.h"
#include "PndGeoSttPar.h"

struct TrackSolution;

class PndPreselectSttHitsTask : public FairTask {
 public:
  PndPreselectSttHitsTask();

  virtual ~PndPreselectSttHitsTask();

  virtual InitStatus Init();

  void SetParContainers();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  void SetPreselector(PndPreselectSttHits *preselector) { fPreselector = preselector; };

  virtual void AddBranchName(TString name)
  {
    fAdditionalBranchNames.push_back(name);
  } 

 protected:

 private:
  PndGeoSttPar *fSttParameters = nullptr; // needed for SttStrawMap
  TClonesArray *fTubeArray = nullptr;     // needed for SttStrawMap

  TClonesArray *fTrackCandArray = nullptr;
  TClonesArray *fSTTHits = nullptr;

  PndPreselectSttHits *fPreselector = nullptr;

  std::vector<TString> fAdditionalBranchNames;

  ClassDef(PndPreselectSttHitsTask, 1);
};