d1/d8a/FPGATrackSimTrackStream_8h_source.html

/*

  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration

*/


#ifndef TRIGFPGATrackSimOBJECTS_FPGATrackSimTRACKSTREAM_H

#define TRIGFPGATrackSimOBJECTS_FPGATrackSimTRACKSTREAM_H


#include "FPGATrackSimTrack.h"


#include <TClonesArray.h>


#include <iostream>

#include <utility>


class FPGATrackSimTrackStream {


public:

  FPGATrackSimTrackStream();

  FPGATrackSimTrackStream(const FPGATrackSimTrackStream &in);

  virtual ~FPGATrackSimTrackStream();

  FPGATrackSimTrackStream & operator = (const FPGATrackSimTrackStream &s);


  const unsigned long& runNumber() const { return m_run_number; }

  const unsigned long& eventNumber() const { return m_event_number; }

  void setRunNumber(const unsigned long& val) { m_run_number = val; }

  void setEventNumber(const unsigned long& val) { m_event_number = val; }


  void addTrack(const FPGATrackSimTrack&);

  FPGATrackSimTrack* getTrack(int) const;

  int findTrack(int trackid, int bankid);

  int getNTracks() const { return m_ntracks; }

  void buildTrackMap();


  void addTrackI(const FPGATrackSimTrack&);

  FPGATrackSimTrack* getTrackI(int) const;

  int getNTracksI() const { return m_ntracksI; }


  int getNConn() const { return m_nconn; }

  int getNExtrapolatedTracks() const { return m_nextrapolatedTracks; }


  void addNCombs(int v) { m_ncombs += v; }

  void addNFits(int v) { m_nfits += v; }

  void addNFitsMajority(int v) { m_nfits_maj += v; }

  void addNFitsMajority_pix(int v) { m_nfits_maj_pix += v; }

  void addNFitsMajority_SCT(int v) { m_nfits_maj_SCT += v; }

  void addNFitsRecovery(int v) { m_nfits_rec += v; }

  void addNAddFitsRecovery(int v) { m_nfits_addrec += v; }

  void addNFitsBad(int v) { m_nfits_bad += v; }

  void addNFitsHWRejected(int v) { m_nfits_rej += v; }

  void addNFitsBadMajority(int v) { m_nfits_badmaj += v; }

  void addNFitsHWRejectedMajority(int v) { m_nfits_rejmaj += v; }

  void addNConnections(int v) { m_nconn += v; }

  void addNExtrapolatedTracks(int v) { m_nextrapolatedTracks += v; }


  int getNCombs() const { return m_ncombs; }

  int getNFits() const { return m_nfits; }

  int getNFitsMajority() const { return m_nfits_maj; }

  int getNFitsMajority_pix() const { return m_nfits_maj_pix; }

  int getNFitsMajority_SCT() const { return m_nfits_maj_SCT; }

  int getNFitsRecovery() const { return m_nfits_rec; }

  int getNAddFitsRecovery() const { return m_nfits_addrec; }

  int getNFitsBad() const { return m_nfits_bad; }

  int getNFitsHWRejected() const { return m_nfits_rej; }

  int getNFitsBadMajority() const { return m_nfits_badmaj; }

  int getNFitsHWRejectedMajority() const { return m_nfits_rejmaj; }

  int getNConnections() const { return m_nconn; }


  void addNCombsI(int v) { m_ncombsI += v; }

  void addNFitsI(int v) { m_nfitsI += v; }

  void addNFitsMajorityI(int v) { m_nfits_majI += v; }

  void addNFitsMajorityI_pix(int v) { m_nfits_majI_pix += v; }

  void addNFitsMajorityI_SCT(int v) { m_nfits_majI_SCT += v; }

  void addNFitsRecoveryI(int v) { m_nfits_recI += v; }

  void addNAddFitsRecoveryI(int v) { m_nfits_addrecI += v; }

  void addNFitsBadI(int v) { m_nfits_badI += v; }

  void addNFitsHWRejectedI(int v) { m_nfits_rejI += v; }

  void addNFitsBadMajorityI(int v) { m_nfits_badmajI += v; }

  void addNFitsHWRejectedMajorityI(int v) { m_nfits_rejmajI += v; }


  int getNCombsI() const { return m_ncombsI; }

  int getNFitsI() const { return m_nfitsI; }

  int getNFitsMajorityI() const { return m_nfits_majI; }

  int getNFitsMajorityI_pix() const { return m_nfits_majI_pix; }

  int getNFitsMajorityI_SCT() const { return m_nfits_majI_SCT; }

  int getNFitsRecoveryI() const { return m_nfits_recI; }

  int getNAddFitsRecoveryI() const { return m_nfits_addrecI; }

  int getNFitsBadI() const { return m_nfits_badI; }

  int getNFitsHWRejectedI() const { return m_nfits_rejI; }

  int getNFitsBadMajorityI() const { return m_nfits_badmajI; }

  int getNFitsHWRejectedMajorityI() const { return m_nfits_rejmajI; }


  void clear();


  int Print(int level = 0, std::ostream& out = std::cout);


private:

  unsigned long m_run_number; // run number

  unsigned long m_event_number; // event number


  int m_ntracks; // number of stored tracks

  TClonesArray* m_tracks; //-> list of stored tracks


  int m_ncombs; // number of combinations

  int m_nfits; // number of fitted combinations

  int m_nfits_maj; // number of fitted combinations in majority roads

  int m_nfits_maj_SCT; // number of fitted combinations in majority roads, missing hit in SCT layer

  int m_nfits_maj_pix; // number of fitted combinations in majority roads, missing hit in pixel layer

  int m_nfits_rec; // number of full fits attempted to recover

  int m_nfits_addrec; // additional fits performed because the recovery

  int m_nfits_bad; // # fits with a bad chi^2

  int m_nfits_rej; // # of fits rejected because the HW

  int m_nfits_badmaj; // # fits with a bad chi^2

  int m_nfits_rejmaj; // # of fits rejected because the HW

  /* PS: m_ntracks, if only good tracks are saved, has to be equal

     to m_nfits-m_nfits_bad-m_nfits_rej */


  int m_ntracksI; // number of stored tracks in the intermediate step


  TClonesArray* m_tracksI; //-> list of stored tracks in the intermedieate step


  // duplicate the counters for the 2nd stage

  int m_ncombsI;

  int m_nfitsI;

  int m_nfits_majI;

  int m_nfits_majI_SCT;

  int m_nfits_majI_pix;

  int m_nfits_recI;

  int m_nfits_addrecI;

  int m_nfits_badI;

  int m_nfits_rejI;

  int m_nfits_badmajI;

  int m_nfits_rejmajI;

  int m_nconn;

  int m_nextrapolatedTracks;


  std::map< std::pair<int, int>, int > m_trackIdMap;

  // Used to optimize the findTrack function.


  ClassDef(FPGATrackSimTrackStream, 1)

};


#endif // TRIGFPGATrackSimOBJECTS_FPGATrackSimTRACKSTREAM_H