PndGeoHitList.h

Go to the documentation of this file.

#ifndef PNDGEOHITLIST_H
#define PNDGEOHITLIST_H

#include "FairHit.h"

#include "TGeoManager.h"
#include "TGeoNode.h"
#include "TGeoMatrix.h"
#include "TGeoVolume.h"
#include "TString.h"
#include "TClonesArray.h"

#include <vector>
#include <iostream>

class PndGeoHitList {
 public:
  PndGeoHitList();
  PndGeoHitList(TString groupName, TString topNodeName, TGeoVolume *baseVolume);
  virtual ~PndGeoHitList();
  PndGeoHitList(const PndGeoHitList &) = delete;
  PndGeoHitList &operator=(const PndGeoHitList &) = delete;

  void CreateNewGroup(TString groupName, TString topNodeName, TGeoVolume *baseVolume);
  void AddHit(FairHit *hit, Bool_t vis = kTRUE);
  void AddHit(Double_t x, Double_t y, Double_t z, Bool_t vis = kTRUE);
  void AddHit(TGeoMatrix *mat, Bool_t vis = kTRUE);
  void AddHits(TClonesArray *hitList, Bool_t vis = kTRUE);
  void SetHits(TClonesArray *hitList, Bool_t vis = kTRUE);
  void RemoveGeoHits(Int_t from);
  void CreateGeoHit(TGeoMatrix *mat);

  TString GetName() const { return fGroupName; };
  const std::vector<TGeoNode *> GetNodes() const { return fNodeList; };
  void SetVisibility(TString NodeName, bool value);
  void SetVisibility(TGeoHMatrix pos, bool value);
  void SetVisibility(Int_t id, bool value);
  void SetAllInvisible();
  Int_t GetNodeId(TString NodeName);
  Int_t GetNodeId(TGeoHMatrix pos);

 private:
  TString fGroupName;
  TString fVolName;
  std::vector<TGeoNode *> fNodeList;
  TGeoVolume *fGroupVolume;
  TGeoVolume *fTopVolume;
  TGeoVolume *fBaseVolume;

  void CreateGeoHits(Int_t number);
  void SetGeoHitsInvisible(Int_t from);

  ClassDef(PndGeoHitList, 1);
};

#endif /*PNDGEOHITLIST_H*/