calcCent.cpp 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. //
  2. // Created by peter on 8/27/19.
  3. //
  4. #include <iostream>
  5. #include <TStopwatch.h>
  6. #include <TFile.h>
  7. #include <TH1F.h>
  8. #include "CentralityCalculator.h"
  9. int main(int argc, char** argv)
  10. {
  11. TString inFileName, CanvasName;
  12. if (argc < 5)
  13. {
  14. std::cerr << "./calcCent -i INPUTFILE -canv CANVASNAME" << std::endl;
  15. return 10;
  16. }
  17. for (int i = 1; i < argc; i++)
  18. {
  19. if (std::string(argv[i]) != "-i" &&
  20. std::string(argv[i]) != "-canv")
  21. {
  22. std::cerr << "\ncalcCent: Unknown parameter: " << argv[i] << std::endl;
  23. return 11;
  24. } else {
  25. if (std::string(argv[i]) == "-i" && i != argc - 1)
  26. {
  27. inFileName = argv[++i];
  28. }
  29. if (std::string(argv[i]) == "-i" && i == argc - 1)
  30. {
  31. std::cerr << "\ncalcCent: Input file name was not specified!" << std::endl;
  32. return 20;
  33. }
  34. if (std::string(argv[i]) == "-canv" && i != argc - 1)
  35. {
  36. CanvasName = argv[++i];
  37. }
  38. if (std::string(argv[i]) == "-canv" && i == argc - 1)
  39. {
  40. std::cerr << "\ncalcCent: Input file name was not specified!" << std::endl;
  41. return 20;
  42. }
  43. }
  44. }
  45. TStopwatch timer;
  46. timer.Start();
  47. TFile *fi = new TFile(inFileName.Data(),"read");
  48. TH1F *hist = (TH1F*) fi->Get("hMult");
  49. CentralityCalculator *cent = new CentralityCalculator();
  50. cent->SetEnergy(200.);
  51. cent->SetModel("PHSD");
  52. cent->SetHistogram(hist);
  53. cent->Calc();
  54. cent->DrawHistogram(CanvasName);
  55. timer.Stop();
  56. timer.Print();
  57. return 0;
  58. }