GGSTHitsReader.h

Go to the documentation of this file.

/*
 * GGSTHitsReader.h
 *
 *  Created on: Nov 8, 2010
 *      Author: Elena Vanuccini
 *  Reworked on: 17 Aug 2011
 *      Author: Nicola Mori
 *      Content: MC truth stripped out. File opening removed.
 */

#ifndef GGSTHITSREADER_H_
#define GGSTHITSREADER_H_

// GGS headers
#include "montecarlo/dataobjs/GGSTHits.h"
#include "montecarlo/readers/GGSTChainReader.h"
#include "utils/GGSNameDecoder.h"

// ROOT headers
#include "TArrayF.h"
#include "TClonesArray.h"
class TBranch;

#include <vector>

class GGSTHitsReader : public GGSTChainReader {

public:
  GGSTHitsReader();

  ~GGSTHitsReader();

  bool SetChain(TChain *hitsChain);

  void GetEntry(Long64_t entry);

  void SetDetector(const char *detector, bool readPartHits = false, bool readPosHits = false);

  bool DetectorExists(const char *detector, bool particleHits = false, bool positionHits = false);

  void ListDetectors();

  std::vector<TString> GetListOfDetectors();

  Int_t GetNHits(const char *detector);

  Int_t GetNHits(int iDet);

  GGSTIntHitBase *GetHit(const char *detector, unsigned int iHit);

  GGSTIntHitBase *GetHit(int iDet, unsigned int iHit);

  std::string GetIntHitsClassName(const char *detector);

  std::string GetIntHitsClassName(int iDet);

  int GetDetectorIndex(const char *detector);

  bool HasPartHits(const char *detector);

  bool HasPosHits(const char *detector);

  float GetIntHitThreshold(const std::string &detector);

  float GetPartHitThreshold(const std::string &detector);

  float GetPosHitThreshold(const std::string &detector);

private:
  TChain *_chain; 
  Long64_t _firstEntryOfCurrentFile;

  std::vector<TClonesArray *> _intHitArrays; 
  std::vector<TObjArray *> _partHitArrays;
  std::vector<TObjArray *> _posHitArrays;
  std::vector<TArrayF *> _timeBins;
  std::vector<bool> _readPartHits;
  std::vector<bool> _readPosHits;
  TClonesArray *_detInfo; 
  std::vector<std::vector<float> *> _thresholds;

  struct Detector {
    std::string name;       
    std::string logVolName; 
    int detInfoArrayIndex;  
  };
  std::vector<Detector> _detectors;

  int _GetDetectorIndexFromStorage(const char *detector);

  int _CheckBranch(const char *branchName);

  TClonesArray *_GetHitArray(int iHitArray);

  GGSTHitDetInfo *_GetDetectorInfo(int iDet);
};

#endif /* GGSTHITSREADER_H_ */