BlineTracer/include/G4BlineEventAction.hh

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 /// \file field/BlineTracer/include/G4BlineEventAction.hh
0027 /// \brief Definition of the G4BlineEventAction class
0028 //
0029 //
0030 //
0031 //
0032 // --------------------------------------------------------------------
0033 //
0034 // G4BlineEventAction
0035 //
0036 // Class description:
0037 //
0038 // Defines the EventAction used during tracing of magnetic field
0039 // lines. During the EndOfEventAction() it stores magneticfield lines
0040 // as polyline and polymarker objects.
0041 // These polyline and polymarker objects can be drawn by using the
0042 // DrawFieldLines() method.
0043
0044 // --------------------------------------------------------------------
0045 // Author: Laurent Desorgher (desorgher@phim.unibe.ch)
0046 //         Created - 2003-10-06
0047 // --------------------------------------------------------------------
0048 #ifndef G4BlineEventAction_h
0049 #define G4BlineEventAction_h 1
0050
0051 #include "G4Colour.hh"
0052 #include "G4UserEventAction.hh"
0053 #include "G4VisAttributes.hh"
0054
0055 class G4Event;
0056 class G4Polyline;
0057 class G4Polymarker;
0058 class G4BlineTracer;
0059
0060 class G4BlineEventAction : public G4UserEventAction
0061 {
0062   public:  // with description
0063     G4BlineEventAction(G4BlineTracer* aBlineTool);
0064     ~G4BlineEventAction() override;
0065
0066     void BeginOfEventAction(const G4Event*) override;
0067     void EndOfEventAction(const G4Event*) override;
0068
0069     void DrawFieldLines(G4double zoom, G4double theta, G4double phi);
0070     void ResetVectorObjectToBeDrawn();
0071
0072   public:  // with description
0073     inline void SetDrawColour(G4Colour aColour) { fDrawColour = aColour; }
0074     inline void SetDrawBline(G4bool aBool) { fDrawBline = aBool; }
0075     inline void SetDrawPoints(G4bool aBool) { fDrawPoints = aBool; }
0076     inline void SetPointSize(G4double aVal) { fPointSize = aVal; }
0077     inline G4bool GetDrawBline() { return fDrawBline; }
0078
0079   public:  // without description
0080            // inline void SetDrawLineWidth(G4double aVal)
0081            //   { fTrajectoryVisAttributes.SetLineWidth(aVal); }
0082            // inline void SetDrawLineStyle(G4VisAttributes::LineStyle aStyle)
0083            //   { fTrajectoryVisAttributes.SetLineStyle(aStyle); }
0084            // Future implementation...
0085   private:
0086     G4BlineTracer* fBlineTool = nullptr;
0087     G4Colour fDrawColour;
0088     G4bool fDrawBline = false;
0089     G4bool fDrawPoints = false;
0090     G4double fPointSize = 1;
0091     std::vector<G4VisAttributes*> fTrajectoryVisAttributes;
0092     std::vector<G4Polyline> fTrajectoryPolyline;
0093     std::vector<G4Polymarker> fTrajectoryPoints;
0094 };
0095
0096 #endif