PndCATrackMCPointContainer.h

Go to the documentation of this file.

/*
 *====================================================================
 *
 *  CBM KF Track Quality
 *
 *  Authors: M.Zyzak
 *
 *  e-mail :
 *
 *====================================================================
 *
 *  KF Fit performance
 *
 *====================================================================
 */

#ifndef _PndCATrackMCPointContainer_h_
#define _PndCATrackMCPointContainer_h_

#include <FairMCPoint.h>

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

#include "PndSttPoint.h"
#include "PndSdsMCPoint.h"

using std::vector;

class PndCATrackMCPointContainer : public TObject {

 public:
  PndCATrackMCPointContainer();
  ~PndCATrackMCPointContainer(){};

  PndSdsMCPoint *GetMvdPoint(Int_t i) { return MvdArray[i]; }
  PndSttPoint *GetSttPoint(Int_t i) { return SttArray[i]; }

  void SetMvdPoint(PndSdsMCPoint *mp) { MvdArray.push_back(mp); }
  void SetSttPoint(PndSttPoint *sp) { SttArray.push_back(sp); }

  int GetNMvdPoints() const { return MvdArray.size(); }
  int GetNSttPoints() const { return SttArray.size(); }

  double GetMvdPointX(int i) { return MvdArray[i]->FairMCPoint::GetX(); }
  double GetMvdPointY(int i) { return MvdArray[i]->FairMCPoint::GetY(); }
  double GetMvdPointZ(int i) { return MvdArray[i]->GetZ(); }
  double GetMvdPointPx(int i) { return MvdArray[i]->GetPx(); }
  double GetMvdPointPy(int i) { return MvdArray[i]->GetPy(); }
  double GetMvdPointPz(int i) { return MvdArray[i]->GetPz(); }

  double GetSttPointX(int i) { return SttArray[i]->FairMCPoint::GetX(); }
  double GetSttPointY(int i) { return SttArray[i]->FairMCPoint::GetY(); }
  double GetSttPointZ(int i) { return SttArray[i]->GetZ(); }
  double GetSttPointPx(int i) { return SttArray[i]->GetPx(); }
  double GetSttPointPy(int i) { return SttArray[i]->GetPy(); }
  double GetSttPointPz(int i) { return SttArray[i]->GetPz(); }

  vector<PndSttPoint *> SttArray;
  vector<PndSdsMCPoint *> MvdArray;

  ClassDef(PndCATrackMCPointContainer, 1);
};

#endif // _PndCATrackMCPointContainer_h_