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 September 2006 00029 // 00030 // Draw trajectories according to attribute. 00031 // 00032 #ifndef G4TRAJECTORYDRAWBYATTRIBUTE_HH 00033 #define G4TRAJECTORYDRAWBYATTRIBUTE_HH 00034 00035 #include "globals.hh" 00036 #include "G4VTrajectoryModel.hh" 00037 #include <map> 00038 00039 class G4VAttValueFilter; 00040 class G4VisTrajContext; 00041 00042 class G4TrajectoryDrawByAttribute : public G4VTrajectoryModel { 00043 00044 public: 00045 00046 // Constructor 00047 G4TrajectoryDrawByAttribute(const G4String& name = "Unspecified", G4VisTrajContext* context=0); 00048 00049 // Destructor 00050 virtual ~G4TrajectoryDrawByAttribute(); 00051 00052 // Draw the trajectory 00053 virtual void Draw(const G4VTrajectory& trajectory, const G4int& i_mode = 0, 00054 const G4bool& visible = true) const; 00055 virtual void Draw(const G4VTrajectory& trajectory, 00056 const G4bool& visible = true) const; 00057 00058 // Print configuration 00059 virtual void Print(std::ostream& ostr) const; 00060 00061 // Configuration methods 00062 void Set(const G4String& attribute); 00063 void AddIntervalContext(const G4String& name, G4VisTrajContext* context); 00064 void AddValueContext(const G4String& name, G4VisTrajContext* context); 00065 00066 private: 00067 00068 enum Config {Interval, SingleValue}; 00069 00070 typedef std::pair<G4String, Config> Pair; 00071 typedef std::map<Pair, G4VisTrajContext*> ContextMap; 00072 00073 // Data members 00074 G4String fAttName; 00075 ContextMap fContextMap; 00076 00077 // Caching 00078 mutable G4bool fFirst; 00079 mutable G4bool fWarnedMissingAttribute; 00080 mutable G4VAttValueFilter* filter; 00081 00082 }; 00083 00084 #endif