CbmEvent.h

Go to the documentation of this file.

 
#ifndef CBMEVENT_H
#define CBMEVENT_H 1
 
#include <Rtypes.h>          // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>      // for Double_t, UInt_t, Int_t
#include <TMatrixFSymfwd.h>  // for TMatrixFSym
#include <TObject.h>         // for TObject
 
#include <map>     // for map, map<>::mapped_type
#include <string>  // for string
#include <vector>  // for vector
 
#include "CbmDefs.h"    // for ECbmDataType, ECbmModuleId::kStsTrack
#include "CbmMatch.h"   // for CbmMatch
#include "CbmVertex.h"  // for CbmVertex, found in core/data/global
 
class CbmEvent : public TObject {
 
public:
  CbmEvent() : CbmEvent(-1, 0., 0.) {};
 
  CbmEvent(Int_t number, Double_t startTime = 0., Double_t endTime = 0.)
    : TObject()
    , fNumber(number)
    , fTimeStart(startTime)
    , fTimeEnd(endTime)
    , fNofData(0)
    , fVertex()
    , fMatch(nullptr)
    , fIndexMap() {}
 
  CbmEvent(CbmEvent&&) = default;
 
  virtual ~CbmEvent() {
    if (fMatch) delete fMatch;
  }
 
  void Clear(Option_t* /*option*/) { fIndexMap.clear(); }
 
 
  void AddData(ECbmDataType type, UInt_t index);
 
 
  void AddStsTrack(UInt_t index) { AddData(ECbmDataType::kStsTrack, index); }
 
 
  UInt_t GetIndex(ECbmDataType type, UInt_t iData);
 
 
  CbmMatch* GetMatch() const { return fMatch; }
 
 
  Int_t GetNofData() const { return fNofData; }
 
 
  Int_t GetNofData(ECbmDataType type) const;
 
 
  Int_t GetNofStsTracks() const { return GetNofData(ECbmDataType::kStsTrack); }
 
 
  Int_t GetNumber() const { return fNumber; }
 
 
  Int_t GetStsTrackIndex(Int_t iTrack) {
    return GetIndex(ECbmDataType::kStsTrack, iTrack);
  }
 
 
  Double_t GetEndTime() const { return fTimeEnd; }
 
 
  Double_t GetStartTime() const { return fTimeStart; }
 
 
  void SetEndTime(Double_t endTime) { fTimeEnd = endTime; }
 
 
  void SetMatch(CbmMatch* match) { fMatch = match; }
 
 
  void SetStartTime(Double_t startTime) { fTimeStart = startTime; }
 
 
  void SetStsTracks(std::vector<UInt_t>& indexVector) {
    fNofData -= fIndexMap[ECbmDataType::kStsTrack].size();
    fIndexMap[ECbmDataType::kStsTrack] = indexVector;
    fNofData += fIndexMap[ECbmDataType::kStsTrack].size();
  }
 
 
  void SetVertex(Double_t x,
                 Double_t y,
                 Double_t z,
                 Double_t chi2,
                 Int_t ndf,
                 Int_t nTracks,
                 const TMatrixFSym& covMat);
 
 
  std::string ToString() const;
 
 
  CbmVertex* GetVertex() { return &fVertex; }
 
 
private:
  Int_t fNumber;        
  Double_t fTimeStart;  
  Double_t fTimeEnd;    
  Int_t fNofData;       
  CbmVertex fVertex;    
  CbmMatch* fMatch;     
 
  std::map<ECbmDataType, std::vector<UInt_t>> fIndexMap;
 
  CbmEvent(const CbmEvent&);
  CbmEvent& operator=(const CbmEvent&);
 
 
  ClassDef(CbmEvent, 2);
};
 
#endif /* CBMEVENT_H_ */