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