G4TrajectoryDrawByCharge.hh

Go to the documentation of this file.
00001 //
00002 // ********************************************************************
00003 // * License and Disclaimer                                           *
00004 // *                                                                  *
00005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
00006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
00007 // * conditions of the Geant4 Software License,  included in the file *
00008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
00009 // * include a list of copyright holders.                             *
00010 // *                                                                  *
00011 // * Neither the authors of this software system, nor their employing *
00012 // * institutes,nor the agencies providing financial support for this *
00013 // * work  make  any representation or  warranty, express or implied, *
00014 // * regarding  this  software system or assume any liability for its *
00015 // * use.  Please see the license in the file  LICENSE  and URL above *
00016 // * for the full disclaimer and the limitation of liability.         *
00017 // *                                                                  *
00018 // * This  code  implementation is the result of  the  scientific and *
00019 // * technical work of the GEANT4 collaboration.                      *
00020 // * By using,  copying,  modifying or  distributing the software (or *
00021 // * any work based  on the software)  you  agree  to acknowledge its *
00022 // * use  in  resulting  scientific  publications,  and indicate your *
00023 // * acceptance of all terms of the Geant4 Software license.          *
00024 // ********************************************************************
00025 //
00026 // $Id$
00027 //
00028 // Jane Tinslay, John Allison, Joseph Perl November 2005
00029 //
00030 // Class Description:
00031 // Trajectory model which colours a trajectory according to  
00032 // charge. 
00033 // Class Description - End:
00034 
00035 #ifndef G4TRAJECTORYDRAWBYCHARGE_HH
00036 #define G4TRAJECTORYDRAWBYCHARGE_HH
00037 
00038 #include "G4Colour.hh"
00039 #include "G4ModelColourMap.hh"
00040 #include "G4VTrajectoryModel.hh"
00041 #include <map>
00042 
00043 class G4TrajectoryDrawByCharge : public G4VTrajectoryModel {
00044 
00045 public: // With description
00046  
00047   enum Charge {Negative=-1, Neutral=0, Positive=1}; 
00048 
00049   G4TrajectoryDrawByCharge(const G4String& name = "Unspecified", G4VisTrajContext* context=0);
00050 
00051   G4TrajectoryDrawByCharge(const G4String& name,
00052                            const G4Colour& positive,
00053                            const G4Colour& negative,
00054                            const G4Colour& neutral);
00055   
00056   virtual ~G4TrajectoryDrawByCharge();
00057 
00058   // Draw method
00059   virtual void Draw(const G4VTrajectory& trajectory, const G4int& i_mode = 0,
00060                     const G4bool& visible = true) const;
00061   virtual void Draw(const G4VTrajectory& trajectory, 
00062                     const G4bool& visible = true) const;
00063 
00064   // Print configuration
00065   virtual void Print(std::ostream& ostr) const;
00066 
00067   void Set(const Charge& charge, const G4Colour& colour);
00068   void Set(const Charge& charge, const G4String& colour);
00069 
00070   void Set(const G4String& charge, const G4Colour& colour);
00071   void Set(const G4String& charge, const G4String& colour);
00072   // Configuration functions 
00073 
00074 private:
00075   
00076   G4bool ConvertToCharge(const G4String&, Charge&);
00077 
00078   // Data member
00079   G4ModelColourMap<Charge> fMap;
00080   
00081 };
00082 
00083 #endif

Generated on Mon May 27 17:50:02 2013 for Geant4 by  doxygen 1.4.7