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

/*
 * PndTrack.h
 *
 *  Created on: 05.03.2009
 *      Author: everybody
 */

#ifndef PNDTRACK_H_
#define PNDTRACK_H_

#include "TObject.h"
#include "PndTrackCand.h"
#include "FairTrackParP.h"
#include "PndDetectorList.h"
#include "FairTimeStamp.h"
#include "TRef.h"

#ifndef __CINT__
#include <boost/serialization/access.hpp>
#include <boost/serialization/base_object.hpp>
#endif //__CINT__

class PndTrack : public FairTimeStamp {
 public:
  PndTrack();
  PndTrack(const FairTrackParP &first, const FairTrackParP &last, const PndTrackCand &cand, Int_t flag = 0, Double_t chi2 = -1., Int_t ndf = 0, Int_t pid = 0, Int_t id = -1,
           Int_t type = -1);

  void Print();

  Int_t GetPidHypo() const { return fPidHypo; }
  Int_t GetFlag() const { return fFlag; } // Quality flag
  Double_t GetChi2() const { return fChi2; }
  Int_t GetNDF() const { return fNDF; }
  Int_t GetRefIndex() const { return fRefIndex; }
  void SetPidHypo(Int_t i) { fPidHypo = i; }
  void SetFlag(Int_t i) { fFlag = i; }
  void SetChi2(Double_t d) { fChi2 = d; }
  void SetNDF(Int_t i) { fNDF = i; }
  void SetRefIndex(TString branch, Int_t i)
  {
    fRefIndex = i;
    SetLink(FairLink(branch, i));
  }
  void SetRefIndex(Int_t i) { fRefIndex = i; }
  void SetTrackCand(const PndTrackCand &cand) { fTrackCand = cand; };
  void SetTrackCandRef(PndTrackCand *candPointer) { fRefToPndTrackCand = candPointer; }
  PndTrackCand *GetPndTrackCandViaRef() { return (PndTrackCand *)fRefToPndTrackCand.GetObject(); }

  PndTrackCand GetTrackCand() { return fTrackCand; }
  PndTrackCand *GetTrackCandPtr() { return &fTrackCand; }
  FairTrackParP GetParamFirst() { return fTrackParamFirst; }
  FairTrackParP GetParamLast() { return fTrackParamLast; }

  enum class TrackType :int {kUndefined, kBarrel, kForward, kBoth};
  void SetTrackType(enum TrackType type) { fTrackType = type; }
  enum TrackType GetTrackType() { return fTrackType; };

  virtual bool equal(FairTimeStamp *data) const
  {
    if (data)
      return false; // dummy
    return false;
  }

  virtual bool operator<(const PndTrack &myTrack) const
  {
    PndTrackCand myCand = ((PndTrack)myTrack).GetTrackCand();
    if (fTimeStamp < myCand.GetTimeStamp())
      return true;
    else if (fTimeStamp > myCand.GetTimeStamp())
      return false;
    return false;
  }

  virtual bool operator>(const PndTrack &myTrack) const
  {
    PndTrackCand myCand = ((PndTrack)myTrack).GetTrackCand();
    if (fTimeStamp > myCand.GetTimeStamp())
      return true;
    else if (fTimeStamp < myCand.GetTimeStamp())
      return false;
    return false;
  }

  virtual bool operator==(const PndTrack &myTrack) const
  {
    if (fFlag == myTrack.fFlag)
      return false; // dummy
    return false;
  }

#ifndef __CINT__ // for BOOST serialization
  template <class Archive>
  void serialize(Archive &ar, const unsigned int version)
  {
    ar &boost::serialization::base_object<FairTimeStamp>(*this);
    //      ar & fTrackParamFirst;
    //      ar & fTrackParamLast;
    ar &fTrackCand;
    ar &fPidHypo;
    ar &fFlag;
    ar &fChi2;
    ar &fNDF;
    ar &fRefIndex;
  }
#endif // for BOOST serialization

 private:
  FairTrackParP fTrackParamFirst;
  FairTrackParP fTrackParamLast;

  PndTrackCand fTrackCand;
  TRef fRefToPndTrackCand;

  Int_t fPidHypo;
  Int_t fFlag; 
  Double_t fChi2;
  Int_t fNDF;
  Int_t fRefIndex;

  TrackType fTrackType;

 public:
  ClassDef(PndTrack, 2)
};

#endif /* PNDTRACK_H_ */