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 // Filter trajectories according to charge. Only registered 00029 // charges will pass the filter. 00030 // 00031 // Jane Tinslay, May 2006 00032 // 00033 #ifndef G4TRAJECTORYCHARGEFILTER_HH 00034 #define G4TRAJECTORYCHARGEFILTER_HH 00035 00036 #include "G4SmartFilter.hh" 00037 #include "G4VTrajectory.hh" 00038 #include <vector> 00039 00040 class G4TrajectoryChargeFilter : public G4SmartFilter<G4VTrajectory> { 00041 00042 public: // With description 00043 00044 // Construct with filter name 00045 G4TrajectoryChargeFilter(const G4String& name = "Unspecified"); 00046 00047 virtual ~G4TrajectoryChargeFilter(); 00048 00049 // Evaluate this trajectory 00050 virtual bool Evaluate(const G4VTrajectory&) const; 00051 00052 // Print configuration 00053 virtual void Print(std::ostream& ostr) const; 00054 00055 // Clear filter 00056 virtual void Clear(); 00057 00058 // Configuration function 00059 void Add(const G4String& particle); 00060 00061 private: 00062 00063 enum MyCharge {Negative=-1, Neutral=0, Positive=1}; 00064 00065 G4bool ConvertToCharge(const G4String&, MyCharge&); 00066 00067 void Add(const MyCharge& chgear); 00068 00069 // Data member 00070 std::vector<MyCharge> fCharges; 00071 00072 }; 00073 00074 #endif