Class Reference for E1039 Core & Analysis Software
PHG4TrackUserInfoV1.cc
Go to the documentation of this file.
1 #include "PHG4TrackUserInfoV1.h"
2 
3 #include <Geant4/G4Track.hh>
4 #include <boost/lexical_cast.hpp>
5 
6 namespace PHG4TrackUserInfo {
7 
8  void SetUserTrackId(G4Track* track, const int usertrackid) {
9 
10  if ( G4VUserTrackInformation* p = track->GetUserInformation() ) {
11  // User info exists, test it for something valid
12  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) ) {
13  pp->SetUserTrackId(usertrackid);
14  } else {
15  std::cout << "Unknown UserTrackInformation stored in track number "
16  << boost::lexical_cast<std::string>(track->GetTrackID())
17  << std::endl;
18  }
19  } else {
20  // User info does not exist, add it.
21  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
22  pp->SetUserTrackId(usertrackid);
23  track->SetUserInformation(pp);
24  }
25  }
26 
27  void SetUserParentId(G4Track* track, const int userparentid) {
28 
29  if ( G4VUserTrackInformation* p = track->GetUserInformation() ) {
30  // User info exists, test it for something valid
31  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) ) {
32  pp->SetUserParentId(userparentid);
33  } else {
34  std::cout << "Unknown UserTrackInformation stored in track number "
35  << boost::lexical_cast<std::string>(track->GetTrackID())
36  << std::endl;
37  }
38  } else {
39  // User info does not exist, add it.
40  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
41  pp->SetUserParentId(userparentid);
42  track->SetUserInformation(pp);
43  }
44  }
45 
46  void SetUserPrimaryId(G4Track* track, const int userprimaryid) {
47 
48  if ( G4VUserTrackInformation* p = track->GetUserInformation() ) {
49  // User info exists, test it for something valid
50  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) ) {
51  pp->SetUserPrimaryId(userprimaryid);
52  } else {
53  std::cout << "Unknown UserTrackInformation stored in track number "
54  << boost::lexical_cast<std::string>(track->GetTrackID())
55  << std::endl;
56  }
57  } else {
58  // User info does not exist, add it.
59  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
60  pp->SetUserPrimaryId(userprimaryid);
61  track->SetUserInformation(pp);
62  }
63  }
64 
65  void SetWanted(G4Track* track, const int trkid)
66  {
67  if ( G4VUserTrackInformation* p = track->GetUserInformation() )
68  {
69  // User info exists, test it for something valid
70  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) )
71  {
72  pp->SetWanted(trkid);
73  }
74  else
75  {
76  std::cout << "Unknown UserTrackInformation stored in track number "
77  << boost::lexical_cast<std::string>(track->GetTrackID())
78  << std::endl;
79  }
80  }
81  else
82  {
83  // User info does not exist, add it.
84  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
85  pp->SetWanted(trkid);
86  track->SetUserInformation(pp);
87  }
88 
89  }
90 
91  void SetKeep(G4Track* track, const int trkid)
92  {
93  if ( G4VUserTrackInformation* p = track->GetUserInformation() )
94  {
95  // User info exists, test it for something valid
96  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) )
97  {
98  pp->SetKeep(trkid);
99  }
100  else
101  {
102  std::cout << "Unknown UserTrackInformation stored in track number "
103  << boost::lexical_cast<std::string>(track->GetTrackID())
104  << std::endl;
105  }
106  }
107  else
108  {
109  // User info does not exist, add it.
110  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
111  pp->SetKeep(trkid);
112  track->SetUserInformation(pp);
113  }
114 
115  }
116 
117  void SetShower(G4Track* track, PHG4Shower* shower)
118  {
119  if ( G4VUserTrackInformation* p = track->GetUserInformation() )
120  {
121  // User info exists, test it for something valid
122  if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) )
123  {
124  pp->SetShower(shower);
125  }
126  else
127  {
128  std::cout << "Unknown UserTrackInformation stored in track number "
129  << boost::lexical_cast<std::string>(track->GetTrackID())
130  << std::endl;
131  }
132  }
133  else
134  {
135  // User info does not exist, add it.
136  PHG4TrackUserInfoV1* pp = new PHG4TrackUserInfoV1();
137  pp->SetShower(shower);
138  track->SetUserInformation(pp);
139  }
140 
141  }
142 
143 
144 };
PHG4TrackUserInfoV1::SetUserTrackId
void SetUserTrackId(const int val)
Definition: PHG4TrackUserInfoV1.h:32
PHG4TrackUserInfoV1::SetShower
void SetShower(PHG4Shower *ptr)
Definition: PHG4TrackUserInfoV1.h:47
PHG4TrackUserInfoV1::SetUserParentId
void SetUserParentId(const int val)
Definition: PHG4TrackUserInfoV1.h:35
PHG4TrackUserInfoV1::SetUserPrimaryId
void SetUserPrimaryId(const int val)
Definition: PHG4TrackUserInfoV1.h:38
PHG4TrackUserInfoV1::SetKeep
void SetKeep(const int val)
Definition: PHG4TrackUserInfoV1.h:44
PHG4TrackUserInfoV1::SetWanted
void SetWanted(const int val)
Definition: PHG4TrackUserInfoV1.h:41
PHG4TrackUserInfo::SetUserPrimaryId
void SetUserPrimaryId(G4Track *track, const int userprimaryid)
Definition: PHG4TrackUserInfoV1.cc:46
PHG4TrackUserInfo::SetWanted
void SetWanted(G4Track *track, const int trkid)
Definition: PHG4TrackUserInfoV1.cc:65
PHG4TrackUserInfo::SetShower
void SetShower(G4Track *track, PHG4Shower *shower)
Definition: PHG4TrackUserInfoV1.cc:117
PHG4TrackUserInfo::SetUserTrackId
void SetUserTrackId(G4Track *track, const int usertrackid)
Definition: PHG4TrackUserInfoV1.cc:8
PHG4TrackUserInfo::SetKeep
void SetKeep(G4Track *track, const int trkid)
Definition: PHG4TrackUserInfoV1.cc:91
PHG4TrackUserInfo::SetUserParentId
void SetUserParentId(G4Track *track, const int userparentid)
Definition: PHG4TrackUserInfoV1.cc:27