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 // Class Description: 00029 // Trajectory model which colours a trajectory according to 00030 // the origin volume 00031 // Class Description - End: 00032 // Jane Tinslay March 2006 00033 00034 #ifndef G4TRAJECTORYDRAWBYORIGINVOLUME 00035 #define G4TRAJECTORYDRAWBYORIGINVOLUME 00036 00037 #include "G4VTrajectoryModel.hh" 00038 #include "G4Colour.hh" 00039 #include "G4ModelColourMap.hh" 00040 #include "G4String.hh" 00041 #include <map> 00042 00043 class G4TrajectoryDrawByOriginVolume : public G4VTrajectoryModel { 00044 00045 public: // With description 00046 00047 G4TrajectoryDrawByOriginVolume(const G4String& name = "Unspecified", G4VisTrajContext* context=0); 00048 00049 virtual ~G4TrajectoryDrawByOriginVolume(); 00050 00051 // Draw method 00052 virtual void Draw(const G4VTrajectory& trajectory, const G4int& i_mode = 0, 00053 const G4bool& visible = true) const; 00054 virtual void Draw(const G4VTrajectory& trajectory, 00055 const G4bool& visible = true) const; 00056 00057 virtual void Print(std::ostream& ostr) const; 00058 // Print configuration 00059 00060 void SetDefault(const G4String&); 00061 void SetDefault(const G4Colour&); 00062 00063 void Set(const G4String& particle, const G4String& colour); 00064 void Set(const G4String& particle, const G4Colour& colour); 00065 // Configuration functions 00066 00067 private: 00068 00069 // Data members 00070 G4ModelColourMap<G4String> fMap; 00071 G4Colour fDefault; 00072 00073 }; 00074 00075 #endif