HWTracks3DEventDisplay.h

Go to the documentation of this file.

/*
 * HWTracks3DEventDisplay.h
 *
 *  Created on: 17 Feb 2022
 *      Author: Valerio Formato
 */

#ifndef HWTracks3DEventDisplay_H_
#define HWTracks3DEventDisplay_H_

// HERD headers
#include "algorithms/display/HWBaseEventDisplay.h"
#include "dataobjects/ClustersColl.h"
#include "dataobjects/Line.h"
#include "dataobjects/Track.h"

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

namespace Herd {
class HWTracks3DEventDisplay : public HWBaseEventDisplay {
public:
  HWTracks3DEventDisplay(std::string name = "Tracks", Color_t color = kYellow);
  virtual ~HWTracks3DEventDisplay() = default;

  void ImportTracks(const std::vector<Herd::Line> &lines);
  void ImportTracks(const std::vector<Herd::Track> &tracks);
  void SetRenderState(bool renderState);

  void SetDrawArrow(bool drawArrow) { m_drawArrow = drawArrow; }

  TEveTrackList *trackList = nullptr;
  std::vector<TEveArrow *> arrows{};

protected:
  Color_t m_color;
  bool m_drawArrow = false;
};
} // namespace Herd

#endif