PndGFRecoHitFactoryCreator.h

Go to the documentation of this file.

#pragma once

#include <MeasurementProducer.h>
#include <MeasurementFactory.h>
#include <AbsMeasurement.h>

class PndGFRecoHitFactoryCreator {
 public:
  PndGFRecoHitFactoryCreator();
  virtual ~PndGFRecoHitFactoryCreator();

  void AddBranch(TString branchName) { fBranches[branchName] = nullptr; }

  Bool_t Init();

  genfit::AbsMeasurementProducer<genfit::AbsMeasurement> *CreateProducer(TString branchName, TClonesArray *tcArray);
  genfit::MeasurementFactory<genfit::AbsMeasurement> *GetFactory() { return fTheRecoHitFactory; }

 protected:
  void FillRecoHitFactory();
  void SetDefaultBranches();
  void InitBranches();

 private:
  genfit::MeasurementFactory<genfit::AbsMeasurement> *fTheRecoHitFactory = nullptr;
  TClonesArray *fTubeArray = nullptr;
  TClonesArray *fFtsTubeArray = nullptr;
  std::map<TString, TClonesArray *> fBranches; //< branchName, matching TClonesArray;
};