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 26 /// \file field/BlineTracer/src/G4BlineEventAction.cc 27 /// \brief Implementation of the G4BlineEventA 27 /// \brief Implementation of the G4BlineEventAction class 28 // 28 // 29 // 29 // 30 // 30 // 31 // 31 // 32 // ------------------------------------------- 32 // -------------------------------------------------------------------- 33 // 33 // 34 // G4BlineEventAction implementation 34 // G4BlineEventAction implementation 35 // 35 // 36 // ------------------------------------------- 36 // -------------------------------------------------------------------- 37 // Author: Laurent Desorgher (desorgher@phim.u 37 // Author: Laurent Desorgher (desorgher@phim.unibe.ch) 38 // Created - 2003-10-06 38 // Created - 2003-10-06 39 // ------------------------------------------- 39 // -------------------------------------------------------------------- 40 40 41 #include "G4BlineEventAction.hh" 41 #include "G4BlineEventAction.hh" 42 << 43 #include "G4BlineTracer.hh" 42 #include "G4BlineTracer.hh" >> 43 44 #include "G4Event.hh" 44 #include "G4Event.hh" >> 45 #include "G4Trajectory.hh" 45 #include "G4EventManager.hh" 46 #include "G4EventManager.hh" >> 47 #include "G4VisManager.hh" >> 48 #include "G4UImanager.hh" 46 #include "G4Polyline.hh" 49 #include "G4Polyline.hh" 47 #include "G4Polymarker.hh" 50 #include "G4Polymarker.hh" 48 #include "G4Trajectory.hh" << 49 #include "G4UImanager.hh" << 50 #include "G4VisManager.hh" << 51 51 52 //....oooOO0OOooo........oooOO0OOooo........oo 52 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 53 53 54 G4BlineEventAction::G4BlineEventAction(G4Bline 54 G4BlineEventAction::G4BlineEventAction(G4BlineTracer* aBlineTool) 55 { 55 { 56 fBlineTool = aBlineTool; << 56 fBlineTool=aBlineTool; 57 } 57 } 58 58 59 //....oooOO0OOooo........oooOO0OOooo........oo 59 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 60 60 61 G4BlineEventAction::~G4BlineEventAction() 61 G4BlineEventAction::~G4BlineEventAction() 62 { 62 { 63 for (size_t i = 0; i < fTrajectoryVisAttribu << 63 for (size_t i=0; i<fTrajectoryVisAttributes.size(); i++) 64 delete fTrajectoryVisAttributes[i]; 64 delete fTrajectoryVisAttributes[i]; 65 } 65 } 66 66 67 //....oooOO0OOooo........oooOO0OOooo........oo 67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 68 68 69 void G4BlineEventAction::BeginOfEventAction(co << 69 void G4BlineEventAction::BeginOfEventAction(const G4Event*) >> 70 { >> 71 } 70 72 71 //....oooOO0OOooo........oooOO0OOooo........oo 73 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 72 74 73 void G4BlineEventAction::EndOfEventAction(cons 75 void G4BlineEventAction::EndOfEventAction(const G4Event* evt) 74 { 76 { 75 G4TrajectoryContainer* trajectoryContainer = << 77 G4TrajectoryContainer * trajectoryContainer = evt->GetTrajectoryContainer(); 76 if (trajectoryContainer) { << 78 if(trajectoryContainer) >> 79 { 77 // visualisation 80 // visualisation 78 // ------------- 81 // ------------- 79 82 80 if (fDrawBline || fDrawPoints) { << 83 if (fDrawBline || fDrawPoints) >> 84 { 81 G4int n_point = (*(evt->GetTrajectoryCon 85 G4int n_point = (*(evt->GetTrajectoryContainer()))[0]->GetPointEntries(); 82 86 83 G4Polyline pPolyline; 87 G4Polyline pPolyline; 84 G4Polymarker stepPoints; 88 G4Polymarker stepPoints; 85 fTrajectoryVisAttributes.push_back(new G 89 fTrajectoryVisAttributes.push_back(new G4VisAttributes(fDrawColour)); 86 stepPoints.SetMarkerType(G4Polymarker::c 90 stepPoints.SetMarkerType(G4Polymarker::circles); 87 stepPoints.SetScreenSize(fPointSize); 91 stepPoints.SetScreenSize(fPointSize); 88 stepPoints.SetFillStyle(G4VMarker::fille 92 stepPoints.SetFillStyle(G4VMarker::filled); 89 stepPoints.SetVisAttributes(fTrajectoryV 93 stepPoints.SetVisAttributes(fTrajectoryVisAttributes.back()); 90 94 91 for (G4int i = 0; i < n_point; i++) { << 95 for(G4int i=0; i<n_point; i++) 92 G4ThreeVector pos = << 96 { 93 ((G4TrajectoryPoint*)((*(evt->GetTra << 97 G4ThreeVector pos = ((G4TrajectoryPoint*) 94 if (fDrawBline) pPolyline.push_back(po << 98 ((*(evt->GetTrajectoryContainer()))[0]->GetPoint(i)))->GetPosition(); 95 if (fDrawPoints) stepPoints.push_back( << 99 if (fDrawBline) pPolyline.push_back( pos); >> 100 if (fDrawPoints) stepPoints.push_back(pos); 96 } 101 } 97 102 98 pPolyline.SetVisAttributes(fTrajectoryVi 103 pPolyline.SetVisAttributes(fTrajectoryVisAttributes.back()); 99 104 100 fTrajectoryPolyline.push_back(pPolyline) 105 fTrajectoryPolyline.push_back(pPolyline); 101 fTrajectoryPoints.push_back(stepPoints); 106 fTrajectoryPoints.push_back(stepPoints); 102 } 107 } 103 } 108 } 104 } 109 } 105 110 106 //....oooOO0OOooo........oooOO0OOooo........oo 111 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 107 112 108 void G4BlineEventAction::DrawFieldLines(G4doub << 113 void G4BlineEventAction:: >> 114 DrawFieldLines( G4double, G4double, G4double ) 109 { 115 { 110 size_t nline = fTrajectoryPolyline.size(); 116 size_t nline = fTrajectoryPolyline.size(); 111 size_t npoints = fTrajectoryPoints.size(); << 117 size_t npoints =fTrajectoryPoints.size(); 112 118 113 G4VVisManager* pVVisManager = G4VVisManager: 119 G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance(); 114 if (!pVVisManager) { << 120 if (!pVVisManager) 115 G4Exception("G4BlineEventAction::DrawField << 121 { 116 "Missing visualisation driver << 122 G4Exception("G4BlineEventAction::DrawFieldLines()", >> 123 "NullPointer", JustWarning, >> 124 "Missing visualisation driver for visualising magnetic field lines!"); 117 return; 125 return; 118 } 126 } 119 127 120 if (nline == 0) { << 128 if (nline ==0) >> 129 { 121 G4cout << "WARNING - G4BlineEventAction::D 130 G4cout << "WARNING - G4BlineEventAction::DrawFieldLines()" << G4endl 122 << " There is nothing to v 131 << " There is nothing to visualise !" << G4endl; 123 return; 132 return; 124 } 133 } 125 ((G4VisManager*)pVVisManager)->GetCurrentSce << 134 ((G4VisManager*)pVVisManager)->GetCurrentSceneHandler()-> ClearStore (); 126 G4UImanager::GetUIpointer()->ApplyCommand("/ << 135 G4UImanager::GetUIpointer () -> ApplyCommand ("/vis/drawVolume"); 127 136 128 for (size_t i = 0; i < nline; i++) << 137 for (size_t i=0;i<nline;i++) 129 pVVisManager->Draw(fTrajectoryPolyline[i]) 138 pVVisManager->Draw(fTrajectoryPolyline[i]); 130 for (size_t i = 0; i < npoints; i++) << 139 for (size_t i=0;i<npoints;i++) 131 pVVisManager->Draw(fTrajectoryPoints[i]); 140 pVVisManager->Draw(fTrajectoryPoints[i]); 132 141 133 // ((G4VisManager*)pVVisManager)->GetCurrent 142 // ((G4VisManager*)pVVisManager)->GetCurrentViewer()->DrawView(); 134 // ((G4VisManager*)pVVisManager)->GetCurrent 143 // ((G4VisManager*)pVVisManager)->GetCurrentViewer()->ShowView(); 135 } 144 } 136 145 137 //....oooOO0OOooo........oooOO0OOooo........oo 146 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 138 147 139 void G4BlineEventAction::ResetVectorObjectToBe 148 void G4BlineEventAction::ResetVectorObjectToBeDrawn() 140 { 149 { 141 fTrajectoryVisAttributes.clear(); 150 fTrajectoryVisAttributes.clear(); 142 fTrajectoryPolyline.clear(); 151 fTrajectoryPolyline.clear(); 143 fTrajectoryPoints.clear(); 152 fTrajectoryPoints.clear(); 144 } 153 } 145 154