packages/NodeDump/DumpTpc_PolyTrackVertexContainer.cc

0001 #include "DumpTpc_PolyTrackVertexContainer.h"
0002
0003 #include <phool/PHIODataNode.h>
0004
0005 #include <tpctrackreco/Tpc_PolyTrackVertex.h>
0006 #include <tpctrackreco/Tpc_PolyTrackVertexContainer.h>
0007
0008 #include <ostream>
0009 #include <string>
0010
0011 using MyNode_t = PHIODataNode<Tpc_PolyTrackVertexContainer>;
0012
0013 DumpTpc_PolyTrackVertexContainer::DumpTpc_PolyTrackVertexContainer(const std::string &NodeName)
0014   : DumpObject(NodeName)
0015 {
0016 }
0017
0018 int DumpTpc_PolyTrackVertexContainer::process_Node(PHNode *myNode)
0019 {
0020   Tpc_PolyTrackVertexContainer *container{nullptr};
0021   MyNode_t *thisNode = static_cast<MyNode_t *>(myNode);  // NOLINT(cppcoreguidelines-pro-type-static-cast-downcast)
0022   if (thisNode)
0023   {
0024     container = thisNode->getData();
0025   }
0026   if (container)
0027   {
0028     unsigned int is = container->size();
0029
0030     *fout << "size: " << is << std::endl;
0031     for (unsigned int i = 0; i < is; ++i)
0032     {
0033       const Tpc_PolyTrackVertex *vertex = container->get_vertex(i);
0034       *fout << "get_track_id(" << i << "): " << vertex->get_track_id() << std::endl;
0035       *fout << "get_source_assembled_track_id(" << i << "): " << vertex->get_source_assembled_track_id() << std::endl;
0036       *fout << "get_dca2d(" << i << "): " << vertex->get_dca2d() << std::endl;
0037       *fout << "get_z0(" << i << "): " << vertex->get_z0() << std::endl;
0038       *fout << "get_pca_valid(" << i << "): " << vertex->get_pca_valid() << std::endl;
0039       *fout << "get_pca_x(" << i << "): " << vertex->get_pca_x() << std::endl;
0040       *fout << "get_pca_y(" << i << "): " << vertex->get_pca_y() << std::endl;
0041       *fout << "get_pca_z(" << i << "): " << vertex->get_pca_z() << std::endl;
0042       *fout << "get_pca_radius(" << i << "): " << vertex->get_pca_radius() << std::endl;
0043       *fout << "get_pca_phi(" << i << "): " << vertex->get_pca_phi() << std::endl;
0044     }
0045
0046     *fout << "get_collision_vertex_valid: " << container->get_collision_vertex_valid() << std::endl;
0047     *fout << "get_collision_min_clusters: " << container->get_collision_min_clusters() << std::endl;
0048     *fout << "get_collision_vertex_count: " << container->get_collision_vertex_count() << std::endl;
0049     for (unsigned int i = 0; i < container->get_collision_vertex_count(); ++i)
0050     {
0051       *fout << "get_collision_x(" << i << "): " << container->get_collision_x(i) << std::endl;
0052       *fout << "get_collision_y(" << i << "): " << container->get_collision_y(i) << std::endl;
0053       *fout << "get_collision_z(" << i << "): " << container->get_collision_z(i) << std::endl;
0054       *fout << "get_collision_z_rms(" << i << "): " << container->get_collision_z_rms(i) << std::endl;
0055       *fout << "get_collision_ntracks(" << i << "): " << container->get_collision_ntracks(i) << std::endl;
0056     }
0057   }
0058   return 0;
0059 }