ViewDrift.hh

Go to the documentation of this file.

#ifndef G_VIEW_DRIFT
#define G_VIEW_DRIFT
 
#include <string>
#include <TGraph.h>
#include <TCanvas.h>
#include <TPolyLine3D.h>
#include <TPolyMarker3D.h>
#include <TView.h>
 
namespace Garfield {
 
/// Visualize drift lines and tracks.
 
class ViewDrift {
 
 public:
  /// Constructor
  ViewDrift();
  /// Destructor
  ~ViewDrift();
 
  /// Set the canvas to be painted on.
  void SetCanvas(TCanvas* c);
 
  /// 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);
 
  /// Switch on/off debugging output.
  void EnableDebugging(const bool on = true) { m_debug = on; }
 
  friend class ViewFEMesh;
 
 private:
  std::string m_className;
 
  // Options
  bool m_debug;
 
  struct Marker {
    double x;
    double y;
    double z;
  };
  // Canvas
  TCanvas* m_canvas;
  bool m_hasExternalCanvas;
 
  // Box dimensions
  double m_xMin, m_yMin, m_zMin;
  double m_xMax, m_yMax, m_zMax;
  // View
  TView* m_view;
 
  struct driftLine {
    std::vector<Marker> vect;
    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;
  TPolyMarker3D* m_excPlot;
  std::vector<Marker> m_ionMarkers;
  TPolyMarker3D* m_ionPlot;
  std::vector<Marker> m_attMarkers;
  TPolyMarker3D* m_attPlot;
 
  double m_markerSizeCluster;
  double m_markerSizeCollision;
 
  void Plot2d(const bool axis);
  void Plot3d(const bool axis);
};
}
#endif