#include <TMath.h>
#include <TError.h>
#include <TROOT.h>
-#define FIT_OPTIONS "RNQS"
+#define FIT_OPTIONS "RNS"
//====================================================================
ULong_t AliForwardUtil::AliROOTRevision()
{
#ifdef ALIROOT_SVN_REVISION
return ALIROOT_SVN_REVISION;
-#else
+#elif defined(ALIROOT_REVISION)
+ static ULong_t ret = 0;
+ if (ret != 0) return ret;
+
+ // Select first 32bits of the 40byte long check-sum
+ TString rev(ALIROOT_REVISION, 8);
+ for (ULong_t i = 0; i < 8; i++) {
+ ULong_t p = 0;
+ switch (rev[i]) {
+ case '0': p = 0; break;
+ case '1': p = 1; break;
+ case '2': p = 2; break;
+ case '3': p = 3; break;
+ case '4': p = 4; break;
+ case '5': p = 5; break;
+ case '6': p = 6; break;
+ case '7': p = 7; break;
+ case '8': p = 8; break;
+ case '9': p = 9; break;
+ case 'a': case 'A': p = 10; break;
+ case 'b': case 'B': p = 11; break;
+ case 'c': case 'C': p = 12; break;
+ case 'd': case 'D': p = 13; break;
+ case 'e': case 'E': p = 14; break;
+ case 'f': case 'F': p = 15; break;
+ }
+ ret |= (p << (32-4*(i+1)));
+ }
+ return ret;
+#else
return 0;
#endif
}
//____________________________________________________________________
ULong_t AliForwardUtil::AliROOTBranch()
{
-#ifdef ALIROOT_SVN_BRANCH
+ // Do something here when we switch to git - sigh!
+#if !defined(ALIROOT_SVN_BRANCH) && !defined(ALIROOT_BRANCH)
+ return 0;
+#endif
static ULong_t ret = 0;
if (ret != 0) return ret;
-
- TString str(ALIROOT_SVN_BRANCH);
+ TString str;
+ TString top;
+#ifdef ALIROOT_SVN_BRANCH
+ str = ALIROOT_SVN_BRANCH;
+ top = "trunk";
+#elif defined(ALIROOT_BRANCH)
+ str = ALIROOT_BRANCH;
+ top = "master";
+#endif
+ if (str.IsNull()) return 0xFFFFFFFF;
if (str[0] == 'v') str.Remove(0,1);
- if (str.EqualTo("trunk")) return ret = 0xFFFFFFFF;
+ if (str.EqualTo(top)) return ret = 0xFFFFFFFF;
TObjArray* tokens = str.Tokenize("-");
- TObjString* pMajor = static_cast<TObjString*>(tokens->At(0));
- TObjString* pMinor = static_cast<TObjString*>(tokens->At(1));
- TObjString* pRelea = static_cast<TObjString*>(tokens->At(2));
- TObjString* pAn = (tokens->GetEntries() > 3 ?
- static_cast<TObjString*>(tokens->At(3)) : 0);
- TString sMajor = pMajor->String().Strip(TString::kLeading, '0');
- TString sMinor = pMinor->String().Strip(TString::kLeading, '0');
- TString sRelea = pRelea->String().Strip(TString::kLeading, '0');
-
+ TObjString* pMajor = tokens->GetEntries()>0 ?
+ (static_cast<TObjString*>(tokens->At(0))) : 0;
+ TObjString* pMinor = tokens->GetEntries()>1 ?
+ (static_cast<TObjString*>(tokens->At(1))) : 0;
+ TObjString* pRelea = tokens->GetEntries() > 2 ?
+ static_cast<TObjString*>(tokens->At(2)) : 0;
+ TObjString* pAn = tokens->GetEntries() > 3 ?
+ static_cast<TObjString*>(tokens->At(3)) : 0;
+ TString sMajor,sMinor,sRelea;
+ if (pMajor) sMajor = pMajor->String().Strip(TString::kLeading, '0');
+ if (pMinor) sMinor = pMinor->String().Strip(TString::kLeading, '0');
+ if (pRelea) sRelea = pRelea->String().Strip(TString::kLeading, '0');
+ //
ret = (((sMajor.Atoi() & 0xFF) << 12) |
((sMinor.Atoi() & 0xFF) << 8) |
((sRelea.Atoi() & 0xFF) << 4) |
(pAn ? 0xAA : 0));
return ret;
-#else
- return 0;
-#endif
}
//====================================================================
{
AliAnalysisManager* am = AliAnalysisManager::GetAnalysisManager();
if (dynamic_cast<AliAODInputHandler*>(am->GetInputEventHandler())) {
- /// ::Info("CheckForAOD", "Found AOD Input handler");
+ // ::Info("CheckForAOD", "Found AOD Input handler");
return 1;
}
if (dynamic_cast<AliAODHandler*>(am->GetOutputEventHandler())) {
//====================================================================
TAxis*
AliForwardUtil::MakeFullIpZAxis(Int_t nCenter)
+{
+ TArrayD bins;
+ MakeFullIpZAxis(nCenter, bins);
+ TAxis* a = new TAxis(bins.GetSize()-1,bins.GetArray());
+ return a;
+}
+//____________________________________________________________________
+void
+AliForwardUtil::MakeFullIpZAxis(Int_t nCenter, TArrayD& bins)
{
// Custom vertex axis that will include satellite vertices
// Satellite vertices are at k*37.5 where k=-10,-9,...,9,10
const Int_t nBins = 2*nSat + nCenter;
const Int_t mBin = nBins / 2;
Double_t dCenter = 2*mCenter / nCenter;
- TArrayD bins(nBins+1);
+ bins.Set(nBins+1);
bins[mBin] = 0;
for (Int_t i = 1; i <= nCenter/2; i++) {
// Assign from the middle out
bins[mBin-o] = -v;
bins[mBin+o] = +v;
}
- TAxis* a = new TAxis(nBins,bins.GetArray());
- return a;
}
+//____________________________________________________________________
+void
+AliForwardUtil::MakeLogScale(Int_t nBins,
+ Int_t minOrder,
+ Int_t maxOrder,
+ TArrayD& bins)
+{
+ Double_t dO = Double_t(maxOrder-minOrder) / nBins;
+ bins.Set(nBins+1);
+ for (Int_t i = 0; i <= nBins; i++) bins[i] = TMath::Power(10, i * dO+minOrder);
+}
+
+//____________________________________________________________________
void
AliForwardUtil::PrintTask(const TObject& o)
{
std::cout << "--- " << t << " " << std::setfill('-')
<< std::setw(maxN-ind-5-t.Length()) << "-" << std::endl;
}
+//____________________________________________________________________
void
AliForwardUtil::PrintName(const char* name)
{
std::cout << std::setfill(' ') << std::left << std::setw(width)
<< n << std::right << std::flush;
}
+//____________________________________________________________________
void
AliForwardUtil::PrintField(const char* name, const char* value, ...)
{
}
//====================================================================
+#if 0 // Moved to separate classes
Int_t AliForwardUtil::fgConvolutionSteps = 100;
Double_t AliForwardUtil::fgConvolutionNSigma = 5;
namespace {
Double_t xiP = pp[AliForwardUtil::ELossFitter::kXi];
Double_t sigmaP = pp[AliForwardUtil::ELossFitter::kSigma];
Double_t cS = pp[AliForwardUtil::ELossFitter::kSigma+1];
- Double_t deltaS = pp[AliForwardUtil::ELossFitter::kSigma+2];
- Double_t xiS = pp[AliForwardUtil::ELossFitter::kSigma+3];
- Double_t sigmaS = pp[AliForwardUtil::ELossFitter::kSigma+4];
+ Double_t deltaS = deltaP; // pp[AliForwardUtil::ELossFitter::kSigma+2];
+ Double_t xiS = pp[AliForwardUtil::ELossFitter::kSigma+2/*3*/];
+ Double_t sigmaS = sigmaP; // pp[AliForwardUtil::ELossFitter::kSigma+4];
return (cP * AliForwardUtil::LandauGaus(x,deltaP,xiP,sigmaP,0) +
cS * AliForwardUtil::LandauGaus(x,deltaS,xiS,sigmaS,0));
// Return:
// @f$ f'_{L}(x;\Delta,\xi) @f$
//
- return TMath::Landau(x, delta - xi * mpshift, xi);
+ Double_t deltaP = delta - xi * mpshift;
+ return TMath::Landau(x, deltaP, xi, true);
}
//____________________________________________________________________
Double_t
// Return:
// @f$ f@f$ evaluated at @f$ x@f$.
//
- Double_t deltap = delta - xi * mpshift;
+ if (xi <= 0) return 0;
+
+ Double_t deltaP = delta; // - sigma * sigmaShift; // + sigma * mpshift;
Double_t sigma2 = sigmaN*sigmaN + sigma*sigma;
Double_t sigma1 = sigmaN == 0 ? sigma : TMath::Sqrt(sigma2);
Double_t xlow = x - fgConvolutionNSigma * sigma1;
Double_t x1 = xlow + (i - .5) * step;
Double_t x2 = xhigh - (i - .5) * step;
- sum += TMath::Landau(x1, deltap, xi, kTRUE) * TMath::Gaus(x, x1, sigma1);
- sum += TMath::Landau(x2, deltap, xi, kTRUE) * TMath::Gaus(x, x2, sigma1);
+ //sum += TMath::Landau(x1, deltap, xi, kTRUE) * TMath::Gaus(x, x1, sigma1);
+ //sum += TMath::Landau(x2, deltap, xi, kTRUE) * TMath::Gaus(x, x2, sigma1);
+ sum += Landau(x1, deltaP, xi) * TMath::Gaus(x, x1, sigma1);
+ sum += Landau(x2, deltaP, xi) * TMath::Gaus(x, x2, sigma1);
}
return step * sum * invSq2pi / sigma1;
}
+namespace {
+ const Double_t sigmaShift = 0.36390; // TMath::Log(TMath::Sqrt(2.));
+ double deltaSigmaShift(Int_t i, Double_t sigma)
+ {
+ return 0; // - sigma * sigmaShift;
+ }
+ void getIPars(Int_t i, Double_t& delta, Double_t& xi, Double_t& sigma)
+ {
+ Double_t dsig = deltaSigmaShift(i, sigma);
+ if (i == 1) {
+ delta += dsig;
+ return; // { delta = delta + xi*mpshift; return; } // Do nothing
+ }
+
+ delta = i * (delta + xi * TMath::Log(i)) + dsig;
+ xi = i * xi;
+ sigma = TMath::Sqrt(Double_t(i)) * sigma;
+ }
+}
+
+
//____________________________________________________________________
Double_t
AliForwardUtil::ILandauGaus(Double_t x, Double_t delta, Double_t xi,
// Return:
// @f$ f_i @f$ evaluated
//
- Double_t deltaI = (i == 1 ? delta : i * (delta + xi * TMath::Log(i)));
- Double_t xiI = i * xi;
- Double_t sigmaI = (i == 1 ? sigma : TMath::Sqrt(Double_t(i))*sigma);
+ Double_t deltaI = delta;
+ Double_t xiI = xi;
+ Double_t sigmaI = sigma;
+ getIPars(i, deltaI, xiI, sigmaI);
if (sigmaI < 1e-10) {
// Fall back to landau
return AliForwardUtil::Landau(x, deltaI, xiI);
fFitResults.Clear();
fFunctions.Clear();
}
+namespace {
+ void setParLimit(TF1* f, Int_t iPar, Bool_t debug,
+ Double_t test, Double_t low, Double_t high)
+ {
+ if (test >= low && test <= high) {
+ if (debug)
+ printf("Fit: Set par limits on %s: %f, %f\n",
+ f->GetParName(iPar), low, high);
+ f->SetParLimits(iPar, low, high);
+ }
+ }
+}
+
//____________________________________________________________________
TF1*
AliForwardUtil::ELossFitter::Fit1Particle(TH1* dist, Double_t sigman)
// Get the bin with maximum
Int_t peakBin = dist->GetMaximumBin();
Double_t peakE = dist->GetBinLowEdge(peakBin);
+ Double_t rmsE = dist->GetRMS();
// Get the low edge
// dist->GetXaxis()->SetRangeUser(fLowCut, peakE);
TF1* landau1 = new TF1("landau1", landauGaus1, minE,maxE,kSigmaN+1);
// Set initial guesses, parameter names, and limits
- landau1->SetParameters(1,peakE,peakE/10,peakE/5,sigman);
+ landau1->SetParameters(intg,peakE,peakE/10,peakE/5,sigman);
landau1->SetParNames("C","#Delta_{p}","#xi", "#sigma", "#sigma_{n}");
landau1->SetNpx(500);
- if (peakE >= minE && peakE <= fMaxRange) {
- // printf("Fit1: Set par limits on Delta: %f, %f\n", minE, fMaxRange);
- landau1->SetParLimits(kDelta, minE, fMaxRange);
- }
- if (peakE/10 >= 0 && peakE <= 0.1) {
- // printf("Fit1: Set par limits on xi: %f, %f\n", 0., 0.1);
- landau1->SetParLimits(kXi, 0.00, 0.1); // Was fMaxRange - too wide
- }
- if (peakE/5 >= 0 && peakE/5 <= 0.1) {
- // printf("Fit1: Set par limits on sigma: %f, %f\n", 0., 0.1);
- landau1->SetParLimits(kSigma, 1e-5, 0.1); // Was fMaxRange - too wide
- }
+ setParLimit(landau1, kDelta, fDebug, peakE, minE, fMaxRange);
+ setParLimit(landau1, kXi, fDebug, peakE, 0, rmsE); // 0.1
+ setParLimit(landau1, kSigma, fDebug, peakE/5, 1e-5, rmsE); // 0.1
if (sigman <= 0) landau1->FixParameter(kSigmaN, 0);
- else {
- // printf("Fit1: Set par limits on sigmaN: %f, %f\n", 0., fMaxRange);
- landau1->SetParLimits(kSigmaN, 0, fMaxRange);
- }
+ else
+ setParLimit(landau1, kSigmaN, fDebug, peakE, 0, rmsE);
+
+ TString opts(Form("%s%s", FIT_OPTIONS, fDebug ? "" : "Q"));
// Do the fit, getting the result object
if (fDebug)
::Info("Fit1Particle", "Fitting in the range %f,%f", minE, maxE);
- TFitResultPtr r = dist->Fit(landau1, FIT_OPTIONS, "", minE, maxE);
+ TFitResultPtr r = dist->Fit(landau1, opts, "", minE, maxE);
+ if (!r.Get()) {
+ ::Warning("Fit1Particle",
+ "No fit returned when processing %s in the range [%f,%f] "
+ "options %s", dist->GetName(), minE, maxE, FIT_OPTIONS);
+ return 0;
+ }
// landau1->SetRange(minE, fMaxRange);
fFitResults.AddAtAndExpand(new TFitResult(*r), 0);
fFunctions.AddAtAndExpand(landau1, 0);
Int_t minEb = dist->GetXaxis()->FindBin(minE);
Int_t maxEb = dist->GetXaxis()->FindBin(maxEi);
+ Double_t rmsE = dist->GetRMS();
Double_t intg = dist->Integral(minEb, maxEb);
if (intg <= 0) {
::Warning("FitNParticle",
r->Parameter(kSigma),
r->Parameter(kSigmaN),
n, a.fArray, minE, maxEi);
- if (minE <= r->Parameter(kDelta) &&
- fMaxRange >= r->Parameter(kDelta)) {
- // Protect against warning from ParameterSettings
- // printf("FitN: Set par limits on Delta: %f, %f\n", minE, fMaxRange);
- landaun->SetParLimits(kDelta, minE, fMaxRange); // Delta
- }
- if (r->Parameter(kXi) >= 0 && r->Parameter(kXi) <= 0.1) {
- // printf("FitN: Set par limits on xi: %f, %f\n", 0., 0.1);
- landaun->SetParLimits(kXi, 0.00, 0.1); // was fMaxRange - too wide
- }
- if (r->Parameter(kSigma) >= 1e-5 && r->Parameter(kSigma) <= 0.1) {
- // printf("FitN: Set par limits on sigma: %f, %f\n", 1e-5, 0.1);
- landaun->SetParLimits(kSigma, 1e-5, 0.1); // was fMaxRange - too wide
- }
- // Check if we're using the noise sigma
+ setParLimit(landaun, kDelta, fDebug, r->Parameter(kDelta), minE, fMaxRange);
+ setParLimit(landaun, kXi, fDebug, r->Parameter(kXi), 0, rmsE); // 0.1
+ setParLimit(landaun, kSigma, fDebug, r->Parameter(kSigma), 1e-5, rmsE); // 0.1
if (sigman <= 0) landaun->FixParameter(kSigmaN, 0);
- else {
- // printf("FitN: Set par limits on sigmaN: %f, %f\n", 0., fMaxRange);
- landaun->SetParLimits(kSigmaN, 0, fMaxRange);
- }
+ else
+ setParLimit(landaun, kSigmaN, fDebug, r->Parameter(kSigmaN), 0, rmsE);
// Set the range and name of the scale parameters
for (UShort_t i = 2; i <= n; i++) {// Take parameters from last fit
- if (a[i-2] >= 0 && a[i-2] <= 1) {
- // printf("FitN: Set par limits on a_%d: %f, %f\n", i, 0., 1.);
- landaun->SetParLimits(kA+i-2, 0,1);
- }
+ setParLimit(landaun, kA+i-2, fDebug, a[i-2], 0, 1);
}
// Do the fit
+ TString opts(Form("%s%s", FIT_OPTIONS, fDebug ? "" : "Q"));
if (fDebug)
::Info("FitNParticle", "Fitting in the range %f,%f (%d)", minE, maxEi, n);
- TFitResultPtr tr = dist->Fit(landaun, FIT_OPTIONS, "", minE, maxEi);
+ TFitResultPtr tr = dist->Fit(landaun, opts, "", minE, maxEi);
// landaun->SetRange(minE, fMaxRange);
fFitResults.AddAtAndExpand(new TFitResult(*tr), n-1);
{
//
// Fit a composite particle signal to the passed energy loss
- // distribution
- //
+ // distribution //
// Parameters:
// dist Data to fit the function to
// sigman If larger than zero, the initial guess of the
/* TFitResultPtr r = */ dist->Fit(seed, FIT_OPTIONS, "", minE, maxE);
maxE = dist->GetXaxis()->GetXmax();
+#if 1
+ TF1* comp = new TF1("composite", landauGausComposite,
+ minE, maxE, kSigma+1+2);
+ comp->SetParNames("C", "#Delta_{p}", "#xi", "#sigma",
+ "C#prime", "#xi#prime");
+ comp->SetParameters(0.8 * seed->GetParameter(kC), // 0 Primary weight
+ seed->GetParameter(kDelta), // 1 Primary Delta
+ seed->GetParameter(kDelta)/10, // 2 primary Xi
+ seed->GetParameter(kDelta)/5, // 3 primary sigma
+ 1.20 * seed->GetParameter(kC), // 5 Secondary weight
+ seed->GetParameter(kXi)); // 7 secondary Xi
+ // comp->SetParLimits(kC, minE, fMaxRange); // C
+ comp->SetParLimits(kDelta, minE, fMaxRange); // Delta
+ comp->SetParLimits(kXi, 0.00, fMaxRange); // Xi
+ comp->SetParLimits(kSigma, 1e-5, fMaxRange); // Sigma
+ // comp->SetParLimits(kSigma+1, minE, fMaxRange); // C
+ comp->SetParLimits(kSigma+2, 0.00, fMaxRange); // Xi'
+#else
TF1* comp = new TF1("composite", landauGausComposite,
minE, maxE, kSigma+1+4);
comp->SetParNames("C", "#Delta_{p}", "#xi", "#sigma",
seed->GetParameter(kDelta), // 6 secondary Delta
seed->GetParameter(kXi), // 7 secondary Xi
seed->GetParameter(kSigma)); // 8 secondary sigma
-
// comp->SetParLimits(kC, minE, fMaxRange); // C
comp->SetParLimits(kDelta, minE, fMaxRange); // Delta
comp->SetParLimits(kXi, 0.00, fMaxRange); // Xi
comp->SetParLimits(kSigma+2, minE/10, fMaxRange); // Delta
comp->SetParLimits(kSigma+3, 0.00, fMaxRange); // Xi
comp->SetParLimits(kSigma+4, 1e-6, fMaxRange); // Sigma
+#endif
comp->SetLineColor(kRed+1);
comp->SetLineWidth(3);
// Do the fit, getting the result object
+ TString opts(Form("%s%s", FIT_OPTIONS, fDebug ? "" : "Q"));
if (fDebug)
::Info("FitComposite", "Fitting composite in the range %f,%f", minE, maxE);
- /* TFitResultPtr r = */ dist->Fit(comp, FIT_OPTIONS, "", minE, maxE);
+ /* TFitResultPtr r = */ dist->Fit(comp, opts, "", minE, maxE);
#if 0
TF1* part1 = static_cast<TF1*>(seed->Clone("part1"));
#endif
return comp;
}
+#endif
//====================================================================
AliForwardUtil::Histos::~Histos()
// Parameters:
// etaAxis Eta axis to use
//
- RebinEta(fFMD1i, etaAxis);
- RebinEta(fFMD2i, etaAxis);
- RebinEta(fFMD2o, etaAxis);
- RebinEta(fFMD3i, etaAxis);
- RebinEta(fFMD3o, etaAxis);
+ if (!fFMD1i) fFMD1i = Make(1, 'i', etaAxis); else RebinEta(fFMD1i, etaAxis);
+ if (!fFMD2i) fFMD2i = Make(2, 'i', etaAxis); else RebinEta(fFMD2i, etaAxis);
+ if (!fFMD2o) fFMD2o = Make(2, 'o', etaAxis); else RebinEta(fFMD2o, etaAxis);
+ if (!fFMD3i) fFMD3i = Make(3, 'i', etaAxis); else RebinEta(fFMD3i, etaAxis);
+ if (!fFMD3o) fFMD3o = Make(3, 'o', etaAxis); else RebinEta(fFMD3o, etaAxis);
}
//____________________________________________________________________
// Parameters:
// option Not used
//
- if (fFMD1i) fFMD1i->Reset(option);
- if (fFMD2i) fFMD2i->Reset(option);
- if (fFMD2o) fFMD2o->Reset(option);
- if (fFMD3i) fFMD3i->Reset(option);
- if (fFMD3o) fFMD3o->Reset(option);
+ if (fFMD1i) { fFMD1i->Reset(option); fFMD1i->ResetBit(kSkipRing); }
+ if (fFMD2i) { fFMD2i->Reset(option); fFMD2i->ResetBit(kSkipRing); }
+ if (fFMD2o) { fFMD2o->Reset(option); fFMD2o->ResetBit(kSkipRing); }
+ if (fFMD3i) { fFMD3i->Reset(option); fFMD3i->ResetBit(kSkipRing); }
+ if (fFMD3o) { fFMD3o->Reset(option); fFMD3o->ResetBit(kSkipRing); }
}
//____________________________________________________________________