| 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 /* 26 /*
27 * =========================================== 27 * ============================================================================
28 * 28 *
29 * Filename: CexmcTrackPoints.cc 29 * Filename: CexmcTrackPoints.cc
30 * 30 *
31 * Description: track points collection 31 * Description: track points collection
32 * 32 *
33 * Version: 1.0 33 * Version: 1.0
34 * Created: 16.11.2009 13:03:03 34 * Created: 16.11.2009 13:03:03
35 * Revision: none 35 * Revision: none
36 * Compiler: gcc 36 * Compiler: gcc
37 * 37 *
38 * Author: Alexey Radkov (), 38 * Author: Alexey Radkov (),
39 * Company: PNPI 39 * Company: PNPI
40 * 40 *
41 * =========================================== 41 * ============================================================================
42 */ 42 */
43 43
44 #include <G4HCofThisEvent.hh> 44 #include <G4HCofThisEvent.hh>
45 #include <G4AffineTransform.hh> 45 #include <G4AffineTransform.hh>
46 #include <G4UnitsTable.hh> 46 #include <G4UnitsTable.hh>
47 #include "CexmcTrackPoints.hh" 47 #include "CexmcTrackPoints.hh"
48 #include "CexmcTrackInfo.hh" 48 #include "CexmcTrackInfo.hh"
49 49
50 50
51 CexmcTrackPoints::CexmcTrackPoints( const G4St 51 CexmcTrackPoints::CexmcTrackPoints( const G4String & name ) :
52 CexmcPrimitiveScorer( name ), hcId( -1 ) 52 CexmcPrimitiveScorer( name ), hcId( -1 )
53 { 53 {
54 } 54 }
55 55
56 56
57 G4int CexmcTrackPoints::GetTrackId( G4Step * 57 G4int CexmcTrackPoints::GetTrackId( G4Step * step )
58 { 58 {
59 /* NB: trackInfo for these tracks shall al 59 /* NB: trackInfo for these tracks shall always exist as far as
60 * track points filter decline tracks with 60 * track points filter decline tracks without it */
61 CexmcTrackInfo * trackInfo( static_cast< 61 CexmcTrackInfo * trackInfo( static_cast< CexmcTrackInfo * >(
62 step->GetT 62 step->GetTrack()->GetUserInformation() ) );
63 63
64 G4int ret( trackInfo->GetTrackType() ); 64 G4int ret( trackInfo->GetTrackType() );
65 65
66 if ( ret == CexmcOutputParticleDecayProduc 66 if ( ret == CexmcOutputParticleDecayProductTrack )
67 ret += trackInfo->GetCopyNumber(); 67 ret += trackInfo->GetCopyNumber();
68 68
69 return ret; 69 return ret;
70 } 70 }
71 71
72 72
73 G4int CexmcTrackPoints::GetIndex( G4Step * s 73 G4int CexmcTrackPoints::GetIndex( G4Step * step )
74 { 74 {
75 return GetTrackId( step ); 75 return GetTrackId( step );
76 } 76 }
77 77
78 78
79 G4bool CexmcTrackPoints::ProcessHits( G4Step 79 G4bool CexmcTrackPoints::ProcessHits( G4Step * step, G4TouchableHistory * )
80 { 80 {
81 G4int index( GetIndex( step ) ); 81 G4int index( GetIndex( step ) );
82 82
83 if ( ( *eventMap )[ index ] ) 83 if ( ( *eventMap )[ index ] )
84 return false; 84 return false;
85 85
86 G4Track * track( step->GetTr 86 G4Track * track( step->GetTrack() );
87 G4ParticleDefinition * particle( track->G 87 G4ParticleDefinition * particle( track->GetDefinition() );
88 CexmcTrackType trackType( CexmcIn 88 CexmcTrackType trackType( CexmcInsipidTrack );
89 89
90 G4StepPoint * preStepPoint( step 90 G4StepPoint * preStepPoint( step->GetPreStepPoint() );
91 G4ThreeVector position( preStepP 91 G4ThreeVector position( preStepPoint->GetPosition() );
92 G4ThreeVector direction( preStep 92 G4ThreeVector direction( preStepPoint->GetMomentumDirection() );
93 93
94 const G4AffineTransform & transform( preS 94 const G4AffineTransform & transform( preStepPoint->GetTouchable()->
95 GetH 95 GetHistory()->GetTopTransform() );
96 96
97 CexmcTrackInfo * trackInfo( static_cast< 97 CexmcTrackInfo * trackInfo( static_cast< CexmcTrackInfo * >(
98 98 track->GetUserInformation() ) );
99 trackType = trackInfo->GetTrackType(); 99 trackType = trackInfo->GetTrackType();
100 100
101 CexmcTrackPointInfo trackPointInfo( trans 101 CexmcTrackPointInfo trackPointInfo( transform.TransformPoint( position ),
102 posit 102 position,
103 trans 103 transform.TransformAxis( direction ),
104 direc 104 direction,
105 preSt 105 preStepPoint->GetMomentum().mag(),
106 parti 106 particle, track->GetTrackID(),
107 track 107 trackType );
108 108
109 eventMap->set( index, trackPointInfo ); 109 eventMap->set( index, trackPointInfo );
110 110
111 return true; 111 return true;
112 } 112 }
113 113
114 114
115 void CexmcTrackPoints::Initialize( G4HCofThis 115 void CexmcTrackPoints::Initialize( G4HCofThisEvent * hcOfEvent )
116 { 116 {
117 eventMap = new CexmcTrackPointsCollection( 117 eventMap = new CexmcTrackPointsCollection( detector->GetName(),
118 118 primitiveName );
119 119
120 if ( hcId < 0 ) 120 if ( hcId < 0 )
121 hcId = GetCollectionID( 0 ); 121 hcId = GetCollectionID( 0 );
122 122
123 hcOfEvent->AddHitsCollection( hcId, eventM 123 hcOfEvent->AddHitsCollection( hcId, eventMap );
124 } 124 }
125 125
126 126
127 void CexmcTrackPoints::EndOfEvent( G4HCofThis 127 void CexmcTrackPoints::EndOfEvent( G4HCofThisEvent * )
128 { 128 {
129 if ( GetVerboseLevel() > 0 ) 129 if ( GetVerboseLevel() > 0 )
130 PrintAll(); 130 PrintAll();
131 } 131 }
132 132
133 133
134 void CexmcTrackPoints::clear( void ) 134 void CexmcTrackPoints::clear( void )
135 { 135 {
136 eventMap->clear(); 136 eventMap->clear();
137 } 137 }
138 138
139 139
140 void CexmcTrackPoints::DrawAll( void ) 140 void CexmcTrackPoints::DrawAll( void )
141 { 141 {
142 } 142 }
143 143
144 144
145 void CexmcTrackPoints::PrintAll( void ) 145 void CexmcTrackPoints::PrintAll( void )
146 { 146 {
147 G4int nmbOfEntries( eventMap->entries() 147 G4int nmbOfEntries( eventMap->entries() );
148 148
149 if ( nmbOfEntries == 0 ) 149 if ( nmbOfEntries == 0 )
150 return; 150 return;
151 151
152 PrintHeader( nmbOfEntries ); 152 PrintHeader( nmbOfEntries );
153 153
154 for ( CexmcTrackPointsCollectionData::iter 154 for ( CexmcTrackPointsCollectionData::iterator
155 itr( eventMap->GetMap 155 itr( eventMap->GetMap()->begin() );
156 itr != ev 156 itr != eventMap->GetMap()->end(); ++itr )
157 { 157 {
158 G4cout << " track id " << itr->f 158 G4cout << " track id " << itr->first << G4endl;
159 G4cout << " , position: " << 159 G4cout << " , position: " <<
160 G4BestUnit( itr->second->posit 160 G4BestUnit( itr->second->positionLocal, "Length" ) << G4endl;
161 G4cout << " , direction: " << 161 G4cout << " , direction: " <<
162 itr->second->directionLocal << 162 itr->second->directionLocal << G4endl;
163 G4cout << " , momentum: " << 163 G4cout << " , momentum: " <<
164 G4BestUnit( itr->second->momen 164 G4BestUnit( itr->second->momentumAmp, "Energy" ) << G4endl;
165 G4cout << " , particle: " << 165 G4cout << " , particle: " <<
166 itr->second->particle->GetPart 166 itr->second->particle->GetParticleName() << G4endl;
167 } 167 }
168 } 168 }
169 169
170 170