Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PHG4InEventReadBack.cc
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4InEventReadBack.cc
1 #include "PHG4InEventReadBack.h"
2 #include "PHG4InEvent.h"
3 #include "PHG4Particle.h" // for PHG4Particle
4 #include "PHG4Particlev1.h"
5 #include "PHG4VtxPointv1.h"
6 
7 #include <vararray/VariableArray.h>
8 #include <vararray/VariableArrayUtils.h>
9 
10 #include <fun4all/Fun4AllReturnCodes.h>
11 
12 #include <phool/getClass.h>
13 #include <phool/PHCompositeNode.h>
14 #include <phool/PHIODataNode.h>
15 #include <phool/PHNode.h> // for PHNode
16 #include <phool/PHNodeIterator.h> // for PHNodeIterator
17 #include <phool/PHObject.h> // for PHObject
18 
19 #include <iostream>
20 
21 using namespace std;
22 
23 PHG4InEventReadBack::PHG4InEventReadBack(const std::string &name): SubsysReco(name)
24 {}
25 
26 int
27 PHG4InEventReadBack::InitRun(PHCompositeNode *topNode)
28 {
29  PHG4InEvent *ineve = findNode::getClass<PHG4InEvent>(topNode,"PHG4INEVENT");
30  if (!ineve)
31  {
32  PHNodeIterator iter( topNode );
33  PHCompositeNode *dstNode;
34  dstNode = dynamic_cast<PHCompositeNode*>(iter.findFirst("PHCompositeNode", "DST" ));
35 
36  ineve = new PHG4InEvent();
37  PHIODataNode<PHObject> *newNode = new PHIODataNode<PHObject>(ineve, "PHG4INEVENT", "PHObject");
38  dstNode->addNode(newNode);
39  }
40 
41  return Fun4AllReturnCodes::EVENT_OK;
42 }
43 
44 int
45 PHG4InEventReadBack::process_event(PHCompositeNode *topNode)
46 {
47  PHG4InEvent *inEvent = findNode::getClass<PHG4InEvent>(topNode,"PHG4INEVENT");
48  if (!inEvent)
49  {
50  cout << "no PHG4INEVENT node found" << endl;
51  return Fun4AllReturnCodes::EVENT_OK;
52  }
53  VariableArray *vtxarray = findNode::getClass<VariableArray>(topNode,"PHG4Vtx_VarArray");
54  if (!vtxarray)
55  {
56  cout << "no PHG4Vtx_VarArray node found" << endl;
57  return Fun4AllReturnCodes::EVENT_OK;
58  }
59 
60  VariableArray *particlearray = findNode::getClass<VariableArray>(topNode,"PHG4Particle_VarArray");
61  if (!particlearray)
62  {
63  cout << "no PHG4Particle_VarArray node found" << endl;
64  return Fun4AllReturnCodes::EVENT_OK;
65  }
66  inEvent->Reset();
67  unsigned int size = vtxarray->get_array_size();
68  const short int *sval = vtxarray->get_array();
69  PHG4VtxPointv1 vtx;
70  while(size > 0)
71  {
72  int vtxid = *sval++;
73  size --;
74  vtx.set_x(VariableArrayUtils::ShortBitsToFloat(*sval++));
75  size --;
76  vtx.set_y(VariableArrayUtils::ShortBitsToFloat(*sval++));
77  size --;
78  vtx.set_z(VariableArrayUtils::ShortBitsToFloat(*sval++));
79  size--;
80  vtx.set_t(VariableArrayUtils::ShortBitsToFloat(*sval++));
81  size--;
82  inEvent->AddVtx(vtxid,vtx);
83  }
84 
85  size = particlearray->get_array_size();
86  sval = particlearray->get_array();
87  while(size > 0)
88  {
89  PHG4Particle *particle = new PHG4Particlev1();
90  int vtxid = *sval++;
91  size --;
92  particle->set_pid(*sval++);
93  size --;
94  particle->set_px(VariableArrayUtils::ShortBitsToFloat(*sval++));
95  size --;
96  particle->set_py(VariableArrayUtils::ShortBitsToFloat(*sval++));
97  size --;
98  particle->set_pz(VariableArrayUtils::ShortBitsToFloat(*sval++));
99  size --;
100  inEvent->AddParticle(vtxid, particle);
101  }
102  // inEvent->identify();
103  return Fun4AllReturnCodes::EVENT_OK;
104 }
105 
106 int
107 PHG4InEventReadBack::End(PHCompositeNode */*topNode*/)
108 {
109  return Fun4AllReturnCodes::EVENT_OK;
110 }