Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 /// \file field/BlineTracer/src/G4BlineEventAc << 27 /// \brief Implementation of the G4BlineEventA << 28 // << 29 // 26 // >> 27 // $Id: G4BlineEventAction.cc,v 1.2 2006-06-29 17:15:09 gunter Exp $ >> 28 // GEANT4 tag $Name: geant4-09-04-patch-02 $ 30 // 29 // 31 // 30 // 32 // ------------------------------------------- 31 // -------------------------------------------------------------------- 33 // 32 // 34 // G4BlineEventAction implementation 33 // G4BlineEventAction implementation 35 // 34 // 36 // ------------------------------------------- 35 // -------------------------------------------------------------------- 37 // Author: Laurent Desorgher (desorgher@phim.u 36 // Author: Laurent Desorgher (desorgher@phim.unibe.ch) 38 // Created - 2003-10-06 37 // Created - 2003-10-06 39 // ------------------------------------------- 38 // -------------------------------------------------------------------- 40 39 41 #include "G4BlineEventAction.hh" 40 #include "G4BlineEventAction.hh" 42 << 43 #include "G4BlineTracer.hh" 41 #include "G4BlineTracer.hh" >> 42 44 #include "G4Event.hh" 43 #include "G4Event.hh" >> 44 #include "G4Trajectory.hh" 45 #include "G4EventManager.hh" 45 #include "G4EventManager.hh" >> 46 #include "G4VisManager.hh" >> 47 #include "G4UImanager.hh" 46 #include "G4Polyline.hh" 48 #include "G4Polyline.hh" 47 #include "G4Polymarker.hh" 49 #include "G4Polymarker.hh" 48 #include "G4Trajectory.hh" << 49 #include "G4UImanager.hh" << 50 #include "G4VisManager.hh" << 51 50 52 //....oooOO0OOooo........oooOO0OOooo........oo << 51 /////////////////////////////////////////////////////////////////////////// 53 52 54 G4BlineEventAction::G4BlineEventAction(G4Bline 53 G4BlineEventAction::G4BlineEventAction(G4BlineTracer* aBlineTool) 55 { 54 { 56 fBlineTool = aBlineTool; << 55 fBlineTool=aBlineTool; 57 } 56 } 58 57 59 //....oooOO0OOooo........oooOO0OOooo........oo << 58 /////////////////////////////////////////////////////////////////////////// 60 59 61 G4BlineEventAction::~G4BlineEventAction() 60 G4BlineEventAction::~G4BlineEventAction() 62 { 61 { 63 for (size_t i = 0; i < fTrajectoryVisAttribu << 62 for (size_t i=0; i<TrajectoryVisAttributes.size(); i++) 64 delete fTrajectoryVisAttributes[i]; << 63 delete TrajectoryVisAttributes[i]; 65 } 64 } 66 65 67 //....oooOO0OOooo........oooOO0OOooo........oo << 66 /////////////////////////////////////////////////////////////////////////// 68 67 69 void G4BlineEventAction::BeginOfEventAction(co << 68 void G4BlineEventAction::BeginOfEventAction(const G4Event*) >> 69 { >> 70 } 70 71 71 //....oooOO0OOooo........oooOO0OOooo........oo << 72 /////////////////////////////////////////////////////////////////////////// 72 73 73 void G4BlineEventAction::EndOfEventAction(cons 74 void G4BlineEventAction::EndOfEventAction(const G4Event* evt) 74 { 75 { 75 G4TrajectoryContainer* trajectoryContainer = << 76 G4TrajectoryContainer * trajectoryContainer = evt->GetTrajectoryContainer(); 76 if (trajectoryContainer) { << 77 G4int n_trajectories = 0; >> 78 if(trajectoryContainer) >> 79 { >> 80 n_trajectories = trajectoryContainer->entries(); >> 81 77 // visualisation 82 // visualisation 78 // ------------- 83 // ------------- 79 << 84 80 if (fDrawBline || fDrawPoints) { << 85 if (DrawBline || DrawPoints) >> 86 { 81 G4int n_point = (*(evt->GetTrajectoryCon 87 G4int n_point = (*(evt->GetTrajectoryContainer()))[0]->GetPointEntries(); 82 << 88 83 G4Polyline pPolyline; 89 G4Polyline pPolyline; 84 G4Polymarker stepPoints; 90 G4Polymarker stepPoints; 85 fTrajectoryVisAttributes.push_back(new G << 91 TrajectoryVisAttributes.push_back(new G4VisAttributes(DrawColour)); 86 stepPoints.SetMarkerType(G4Polymarker::c 92 stepPoints.SetMarkerType(G4Polymarker::circles); 87 stepPoints.SetScreenSize(fPointSize); << 93 stepPoints.SetScreenSize(PointSize); 88 stepPoints.SetFillStyle(G4VMarker::fille 94 stepPoints.SetFillStyle(G4VMarker::filled); 89 stepPoints.SetVisAttributes(fTrajectoryV << 95 stepPoints.SetVisAttributes(TrajectoryVisAttributes.back()); 90 96 91 for (G4int i = 0; i < n_point; i++) { << 97 for(G4int i=0; i<n_point; i++) 92 G4ThreeVector pos = << 98 { 93 ((G4TrajectoryPoint*)((*(evt->GetTra << 99 G4ThreeVector pos = ((G4TrajectoryPoint*) 94 if (fDrawBline) pPolyline.push_back(po << 100 ((*(evt->GetTrajectoryContainer()))[0]->GetPoint(i)))->GetPosition(); 95 if (fDrawPoints) stepPoints.push_back( << 101 if (DrawBline) pPolyline.push_back( pos); >> 102 if (DrawPoints) stepPoints.push_back(pos); 96 } 103 } 97 104 98 pPolyline.SetVisAttributes(fTrajectoryVi << 105 pPolyline.SetVisAttributes(TrajectoryVisAttributes.back()); 99 106 100 fTrajectoryPolyline.push_back(pPolyline) << 107 TrajectoryPolyline.push_back(pPolyline); 101 fTrajectoryPoints.push_back(stepPoints); << 108 TrajectoryPoints.push_back(stepPoints); 102 } 109 } 103 } 110 } 104 } 111 } 105 112 106 //....oooOO0OOooo........oooOO0OOooo........oo << 113 /////////////////////////////////////////////////////////////////////////// 107 114 108 void G4BlineEventAction::DrawFieldLines(G4doub << 115 void G4BlineEventAction:: >> 116 DrawFieldLines( G4double, G4double, G4double ) 109 { 117 { 110 size_t nline = fTrajectoryPolyline.size(); << 118 size_t nline = TrajectoryPolyline.size(); 111 size_t npoints = fTrajectoryPoints.size(); << 119 size_t npoints =TrajectoryPoints.size(); 112 120 113 G4VVisManager* pVVisManager = G4VVisManager: 121 G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance(); 114 if (!pVVisManager) { << 122 if (!pVVisManager) 115 G4Exception("G4BlineEventAction::DrawField << 123 { 116 "Missing visualisation driver << 124 G4Exception("G4BlineEventAction::DrawFieldLines()", >> 125 "NullPointer", JustWarning, >> 126 "Missing visualisation driver for visualising magnetic field lines!"); 117 return; 127 return; 118 } 128 } 119 << 129 120 if (nline == 0) { << 130 if (nline ==0) >> 131 { 121 G4cout << "WARNING - G4BlineEventAction::D 132 G4cout << "WARNING - G4BlineEventAction::DrawFieldLines()" << G4endl 122 << " There is nothing to v 133 << " There is nothing to visualise !" << G4endl; 123 return; 134 return; 124 } 135 } 125 ((G4VisManager*)pVVisManager)->GetCurrentSce << 136 ((G4VisManager*)pVVisManager)->GetCurrentSceneHandler()-> ClearStore (); 126 G4UImanager::GetUIpointer()->ApplyCommand("/ << 137 G4UImanager::GetUIpointer () -> ApplyCommand ("/vis/drawVolume"); 127 138 128 for (size_t i = 0; i < nline; i++) << 139 for (size_t i=0;i<nline;i++) 129 pVVisManager->Draw(fTrajectoryPolyline[i]) << 140 pVVisManager->Draw(TrajectoryPolyline[i]); 130 for (size_t i = 0; i < npoints; i++) << 141 for (size_t i=0;i<npoints;i++) 131 pVVisManager->Draw(fTrajectoryPoints[i]); << 142 pVVisManager->Draw(TrajectoryPoints[i]); 132 143 133 // ((G4VisManager*)pVVisManager)->GetCurrent << 144 // ((G4VisManager*)pVVisManager)->GetCurrentViewer()->DrawView(); 134 // ((G4VisManager*)pVVisManager)->GetCurrent 145 // ((G4VisManager*)pVVisManager)->GetCurrentViewer()->ShowView(); 135 } 146 } 136 147 137 //....oooOO0OOooo........oooOO0OOooo........oo << 148 /////////////////////////////////////////////////////////////////////////// 138 149 139 void G4BlineEventAction::ResetVectorObjectToBe 150 void G4BlineEventAction::ResetVectorObjectToBeDrawn() 140 { 151 { 141 fTrajectoryVisAttributes.clear(); << 152 TrajectoryVisAttributes.clear(); 142 fTrajectoryPolyline.clear(); << 153 TrajectoryPolyline.clear(); 143 fTrajectoryPoints.clear(); << 154 TrajectoryPoints.clear(); 144 } 155 } 145 156