PndCurlingTrackFinder.h

Go to the documentation of this file.

/*
 * PndCurlingTrackFinder.h
 *
 *  Created on: 06.07.2021
 *      Author: tstockmanns
 */

#pragma once

#include "TClonesArray.h"
#include "TVector3.h"
#include <vector>
#include <iostream>

#include "PndSttStrawMap.h"
#include "PndSttGeometryMap.h"
#include "PndSttHit.h"

#include "PndPreselectSttHits.h"
#include "PndSttCA.h"
#include "PndApolloniusTriplet.h"
using namespace PndApollonius;

class PndCurlingTrackFinder {
 public:
  PndCurlingTrackFinder(TClonesArray *tubeArray);
  // PndCurlingTrackFinder(){  }
  virtual ~PndCurlingTrackFinder();

  void Reset();
  void FindTracks();
  std::vector<TripletSolution> GetSolutions() const { return fSolutions; }

  virtual void SetWithTubeReduction(bool reduction) { fWithTubeReduction = reduction; };
  virtual void SetCATrackFinder(PndSttCA *CAFinder) { fCATrackFinder = CAFinder; };
  virtual void SetUnassignedSttHits(std::vector<PndSttHit *> &hits) { fUnassignedSttHits = hits; };
  virtual void SetSttHits(std::vector<PndSttHit *> &hits) { fSttHits = hits; };
  virtual void SetBranchMap(std::map<TString, TClonesArray *> &map) { fBranchMap = map; };

  std::vector<TripletSolution> GetTripletTracks() const { return fTripletTracks; }
  std::vector<TripletSolution> GetContinuousTripletTracks() const { return fContinuousTripletTracks; }
  std::vector<TripletSolution> GetTripletTracksAfterAdding() const { return fTripletTracksAfterAdding; }

 protected:
  TripletValues GetTripletsCurledTracks(std::map<int, std::vector<std::vector<PndSttHit *>>> &tubeStructure);
  TripletValues GetTripletsCurledTracks2();

  // std::map<int, std::vector<std::vector<PndSttHit *>>> DeleteFromCurlingPool(std::vector<TripletSolution> &combinedSolutions, std::map<int, std::vector<std::vector<PndSttHit
  // *>>> &CombinedTubeStructure);
  int GetMidIndex(int firstIndex, int lastIndex);

 private:
  ApolloniusTripletFunctions fFunc;
  std::vector<TripletSolution> fSolutions;
  std::vector<TripletSolution> fTripletTracks;
  std::vector<TripletSolution> fContinuousTripletTracks;
  std::vector<TripletSolution> fTripletTracksAfterAdding;

  std::vector<PndSttHit *> fUnassignedSttHits;
  std::vector<PndSttHit *> fSttHits;
  int fAllHitsCounter;
  bool fIsStrongCurling;
  bool fWithTubeReduction;
  bool fWithCombiReduction = false;
  PndSttCA *fCATrackFinder = nullptr;
  PndSttGeometryMap *fGeometryMap = nullptr; 
  PndSttStrawMap *fStrawMap = nullptr;       
  std::map<TString, TClonesArray *> fBranchMap;
  std::map<FairLink, int> fMapHitstoCATracklet;
  int fBeforeSkewedRowNumber = 7;
  int fAfterSkewedRowNumber = 16;
};