main.cpp 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. #include <iostream>
  2. #include <vector>
  3. #include <string>
  4. #include <TString.h>
  5. #include <TH1D.h>
  6. #include <TH1I.h>
  7. #include <TH2D.h>
  8. #include <TProfile.h>
  9. #include <TProfile2D.h>
  10. #include <TFile.h>
  11. #include <TStopwatch.h>
  12. #include <qaParticle.h>
  13. #include <qaEvent.h>
  14. #include <qaReader_manager.h>
  15. #include <qaReader_smash_root.h>
  16. #include <qaReader_mcpico.h>
  17. #include <Utility.h>
  18. #include <qaWriter_manager.h>
  19. #include <qaWriter_analysistree2.h>
  20. #ifdef _MCINI_
  21. #include <qaReader_mcini.h>
  22. #endif
  23. #ifdef _PHQMD_
  24. #include <qaReader_phqmd.h>
  25. #endif
  26. #ifdef _HSD_ROOT_
  27. #include <qaReader_hsd_root.h>
  28. #endif
  29. // AnalysisTree headers
  30. #include <AnalysisTree/Configuration.hpp>
  31. #include <AnalysisTree/DataHeader.hpp>
  32. #include <AnalysisTree/EventHeader.hpp>
  33. #include <AnalysisTree/Detector.hpp>
  34. #include <AnalysisTree/Matching.hpp>
  35. int main(int argc, char **argv)
  36. {
  37. TString iFileName, oFileName;
  38. TString outputFormat = "analysistree_v2";
  39. std::string coll_system = "NN";
  40. float coll_energy = -1.;
  41. if (argc < 7)
  42. {
  43. std::cerr << "./convert -i input.list -o output.root -input_format [INPUT_FORMAT] [OPTIONAL: -output_format [OUTPUT_FORMAT] -system [SYSTEM] -energy [ENERGY]]" << std::endl;
  44. std::cerr << "Available input formats:" << std::endl;
  45. std::cerr << "\tmcpico - simple custom ROOT format to store model data" << std::endl;
  46. std::cerr << "\tparticle - ROOT format that is used by the SMASH model" << std::endl;
  47. #ifdef _MCINI_
  48. std::cerr << "\tmcini - custom ROOT format to store both initial state and final state (UniGen data format) model data" << std::endl;
  49. #endif
  50. #ifdef _PHQMD_
  51. std::cerr << "\tphqmd - custom ROOT format to store PHQMD (with MST) model data" << std::endl;
  52. #endif
  53. #ifdef _HSD_ROOT_
  54. std::cerr << "\thsd - custom ROOT format to store HSD model data" << std::endl;
  55. #endif
  56. std::cerr << "Available output formats:" << std::endl;
  57. std::cerr << "\tanalysistree_v2 - AnalysisTree ver. 2 data format" << std::endl;
  58. return 1;
  59. }
  60. for (int i = 1; i < argc; i++)
  61. {
  62. if (std::string(argv[i]) != "-i" &&
  63. std::string(argv[i]) != "-o" &&
  64. std::string(argv[i]) != "-input_format" &&
  65. std::string(argv[i]) != "-output_format" &&
  66. std::string(argv[i]) != "-system" &&
  67. std::string(argv[i]) != "-energy")
  68. {
  69. std::cerr << "\n[ERROR]: Unknown parameter " << i << ": " << argv[i] << std::endl;
  70. return 2;
  71. }
  72. else
  73. {
  74. if (std::string(argv[i]) == "-i" && i != argc - 1)
  75. {
  76. iFileName = argv[++i];
  77. continue;
  78. }
  79. if (std::string(argv[i]) == "-i" && i == argc - 1)
  80. {
  81. std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
  82. return 1;
  83. }
  84. if (std::string(argv[i]) == "-o" && i != argc - 1)
  85. {
  86. oFileName = argv[++i];
  87. continue;
  88. }
  89. if (std::string(argv[i]) == "-o" && i == argc - 1)
  90. {
  91. std::cerr << "\n[ERROR]: Output file name was not specified " << std::endl;
  92. return 1;
  93. }
  94. if (std::string(argv[i]) == "-input_format" && i != argc - 1)
  95. {
  96. qaUtility::GetInstance()->format = argv[++i];
  97. continue;
  98. }
  99. if (std::string(argv[i]) == "-input_format" && i == argc - 1)
  100. {
  101. std::cerr << "\n[ERROR]: Input file format was not specified " << std::endl;
  102. return 1;
  103. }
  104. if (std::string(argv[i]) == "-output_format" && i != argc - 1)
  105. {
  106. outputFormat = argv[++i];
  107. continue;
  108. }
  109. if (std::string(argv[i]) == "-output_format" && i == argc - 1)
  110. {
  111. std::cerr << "\n[ERROR]: Output file format was not specified " << std::endl;
  112. return 1;
  113. }
  114. if (std::string(argv[i]) == "-system" && i != argc - 1)
  115. {
  116. coll_system = std::string(argv[++i]);
  117. continue;
  118. }
  119. if (std::string(argv[i]) == "-system" && i == argc - 1)
  120. {
  121. std::cerr << "\n[ERROR]: System was not specified " << std::endl;
  122. return 1;
  123. }
  124. if (std::string(argv[i]) == "-energy" && i != argc - 1)
  125. {
  126. coll_energy = std::stof(std::string(argv[++i]));
  127. continue;
  128. }
  129. if (std::string(argv[i]) == "-energy" && i == argc - 1)
  130. {
  131. std::cerr << "\n[ERROR]: Energy was not specified " << std::endl;
  132. return 1;
  133. }
  134. }
  135. }
  136. TStopwatch timer;
  137. timer.Start();
  138. qaReader_manager *readerManager;
  139. if (qaUtility::GetInstance()->format == "mcpico")
  140. {
  141. readerManager = new qaReader_mcpico();
  142. }
  143. #ifdef _MCINI_
  144. if (qaUtility::GetInstance()->format == "mcini")
  145. {
  146. readerManager = new qaReader_mcini();
  147. }
  148. #endif
  149. #ifdef _PHQMD_
  150. if (qaUtility::GetInstance()->format == "phqmd")
  151. {
  152. readerManager = new qaReader_phqmd();
  153. }
  154. #endif
  155. #ifdef _HSD_ROOT_
  156. if (qaUtility::GetInstance()->format == "hsd")
  157. {
  158. readerManager = new qaReader_hsd_root();
  159. }
  160. #endif
  161. if (qaUtility::GetInstance()->format == "particles")
  162. {
  163. readerManager = new qaReader_smash_root();
  164. }
  165. if (!readerManager)
  166. {
  167. std::cerr << "This input format is not found!" << std::endl;
  168. return 20;
  169. }
  170. qaWriter_manager *writerManager;
  171. if (outputFormat == "analysistree_v2")
  172. {
  173. writerManager = new qaWriter_analysistree2();
  174. }
  175. if (!writerManager)
  176. {
  177. std::cerr << "This output format is not found!" << std::endl;
  178. return 30;
  179. }
  180. readerManager->SetChain(iFileName.Data());
  181. Long64_t Nentries = readerManager->GetEntries();
  182. Int_t Nparticles;
  183. qaEvent *event = nullptr;
  184. qaParticle *particle = nullptr;
  185. writerManager->Init(oFileName.Data(), "modelTree", coll_system, coll_energy);
  186. for (Long64_t iev = 0; iev < Nentries; iev++)
  187. {
  188. if (iev % 1000 == 0)
  189. std::cout << "Event [" << iev << "/" << Nentries << "]" << std::endl;
  190. event = (qaEvent *)readerManager->ReadEvent(iev);
  191. if (!event)
  192. continue;
  193. writerManager->WriteEvent(event);
  194. Nparticles = event->GetNparticles();
  195. for (int iparticle = 0; iparticle < Nparticles; iparticle++)
  196. {
  197. particle = readerManager->ReadParticle(iparticle);
  198. if (!particle)
  199. continue;
  200. writerManager->WriteParticle(particle);
  201. delete particle;
  202. }
  203. writerManager->FillTree();
  204. delete event;
  205. }
  206. writerManager->WriteTree();
  207. timer.Stop();
  208. timer.Print();
  209. return 0;
  210. }