HWMCTruthDisplay.h

Go to the documentation of this file.

/*
 * HWMCTruthDisplay.h
 *
 *  Created on: 13 Nov 2019
 *      Author: Valerio Formato
 */

// HERD headers
#include "algorithms/display/HWBaseEventDisplay.h"
#include "dataobjects/MCTruth.h"

// ROOT headers
#include "TEveArrow.h"
#include "TEveTrack.h"

#ifndef HWMCTruthDisplay_H_
#define HWMCTruthDisplay_H_

namespace Herd {
class HWMCTruthDisplay : public HWBaseEventDisplay {
public:
  HWMCTruthDisplay() : HWBaseEventDisplay("MC truth"), trackList(nullptr), arrow(nullptr){};
  ~HWMCTruthDisplay(){};

  void ImportMCTruth(const MCTruth &mcTruth);
  void MakeTracks();
  void SetRenderState(bool renderState);

  TEveTrackList *trackList;
  TEveArrow *arrow;

private:
  bool tracksReady = false;
};
} // namespace Herd

#endif