FemtoDstAnalyzer_Resolution.C 90 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149
  1. /**
  2. * \brief Example of how to read a file (list of files) using StFemtoEvent classes
  3. *
  4. * RunFemtoDstAnalyzer.C is an example of reading FemtoDst format.
  5. * One can use either FemtoDst file or a list of femtoDst files (inFile.lis or
  6. * inFile.list) as an input, and preform physics analysis
  7. *
  8. * \author Grigory Nigmatkulov
  9. * \date May 29, 2018
  10. */
  11. // This is needed for calling standalone classes
  12. #define _VANILLA_ROOT_
  13. // C++ headers
  14. #include <iostream>
  15. #include <vector>
  16. #include <map>
  17. // ROOT headers
  18. #include "TROOT.h"
  19. #include "TFile.h"
  20. #include "TChain.h"
  21. #include "TVector2.h"
  22. #include "TVector3.h"
  23. #include "TTree.h"
  24. #include "TSystem.h"
  25. #include "TH1.h"
  26. #include "TH2.h"
  27. #include "TMath.h"
  28. #include "TProfile2D.h"
  29. #include "TStopwatch.h"
  30. #include "TRandom3.h"
  31. // FemtoDst headers
  32. #include "StFemtoDstReader.h"
  33. #include "StFemtoDst.h"
  34. #include "StFemtoEvent.h"
  35. #include "StFemtoTrack.h"
  36. // Load libraries (for ROOT_VERSTION_CODE >= 393215)
  37. #if ROOT_VERSION_CODE >= ROOT_VERSION(6, 0, 0)
  38. R__LOAD_LIBRARY(/ mnt / pool / rhic / 4 / parfenovpeter / STAR / build / libStFemtoDst.so)
  39. #endif
  40. #include "Constants.h"
  41. // inFile - is a name of name.FemtoDst.root file or a name
  42. // of a name.lis(t) files, that contains a list of
  43. // name1.FemtoDst.root, name2.FemtoDst.root, ... files
  44. //Used functions (see them below)
  45. Bool_t isGoodEvent(StFemtoEvent *const &event);
  46. Bool_t isGoodTrack(StFemtoTrack *const &track, Float_t _energy, TVector3 pVtx);
  47. Bool_t isGoodTrackFlow(StFemtoTrack *const &track, Float_t _energy, TVector3 pVtx);
  48. Double_t GetWeight(StFemtoTrack *const &track);
  49. TVector2 CalcQvector(StFemtoTrack *const &track, Int_t _harm);
  50. Double_t AngleShift(Double_t Psi, Double_t order);
  51. Int_t GetVzBin(Double_t vtxZ);
  52. Float_t BBC_GetPhi(const Int_t eastWest, const Int_t tileId);
  53. Double_t GetZDCPosition(Int_t eastwest, Int_t verthori, Int_t strip);
  54. //_________________
  55. void FemtoDstAnalyzer_Resolution(const Char_t *inFile = "st_physics_12150008_raw_4030001.femtoDst.root",
  56. const Char_t *outFile = "./oResTest.root",
  57. const Char_t *reCentFile = "",
  58. const Char_t *shiftFile = "",
  59. const Char_t *gainBBC = "",
  60. const Char_t *reCentFileBBC = "",
  61. const Char_t *shiftFileBBC = "",
  62. const Char_t *reCentFileZDC = "",
  63. const Char_t *shiftFileZDC = "")
  64. {
  65. std::cout << "Hi! Lets do some physics, Master!" << std::endl;
  66. TStopwatch timer;
  67. timer.Start();
  68. #if ROOT_VERSION_CODE < ROOT_VERSION(6, 0, 0)
  69. gSystem->Load("../libStFemtoDst.so");
  70. #endif
  71. StFemtoDstReader *femtoReader = new StFemtoDstReader(inFile);
  72. femtoReader->Init();
  73. // This is a way if you want to spead up IO
  74. std::cout << "Explicit read status for some branches" << std::endl;
  75. femtoReader->SetStatus("*", 0);
  76. femtoReader->SetStatus("Event", 1);
  77. femtoReader->SetStatus("Track", 1);
  78. std::cout << "Status has been set" << std::endl;
  79. std::cout << "Now I know what to read, Master!" << std::endl;
  80. if (!femtoReader->chain())
  81. {
  82. std::cout << "No chain has been found." << std::endl;
  83. }
  84. Long64_t eventsInTree = femtoReader->tree()->GetEntries();
  85. std::cout << "eventsInTree: " << eventsInTree << std::endl;
  86. Long64_t events2read = femtoReader->chain()->GetEntries();
  87. std::cout << "Number of events to read: " << events2read << std::endl;
  88. //Read ReCentering <Q> vectors
  89. TFile *fiReCent = new TFile(reCentFile, "read");
  90. std::map<Double_t,Double_t> p_Qx_FULL_EP[fNharmonics][fArraySize][fNCent];
  91. std::map<Double_t,Double_t> p_Qy_FULL_EP[fNharmonics][fArraySize][fNCent];
  92. std::map<Double_t,Double_t> p_Qx_FULL_SP[fNharmonics][fArraySize][fNCent];
  93. std::map<Double_t,Double_t> p_Qy_FULL_SP[fNharmonics][fArraySize][fNCent];
  94. std::map<Double_t,Double_t> p_Qx_EAST_EP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  95. std::map<Double_t,Double_t> p_Qy_EAST_EP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  96. std::map<Double_t,Double_t> p_Qx_EAST_SP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  97. std::map<Double_t,Double_t> p_Qy_EAST_SP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  98. std::map<Double_t,Double_t> p_Qx_WEST_EP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  99. std::map<Double_t,Double_t> p_Qy_WEST_EP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  100. std::map<Double_t,Double_t> p_Qx_WEST_SP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  101. std::map<Double_t,Double_t> p_Qy_WEST_SP[fNharmonics][fArraySize][fNCent][NEtaGaps];
  102. TProfile2D *prof;
  103. double binCont;
  104. std::cout << "Reading file for TPC recentering..." << std::endl;
  105. for (int iHarm = 0; iHarm < fNharmonics; iHarm++)
  106. {
  107. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  108. {
  109. std::cout << "iHarm: " << iHarm << " iVtx: " << iVtx << std::endl;
  110. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_FULL_EP%i", iHarm, iVtx));
  111. for (int i=0; i<fNBins; i++)
  112. {
  113. for (int iCent=0; iCent<fNCent; iCent++)
  114. {
  115. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  116. if (binCont == 0.) continue;
  117. p_Qx_FULL_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  118. }
  119. }
  120. delete prof;
  121. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_FULL_EP%i", iHarm, iVtx));
  122. for (int i=0; i<fNBins; i++)
  123. {
  124. for (int iCent=0; iCent<fNCent; iCent++)
  125. {
  126. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  127. if (binCont == 0.) continue;
  128. p_Qy_FULL_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  129. }
  130. }
  131. delete prof;
  132. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_FULL_SP%i", iHarm, iVtx));
  133. for (int i=0; i<fNBins; i++)
  134. {
  135. for (int iCent=0; iCent<fNCent; iCent++)
  136. {
  137. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  138. if (binCont == 0.) continue;
  139. p_Qx_FULL_SP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  140. }
  141. }
  142. delete prof;
  143. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_FULL_SP%i", iHarm, iVtx));
  144. for (int i=0; i<fNBins; i++)
  145. {
  146. for (int iCent=0; iCent<fNCent; iCent++)
  147. {
  148. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  149. if (binCont == 0.) continue;
  150. p_Qy_FULL_SP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  151. }
  152. }
  153. delete prof;
  154. for (int iEtaGap = 0; iEtaGap < NEtaGaps; iEtaGap++)
  155. {
  156. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_EAST_EP%i_gap%i", iHarm, iVtx, iEtaGap));
  157. for (int i=0; i<fNBins; i++)
  158. {
  159. for (int iCent=0; iCent<fNCent; iCent++)
  160. {
  161. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  162. if (binCont == 0.) continue;
  163. p_Qx_EAST_EP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  164. }
  165. }
  166. delete prof;
  167. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_EAST_EP%i_gap%i", iHarm, iVtx, iEtaGap));
  168. for (int i=0; i<fNBins; i++)
  169. {
  170. for (int iCent=0; iCent<fNCent; iCent++)
  171. {
  172. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  173. if (binCont == 0.) continue;
  174. p_Qy_EAST_EP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  175. }
  176. }
  177. delete prof;
  178. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_EAST_SP%i_gap%i", iHarm, iVtx, iEtaGap));
  179. for (int i=0; i<fNBins; i++)
  180. {
  181. for (int iCent=0; iCent<fNCent; iCent++)
  182. {
  183. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  184. if (binCont == 0.) continue;
  185. p_Qx_EAST_SP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  186. }
  187. }
  188. delete prof;
  189. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_EAST_SP%i_gap%i", iHarm, iVtx, iEtaGap));
  190. for (int i=0; i<fNBins; i++)
  191. {
  192. for (int iCent=0; iCent<fNCent; iCent++)
  193. {
  194. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  195. if (binCont == 0.) continue;
  196. p_Qy_EAST_SP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  197. }
  198. }
  199. delete prof;
  200. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_WEST_EP%i_gap%i", iHarm, iVtx, iEtaGap));
  201. for (int i=0; i<fNBins; i++)
  202. {
  203. for (int iCent=0; iCent<fNCent; iCent++)
  204. {
  205. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  206. if (binCont == 0.) continue;
  207. p_Qx_WEST_EP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  208. }
  209. }
  210. delete prof;
  211. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_WEST_EP%i_gap%i", iHarm, iVtx, iEtaGap));
  212. for (int i=0; i<fNBins; i++)
  213. {
  214. for (int iCent=0; iCent<fNCent; iCent++)
  215. {
  216. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  217. if (binCont == 0.) continue;
  218. p_Qy_WEST_EP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  219. }
  220. }
  221. delete prof;
  222. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%ix_WEST_SP%i_gap%i", iHarm, iVtx, iEtaGap));
  223. for (int i=0; i<fNBins; i++)
  224. {
  225. for (int iCent=0; iCent<fNCent; iCent++)
  226. {
  227. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  228. if (binCont == 0.) continue;
  229. p_Qx_WEST_SP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  230. }
  231. }
  232. delete prof;
  233. prof = (TProfile2D *)fiReCent->Get(Form("p_Q%iy_WEST_SP%i_gap%i", iHarm, iVtx, iEtaGap));
  234. for (int i=0; i<fNBins; i++)
  235. {
  236. for (int iCent=0; iCent<fNCent; iCent++)
  237. {
  238. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  239. if (binCont == 0.) continue;
  240. p_Qy_WEST_SP[iHarm][iVtx][iCent][iEtaGap][(Double_t) (runId_min+i)] = binCont;
  241. }
  242. }
  243. delete prof;
  244. }
  245. }
  246. }
  247. //Read Shift coefficients
  248. TFile *fiShift = new TFile(shiftFile, "read");
  249. std::map<Double_t,Double_t> p_Cos2_FULL_EP[fArraySize][NShiftOrderMax][fNCent];
  250. std::map<Double_t,Double_t> p_Sin2_FULL_EP[fArraySize][NShiftOrderMax][fNCent];
  251. std::map<Double_t,Double_t> p_Cos2_FULL_SP[fArraySize][NShiftOrderMax][fNCent];
  252. std::map<Double_t,Double_t> p_Sin2_FULL_SP[fArraySize][NShiftOrderMax][fNCent];
  253. std::map<Double_t,Double_t> p_Cos2_EAST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  254. std::map<Double_t,Double_t> p_Sin2_EAST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  255. std::map<Double_t,Double_t> p_Cos2_EAST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  256. std::map<Double_t,Double_t> p_Sin2_EAST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  257. std::map<Double_t,Double_t> p_Cos2_WEST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  258. std::map<Double_t,Double_t> p_Sin2_WEST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  259. std::map<Double_t,Double_t> p_Cos2_WEST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  260. std::map<Double_t,Double_t> p_Sin2_WEST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  261. std::map<Double_t,Double_t> p_Cos3_FULL_EP[fArraySize][NShiftOrderMax][fNCent];
  262. std::map<Double_t,Double_t> p_Sin3_FULL_EP[fArraySize][NShiftOrderMax][fNCent];
  263. std::map<Double_t,Double_t> p_Cos3_FULL_SP[fArraySize][NShiftOrderMax][fNCent];
  264. std::map<Double_t,Double_t> p_Sin3_FULL_SP[fArraySize][NShiftOrderMax][fNCent];
  265. std::map<Double_t,Double_t> p_Cos3_EAST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  266. std::map<Double_t,Double_t> p_Sin3_EAST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  267. std::map<Double_t,Double_t> p_Cos3_EAST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  268. std::map<Double_t,Double_t> p_Sin3_EAST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  269. std::map<Double_t,Double_t> p_Cos3_WEST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  270. std::map<Double_t,Double_t> p_Sin3_WEST_EP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  271. std::map<Double_t,Double_t> p_Cos3_WEST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  272. std::map<Double_t,Double_t> p_Sin3_WEST_SP[fArraySize][NShiftOrderMax][fNCent][NEtaGaps];
  273. std::cout << "Reading file for TPC flattening..." << std::endl;
  274. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  275. {
  276. for (int iShift = 0; iShift < NShiftOrderMax; iShift++)
  277. {
  278. std::cout << "iVtx: " << iVtx << " iShiftOrder: " << iShift << std::endl;
  279. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_FULL_EP%i_shiftOrder_%i", 2, iVtx, iShift));
  280. for (int i=0; i<fNBins; i++)
  281. {
  282. for (int iCent=0; iCent<fNCent; iCent++)
  283. {
  284. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  285. if (binCont == 0.) continue;
  286. p_Cos2_FULL_EP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  287. }
  288. }
  289. delete prof;
  290. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_FULL_EP%i_shiftOrder_%i", 2, iVtx, iShift));
  291. for (int i=0; i<fNBins; i++)
  292. {
  293. for (int iCent=0; iCent<fNCent; iCent++)
  294. {
  295. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  296. if (binCont == 0.) continue;
  297. p_Sin2_FULL_EP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  298. }
  299. }
  300. delete prof;
  301. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_FULL_SP%i_shiftOrder_%i", 2, iVtx, iShift));
  302. for (int i=0; i<fNBins; i++)
  303. {
  304. for (int iCent=0; iCent<fNCent; iCent++)
  305. {
  306. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  307. if (binCont == 0.) continue;
  308. p_Cos2_FULL_SP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  309. }
  310. }
  311. delete prof;
  312. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_FULL_SP%i_shiftOrder_%i", 2, iVtx, iShift));
  313. for (int i=0; i<fNBins; i++)
  314. {
  315. for (int iCent=0; iCent<fNCent; iCent++)
  316. {
  317. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  318. if (binCont == 0.) continue;
  319. p_Sin2_FULL_SP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  320. }
  321. }
  322. delete prof;
  323. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_FULL_EP%i_shiftOrder_%i", 3, iVtx, iShift));
  324. for (int i=0; i<fNBins; i++)
  325. {
  326. for (int iCent=0; iCent<fNCent; iCent++)
  327. {
  328. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  329. if (binCont == 0.) continue;
  330. p_Cos3_FULL_EP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  331. }
  332. }
  333. delete prof;
  334. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_FULL_EP%i_shiftOrder_%i", 3, iVtx, iShift));
  335. for (int i=0; i<fNBins; i++)
  336. {
  337. for (int iCent=0; iCent<fNCent; iCent++)
  338. {
  339. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  340. if (binCont == 0.) continue;
  341. p_Sin3_FULL_EP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  342. }
  343. }
  344. delete prof;
  345. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_FULL_SP%i_shiftOrder_%i", 3, iVtx, iShift));
  346. for (int i=0; i<fNBins; i++)
  347. {
  348. for (int iCent=0; iCent<fNCent; iCent++)
  349. {
  350. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  351. if (binCont == 0.) continue;
  352. p_Cos3_FULL_SP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  353. }
  354. }
  355. delete prof;
  356. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_FULL_SP%i_shiftOrder_%i", 3, iVtx, iShift));
  357. for (int i=0; i<fNBins; i++)
  358. {
  359. for (int iCent=0; iCent<fNCent; iCent++)
  360. {
  361. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  362. if (binCont == 0.) continue;
  363. p_Sin3_FULL_SP[iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  364. }
  365. }
  366. delete prof;
  367. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  368. {
  369. // for 2nd harmonics
  370. // EAST
  371. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_EAST_EP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  372. for (int i=0; i<fNBins; i++)
  373. {
  374. for (int iCent=0; iCent<fNCent; iCent++)
  375. {
  376. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  377. if (binCont == 0.) continue;
  378. p_Cos2_EAST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  379. }
  380. }
  381. delete prof;
  382. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_EAST_EP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  383. for (int i=0; i<fNBins; i++)
  384. {
  385. for (int iCent=0; iCent<fNCent; iCent++)
  386. {
  387. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  388. if (binCont == 0.) continue;
  389. p_Sin2_EAST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  390. }
  391. }
  392. delete prof;
  393. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_EAST_SP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  394. for (int i=0; i<fNBins; i++)
  395. {
  396. for (int iCent=0; iCent<fNCent; iCent++)
  397. {
  398. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  399. if (binCont == 0.) continue;
  400. p_Cos2_EAST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  401. }
  402. }
  403. delete prof;
  404. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_EAST_SP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  405. for (int i=0; i<fNBins; i++)
  406. {
  407. for (int iCent=0; iCent<fNCent; iCent++)
  408. {
  409. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  410. if (binCont == 0.) continue;
  411. p_Sin2_EAST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  412. }
  413. }
  414. delete prof;
  415. // WEST
  416. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_WEST_EP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  417. for (int i=0; i<fNBins; i++)
  418. {
  419. for (int iCent=0; iCent<fNCent; iCent++)
  420. {
  421. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  422. if (binCont == 0.) continue;
  423. p_Cos2_WEST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  424. }
  425. }
  426. delete prof;
  427. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_WEST_EP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  428. for (int i=0; i<fNBins; i++)
  429. {
  430. for (int iCent=0; iCent<fNCent; iCent++)
  431. {
  432. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  433. if (binCont == 0.) continue;
  434. p_Sin2_WEST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  435. }
  436. }
  437. delete prof;
  438. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_WEST_SP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  439. for (int i=0; i<fNBins; i++)
  440. {
  441. for (int iCent=0; iCent<fNCent; iCent++)
  442. {
  443. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  444. if (binCont == 0.) continue;
  445. p_Cos2_WEST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  446. }
  447. }
  448. delete prof;
  449. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_WEST_SP%i_shiftOrder_%i_gap%i", 2, iVtx, iShift, iGap));
  450. for (int i=0; i<fNBins; i++)
  451. {
  452. for (int iCent=0; iCent<fNCent; iCent++)
  453. {
  454. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  455. if (binCont == 0.) continue;
  456. p_Sin2_WEST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  457. }
  458. }
  459. delete prof;
  460. // for 3rd harmonics
  461. // EAST
  462. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_EAST_EP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  463. for (int i=0; i<fNBins; i++)
  464. {
  465. for (int iCent=0; iCent<fNCent; iCent++)
  466. {
  467. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  468. if (binCont == 0.) continue;
  469. p_Cos3_EAST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  470. }
  471. }
  472. delete prof;
  473. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_EAST_EP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  474. for (int i=0; i<fNBins; i++)
  475. {
  476. for (int iCent=0; iCent<fNCent; iCent++)
  477. {
  478. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  479. if (binCont == 0.) continue;
  480. p_Sin3_EAST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  481. }
  482. }
  483. delete prof;
  484. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_EAST_SP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  485. for (int i=0; i<fNBins; i++)
  486. {
  487. for (int iCent=0; iCent<fNCent; iCent++)
  488. {
  489. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  490. if (binCont == 0.) continue;
  491. p_Cos3_EAST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  492. }
  493. }
  494. delete prof;
  495. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_EAST_SP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  496. for (int i=0; i<fNBins; i++)
  497. {
  498. for (int iCent=0; iCent<fNCent; iCent++)
  499. {
  500. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  501. if (binCont == 0.) continue;
  502. p_Sin3_EAST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  503. }
  504. }
  505. delete prof;
  506. // WEST
  507. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_WEST_EP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  508. for (int i=0; i<fNBins; i++)
  509. {
  510. for (int iCent=0; iCent<fNCent; iCent++)
  511. {
  512. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  513. if (binCont == 0.) continue;
  514. p_Cos3_WEST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  515. }
  516. }
  517. delete prof;
  518. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_WEST_EP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  519. for (int i=0; i<fNBins; i++)
  520. {
  521. for (int iCent=0; iCent<fNCent; iCent++)
  522. {
  523. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  524. if (binCont == 0.) continue;
  525. p_Sin3_WEST_EP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  526. }
  527. }
  528. delete prof;
  529. prof = (TProfile2D *)fiShift->Get(Form("p_Cos%i_WEST_SP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  530. for (int i=0; i<fNBins; i++)
  531. {
  532. for (int iCent=0; iCent<fNCent; iCent++)
  533. {
  534. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  535. if (binCont == 0.) continue;
  536. p_Cos3_WEST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  537. }
  538. }
  539. delete prof;
  540. prof = (TProfile2D *)fiShift->Get(Form("p_Sin%i_WEST_SP%i_shiftOrder_%i_gap%i", 3, iVtx, iShift, iGap));
  541. for (int i=0; i<fNBins; i++)
  542. {
  543. for (int iCent=0; iCent<fNCent; iCent++)
  544. {
  545. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  546. if (binCont == 0.) continue;
  547. p_Sin3_WEST_SP[iVtx][iShift][iCent][iGap][(Double_t) (runId_min+i)] = binCont;
  548. }
  549. }
  550. delete prof;
  551. }
  552. }
  553. }
  554. //Manage gain correction
  555. TProfile2D *BBCgain_East[fArraySize];
  556. TProfile2D *BBCgain_West[fArraySize];
  557. TFile *fiGain = new TFile(gainBBC, "read");
  558. fiGain->cd();
  559. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  560. {
  561. BBCgain_East[iVtx] = (TProfile2D *)fiGain->Get(Form("p_gainBBC_East_Vtx%i", iVtx));
  562. BBCgain_West[iVtx] = (TProfile2D *)fiGain->Get(Form("p_gainBBC_West_Vtx%i", iVtx));
  563. }
  564. //Manage recentering BBC
  565. std::map<Double_t,Double_t> p_BBC_Qx_Full_EP[fNharmonics][fArraySize][fNCent];
  566. std::map<Double_t,Double_t> p_BBC_Qy_Full_EP[fNharmonics][fArraySize][fNCent];
  567. std::map<Double_t,Double_t> p_BBC_Qx_East_EP[fNharmonics][fArraySize][fNCent];
  568. std::map<Double_t,Double_t> p_BBC_Qy_East_EP[fNharmonics][fArraySize][fNCent];
  569. std::map<Double_t,Double_t> p_BBC_Qx_West_EP[fNharmonics][fArraySize][fNCent];
  570. std::map<Double_t,Double_t> p_BBC_Qy_West_EP[fNharmonics][fArraySize][fNCent];
  571. TFile *fiReCentBBC = new TFile(reCentFileBBC, "read");
  572. std::cout << "Reading file for BBC recentering..." << std::endl;
  573. for (int iHarm = 0; iHarm < fNharmonics; iHarm++)
  574. {
  575. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  576. {
  577. std::cout << "iHarm: " << iHarm << " iVtx: " << iVtx << std::endl;
  578. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qx_Full_EP%i_vtx%i", iHarm, iVtx));
  579. for (int i=0; i<fNBins; i++)
  580. {
  581. for (int iCent=0; iCent<fNCent; iCent++)
  582. {
  583. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  584. if (binCont == 0.) continue;
  585. p_BBC_Qx_Full_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  586. }
  587. }
  588. delete prof;
  589. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qy_Full_EP%i_vtx%i", iHarm, iVtx));
  590. for (int i=0; i<fNBins; i++)
  591. {
  592. for (int iCent=0; iCent<fNCent; iCent++)
  593. {
  594. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  595. if (binCont == 0.) continue;
  596. p_BBC_Qy_Full_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  597. }
  598. }
  599. delete prof;
  600. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qx_East_EP%i_vtx%i", iHarm, iVtx));
  601. for (int i=0; i<fNBins; i++)
  602. {
  603. for (int iCent=0; iCent<fNCent; iCent++)
  604. {
  605. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  606. if (binCont == 0.) continue;
  607. p_BBC_Qx_East_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  608. }
  609. }
  610. delete prof;
  611. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qy_East_EP%i_vtx%i", iHarm, iVtx));
  612. for (int i=0; i<fNBins; i++)
  613. {
  614. for (int iCent=0; iCent<fNCent; iCent++)
  615. {
  616. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  617. if (binCont == 0.) continue;
  618. p_BBC_Qy_East_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  619. }
  620. }
  621. delete prof;
  622. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qx_West_EP%i_vtx%i", iHarm, iVtx));
  623. for (int i=0; i<fNBins; i++)
  624. {
  625. for (int iCent=0; iCent<fNCent; iCent++)
  626. {
  627. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  628. if (binCont == 0.) continue;
  629. p_BBC_Qx_West_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  630. }
  631. }
  632. delete prof;
  633. prof = (TProfile2D *)fiReCentBBC->Get(Form("p_BBC_Qy_West_EP%i_vtx%i", iHarm, iVtx));
  634. for (int i=0; i<fNBins; i++)
  635. {
  636. for (int iCent=0; iCent<fNCent; iCent++)
  637. {
  638. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  639. if (binCont == 0.) continue;
  640. p_BBC_Qy_West_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  641. }
  642. }
  643. delete prof;
  644. }
  645. }
  646. //Manage ShiftCorr BBC
  647. std::map<Double_t,Double_t> p_BBC_Cos_Full_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  648. std::map<Double_t,Double_t> p_BBC_Sin_Full_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  649. std::map<Double_t,Double_t> p_BBC_Cos_East_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  650. std::map<Double_t,Double_t> p_BBC_Sin_East_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  651. std::map<Double_t,Double_t> p_BBC_Cos_West_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  652. std::map<Double_t,Double_t> p_BBC_Sin_West_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  653. TFile *fiShiftBBC = new TFile(shiftFileBBC, "read");
  654. fiShiftBBC->cd();
  655. std::cout << "Reading file for BBC shift corrections..." << std::endl;
  656. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  657. {
  658. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  659. {
  660. for (int iShift = 0; iShift < NShiftOrderMax*3; iShift++)
  661. {
  662. std::cout << "iHarm: " << iHarm << " iVtx: " << iVtx << " iShiftOrder: " << iShift << std::endl;
  663. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Cos%i_Full_EP%i_vtx%i", iShift, iHarm, iVtx));
  664. for (int i=0; i<fNBins; i++)
  665. {
  666. for (int iCent=0; iCent<fNCent; iCent++)
  667. {
  668. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  669. if (binCont == 0.) continue;
  670. p_BBC_Cos_Full_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  671. }
  672. }
  673. delete prof;
  674. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Sin%i_Full_EP%i_vtx%i", iShift, iHarm, iVtx));
  675. for (int i=0; i<fNBins; i++)
  676. {
  677. for (int iCent=0; iCent<fNCent; iCent++)
  678. {
  679. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  680. if (binCont == 0.) continue;
  681. p_BBC_Sin_Full_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  682. }
  683. }
  684. delete prof;
  685. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Cos%i_East_EP%i_vtx%i", iShift, iHarm, iVtx));
  686. for (int i=0; i<fNBins; i++)
  687. {
  688. for (int iCent=0; iCent<fNCent; iCent++)
  689. {
  690. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  691. if (binCont == 0.) continue;
  692. p_BBC_Cos_East_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  693. }
  694. }
  695. delete prof;
  696. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Sin%i_East_EP%i_vtx%i", iShift, iHarm, iVtx));
  697. for (int i=0; i<fNBins; i++)
  698. {
  699. for (int iCent=0; iCent<fNCent; iCent++)
  700. {
  701. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  702. if (binCont == 0.) continue;
  703. p_BBC_Sin_East_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  704. }
  705. }
  706. delete prof;
  707. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Cos%i_West_EP%i_vtx%i", iShift, iHarm, iVtx));
  708. for (int i=0; i<fNBins; i++)
  709. {
  710. for (int iCent=0; iCent<fNCent; iCent++)
  711. {
  712. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  713. if (binCont == 0.) continue;
  714. p_BBC_Cos_West_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  715. }
  716. }
  717. delete prof;
  718. prof = (TProfile2D *)fiShiftBBC->Get(Form("p_BBC_Sin%i_West_EP%i_vtx%i", iShift, iHarm, iVtx));
  719. for (int i=0; i<fNBins; i++)
  720. {
  721. for (int iCent=0; iCent<fNCent; iCent++)
  722. {
  723. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  724. if (binCont == 0.) continue;
  725. p_BBC_Sin_West_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  726. }
  727. }
  728. delete prof;
  729. }
  730. }
  731. }
  732. //Manage recentering ZDC
  733. std::map<Double_t,Double_t> p_ZDC_Qx_Full_EP[fNharmonics][fArraySize][fNCent];
  734. std::map<Double_t,Double_t> p_ZDC_Qy_Full_EP[fNharmonics][fArraySize][fNCent];
  735. std::map<Double_t,Double_t> p_ZDC_Qx_East_EP[fNharmonics][fArraySize][fNCent];
  736. std::map<Double_t,Double_t> p_ZDC_Qy_East_EP[fNharmonics][fArraySize][fNCent];
  737. std::map<Double_t,Double_t> p_ZDC_Qx_West_EP[fNharmonics][fArraySize][fNCent];
  738. std::map<Double_t,Double_t> p_ZDC_Qy_West_EP[fNharmonics][fArraySize][fNCent];
  739. TFile *fiReCentZDC = new TFile(reCentFileZDC, "read");
  740. std::cout << "Reading file for ZDC recentering..." << std::endl;
  741. for (int iHarm = 0; iHarm < fNharmonics; iHarm++)
  742. {
  743. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  744. {
  745. std::cout << "iHarm: " << iHarm << " iVtx: " << iVtx << std::endl;
  746. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qx_Full_EP%i_vtx%i", iHarm, iVtx));
  747. for (int i=0; i<fNBins; i++)
  748. {
  749. for (int iCent=0; iCent<fNCent; iCent++)
  750. {
  751. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  752. if (binCont == 0.) continue;
  753. p_ZDC_Qx_Full_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  754. }
  755. }
  756. delete prof;
  757. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qy_Full_EP%i_vtx%i", iHarm, iVtx));
  758. for (int i=0; i<fNBins; i++)
  759. {
  760. for (int iCent=0; iCent<fNCent; iCent++)
  761. {
  762. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  763. if (binCont == 0.) continue;
  764. p_ZDC_Qy_Full_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  765. }
  766. }
  767. delete prof;
  768. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qx_East_EP%i_vtx%i", iHarm, iVtx));
  769. for (int i=0; i<fNBins; i++)
  770. {
  771. for (int iCent=0; iCent<fNCent; iCent++)
  772. {
  773. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  774. if (binCont == 0.) continue;
  775. p_ZDC_Qx_East_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  776. }
  777. }
  778. delete prof;
  779. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qy_East_EP%i_vtx%i", iHarm, iVtx));
  780. for (int i=0; i<fNBins; i++)
  781. {
  782. for (int iCent=0; iCent<fNCent; iCent++)
  783. {
  784. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  785. if (binCont == 0.) continue;
  786. p_ZDC_Qy_East_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  787. }
  788. }
  789. delete prof;
  790. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qx_West_EP%i_vtx%i", iHarm, iVtx));
  791. for (int i=0; i<fNBins; i++)
  792. {
  793. for (int iCent=0; iCent<fNCent; iCent++)
  794. {
  795. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  796. if (binCont == 0.) continue;
  797. p_ZDC_Qx_West_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  798. }
  799. }
  800. delete prof;
  801. prof = (TProfile2D *)fiReCentZDC->Get(Form("p_ZDC_Qy_West_EP%i_vtx%i", iHarm, iVtx));
  802. for (int i=0; i<fNBins; i++)
  803. {
  804. for (int iCent=0; iCent<fNCent; iCent++)
  805. {
  806. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  807. if (binCont == 0.) continue;
  808. p_ZDC_Qy_West_EP[iHarm][iVtx][iCent][(Double_t) (runId_min+i)] = binCont;
  809. }
  810. }
  811. delete prof;
  812. }
  813. }
  814. //Manage ShiftCorr ZDC
  815. std::map<Double_t,Double_t> p_ZDC_Cos_Full_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  816. std::map<Double_t,Double_t> p_ZDC_Sin_Full_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  817. std::map<Double_t,Double_t> p_ZDC_Cos_East_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  818. std::map<Double_t,Double_t> p_ZDC_Sin_East_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  819. std::map<Double_t,Double_t> p_ZDC_Cos_West_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  820. std::map<Double_t,Double_t> p_ZDC_Sin_West_EP[fNharmonics + 1][fArraySize][NShiftOrderMax*3][fNCent];
  821. TFile *fiShiftZDC = new TFile(shiftFileZDC, "read");
  822. fiShiftZDC->cd();
  823. std::cout << "Reading file for ZDC shift corrections..." << std::endl;
  824. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  825. {
  826. for (int iVtx = 0; iVtx < fArraySize; iVtx++)
  827. {
  828. for (int iShift = 0; iShift < NShiftOrderMax*3; iShift++)
  829. {
  830. std::cout << "iHarm: " << iHarm << " iVtx: " << iVtx << " iShiftOrder: " << iShift << std::endl;
  831. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Cos%i_Full_EP%i_vtx%i", iShift, iHarm, iVtx));
  832. for (int i=0; i<fNBins; i++)
  833. {
  834. for (int iCent=0; iCent<fNCent; iCent++)
  835. {
  836. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  837. if (binCont == 0.) continue;
  838. p_ZDC_Cos_Full_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  839. }
  840. }
  841. delete prof;
  842. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Sin%i_Full_EP%i_vtx%i", iShift, iHarm, iVtx));
  843. for (int i=0; i<fNBins; i++)
  844. {
  845. for (int iCent=0; iCent<fNCent; iCent++)
  846. {
  847. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  848. if (binCont == 0.) continue;
  849. p_ZDC_Sin_Full_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  850. }
  851. }
  852. delete prof;
  853. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Cos%i_East_EP%i_vtx%i", iShift, iHarm, iVtx));
  854. for (int i=0; i<fNBins; i++)
  855. {
  856. for (int iCent=0; iCent<fNCent; iCent++)
  857. {
  858. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  859. if (binCont == 0.) continue;
  860. p_ZDC_Cos_East_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  861. }
  862. }
  863. delete prof;
  864. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Sin%i_East_EP%i_vtx%i", iShift, iHarm, iVtx));
  865. for (int i=0; i<fNBins; i++)
  866. {
  867. for (int iCent=0; iCent<fNCent; iCent++)
  868. {
  869. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  870. if (binCont == 0.) continue;
  871. p_ZDC_Sin_East_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  872. }
  873. }
  874. delete prof;
  875. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Cos%i_West_EP%i_vtx%i", iShift, iHarm, iVtx));
  876. for (int i=0; i<fNBins; i++)
  877. {
  878. for (int iCent=0; iCent<fNCent; iCent++)
  879. {
  880. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  881. if (binCont == 0.) continue;
  882. p_ZDC_Cos_West_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  883. }
  884. }
  885. delete prof;
  886. prof = (TProfile2D *)fiShiftZDC->Get(Form("p_ZDC_Sin%i_West_EP%i_vtx%i", iShift, iHarm, iVtx));
  887. for (int i=0; i<fNBins; i++)
  888. {
  889. for (int iCent=0; iCent<fNCent; iCent++)
  890. {
  891. binCont = prof->GetBinContent(prof->FindBin((Double_t) (runId_min+i),(Double_t) iCent));
  892. if (binCont == 0.) continue;
  893. p_ZDC_Sin_West_EP[iHarm][iVtx][iShift][iCent][(Double_t) (runId_min+i)] = binCont;
  894. }
  895. }
  896. delete prof;
  897. }
  898. }
  899. }
  900. //Manage output TProfiles
  901. TProfile *p_res2_EP[NEtaGaps];
  902. TProfile *p_res3_EP[NEtaGaps];
  903. TProfile *p_res2_SP[NEtaGaps];
  904. TProfile *p_res3_SP[NEtaGaps];
  905. TProfile *p_res2_SP_Norm[NEtaGaps];
  906. TProfile *p_res3_SP_Norm[NEtaGaps];
  907. // <Res> vs RunId & centrality - for RunQA
  908. // In this case, standard EP from 2-sub(TPC) method is used
  909. TProfile2D *p_res2_vs_run;
  910. TProfile2D *p_res3_vs_run;
  911. p_res2_vs_run = new TProfile2D(Form("p_res2_vs_run"),Form("p_res2_vs_run"),fNBins, runId_min - 0.5, runId_max + 0.5, 16, -0.5, 15.5);
  912. p_res3_vs_run = new TProfile2D(Form("p_res3_vs_run"),Form("p_res3_vs_run"),fNBins, runId_min - 0.5, runId_max + 0.5, 16, -0.5, 15.5);
  913. // 3-sub method
  914. //
  915. // A - TPC-full, B - BBC-East, C - BBC-West
  916. TProfile *p_res2_EP_3sub_AB = new TProfile(Form("p_res2_EP_3sub_AB"), Form("p_res2_EP_3sub_AB"), 16, -0.5, 15.5);
  917. TProfile *p_res2_EP_3sub_AC = new TProfile(Form("p_res2_EP_3sub_AC"), Form("p_res2_EP_3sub_AC"), 16, -0.5, 15.5);
  918. TProfile *p_res2_EP_3sub_BC = new TProfile(Form("p_res2_EP_3sub_BC"), Form("p_res2_EP_3sub_BC"), 16, -0.5, 15.5);
  919. TProfile *p_res3_EP_3sub_AB = new TProfile(Form("p_res3_EP_3sub_AB"), Form("p_res3_EP_3sub_AB"), 16, -0.5, 15.5);
  920. TProfile *p_res3_EP_3sub_AC = new TProfile(Form("p_res3_EP_3sub_AC"), Form("p_res3_EP_3sub_AC"), 16, -0.5, 15.5);
  921. TProfile *p_res3_EP_3sub_BC = new TProfile(Form("p_res3_EP_3sub_BC"), Form("p_res3_EP_3sub_BC"), 16, -0.5, 15.5);
  922. // A - BBC-full, B - TPC-East, C - TPC-West
  923. TProfile *p_res2_EP_3sub1_AB = new TProfile(Form("p_res2_EP_3sub1_AB"), Form("p_res2_EP_3sub1_AB"), 16, -0.5, 15.5);
  924. TProfile *p_res2_EP_3sub1_AC = new TProfile(Form("p_res2_EP_3sub1_AC"), Form("p_res2_EP_3sub1_AC"), 16, -0.5, 15.5);
  925. TProfile *p_res2_EP_3sub1_BC = new TProfile(Form("p_res2_EP_3sub1_BC"), Form("p_res2_EP_3sub1_BC"), 16, -0.5, 15.5);
  926. TProfile *p_res3_EP_3sub1_AB = new TProfile(Form("p_res3_EP_3sub1_AB"), Form("p_res3_EP_3sub1_AB"), 16, -0.5, 15.5);
  927. TProfile *p_res3_EP_3sub1_AC = new TProfile(Form("p_res3_EP_3sub1_AC"), Form("p_res3_EP_3sub1_AC"), 16, -0.5, 15.5);
  928. TProfile *p_res3_EP_3sub1_BC = new TProfile(Form("p_res3_EP_3sub1_BC"), Form("p_res3_EP_3sub1_BC"), 16, -0.5, 15.5);
  929. TProfile *p_resBBC_EP[fNharmonics+1];
  930. TProfile *p_resBBC_EP_firstHarm[fNharmonics+1];
  931. TProfile *p_resTPCBBCEast = new TProfile("p_resTPCBBCEast","p_resTPCBBCEast",16,-0.5,15.5);
  932. TProfile *p_resTPCBBCWest = new TProfile("p_resTPCBBCWest","p_resTPCBBCWest",16,-0.5,15.5);
  933. TProfile *p_resTPCWestBBCEast = new TProfile("p_resTPCWestBBCEast","p_resTPCWestBBCEast",16,-0.5,15.5);
  934. TProfile *p_resTPCEastBBCWest = new TProfile("p_resTPCEastBBCWest","p_resTPCEastBBCWest",16,-0.5,15.5);
  935. TProfile *p_resZDC_EP[fNharmonics+1]; // ZDC E,W
  936. TProfile *p_resZDC1_3sub[fNharmonics+1]; // ZDC E,W & TPC
  937. TProfile *p_resZDC2_3sub[fNharmonics+1]; // TPC E,W & ZDC
  938. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  939. {
  940. p_res2_EP[iGap] = new TProfile(Form("p_res2_EP_gap%i", iGap), Form("p_res2_EP_gap%i", iGap), 16, -0.5, 15.5);
  941. p_res3_EP[iGap] = new TProfile(Form("p_res3_EP_gap%i", iGap), Form("p_res3_EP_gap%i", iGap), 16, -0.5, 15.5);
  942. p_res2_SP[iGap] = new TProfile(Form("p_res2_SP_gap%i", iGap), Form("p_res2_SP_gap%i", iGap), 16, -0.5, 15.5);
  943. p_res3_SP[iGap] = new TProfile(Form("p_res3_SP_gap%i", iGap), Form("p_res3_SP_gap%i", iGap), 16, -0.5, 15.5);
  944. p_res2_SP_Norm[iGap] = new TProfile(Form("p_res2_SP_Norm_gap%i", iGap), Form("p_res2_SP_Norm_gap%i", iGap), 16, -0.5, 15.5);
  945. p_res3_SP_Norm[iGap] = new TProfile(Form("p_res3_SP_Norm_gap%i", iGap), Form("p_res3_SP_Norm_gap%i", iGap), 16, -0.5, 15.5);
  946. }
  947. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  948. {
  949. p_resBBC_EP[iHarm] = new TProfile(Form("p_resBBC_EP%i", iHarm), Form("p_resBBC_EP%i", iHarm), 16, -0.5, 15.5);
  950. p_resBBC_EP_firstHarm[iHarm] = new TProfile(Form("p_resBBC_EP_firstHarm%i", iHarm), Form("p_resBBC_EP_firstHarm%i", iHarm), 16, -0.5, 15.5);
  951. p_resZDC_EP[iHarm] = new TProfile(Form("p_resZDC_EP%i", iHarm), Form("p_resZDC_EP%i", iHarm), 16, -0.5, 15.5);
  952. p_resZDC1_3sub[iHarm] = new TProfile(Form("p_resZDC1_3sub%i", iHarm), Form("p_resZDC1_3sub%i", iHarm), 16, -0.5, 15.5);
  953. p_resZDC2_3sub[iHarm] = new TProfile(Form("p_resZDC2_3sub%i", iHarm), Form("p_resZDC2_3sub%i", iHarm), 16, -0.5, 15.5);
  954. }
  955. //Counters
  956. Int_t fUsedTracks = 0;
  957. Int_t fQcountFull = 0;
  958. Int_t fQcountFullEast = 0;
  959. Int_t fQcountFullWest = 0;
  960. Int_t fQcountWest[NEtaGaps];
  961. Int_t fQcountEast[NEtaGaps];
  962. //Q-vectors
  963. TVector2 fQv2FullEP;
  964. TVector2 fQv2FullSP;
  965. TVector2 fQv3FullEP;
  966. TVector2 fQv3FullSP;
  967. TVector2 fQv2WestEP[NEtaGaps];
  968. TVector2 fQv2EastEP[NEtaGaps];
  969. TVector2 fQv2WestSP[NEtaGaps];
  970. TVector2 fQv2EastSP[NEtaGaps];
  971. TVector2 fQv3WestEP[NEtaGaps];
  972. TVector2 fQv3EastEP[NEtaGaps];
  973. TVector2 fQv3WestSP[NEtaGaps];
  974. TVector2 fQv3EastSP[NEtaGaps];
  975. TVector2 QvTMP;
  976. TVector2 fQv2WestSP_Shift[NEtaGaps];
  977. TVector2 fQv2EastSP_Shift[NEtaGaps];
  978. TVector2 fQv3WestSP_Shift[NEtaGaps];
  979. TVector2 fQv3EastSP_Shift[NEtaGaps];
  980. Double_t weight;
  981. Int_t VtxSign;
  982. Double_t psiShiftedTMP;
  983. Double_t PsiSin;
  984. Double_t PsiCos;
  985. Long64_t goodEventCounter = 0;
  986. Double_t Psi2FullEP, Psi3FullEP;
  987. Double_t Psi2EastEP[NEtaGaps], Psi3EastEP[NEtaGaps];
  988. Double_t Psi2WestEP[NEtaGaps], Psi3WestEP[NEtaGaps];
  989. Double_t Psi2EastSP[NEtaGaps], Psi3EastSP[NEtaGaps];
  990. Double_t Psi2WestSP[NEtaGaps], Psi3WestSP[NEtaGaps];
  991. Double_t Psi_ReCenter, dPsi, psiEP_BBC_Full[fNharmonics+1];
  992. Int_t bin;
  993. Double_t res2EP[NEtaGaps], res3EP[NEtaGaps];
  994. Double_t res2SP[NEtaGaps], res3SP[NEtaGaps];
  995. Double_t res2EP_3subAB, res2EP_3subAC, res2EP_3subBC;
  996. Double_t res3EP_3subAB, res3EP_3subAC, res3EP_3subBC;
  997. Double_t res2EP_3sub1AB, res2EP_3sub1AC, res2EP_3sub1BC;
  998. Double_t res3EP_3sub1AB, res3EP_3sub1AC, res3EP_3sub1BC;
  999. // BBC related EP determination
  1000. Double_t qxEastEP[fNharmonics + 1], qyEastEP[fNharmonics + 1], qxWestEP[fNharmonics + 1], qyWestEP[fNharmonics + 1];
  1001. Double_t adcEast, adcWest, adcSumEast, adcSumWest, adcNormEastInner, adcNormWestInner, adcNormEastOuter, adcNormWestOuter;
  1002. Int_t adcBin, shiftBin;
  1003. Double_t psiEP_rec, psiEP_BBC_East[fNharmonics+1], psiEP_BBC_West[fNharmonics+1], shiftCos, shiftSin;
  1004. Double_t resEastWestBBC[fNharmonics + 1], resTPCEastBBC[fNharmonics + 1], resTPCWestBBC[fNharmonics + 1], resTPCWestBBCEast[fNharmonics+1], resTPCEastBBCWest[fNharmonics+1];
  1005. Double_t resEastWestBBCFirstHarm[fNharmonics + 1];
  1006. // ZDC related EP determination
  1007. Double_t psiEP_ZDC_Full[fNharmonics+1], psiEP_ZDC_East[fNharmonics+1], psiEP_ZDC_West[fNharmonics+1];
  1008. TVector2 qRecEastZDC[fNharmonics+1], qRecWestZDC[fNharmonics+1], qRecFullZDC[fNharmonics+1];
  1009. Double_t qxEast, qyEast, adcxEast, adcyEast;
  1010. Double_t qxWest, qyWest, adcxWest, adcyWest;
  1011. Double_t qxFull, qyFull, adcxFull, adcyFull;
  1012. Double_t qxRecEast, qyRecEast;
  1013. Double_t qxRecWest, qyRecWest;
  1014. Double_t qxRecFull, qyRecFull;
  1015. Double_t resEastWestZDC[fNharmonics + 1];
  1016. // Loop over events
  1017. for (Long64_t iEvent = 0; iEvent < events2read; iEvent++)
  1018. {
  1019. if ((iEvent + 1) % 1000 == 0)
  1020. {
  1021. std::cout << "Working on event #[" << (iEvent + 1)
  1022. << "/" << events2read << "]" << std::endl;
  1023. }
  1024. Bool_t readEvent = femtoReader->readFemtoEvent(iEvent);
  1025. if (!readEvent)
  1026. {
  1027. std::cout << "Something went wrong, Master! Nothing to analyze..." << std::endl;
  1028. break;
  1029. }
  1030. // Retrieve femtoDst
  1031. StFemtoDst *dst = femtoReader->femtoDst();
  1032. // Retrieve event information
  1033. StFemtoEvent *event = dst->event();
  1034. if (!event)
  1035. {
  1036. std::cout << "Something went wrong, Master! Event is hiding from me..." << std::endl;
  1037. break;
  1038. }
  1039. // Event selection
  1040. if (!isGoodEvent(event))
  1041. continue;
  1042. goodEventCounter++;
  1043. TVector3 pVtx = event->primaryVertex();
  1044. // Track analysis
  1045. Int_t nTracks = dst->numberOfTracks();
  1046. //Counters
  1047. fUsedTracks = 0;
  1048. fQcountFull = 0;
  1049. fQcountFullEast = 0;
  1050. fQcountFullWest = 0;
  1051. psiShiftedTMP = 0.;
  1052. PsiSin = 0.;
  1053. PsiCos = 0.;
  1054. fQv2FullEP.Set(0., 0.);
  1055. fQv2FullSP.Set(0., 0.);
  1056. fQv3FullEP.Set(0., 0.);
  1057. fQv3FullSP.Set(0., 0.);
  1058. QvTMP.Set(0., 0.);
  1059. for (int i = 0; i < NEtaGaps; i++)
  1060. {
  1061. fQcountWest[i] = 0;
  1062. fQcountEast[i] = 0;
  1063. fQv2WestEP[i].Set(0., 0.);
  1064. fQv2EastEP[i].Set(0., 0.);
  1065. fQv2WestSP[i].Set(0., 0.);
  1066. fQv2EastSP[i].Set(0., 0.);
  1067. fQv3WestEP[i].Set(0., 0.);
  1068. fQv3EastEP[i].Set(0., 0.);
  1069. fQv3WestSP[i].Set(0., 0.);
  1070. fQv3EastSP[i].Set(0., 0.);
  1071. fQv2WestSP_Shift[i].Set(0., 0.);
  1072. fQv2EastSP_Shift[i].Set(0., 0.);
  1073. fQv3WestSP_Shift[i].Set(0., 0.);
  1074. fQv3EastSP_Shift[i].Set(0., 0.);
  1075. }
  1076. // Vertex sign
  1077. VtxSign = GetVzBin(pVtx.Z());
  1078. if (VtxSign == -1)
  1079. continue;
  1080. //BBC EP
  1081. Double_t adcNormEastInner = BBCgain_East[VtxSign]->Integral(BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),1,6)/6.;
  1082. Double_t adcNormEastOuter = BBCgain_East[VtxSign]->Integral(BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),7,16)/10.;
  1083. Double_t adcNormWestInner = BBCgain_West[VtxSign]->Integral(BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),1,6)/6.;
  1084. Double_t adcNormWestOuter = BBCgain_West[VtxSign]->Integral(BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),BBCgain_East[VtxSign]->GetXaxis()->FindBin(event->runId()),7,16)/10.;
  1085. float egain[16]={
  1086. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),1))/adcNormEastInner),
  1087. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),2))/adcNormEastInner),
  1088. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),3))/adcNormEastInner),
  1089. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),4))/adcNormEastInner),
  1090. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),5))/adcNormEastInner),
  1091. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),6))/adcNormEastInner),
  1092. (float)((0.5)*BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),7))/adcNormEastOuter),
  1093. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),8))/adcNormEastOuter),
  1094. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),9))/adcNormEastOuter),
  1095. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),10))/adcNormEastOuter),
  1096. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),11))/adcNormEastOuter),
  1097. (float)((0.5)*BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),12))/adcNormEastOuter),
  1098. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),13))/adcNormEastOuter),
  1099. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),14))/adcNormEastOuter),
  1100. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),15))/adcNormEastOuter),
  1101. (float)(BBCgain_East[VtxSign]->GetBinContent(BBCgain_East[VtxSign]->FindBin(event->runId(),16))/adcNormEastOuter)
  1102. };
  1103. float wgain[16]={
  1104. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),1))/adcNormWestInner),
  1105. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),2))/adcNormWestInner),
  1106. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),3))/adcNormWestInner),
  1107. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),4))/adcNormWestInner),
  1108. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),5))/adcNormWestInner),
  1109. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),6))/adcNormWestInner),
  1110. (float)((0.5)*BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),7))/adcNormWestOuter),
  1111. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),8))/adcNormWestOuter),
  1112. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),9))/adcNormWestOuter),
  1113. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),10))/adcNormWestOuter),
  1114. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),11))/adcNormWestOuter),
  1115. (float)((0.5)*BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),12))/adcNormWestOuter),
  1116. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),13))/adcNormWestOuter),
  1117. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),14))/adcNormWestOuter),
  1118. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),15))/adcNormWestOuter),
  1119. (float)(BBCgain_West[VtxSign]->GetBinContent(BBCgain_West[VtxSign]->FindBin(event->runId(),16))/adcNormWestOuter)
  1120. };
  1121. float nHitE_Gain[16], nHitW_Gain[16];
  1122. for(int i=0;i<16;i++){
  1123. nHitE_Gain[i] = event->bbcAdcEast(i)/egain[i];
  1124. nHitW_Gain[i] = event->bbcAdcWest(i)/wgain[i];
  1125. }
  1126. Double_t psiWest[fNharmonics+1];
  1127. Double_t psiEast[fNharmonics+1];
  1128. Double_t psiFull[fNharmonics+1];
  1129. TVector2 qGainEast[fNharmonics+1], qGainWest[fNharmonics+1], qGainFull[fNharmonics+1];
  1130. TVector2 qRecEast[fNharmonics+1], qRecWest[fNharmonics+1], qRecFull[fNharmonics+1];
  1131. for (int iHarm=0; iHarm<fNharmonics+1; iHarm++)
  1132. {
  1133. //BBC Full
  1134. //v1*y > 0 by conventions most reasonable to the STAR event plane
  1135. //therefore the west event plane has the right sign for the event plane
  1136. //more simply (Psi_w = 0 => QWest . x_STAR > 0) and (Psi_e = 0 => QEast . x_STAR < 0)
  1137. //The east event plane points opposite that of the west and the west finds the "correct" EP for the STAR coordinates, use West-East
  1138. Double_t QVector_GainX = 0., QVector_GainY = 0.;
  1139. Double_t EventPlaneGainFull = 0.;
  1140. Float_t eXFsum_Gain = 0., wXFsum_Gain = 0., eYFsum_Gain = 0., wYFsum_Gain = 0., eWFgt_Gain=0., wWFgt_Gain = 0.;
  1141. TVector2 QFullGain;
  1142. for(int iTile = 0; iTile < 16; iTile++) {
  1143. eXFsum_Gain += cos((iHarm+1.)*BBC_GetPhi(0,iTile))*nHitE_Gain[iTile];
  1144. wXFsum_Gain += cos((iHarm+1.)*BBC_GetPhi(1,iTile))*nHitW_Gain[iTile];
  1145. eYFsum_Gain += sin((iHarm+1.)*BBC_GetPhi(0,iTile))*nHitE_Gain[iTile];
  1146. wYFsum_Gain += sin((iHarm+1.)*BBC_GetPhi(1,iTile))*nHitW_Gain[iTile];
  1147. eWFgt_Gain += nHitE_Gain[iTile];
  1148. wWFgt_Gain += nHitW_Gain[iTile];
  1149. }
  1150. QVector_GainX=(eWFgt_Gain > 0. && wWFgt_Gain > 0.) ? wXFsum_Gain/wWFgt_Gain + eXFsum_Gain/eWFgt_Gain:0.;
  1151. QVector_GainY=(eWFgt_Gain > 0. && wWFgt_Gain > 0.) ? wYFsum_Gain/wWFgt_Gain + eYFsum_Gain/eWFgt_Gain:0.;
  1152. QFullGain.Set(QVector_GainX,QVector_GainY);
  1153. if(QFullGain.Mod()){
  1154. EventPlaneGainFull = QFullGain.Phi();
  1155. if(EventPlaneGainFull < 0.0) {EventPlaneGainFull += 2.*TMath::Pi()/(iHarm+1.);}
  1156. }
  1157. psiFull[iHarm] = EventPlaneGainFull;
  1158. qGainFull[iHarm] = QFullGain;
  1159. //BBC West
  1160. //v1*y > 0 by conventions most reasonable to the STAR event plane
  1161. //therefore the west event plane has the right sign for the event plane
  1162. //more simply (Psi_w = 0 => QWest . x_STAR > 0) and (Psi_e = 0 => QEast . x_STAR < 0)
  1163. Double_t QVectorW_GainX = 0., QVectorW_GainY = 0.;
  1164. Double_t EventPlaneGainWest = 0.;
  1165. Float_t wXsum_Gain = 0., wYsum_Gain = 0., wWgt_Gain = 0.;
  1166. TVector2 QWestGain;
  1167. for(int iTile = 0; iTile < 16; iTile++) {
  1168. wXsum_Gain += cos((iHarm+1.)*BBC_GetPhi(1,iTile))*nHitW_Gain[iTile];
  1169. wYsum_Gain += sin((iHarm+1.)*BBC_GetPhi(1,iTile))*nHitW_Gain[iTile];
  1170. wWgt_Gain += nHitW_Gain[iTile];
  1171. }
  1172. QVectorW_GainX = (wWgt_Gain > 0.0) ? wXsum_Gain/wWgt_Gain:0.0;
  1173. QVectorW_GainY = (wWgt_Gain > 0.0) ? wYsum_Gain/wWgt_Gain:0.0;
  1174. QWestGain.Set(QVectorW_GainX,QVectorW_GainY);
  1175. if(QWestGain.Mod()){
  1176. EventPlaneGainWest = QWestGain.Phi();
  1177. if(EventPlaneGainWest < 0.0) {EventPlaneGainWest += 2.*TMath::Pi()/(iHarm+1.);}
  1178. }
  1179. psiWest[iHarm] = EventPlaneGainWest;
  1180. qGainWest[iHarm] = QWestGain;
  1181. //BBC East
  1182. //in BBC_GetPhi the fact that the tile numbering scheme is reversed about the y axis for the east BBC is accounted for
  1183. //The east event plane points opposite that of the west since the spectators are on different sides -- see west comments
  1184. Double_t QVectorE_GainX = 0., QVectorE_GainY = 0.;
  1185. Double_t EventPlaneGainEast = 0.;
  1186. Float_t eXsum_Gain = 0., eYsum_Gain = 0., eWgt_Gain = 0.;
  1187. TVector2 QEastGain;
  1188. for(int iTile = 0; iTile < 16; iTile++) {
  1189. eXsum_Gain += cos((iHarm+1.)*BBC_GetPhi(0,iTile))*nHitE_Gain[iTile];
  1190. eYsum_Gain += sin((iHarm+1.)*BBC_GetPhi(0,iTile))*nHitE_Gain[iTile];
  1191. eWgt_Gain += nHitE_Gain[iTile];
  1192. }
  1193. QVectorE_GainX = (eWgt_Gain > 0.0) ? eXsum_Gain/eWgt_Gain:0.0;
  1194. QVectorE_GainY = (eWgt_Gain > 0.0) ? eYsum_Gain/eWgt_Gain:0.0;
  1195. QEastGain.Set(QVectorE_GainX,QVectorE_GainY);
  1196. if(QEastGain.Mod()){
  1197. EventPlaneGainEast = QEastGain.Phi();
  1198. if(EventPlaneGainEast < 0.0) {EventPlaneGainEast += 2.*TMath::Pi()/(iHarm+1.);}
  1199. }
  1200. psiEast[iHarm] = EventPlaneGainEast;
  1201. qGainEast[iHarm] = QEastGain;
  1202. }
  1203. // Apply recentering
  1204. for (int iHarm = 0; iHarm < fNharmonics; iHarm++)
  1205. {
  1206. TVector2 recFullTMP((Double_t)p_BBC_Qx_Full_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()],
  1207. (Double_t)p_BBC_Qy_Full_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1208. TVector2 recEastTMP((Double_t)p_BBC_Qx_East_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()],
  1209. (Double_t)p_BBC_Qy_East_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1210. TVector2 recWestTMP((Double_t)p_BBC_Qx_West_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()],
  1211. (Double_t)p_BBC_Qy_West_EP[iHarm][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1212. qRecEast[iHarm] = qGainEast[iHarm] - recEastTMP;
  1213. qRecWest[iHarm] = qGainWest[iHarm] - recWestTMP;
  1214. qRecFull[iHarm] = qGainFull[iHarm] - recFullTMP;
  1215. }
  1216. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  1217. {
  1218. // Full BBC
  1219. if (qRecFull[iHarm].Mod())
  1220. {
  1221. psiEP_rec = qRecFull[iHarm].Phi() / (iHarm+1.);
  1222. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1223. }
  1224. dPsi = 0.;
  1225. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1226. {
  1227. shiftCos = p_BBC_Cos_Full_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1228. shiftSin = p_BBC_Sin_Full_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1229. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1230. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1231. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1232. }
  1233. psiEP_BBC_Full[iHarm] = psiEP_rec + dPsi;
  1234. if(psiEP_BBC_Full[iHarm] < 0.0) psiEP_BBC_Full[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1235. if(psiEP_BBC_Full[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_BBC_Full[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1236. // East BBC
  1237. if (qRecEast[iHarm].Mod())
  1238. {
  1239. psiEP_rec = qRecEast[iHarm].Phi() / (iHarm+1.);
  1240. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1241. }
  1242. dPsi = 0.;
  1243. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1244. {
  1245. shiftCos = p_BBC_Cos_East_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1246. shiftSin = p_BBC_Sin_East_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1247. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1248. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1249. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1250. }
  1251. psiEP_BBC_East[iHarm] = psiEP_rec + dPsi;
  1252. // West BBC
  1253. if (qRecWest[iHarm].Mod())
  1254. {
  1255. psiEP_rec = qRecWest[iHarm].Phi() / (iHarm+1.);
  1256. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1257. }
  1258. dPsi = 0.;
  1259. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1260. {
  1261. shiftCos = p_BBC_Cos_West_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1262. shiftSin = p_BBC_Sin_West_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1263. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1264. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1265. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1266. }
  1267. psiEP_BBC_West[iHarm] = psiEP_rec + dPsi;
  1268. if(psiEP_BBC_East[iHarm] < 0.0) psiEP_BBC_East[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1269. if(psiEP_BBC_East[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_BBC_East[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1270. if(psiEP_BBC_West[iHarm] < 0.0) psiEP_BBC_West[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1271. if(psiEP_BBC_West[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_BBC_West[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1272. }
  1273. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  1274. {
  1275. resEastWestBBC[iHarm] = TMath::Cos( (iHarm+1.) * AngleShift(psiEP_BBC_East[iHarm]-psiEP_BBC_West[iHarm],(iHarm+1.)) );
  1276. // resEastWestBBC[iHarm] = TMath::Cos( (iHarm+1.) * (qRecEast[iHarm].Phi()-qRecWest[iHarm].Phi()) );
  1277. // resEastWestBBC[iHarm] = TMath::Cos( (iHarm+1.) * (qGainEast[iHarm].Phi()-qGainWest[iHarm].Phi()) );
  1278. if (iHarm==0)
  1279. {
  1280. resEastWestBBCFirstHarm[0] = TMath::Cos( 1. * (psiEP_BBC_East[0]-psiEP_BBC_West[0]) );
  1281. resEastWestBBCFirstHarm[1] = TMath::Cos( 2. * (psiEP_BBC_East[0]-psiEP_BBC_West[0]) );
  1282. resEastWestBBCFirstHarm[2] = TMath::Cos( 3. * (psiEP_BBC_East[0]-psiEP_BBC_West[0]) );
  1283. }
  1284. }
  1285. // ZDC EP
  1286. if (event->zdcSumAdcEast() < 1.) continue;
  1287. if (event->zdcSumAdcWest() < 1.) continue;
  1288. qxEast = 0.;
  1289. qyEast = 0.;
  1290. adcxEast = 0.;
  1291. adcyEast = 0.;
  1292. qxWest = 0.;
  1293. qyWest = 0.;
  1294. adcxWest = 0.;
  1295. adcyWest = 0.;
  1296. qxFull = 0.;
  1297. qyFull = 0.;
  1298. adcxFull = 0.;
  1299. adcyFull = 0.;
  1300. // Vertical ZDC-SMD
  1301. for (int iStrip=0; iStrip<fNZdcSmdStripsVertical; iStrip++)
  1302. {
  1303. qxEast += event->zdcSmdEastVertical(iStrip) * GetZDCPosition(0,0,iStrip);
  1304. adcxEast += event->zdcSmdEastVertical(iStrip);
  1305. qxWest += event->zdcSmdWestVertical(iStrip) * GetZDCPosition(1,0,iStrip);
  1306. adcxWest += event->zdcSmdWestVertical(iStrip);
  1307. }
  1308. // Horizontal ZDC-SMD
  1309. for (int iStrip=0; iStrip<fNZdcSmdStripsHorizontal; iStrip++)
  1310. {
  1311. qyEast += event->zdcSmdEastHorizontal(iStrip) * GetZDCPosition(0,1,iStrip);
  1312. adcyEast += event->zdcSmdEastHorizontal(iStrip);
  1313. qyWest += event->zdcSmdWestHorizontal(iStrip) * GetZDCPosition(1,1,iStrip);
  1314. adcyWest += event->zdcSmdWestHorizontal(iStrip);
  1315. }
  1316. if (adcxEast > 0) qxEast /= adcxEast;
  1317. else qxEast = -999.;
  1318. if (adcyEast > 0) qyEast /= adcyEast;
  1319. else qyEast = -999.;
  1320. if (adcxWest > 0) qxWest /= adcxWest;
  1321. else qxWest = -999.;
  1322. if (adcyWest > 0) qyWest /= adcyWest;
  1323. else qyWest = -999.;
  1324. // Full ZDC-SMD Q-vector
  1325. if (qxEast == -999. || qxWest == -999.) qxFull = -999.;
  1326. else qxFull = qxEast - qxWest;
  1327. if (qyEast == -999. || qyWest == -999.) qyFull = -999.;
  1328. else qyFull = qyEast - qyWest;
  1329. if (qxEast != -999.) qxRecEast = qxEast - (Double_t)p_ZDC_Qx_East_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1330. if (qyEast != -999.) qyRecEast = qyEast - (Double_t)p_ZDC_Qy_East_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1331. if (qxWest != -999.) qxRecWest = qxWest - (Double_t)p_ZDC_Qx_West_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1332. if (qyWest != -999.) qyRecWest = qyWest - (Double_t)p_ZDC_Qy_West_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1333. if (qxFull != -999.) qxRecFull = qxFull - (Double_t)p_ZDC_Qx_Full_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1334. if (qyFull != -999.) qyRecFull = qyFull - (Double_t)p_ZDC_Qy_Full_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()];
  1335. if (qxEast == -999.) qxRecEast = qxEast;
  1336. if (qyEast == -999.) qyRecEast = qyEast;
  1337. if (qxWest == -999.) qxRecWest = qxWest;
  1338. if (qyWest == -999.) qyRecWest = qyWest;
  1339. if (qxFull == -999.) qxRecFull = qxFull;
  1340. if (qyFull == -999.) qyRecFull = qyFull;
  1341. PsiSin = 0.; PsiCos = 0.; psiShiftedTMP = 0.;
  1342. for (int iHarm = 0; iHarm < fNharmonics+1; iHarm++)
  1343. {
  1344. qRecEastZDC[iHarm].Set(qxRecEast,qyRecEast);
  1345. qRecWestZDC[iHarm].Set(qxRecWest,qyRecWest);
  1346. qRecFullZDC[iHarm].Set(qxRecFull,qyRecFull);
  1347. // Full ZDC
  1348. if (qRecFullZDC[iHarm].Mod())
  1349. {
  1350. psiEP_rec = qRecFullZDC[iHarm].Phi() / (iHarm+1.);
  1351. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1352. }
  1353. dPsi = 0.;
  1354. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1355. {
  1356. shiftCos = p_ZDC_Cos_Full_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1357. shiftSin = p_ZDC_Sin_Full_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1358. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1359. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1360. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1361. }
  1362. psiEP_ZDC_Full[iHarm] = psiEP_rec + dPsi;
  1363. if(psiEP_ZDC_Full[iHarm] < 0.0) psiEP_ZDC_Full[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1364. if(psiEP_ZDC_Full[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_ZDC_Full[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1365. // East ZDC
  1366. if (qRecEastZDC[iHarm].Mod())
  1367. {
  1368. psiEP_rec = qRecEastZDC[iHarm].Phi() / (iHarm+1.);
  1369. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1370. }
  1371. dPsi = 0.;
  1372. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1373. {
  1374. shiftCos = p_ZDC_Cos_East_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1375. shiftSin = p_ZDC_Sin_East_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1376. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1377. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1378. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1379. }
  1380. psiEP_ZDC_East[iHarm] = psiEP_rec + dPsi;
  1381. if(psiEP_ZDC_East[iHarm] < 0.0) psiEP_ZDC_East[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1382. if(psiEP_ZDC_East[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_ZDC_East[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1383. // West ZDC
  1384. if (qRecWestZDC[iHarm].Mod())
  1385. {
  1386. psiEP_rec = qRecWestZDC[iHarm].Phi() / (iHarm+1.);
  1387. if (psiEP_rec < 0.) {psiEP_rec += 2*TMath::Pi()/(iHarm+1.);}
  1388. }
  1389. dPsi = 0.;
  1390. for (int iK = 0; iK < NShiftOrderMax*3; iK++)
  1391. {
  1392. shiftCos = p_ZDC_Cos_West_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1393. shiftSin = p_ZDC_Sin_West_EP[iHarm][VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1394. dPsi += (1. / (iHarm+1.)) * (2. / (iK + 1.)) *
  1395. (-1. * shiftSin * cos((iHarm+1.) * (iK + 1.) * psiEP_rec) +
  1396. 1. * shiftCos * sin((iHarm+1.) * (iK + 1.) * psiEP_rec));
  1397. }
  1398. psiEP_ZDC_West[iHarm] = psiEP_rec + dPsi;
  1399. if(psiEP_ZDC_West[iHarm] < 0.0) psiEP_ZDC_West[iHarm] += 2.*TMath::Pi()/(iHarm+1.);
  1400. if(psiEP_ZDC_West[iHarm] > 2.*TMath::Pi()/(iHarm+1.)) psiEP_ZDC_West[iHarm] -= 2.*TMath::Pi()/(iHarm+1.);
  1401. }
  1402. for (int iHarm = 0; iHarm < fNharmonics + 1; iHarm++)
  1403. {
  1404. resEastWestZDC[iHarm] = TMath::Cos( (iHarm+1.) * AngleShift(psiEP_ZDC_East[iHarm]-psiEP_ZDC_West[iHarm],(iHarm+1.)) );
  1405. }
  1406. // Track loop
  1407. for (Int_t iTrk = 0; iTrk < nTracks; iTrk++)
  1408. {
  1409. // Retrieve i-th femto track
  1410. StFemtoTrack *femtoTrack = dst->track(iTrk);
  1411. if (!femtoTrack)
  1412. continue;
  1413. // Must be a primary track
  1414. if (!femtoTrack->isPrimary())
  1415. continue;
  1416. //Track selection
  1417. if (!isGoodTrack(femtoTrack, energy, pVtx))
  1418. continue;
  1419. fUsedTracks++;
  1420. //Fill recentered Q-vectors for all TPC (FULL)
  1421. weight = GetWeight(femtoTrack);
  1422. QvTMP.SetX(p_Qx_FULL_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1423. QvTMP.SetY(p_Qx_FULL_EP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1424. fQv2FullEP += weight*(CalcQvector(femtoTrack,2) - QvTMP);
  1425. QvTMP.Set(0.,0.);
  1426. QvTMP.SetX(p_Qx_FULL_EP[1][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1427. QvTMP.SetY(p_Qy_FULL_EP[1][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1428. fQv3FullEP += weight*(CalcQvector(femtoTrack,3) - QvTMP);
  1429. QvTMP.Set(0.,0.);
  1430. QvTMP.SetX(p_Qx_FULL_SP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1431. QvTMP.SetY(p_Qy_FULL_SP[0][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1432. fQv2FullSP += 1.*(CalcQvector(femtoTrack,2) - QvTMP);
  1433. QvTMP.Set(0.,0.);
  1434. QvTMP.SetX(p_Qx_FULL_SP[1][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1435. QvTMP.SetY(p_Qy_FULL_SP[1][VtxSign][(Int_t)event->cent16()][(Double_t)event->runId()]);
  1436. fQv3FullSP += 1.*(CalcQvector(femtoTrack,3) - QvTMP);
  1437. QvTMP.Set(0.,0.);
  1438. fQcountFull++;
  1439. if (femtoTrack->eta() >= 0.)
  1440. {
  1441. fQcountFullWest++;
  1442. }
  1443. else
  1444. {
  1445. fQcountFullEast++;
  1446. }
  1447. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  1448. {
  1449. //fill Qvectors from EAST arm
  1450. if (femtoTrack->eta() > -1. * cutEta && femtoTrack->eta() < cutEtaGap[iGap])
  1451. {
  1452. QvTMP.SetX(p_Qx_EAST_EP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1453. QvTMP.SetY(p_Qy_EAST_EP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1454. fQv2EastEP[iGap] += weight * (CalcQvector(femtoTrack, 2) - QvTMP);
  1455. QvTMP.Set(0., 0.);
  1456. QvTMP.SetX(p_Qx_EAST_EP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1457. QvTMP.SetY(p_Qy_EAST_EP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1458. fQv3EastEP[iGap] += weight * (CalcQvector(femtoTrack, 3) - QvTMP);
  1459. QvTMP.Set(0., 0.);
  1460. QvTMP.SetX(p_Qx_EAST_SP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1461. QvTMP.SetY(p_Qy_EAST_SP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1462. fQv2EastSP[iGap] += 1.*(CalcQvector(femtoTrack,2) - QvTMP);
  1463. QvTMP.Set(0.,0.);
  1464. QvTMP.SetX(p_Qx_EAST_SP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1465. QvTMP.SetY(p_Qy_EAST_SP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1466. fQv3EastSP[iGap] += 1.*(CalcQvector(femtoTrack,3) - QvTMP);
  1467. QvTMP.Set(0.,0.);
  1468. fQcountEast[iGap]++;
  1469. }
  1470. //fill Qvectors from WEST arm
  1471. if (femtoTrack->eta() < 1. * cutEta && femtoTrack->eta() > cutEtaGap[iGap])
  1472. {
  1473. QvTMP.SetX(p_Qx_WEST_EP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1474. QvTMP.SetY(p_Qy_WEST_EP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1475. fQv2WestEP[iGap] += weight * (CalcQvector(femtoTrack, 2) - QvTMP);
  1476. QvTMP.Set(0., 0.);
  1477. QvTMP.SetX(p_Qx_WEST_EP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1478. QvTMP.SetY(p_Qy_WEST_EP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1479. fQv3WestEP[iGap] += weight * (CalcQvector(femtoTrack, 3) - QvTMP);
  1480. QvTMP.Set(0., 0.);
  1481. QvTMP.SetX(p_Qx_WEST_SP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1482. QvTMP.SetY(p_Qy_WEST_SP[0][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1483. fQv2WestSP[iGap] += 1.*(CalcQvector(femtoTrack,2) - QvTMP);
  1484. QvTMP.Set(0.,0.);
  1485. QvTMP.SetX(p_Qx_WEST_SP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1486. QvTMP.SetY(p_Qy_WEST_SP[1][VtxSign][(Int_t)event->cent16()][iGap][(Double_t)event->runId()]);
  1487. fQv3WestSP[iGap] += 1.*(CalcQvector(femtoTrack,3) - QvTMP);
  1488. QvTMP.Set(0.,0.);
  1489. fQcountWest[iGap]++;
  1490. }
  1491. }
  1492. // Check if track has TOF signal
  1493. if (femtoTrack->isTofTrack())
  1494. {
  1495. } //if( isTofTrack() )
  1496. } //for(Int_t iTrk=0; iTrk<nTracks; iTrk++)
  1497. //Getting PsiEP with recentering + shift correction
  1498. Psi_ReCenter = 0.;
  1499. Double_t mCos, mSin;
  1500. if (fQcountFullWest + fQcountFullEast > cutNcountQvFull)
  1501. {
  1502. Psi2FullEP = 0.;
  1503. Psi3FullEP = 0.;
  1504. //FULL 2-nd order
  1505. Psi_ReCenter = TMath::ATan2(fQv2FullEP.Y(), fQv2FullEP.X()) / 2.;
  1506. dPsi = 0.;
  1507. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1508. {
  1509. mCos = p_Cos2_FULL_EP[VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1510. mSin = p_Cos2_FULL_EP[VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1511. dPsi += (1. / 2.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff2.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff2.at(iK) * Psi_ReCenter));
  1512. }
  1513. Psi2FullEP = AngleShift(Psi_ReCenter + dPsi, 2.);
  1514. //FULL 3-nd order
  1515. Psi_ReCenter = TMath::ATan2(fQv3FullEP.Y(), fQv3FullEP.X()) / 3.;
  1516. dPsi = 0.;
  1517. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1518. {
  1519. mCos = p_Cos3_FULL_EP[VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1520. mSin = p_Cos3_FULL_EP[VtxSign][iK][(Int_t)event->cent16()][(Double_t)event->runId()];
  1521. dPsi += (1. / 3.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff3.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff3.at(iK) * Psi_ReCenter));
  1522. }
  1523. Psi3FullEP = AngleShift(Psi_ReCenter + dPsi, 3.);
  1524. }
  1525. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  1526. {
  1527. if (fQcountEast[iGap] > cutNcountQvGap &&
  1528. fQcountWest[iGap] > cutNcountQvGap)
  1529. {
  1530. Psi2EastEP[iGap] = 0.;
  1531. Psi3EastEP[iGap] = 0.;
  1532. Psi2WestEP[iGap] = 0.;
  1533. Psi3WestEP[iGap] = 0.;
  1534. Psi2EastSP[iGap] = 0.;
  1535. Psi3EastSP[iGap] = 0.;
  1536. Psi2WestSP[iGap] = 0.;
  1537. Psi3WestSP[iGap] = 0.;
  1538. //EAST 2-nd order
  1539. Psi_ReCenter = TMath::ATan2(fQv2EastEP[iGap].Y(), fQv2EastEP[iGap].X()) / 2.;
  1540. dPsi = 0.;
  1541. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1542. {
  1543. mCos = p_Cos2_EAST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1544. mSin = p_Cos2_EAST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1545. dPsi += (1. / 2.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff2.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff2.at(iK) * Psi_ReCenter));
  1546. }
  1547. Psi2EastEP[iGap] = AngleShift(Psi_ReCenter + dPsi, 2.);
  1548. Psi_ReCenter = TMath::ATan2(fQv2EastSP[iGap].Y(), fQv2EastSP[iGap].X()) / 2.;
  1549. dPsi = 0.;
  1550. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1551. {
  1552. mCos = p_Cos2_EAST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1553. mSin = p_Cos2_EAST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1554. dPsi += (1. / 2.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff2.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff2.at(iK) * Psi_ReCenter));
  1555. }
  1556. Psi2EastSP[iGap] = AngleShift(Psi_ReCenter + dPsi, 2.);
  1557. fQv2EastSP_Shift[iGap].Set(fQv2EastSP[iGap].Mod()*TMath::Cos(2.*Psi2EastSP[iGap]),fQv2EastSP[iGap].Mod()*TMath::Sin(2.*Psi2EastSP[iGap]));
  1558. //WEST 2-nd order
  1559. Psi_ReCenter = TMath::ATan2(fQv2WestEP[iGap].Y(), fQv2WestEP[iGap].X()) / 2.;
  1560. dPsi = 0.;
  1561. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1562. {
  1563. mCos = p_Cos2_WEST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1564. mSin = p_Cos2_WEST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1565. dPsi += (1. / 2.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff2.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff2.at(iK) * Psi_ReCenter));
  1566. }
  1567. Psi2WestEP[iGap] = AngleShift(Psi_ReCenter + dPsi, 2.);
  1568. Psi_ReCenter = TMath::ATan2(fQv2WestSP[iGap].Y(), fQv2WestSP[iGap].X()) / 2.;
  1569. dPsi = 0.;
  1570. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1571. {
  1572. mCos = p_Cos2_WEST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1573. mSin = p_Cos2_WEST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1574. dPsi += (1. / 2.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff2.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff2.at(iK) * Psi_ReCenter));
  1575. }
  1576. Psi2WestSP[iGap] = AngleShift(Psi_ReCenter + dPsi, 2.);
  1577. fQv2WestSP_Shift[iGap].Set(fQv2WestSP[iGap].Mod()*TMath::Cos(2.*Psi2WestSP[iGap]),fQv2WestSP[iGap].Mod()*TMath::Sin(2.*Psi2WestSP[iGap]));
  1578. //EAST 3-nd order
  1579. Psi_ReCenter = TMath::ATan2(fQv3EastEP[iGap].Y(), fQv3EastEP[iGap].X()) / 3.;
  1580. dPsi = 0.;
  1581. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1582. {
  1583. mCos = p_Cos3_EAST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1584. mSin = p_Cos3_EAST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1585. dPsi += (1. / 3.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff3.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff3.at(iK) * Psi_ReCenter));
  1586. }
  1587. Psi3EastEP[iGap] = AngleShift(Psi_ReCenter + dPsi, 3.);
  1588. Psi_ReCenter = TMath::ATan2(fQv3EastSP[iGap].Y(), fQv3EastSP[iGap].X()) / 3.;
  1589. dPsi = 0.;
  1590. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1591. {
  1592. mCos = p_Cos3_EAST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1593. mSin = p_Cos3_EAST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1594. dPsi += (1. / 3.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff3.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff3.at(iK) * Psi_ReCenter));
  1595. }
  1596. Psi3EastSP[iGap] = AngleShift(Psi_ReCenter + dPsi, 3.);
  1597. fQv3EastSP_Shift[iGap].Set(fQv3EastSP[iGap].Mod()*TMath::Cos(3.*Psi3EastSP[iGap]),fQv3EastSP[iGap].Mod()*TMath::Sin(3.*Psi3EastSP[iGap]));
  1598. //WEST 3-nd order
  1599. Psi_ReCenter = TMath::ATan2(fQv3WestEP[iGap].Y(), fQv3WestEP[iGap].X()) / 3.;
  1600. dPsi = 0.;
  1601. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1602. {
  1603. mCos = p_Cos3_WEST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1604. mSin = p_Cos3_WEST_EP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1605. dPsi += (1. / 3.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff3.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff3.at(iK) * Psi_ReCenter));
  1606. }
  1607. Psi3WestEP[iGap] = AngleShift(Psi_ReCenter + dPsi, 3.);
  1608. Psi_ReCenter = TMath::ATan2(fQv3WestSP[iGap].Y(), fQv3WestSP[iGap].X()) / 3.;
  1609. dPsi = 0.;
  1610. for (int iK = 0; iK < NShiftOrderMax; iK++)
  1611. {
  1612. mCos = p_Cos3_WEST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1613. mSin = p_Cos3_WEST_SP[VtxSign][iK][(Int_t)event->cent16()][iGap][(Double_t)event->runId()];
  1614. dPsi += (1. / 3.) * (2. / (iK + 1)) * (-1. * mSin * TMath::Cos(shiftOrderCoeff3.at(iK) * Psi_ReCenter) + mCos * TMath::Sin(shiftOrderCoeff3.at(iK) * Psi_ReCenter));
  1615. }
  1616. Psi3WestSP[iGap] = AngleShift(Psi_ReCenter + dPsi, 3.);
  1617. fQv3WestSP_Shift[iGap].Set(fQv3WestSP[iGap].Mod()*TMath::Cos(3.*Psi3WestSP[iGap]),fQv3WestSP[iGap].Mod()*TMath::Sin(3.*Psi3WestSP[iGap]));
  1618. }
  1619. }
  1620. //RESOLUTION CALCULATIONS
  1621. if (fQcountFullWest + fQcountFullEast > cutNcountQvFull)
  1622. {
  1623. res2EP_3subAB = TMath::Cos(2 * (Psi2FullEP - psiEP_BBC_East[1]));
  1624. res2EP_3subAC = TMath::Cos(2 * (Psi2FullEP - psiEP_BBC_West[1]));
  1625. res2EP_3subBC = resEastWestBBC[1];
  1626. res2EP_3sub1AB = TMath::Cos(2 * (Psi2EastEP[0] - psiEP_BBC_Full[1]));
  1627. res2EP_3sub1AC = TMath::Cos(2 * (Psi2WestEP[0] - psiEP_BBC_Full[1]));
  1628. res2EP_3sub1BC = TMath::Cos(2 * (Psi2EastEP[0] - Psi2WestEP[0]));
  1629. res3EP_3subAB = TMath::Cos(3 * (Psi3FullEP - psiEP_BBC_East[2]));
  1630. res3EP_3subAC = TMath::Cos(3 * (Psi3FullEP - psiEP_BBC_West[2]));
  1631. res3EP_3subBC = resEastWestBBC[2];
  1632. res3EP_3sub1AB = TMath::Cos(3 * (Psi3EastEP[0] - psiEP_BBC_Full[2]));
  1633. res3EP_3sub1AC = TMath::Cos(3 * (Psi3WestEP[0] - psiEP_BBC_Full[2]));
  1634. res3EP_3sub1BC = TMath::Cos(3 * (Psi3EastEP[0] - Psi3WestEP[0]));
  1635. p_res2_EP_3sub_AB->Fill(event->cent16(), res2EP_3subAB);
  1636. p_res2_EP_3sub_AC->Fill(event->cent16(), res2EP_3subAC);
  1637. p_res2_EP_3sub_BC->Fill(event->cent16(), res2EP_3subBC);
  1638. p_res2_EP_3sub1_AB->Fill(event->cent16(), res2EP_3sub1AB);
  1639. p_res2_EP_3sub1_AC->Fill(event->cent16(), res2EP_3sub1AC);
  1640. p_res2_EP_3sub1_BC->Fill(event->cent16(), res2EP_3sub1BC);
  1641. p_res3_EP_3sub_AB->Fill(event->cent16(), res3EP_3subAB);
  1642. p_res3_EP_3sub_AC->Fill(event->cent16(), res3EP_3subAC);
  1643. p_res3_EP_3sub_BC->Fill(event->cent16(), res3EP_3subBC);
  1644. p_res3_EP_3sub1_AB->Fill(event->cent16(), res3EP_3sub1AB);
  1645. p_res3_EP_3sub1_AC->Fill(event->cent16(), res3EP_3sub1AC);
  1646. p_res3_EP_3sub1_BC->Fill(event->cent16(), res3EP_3sub1BC);
  1647. }
  1648. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  1649. {
  1650. if (fQcountEast[iGap] > cutNcountQvGap &&
  1651. fQcountWest[iGap] > cutNcountQvGap)
  1652. {
  1653. res2EP[iGap] = TMath::Cos(2 * (Psi2WestEP[iGap] - Psi2EastEP[iGap]));
  1654. res3EP[iGap] = TMath::Cos(3 * (Psi3WestEP[iGap] - Psi3EastEP[iGap]));
  1655. res2SP[iGap] = (Double_t) (fQv2WestSP_Shift[iGap] * fQv2EastSP_Shift[iGap]);
  1656. res3SP[iGap] = (Double_t) (fQv3WestSP_Shift[iGap] * fQv3EastSP_Shift[iGap]);
  1657. p_res2_EP[iGap]->Fill(event->cent16(), res2EP[iGap]);
  1658. p_res3_EP[iGap]->Fill(event->cent16(), res3EP[iGap]);
  1659. p_res2_SP[iGap]->Fill(event->cent16(), res2SP[iGap]);
  1660. p_res3_SP[iGap]->Fill(event->cent16(), res3SP[iGap]);
  1661. p_res2_SP_Norm[iGap]->Fill(event->cent16(), fQv2WestSP_Shift[iGap].Mod() * fQv2EastSP_Shift[iGap].Mod());
  1662. p_res3_SP_Norm[iGap]->Fill(event->cent16(), fQv3WestSP_Shift[iGap].Mod() * fQv3EastSP_Shift[iGap].Mod());
  1663. if (iGap == 0)
  1664. {
  1665. p_res2_vs_run->Fill(event->runId(),event->cent16(),res2EP[iGap]);
  1666. p_res3_vs_run->Fill(event->runId(),event->cent16(),res3EP[iGap]);
  1667. }
  1668. if (iGap == 0)
  1669. {
  1670. resTPCEastBBC[1] = TMath::Cos( (1.+1.) * (psiEP_BBC_East[1]-Psi2EastEP[iGap]) );
  1671. resTPCWestBBC[1] = TMath::Cos( (1.+1.) * (psiEP_BBC_West[1]-Psi2WestEP[iGap]) );
  1672. resTPCEastBBCWest[1] = TMath::Cos( (1.+1.) * (psiEP_BBC_West[1]-Psi2EastEP[iGap]) );
  1673. resTPCWestBBCEast[1] = TMath::Cos( (1.+1.) * (psiEP_BBC_East[1]-Psi2WestEP[iGap]) );
  1674. p_resTPCBBCEast->Fill(event->cent16(),resTPCEastBBC[1]);
  1675. p_resTPCBBCWest->Fill(event->cent16(),resTPCWestBBC[1]);
  1676. p_resTPCWestBBCEast->Fill(event->cent16(),resTPCWestBBCEast[1]);
  1677. p_resTPCEastBBCWest->Fill(event->cent16(),resTPCEastBBCWest[1]);
  1678. }
  1679. }
  1680. }
  1681. for (int iHarm=0;iHarm<fNharmonics+1;iHarm++)
  1682. {
  1683. p_resBBC_EP[iHarm]->Fill(event->cent16(), resEastWestBBC[iHarm]);
  1684. p_resBBC_EP_firstHarm[iHarm]->Fill(event->cent16(), resEastWestBBCFirstHarm[iHarm]);
  1685. p_resZDC_EP[iHarm]->Fill(event->cent16(), resEastWestZDC[iHarm]);
  1686. }
  1687. } //for(Long64_t iEvent=0; iEvent<events2read; iEvent++)
  1688. femtoReader->Finish();
  1689. TFile *output = new TFile(outFile, "recreate");
  1690. for (int iGap = 0; iGap < NEtaGaps; iGap++)
  1691. {
  1692. p_res2_EP[iGap]->Write();
  1693. p_res3_EP[iGap]->Write();
  1694. p_res2_SP[iGap]->Write();
  1695. p_res3_SP[iGap]->Write();
  1696. p_res2_SP_Norm[iGap]->Write();
  1697. p_res3_SP_Norm[iGap]->Write();
  1698. }
  1699. for (int iHarm=0;iHarm<fNharmonics+1;iHarm++)
  1700. {
  1701. p_resBBC_EP[iHarm]->Write();
  1702. p_resBBC_EP_firstHarm[iHarm]->Write();
  1703. }
  1704. for (int iHarm=0;iHarm<fNharmonics+1;iHarm++)
  1705. {
  1706. p_resZDC_EP[iHarm]->Write();
  1707. }
  1708. p_resTPCBBCEast->Write();
  1709. p_resTPCBBCWest->Write();
  1710. p_resTPCWestBBCEast->Write();
  1711. p_resTPCEastBBCWest->Write();
  1712. p_res2_EP_3sub_AB->Write();
  1713. p_res2_EP_3sub_AC->Write();
  1714. p_res2_EP_3sub_BC->Write();
  1715. p_res3_EP_3sub_AB->Write();
  1716. p_res3_EP_3sub_AC->Write();
  1717. p_res3_EP_3sub_BC->Write();
  1718. p_res2_EP_3sub1_AB->Write();
  1719. p_res2_EP_3sub1_AC->Write();
  1720. p_res2_EP_3sub1_BC->Write();
  1721. p_res3_EP_3sub1_AB->Write();
  1722. p_res3_EP_3sub1_AC->Write();
  1723. p_res3_EP_3sub1_BC->Write();
  1724. p_res2_vs_run->Write();
  1725. p_res3_vs_run->Write();
  1726. output->Close();
  1727. std::cout << "I'm done with analysis. We'll have a Nobel Prize, Master!"
  1728. << std::endl;
  1729. std::cout << "Good events: " << goodEventCounter << std::endl;
  1730. timer.Stop();
  1731. timer.Print();
  1732. }
  1733. Bool_t isGoodEvent(StFemtoEvent *const &event)
  1734. {
  1735. if (!event)
  1736. return false;
  1737. if (event == nullptr)
  1738. return false;
  1739. if (event->primaryVertex().Perp() > cutVtxR)
  1740. return false;
  1741. if (TMath::Abs(event->primaryVertex().Z()) > cutVtxZEnergy.at(energy))
  1742. return false;
  1743. if ((energy == 200.) && TMath::Abs(event->primaryVertex().Z() - event->vpdVz()) > cutVpdVz)
  1744. return false;
  1745. return true;
  1746. }
  1747. Bool_t isGoodTrack(StFemtoTrack *const &track, Float_t _energy, TVector3 pVtx)
  1748. {
  1749. if (!track)
  1750. return false;
  1751. // if (!track->isPrimary()) return false;
  1752. if (track->nHitsFit() < cutNhits)
  1753. return false;
  1754. if (track->nHitsPoss() <= cutNhitsPoss)
  1755. return false;
  1756. if ((Double_t)track->nHitsFit() / track->nHitsPoss() < cutNhitsRatio)
  1757. return false;
  1758. if (TMath::Abs(track->eta()) >= cutEta)
  1759. return false;
  1760. if (track->pt() <= cutPtMin.at(_energy))
  1761. return false;
  1762. if (track->pt() > cutPtMax)
  1763. return false;
  1764. if (track->ptot() > cutPMax)
  1765. return false;
  1766. if (track->gDCA(pVtx).Mag() >= cutDCA.at(_energy))
  1767. return false;
  1768. return true;
  1769. }
  1770. Bool_t isGoodTrackFlow(StFemtoTrack *const &track, Float_t _energy, TVector3 pVtx)
  1771. {
  1772. if (!track)
  1773. return false;
  1774. // if (!track->isPrimary()) return false;
  1775. if (track->nHitsFit() < cutNhits)
  1776. return false;
  1777. if (track->nHitsPoss() <= cutNhitsPoss)
  1778. return false;
  1779. if ((Double_t)track->nHitsFit() / track->nHitsPoss() < cutNhitsRatio)
  1780. return false;
  1781. if (TMath::Abs(track->eta()) >= cutEta)
  1782. return false;
  1783. if (track->pt() <= cutPtMin.at(_energy))
  1784. return false;
  1785. //if (track->pt() > cutPtMax) return false;
  1786. if (track->ptot() > cutPMax)
  1787. return false;
  1788. if (track->gDCA(pVtx).Mag() >= cutDCA.at(_energy))
  1789. return false;
  1790. return true;
  1791. }
  1792. Double_t GetWeight(StFemtoTrack *const &track)
  1793. {
  1794. Double_t weight;
  1795. if (track->pt() <= cutPtWeightEP)
  1796. {
  1797. weight = track->pt();
  1798. }
  1799. else
  1800. {
  1801. weight = cutPtWeightEP;
  1802. }
  1803. return weight;
  1804. }
  1805. TVector2 CalcQvector(StFemtoTrack *const &track, Int_t _harm)
  1806. {
  1807. TVector2 qv(0., 0.);
  1808. qv.Set(TMath::Cos(_harm * track->phi()), TMath::Sin(_harm * track->phi()));
  1809. return qv;
  1810. }
  1811. Double_t AngleShift(Double_t Psi, Double_t order)
  1812. {
  1813. Double_t PsiCorr = Psi;
  1814. if (PsiCorr > TMath::Pi() / order)
  1815. {
  1816. PsiCorr = Psi - 2. * TMath::Pi() / order;
  1817. }
  1818. if (PsiCorr < -1. * TMath::Pi() / order)
  1819. {
  1820. PsiCorr = Psi + 2. * TMath::Pi() / order;
  1821. }
  1822. return PsiCorr;
  1823. }
  1824. Int_t GetVzBin(Double_t vtxZ)
  1825. {
  1826. for (Int_t i = 0; i < fArraySize; i++)
  1827. {
  1828. if (vtxZ >= VtxBins[i] && vtxZ < VtxBins[i + 1])
  1829. return i;
  1830. }
  1831. return -1;
  1832. }
  1833. //--------------------------------------------------------------------------------------------------------------------------//
  1834. //this function simply connects the gain values read in to the BBC azimuthal distribution
  1835. //since tiles 7 and 9 (+ 13 and 15) share a gain value it is ambiguous how to assign the geometry here
  1836. //I prefer assigning the angle between the tiles thus "greying out" the adcs.
  1837. //Others have assigned all of the adc to one (exclusive) or the the other.
  1838. Float_t BBC_GetPhi(const Int_t eastWest, const Int_t tileId)
  1839. {
  1840. //float GetPhiInBBC(int eastWest, int bbcN) { //tileId=0 to 23
  1841. const float Pi = TMath::Pi();
  1842. const float Phi_div = Pi / 6;
  1843. float bbc_phi = Phi_div;
  1844. switch(tileId) {
  1845. case 0: bbc_phi = 3.*Phi_div;
  1846. break;
  1847. case 1: bbc_phi = Phi_div;
  1848. break;
  1849. case 2: bbc_phi = -1.*Phi_div;
  1850. break;
  1851. case 3: bbc_phi = -3.*Phi_div;
  1852. break;
  1853. case 4: bbc_phi = -5.*Phi_div;
  1854. break;
  1855. case 5: bbc_phi = 5.*Phi_div;
  1856. break;
  1857. //case 6: bbc_phi= (mRndm.Rndm() > 0.5) ? 2.*Phi_div:4.*Phi_div; //tiles 7 and 9 are gained together we randomly assign the gain to one XOR the other
  1858. case 6: bbc_phi = 3.*Phi_div;
  1859. break;
  1860. case 7: bbc_phi = 3.*Phi_div;
  1861. break;
  1862. case 8: bbc_phi = Phi_div;
  1863. break;
  1864. case 9: bbc_phi = 0.;
  1865. break;
  1866. case 10: bbc_phi = -1.*Phi_div;
  1867. break;
  1868. //case 11: bbc_phi = (mRndm.Rndm() > 0.5) ? -2.*Phi_div:-4.*Phi_div; //tiles 13 and 15 are gained together
  1869. case 11: bbc_phi = -3.*Phi_div;
  1870. break;
  1871. case 12: bbc_phi = -3.*Phi_div;
  1872. break;
  1873. case 13: bbc_phi = -5.*Phi_div;
  1874. break;
  1875. case 14: bbc_phi = Pi;
  1876. break;
  1877. case 15: bbc_phi = 5.*Phi_div;
  1878. break;
  1879. }
  1880. //if we're looking at the east BBC we need to flip around x in the STAR coordinates,
  1881. //a line parallel to the beam would go through tile 1 on the W BBC and tile 3 on the
  1882. if(0 == eastWest){
  1883. if (bbc_phi > -0.001){ //this is not a >= since we are talking about finite adcs -- not to important
  1884. bbc_phi = Pi - bbc_phi;
  1885. }
  1886. else {
  1887. bbc_phi= -Pi - bbc_phi;
  1888. }
  1889. }
  1890. if(bbc_phi < 0.0) bbc_phi += 2.*Pi;
  1891. if(bbc_phi > 2.*Pi) bbc_phi -= 2.*Pi;
  1892. return bbc_phi;
  1893. }
  1894. Double_t GetZDCPosition(Int_t eastwest, Int_t verthori, Int_t strip)
  1895. // Get position of each slat;strip starts from 0
  1896. {
  1897. Double_t zdcsmd_x[7] = {0.5,2,3.5,5,6.5,8,9.5};
  1898. Double_t zdcsmd_y[8] = {1.25,3.25,5.25,7.25,9.25,11.25,13.25,15.25};
  1899. Double_t mZDCSMDCenterex = 4.72466;
  1900. Double_t mZDCSMDCenterey = 5.53629;
  1901. Double_t mZDCSMDCenterwx = 4.39604;
  1902. Double_t mZDCSMDCenterwy = 5.19968;
  1903. if(eastwest==0 && verthori==0) return zdcsmd_x[strip]-mZDCSMDCenterex;
  1904. if(eastwest==1 && verthori==0) return mZDCSMDCenterwx-zdcsmd_x[strip];
  1905. if(eastwest==0 && verthori==1) return zdcsmd_y[strip]/sqrt(2.)-mZDCSMDCenterey;
  1906. if(eastwest==1 && verthori==1) return zdcsmd_y[strip]/sqrt(2.)-mZDCSMDCenterwy;
  1907. return -999.;
  1908. }