PndPreselectSttHitsTask.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".              *
//****************************************************************************

//  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);
};