ViewDrift.hh

Go to the documentation of this file.

#ifndef G_VIEW_DRIFT
#define G_VIEW_DRIFT
 
#include <memory>
#include <string>
#include <vector>
 
#include <TGraph.h>
#include <TPolyLine3D.h>
#include <TPolyMarker3D.h>
#include <TView.h>
 
#include "ViewBase.hh"
 
namespace Garfield {
 
/// Visualize drift lines and tracks.
 
class ViewDrift : public ViewBase {
 public:
  /// Constructor.
  ViewDrift();
  /// Destructor.
  ~ViewDrift();
 
  /// Set the region to be plotted.
  void SetArea(const double xmin, const double ymin, const double zmin,
               const double xmax, const double ymax, const double zmax);
  /// Delete existing drift lines, tracks and markers.
  void Clear();
 
  /// Draw the drift lines.
  void Plot(const bool twod = false, const bool axis = true);
 
  /// Set the size of the cluster markers (see TAttMarker).
  void SetClusterMarkerSize(const double size);
  /// Set the size of the collision markers (see TAttMarker).
  void SetCollisionMarkerSize(const double size);
 
  // Functions used by the transport classes.
  void NewElectronDriftLine(const unsigned int np, int& id, const double x0,
                            const double y0, const double z0);
  void NewHoleDriftLine(const unsigned int np, int& id, const double x0,
                        const double y0, const double z0);
  void NewIonDriftLine(const unsigned int np, int& id, const double x0,
                       const double y0, const double z0);
  void NewPhotonTrack(const double x0, const double y0, const double z0,
                      const double x1, const double y1, const double z1);
  void NewChargedParticleTrack(const unsigned int np, int& id, const double x0,
                               const double y0, const double z0);
 
  void SetDriftLinePoint(const unsigned int iL, const unsigned int iP,
                         const double x, const double y, const double z);
  void AddDriftLinePoint(const unsigned int iL, const double x, const double y,
                         const double z);
  void SetTrackPoint(const unsigned int iL, const unsigned int iP,
                     const double x, const double y, const double z);
  void AddTrackPoint(const unsigned int iL, const double x, const double y,
                     const double z);
  void AddExcitationMarker(const double x, const double y, const double z);
  void AddIonisationMarker(const double x, const double y, const double z);
  void AddAttachmentMarker(const double x, const double y, const double z);
 
  friend class ViewFEMesh;
 
 private:
 
  struct Marker {
    double x;
    double y;
    double z;
  };
 
  // Box dimensions
  double m_xMin = -1., m_yMin = -1., m_zMin = -1.;
  double m_xMax = 1., m_yMax = 1., m_zMax = 1.;
 
  // View
  std::unique_ptr<TView> m_view;
 
  struct DriftLine {
    std::vector<Marker> points;
    int n;  // what kind of particle?
  };
  std::vector<DriftLine> m_driftLines;
  std::vector<TPolyLine3D> m_driftLinePlots;
 
  std::vector<std::vector<Marker> > m_tracks;
  std::vector<TPolyMarker3D> m_trackPlots;
  std::vector<TPolyLine3D> m_trackLinePlots;
 
  std::vector<Marker> m_excMarkers;
  std::unique_ptr<TPolyMarker3D> m_excPlot;
  std::vector<Marker> m_ionMarkers;
  std::unique_ptr<TPolyMarker3D> m_ionPlot;
  std::vector<Marker> m_attMarkers;
  std::unique_ptr<TPolyMarker3D> m_attPlot;
 
  double m_markerSizeCluster = 1.;
  double m_markerSizeCollision = 1.;
 
  void Plot2d(const bool axis);
  void Plot3d(const bool axis);
};
}
#endif