2 * Get a collection from a file directory
4 * @param dir Parent directory
5 * @param name Name of collection
7 * @return collection or null
9 TCollection* GetCollection(TDirectory* dir, const TString& name)
12 Error("GetCollection", "No parent directory for %s", name.Data());
16 dir->GetObject(name, ret);
18 Error("GetCollection", "Couldn't find %s in %s",
19 name.Data(), dir->GetName());
25 * Get an object from a collection. Optionally, we check that the
26 * type of the possibly found object matches the request.
28 * @param parent Parent collection
29 * @param name Name of object
30 * @param cls If specified, check that the found object (if any) is of
33 * @return Found object (possibly type-checked) or null
35 TObject* GetObject(const TCollection* parent, const TString& name,
39 Error("GetObject", "No parent collection for %s", name.Data());
42 TObject* ret = parent->FindObject(name);
44 Error("GetObject", "Couldn't find %s in %s",
45 name.Data(), parent->GetName());
48 if (cls && !ret->IsA()->InheritsFrom(cls)) {
49 Error("GetObject", "%s in %s is a %s, not a %s", name.Data(),
50 parent->GetName(), ret->ClassName(), cls->GetName());
56 * Get a collection contained in another collection
58 * @param parent Parent collection
59 * @param name Name of collection to find
61 * @return Found collection or null
63 TCollection* GetCollection(const TCollection* parent, const TString& name)
65 TObject* o = GetObject(parent, name, TCollection::Class());
67 return static_cast<TCollection*>(o);
70 * Re-run the energy loss fitter on a merged output file
72 * @param input File name of merged output file
73 * @param output If specified, the file the new results are written
74 * to. If this is not specified, it defaults to the name of the input
75 * file with "_rerun" attached to the base name
77 * @param forceSet Forcibly set things
78 * @param input Input file
79 * @param output Output file
81 void RerunELossFits(Bool_t forceSet=false,
82 const TString& input="forward_eloss.root",
84 const TString& output="")
86 const char* fwd = "$ALICE_ROOT/PWGLF/FORWARD/analysis2";
87 gROOT->Macro(Form("%s/scripts/LoadLibs.C", fwd));
91 TString outName(output);
92 if (outName.IsNull()) {
94 outName.ReplaceAll(".root", "_rerun.root");
98 // --- Open input file ---------------------------------------------
99 inFile = TFile::Open(input, "READ");
101 throw TString::Format("Failed to open %s", input.Data());
103 // --- InFiled input collections --------------------------------------
104 TCollection* inFwdSum = GetCollection(inFile, "ForwardELossSums");
105 if (!inFwdSum) throw TString("Cannot proceed without sums");
107 TCollection* inFwdRes = GetCollection(inFile, "ForwardELossResults");
108 if (!inFwdRes) throw TString("Cannot proceed with merged list");
110 TCollection* inEFSum = GetCollection(inFwdRes, "fmdEnergyFitter");
111 if (!inEFSum) throw TString("Cannot proceed without accumulated data");
113 TCollection* inEFRes = GetCollection(inFwdRes, "fmdEnergyFitter");
114 if (!inEFRes) throw TString("Cannot proceed without previous results");
116 // --- Open output file --------------------------------------------
117 outFile = TFile::Open(outName, "RECREATE");
119 throw TString::Format("Failed to open %s", outName.Data());
121 // --- Write copy of sum collection to output --------------------
122 TCollection* outFwdSum = static_cast<TCollection*>(inFwdSum->Clone());
124 outFwdSum->Write(inFwdSum->GetName(), TObject::kSingleKey);
126 // --- Make our fitter object ------------------------------------
127 AliFMDEnergyFitter* fitter = new AliFMDEnergyFitter("energy");
128 fitter->SetDoFits(true);
129 fitter->SetEnableDeltaShift(shift);
131 if (forceSet || !fitter->ReadParameters(inEFSum)) {
132 Printf("Forced settings");
134 const TAxis* etaAxis = static_cast<TAxis*>(GetObject(inEFSum,"etaAxis"));
135 if (!etaAxis) throw TString("Cannot proceed without eta axis");
136 fitter->SetEtaAxis(*etaAxis);
138 // Set maximum energy loss to consider
140 // Set number of energy loss bins
141 fitter->SetNEbins(500);
142 // Set whether to use increasing bin sizes
143 // fitter->SetUseIncreasingBins(true);
144 // Set whether to do fit the energy distributions
145 fitter->SetDoFits(kTRUE);
146 // Set whether to make the correction object
147 fitter->SetDoMakeObject(kTRUE);
148 // Set the low cut used for energy
149 fitter->SetLowCut(0.4);
150 // Set the number of bins to subtract from maximum of distributions
151 // to get the lower bound of the fit range
152 fitter->SetFitRangeBinWidth(4);
153 // Set the maximum number of landaus to try to fit (max 5)
154 fitter->SetNParticles(5);
155 // Set the minimum number of entries in the distribution before
156 // trying to fit to the data - 10k seems the least we can do
157 fitter->SetMinEntries(10000);
158 // fitter->SetMaxChi2PerNDF(10);
162 fitter->SetStoreResiduals(AliFMDEnergyFitter::kResidualSquareDifference);
163 fitter->SetRegularizationCut(1e6); // Lower by factor 3
164 // Set the number of bins to subtract from maximum of distributions
165 // to get the lower bound of the fit range
166 // fitter->SetFitRangeBinWidth(2);
167 // Skip all of FMD2 and 3
168 // fitter->SetSkips(AliFMDEnergyFitter::kFMD2|AliFMDEnergyFitter::kFMD3);
170 // --- Now do the fits -------------------------------------------
173 fitter->Fit(static_cast<TList*>(outFwdSum));
175 // --- Copy full result folder -----------------------------------
176 TCollection* outFwdRes = static_cast<TCollection*>(inFwdRes->Clone());
178 TCollection* outEFRes = GetCollection(outFwdRes, "fmdEnergyFitter");
179 outFwdRes->Remove(outEFRes);
180 // Make our new fit results folder, and add it to results folder
181 TCollection* tmp = GetCollection(outFwdSum, "fmdEnergyFitter");
182 outEFRes = static_cast<TCollection*>(tmp->Clone());
183 outEFRes->Add(new TNamed("refitted", "Refit of the data"));
184 outFwdRes->Add(outEFRes);
186 // --- Write out new results folder ------------------------------
188 outFwdRes->Write(inFwdRes->GetName(), TObject::kSingleKey);
189 Printf("Wrote results to \"%s\" (%s)", outName.Data(), outFile->GetName());
192 catch (const TString& e) {
193 Error("RerunELossFits", e);
195 if (inFile) inFile->Close();
197 Printf("Wrote new output to \"%s\"", outName.Data());
202 gROOT->Macro(Form("%s/corrs/DrawCorrELoss.C(false,false,\"%s\")",
203 fwd, outName.Data()));