HWCaloEventDisplay.h

Go to the documentation of this file.

/*
 * HWCaloEventDisplay.h
 *
 *  Created on: 08 May 2019
 *      Author: Valerio Formato
 */
 
// HERD headers
#include "herd/analysis/algorithms/display/HWBaseEventDisplay.h"
#include "herd/analysis/algorithms/display/VPalette.h"
#include "herd/analysis/dataobjects/CaloGeoParams.h"
#include "herd/analysis/dataobjects/CaloHits.h"
 
// ROOT headers
#include <TEveGeoNode.h>
#include <TEveGeoShape.h>
 
// c++ headers
#include <unordered_map>
 
#ifndef HWCaloEventDisplay_H_
#define HWCaloEventDisplay_H_
 
namespace Herd {
class HWCaloEventDisplay : public HWBaseEventDisplay {
public:
  HWCaloEventDisplay();
  ~HWCaloEventDisplay() = default;
 
  void GenerateHashTable(TEveGeoNode *node);
  void ImportCaloHits(const CaloHits &caloHits, const CaloGeoParams &caloParams);
  void SetRenderState(bool renderState);
 
  TEveElementList *CaloHits3D;
  TEveElementList *CaloHitsXZ;
  TEveElementList *CaloHitsYZ;
  TEveElementList *CaloHitsXY;
 
private:
  std::unordered_map<std::string, TEveGeoShape *> m_EGSMap;
  std::map<RefFrame::View, std::unique_ptr<VPalette>> m_palettes;
  std::unique_ptr<VPalette> m_3Dpalette;
 
  struct CaloProjHit {
    float eDep;
    float u;
    float v;
  };
 
public:
  using projHitIteratorBase = std::pair<std::pair<int, int>, CaloProjHit>;
  using CaloProjHits = std::map<std::pair<int, int>, CaloProjHit>;
  void Draw3DHits(const CaloHits &caloHits, const CaloGeoParams &caloParams);
  void DrawProjectedHits(const CaloProjHits &projHits, RefFrame::View view);
 
  CaloProjHits ProjectHits(const CaloHits &hits, const CaloGeoParams &geoParams, RefFrame::Axis dir1,
                           RefFrame::Axis dir2);
};
} // namespace Herd
 
#endif