KVZALineFinder.cpp

//Created by KVClassFactory on Tue Dec  4 09:49:15 2012
//Author: dgruyer
 
#include "KVZALineFinder.h"
#include "KVIdentificationResult.h"
#include "KVSpiderLine.h"
#include "TSystem.h"
#include "KVIDZALine.h"
#include "KVIDGridEditor.h"
 
ClassImp(KVZALineFinder)
 
 
 
 
KVZALineFinder::KVZALineFinder(KVIDZAGrid* gg, TH2* hh)
{
   // constructor
   fGrid  = gg;
   fHisto = hh;
 
   fGeneratedGrid = 0;
   fLinearHisto   = 0;
   //  fCanvas        = 0;
 
   fPoints  = new TGraph;
   fPoints->SetMarkerStyle(21);
   fNPoints = 0;
   fLines = new TList;
   SetAList("1,4,7,9,11,12,14,16,19,21,23,25,27,29,31,34,37,40,41");
   fBinsByZ = 60;
}
 
 
 
 
 
KVZALineFinder::KVZALineFinder(const KVZALineFinder& obj)  : KVBase()
{
   // Copy constructor
   // This ctor is used to make a copy of an existing object (for example
   // when a method returns an object), and it is always a good idea to
   // implement it.
   // If your class allocates memory in its constructor(s) then it is ESSENTIAL :-)
 
   obj.Copy(*this);
}
 
 
 
 
KVZALineFinder::~KVZALineFinder()
{
   // Destructor
}
 
 
 
 
void KVZALineFinder::Copy(TObject& obj) const
{
   // This method copies the current state of 'this' object into 'obj'
   // You should add here any member variables, for example:
   //    (supposing a member variable KVZALineFinder::fToto)
   //    CastedObj.fToto = fToto;
   // or
   //    CastedObj.SetToto( GetToto() );
 
   KVBase::Copy(obj);
   //KVZALineFinder& CastedObj = (KVZALineFinder&)obj;
}
 
 
 
 
void KVZALineFinder::SetAList(const char* Alist)
{
   Int_t zmax  = (Int_t)(((KVIDentifier*)fGrid->GetIdentifiers()->Last())->GetPID() + 1);
   KVNucleus nuc;
 
   Int_t Z = 1;
   fAList.clear();
   KVNumberList Al(Alist);
   Al.Begin();
   while (!Al.End()) {
      Int_t A = Al.Next();
      if (A <= 0) A = nuc.GetAFromZ(Z, fGrid->GetMassFormula());
      fAList.push_back(A);
      Z++;
   }
 
   for (int i = Z + 1; i <= zmax; i++) fAList.push_back(nuc.GetAFromZ(i, fGrid->GetMassFormula()));
}
 
 
 
 
TH2* KVZALineFinder::LinearizeHisto(Int_t nZbin)
{
   Double_t zmin  = ((KVIDentifier*)fGrid->GetIdentifiers()->First())->GetPID() - 1.0;
   Double_t zmax  = ((KVIDentifier*)fGrid->GetIdentifiers()->Last())->GetPID() + 1.0;
   Int_t    zbins = (Int_t)(zmax - zmin) * nZbin;
 
   Double_t emin  = fHisto->GetXaxis()->GetXmin();
   Double_t emax  = fHisto->GetXaxis()->GetXmax();
   Int_t    ebins = fHisto->GetXaxis()->GetNbins();
 
   if (fLinearHisto) delete fLinearHisto;
   fLinearHisto = new TH2F("fLinearHisto", "fLinearHisto", ebins, emin, emax, zbins, zmin, zmax);
 
   //    if(fConvertHisto) delete fConvertHisto ;
   fConvertHisto = new TH2F("fConvertHisto", "fConvertHisto", ebins, emin, emax, zbins, zmin, zmax);
 
   KVIdentificationResult* idr = new KVIdentificationResult;
 
   Int_t events_read = 0;
   for (int i = 1; i <= fHisto->GetNbinsX(); i++) {
      for (int j = 1; j <= fHisto->GetNbinsY(); j++) {
         Stat_t poids = fHisto->GetBinContent(i, j);
         if (poids == 0) continue;
 
         Axis_t x0 = fHisto->GetXaxis()->GetBinCenter(i);
         Axis_t y0 = fHisto->GetYaxis()->GetBinCenter(j);
         Axis_t wx = fHisto->GetXaxis()->GetBinWidth(i);
         Axis_t wy = fHisto->GetYaxis()->GetBinWidth(j);
 
         Double_t x, y;
         Int_t kmax = (Int_t) TMath::Min(20., poids);
         Double_t weight = (kmax == 20 ? poids / 20. : 1.);
         for (int k = 0; k < kmax; k++) {
            x = gRandom->Uniform(x0 - .5 * wx, x0 + .5 * wx);
            y = gRandom->Uniform(y0 - .5 * wy, y0 + .5 * wy);
            if (fGrid->IsIdentifiable(x, y)) {
               fGrid->Identify(x, y, idr);
               Float_t PID = idr->PID;
               if (idr->Aident) PID = (idr->Z + 0.1 * (idr->PID - 2. * idr->Z));
               fLinearHisto->Fill(x, PID, weight);
               //                    fConvertHisto->SetPoint(fConvertHisto->GetN(),x0, nuc.GetPID(), y0);
            }
            events_read += (Int_t) poids;
            IncrementLinear((Float_t) events_read);
            gSystem->ProcessEvents();
         }
         if (fGrid->IsIdentifiable(x0, y0)) {
            fGrid->Identify(x0, y0, idr);
            Float_t PID = idr->PID;
            if (idr->Aident) PID = (idr->Z + 0.1 * (idr->PID - 2. * idr->Z));
            Int_t ibin = fConvertHisto->FindBin(x0, PID);
            fConvertHisto->SetBinContent(ibin, y0);
         }
      }
   }
 
   delete idr;
 
   //    new KVCanvas;
   //    fConvertHisto->Draw("colz");
   //    cout << fConvertHisto->GetN() << endl;
 
   return fLinearHisto;
}
 
 
 
 
void KVZALineFinder::FindZLine(Int_t zz)
{
   fLinearHisto->SetAxisRange(zz - 0.5, zz + 0.5, "Y");
 
   KVIDLine* line = (KVIDLine*)fGrid->GetIdentifier(zz, 2 * zz + 1); // A=2*zz+1 : dummy, A is ignored in this case
   if (!line) {
      int i = 1;
      while (!(line = (KVIDLine*)fGrid->GetIdentifier(zz + i, 2 * zz + 1))) i++;
   }
   if (!line) return;
 
   Double_t lX, lY;
   line->GetStartPoint(lX, lY);
   Int_t xbmin = 1;
   line->GetEndPoint(lX, lY);
   Int_t xbmax = fLinearHisto->GetXaxis()->FindBin(lX);
   Int_t width = (Int_t)((xbmax - xbmin) * 1.0 / 100.);
 
 
   Int_t widthmax = (Int_t)((xbmax - xbmin) * 1.0 / 30.);
 
   KVIDZALine* TheLine = 0;
   TheLine = (KVIDZALine*)((KVIDZAGrid*)fGeneratedGrid)->NewLine("ID");
   TheLine->SetZ(zz);
   TheLine->SetA(fAList.at(zz - 1));
 
   TH1* projey = 0;
   Int_t i = 0;
   Double_t lasty = zz;
   for (int xx = xbmin; xx < xbmax; xx += width) {
      projey = fLinearHisto->ProjectionY("ProjectionAfterLin", TMath::Max(xx - width / 2, xbmin), xx + width / 2);
      Double_t xline = fLinearHisto->GetBinCenter(xx);
      Double_t yline = projey->GetMean();
      if ((yline > zz - 0.5) && (yline < zz + 0.5)) {
         if (i == 0) lasty = yline;
         if (TMath::Abs(yline - lasty) < 0.2) {
            TheLine->SetPoint(i, xline, yline);
            lasty = yline;
            i++;
            if (width < widthmax) width *= 1.2;
         }
      }
      delete projey;
   }
 
   if (TheLine->GetN() > 5) fGeneratedGrid->Add("ID", TheLine);
}
 
 
 
 
void KVZALineFinder::FindALine(Int_t zz, Int_t width)
{
   fLinearHisto->SetAxisRange(zz - 0.5, zz + 0.5, "Y");
 
   KVIDLine* line = (KVIDLine*)fGrid->GetIdentifier(zz, 2 * zz + 1); // A=2*zz+1 : dummy, A is ignored in this case
   if (!line) {
      int i = 1;
      while (!(line = (KVIDLine*)fGrid->GetIdentifier(zz + i, 2 * zz + 1))) i++;
   }
   if (!line) return;
 
   Double_t lX, lY;
   line->GetStartPoint(lX, lY);
   Int_t xbmin = 0;//fLinearHisto->GetYaxis()->FindBin(lX);
   line->GetEndPoint(lX, lY);
   Int_t xbmax = fLinearHisto->GetXaxis()->FindBin(lX);
 
   // create lines
   TList Lines;
   KVSpiderLine* tmp = 0;
 
   fLinearHisto->SetAxisRange(fLinearHisto->GetXaxis()->GetBinCenter(50), lX, "X"); //fLinearHisto->GetXaxis()->GetXmax(),"X");
   TH1* tmph = fLinearHisto->ProjectionX(Form("tmph%d", zz));
   Int_t startBin = (Int_t)(tmph->GetMaximumBin() * 0.95);
   delete tmph;
 
   TH1* projey = 0;
   if (startBin) {
      projey = fLinearHisto->ProjectionY("ProjectionAfterLin", startBin - width * 3, startBin + width * 3);
      int nfound = fSpectrum.Search(projey, 0.05, "goff", 0.0001);
      Info("FindALine", "%d peack found...", nfound);
#if ROOT_VERSION_CODE > ROOT_VERSION(5,99,01)
      Double_t* xpeaks = fSpectrum.GetPositionX();
      Double_t* ypeaks = fSpectrum.GetPositionY();
#else
      Float_t* xpeaks = fSpectrum.GetPositionX();
      Float_t* ypeaks = fSpectrum.GetPositionY();
#endif
      for (int p = 0; p < nfound; p++) {
         if (p > 8) break;
         if (ypeaks[p] < 10) continue;
         Double_t xline = fLinearHisto->GetXaxis()->GetBinCenter(startBin);
         Double_t yline = xpeaks[p];
         KVSpiderLine* tmp = 0;
         TIter next(&Lines);
         while ((tmp = (KVSpiderLine*)next())) {
            Info("FindALine", "line found but I don't know why...");
            if (TMath::Abs(tmp->GetY() - yline) < 0.05) continue;
         }
         tmp = new KVSpiderLine(zz, -1);
         Lines.AddLast(tmp);
         tmp->AddPoint(xline, yline);
         fPoints->SetPoint(fNPoints, xline, yline);
         fNPoints++;
      }
      if (projey) delete projey;
   }
   else Error("FindALine", "not starting bin indicated...");
   SortLines(&Lines);
 
   Int_t nLines = Lines.GetSize();
   tmp = 0;
   for (int xx = startBin - width; xx > xbmin; xx -= width) {
      projey = fLinearHisto->ProjectionY("ProjectionAfterLin", xx - width / 2, xx + width / 2);
      int nfound = fSpectrum.Search(projey, 0.05, "goff", 0.02);
#if ROOT_VERSION_CODE > ROOT_VERSION(5,99,01)
      Double_t* xpeaks = fSpectrum.GetPositionX();
      Double_t* ypeaks = fSpectrum.GetPositionY();
#else
      Float_t* xpeaks = fSpectrum.GetPositionX();
      Float_t* ypeaks = fSpectrum.GetPositionY();
#endif
      for (int p = 0; p < nfound; p++) {
         if (p >= nLines + 1) continue;
         if (ypeaks[p] < 5) continue;
         Double_t xline = fLinearHisto->GetXaxis()->GetBinCenter(xx);
         Double_t yline = xpeaks[p];
         KVSpiderLine* tmp = 0;
         TIter next(&Lines);
         while ((tmp = (KVSpiderLine*)next())) {
            if ((TMath::Abs(tmp->GetY() - yline) < 0.05)) break;
         }
         if (tmp) {
            if ((TMath::Abs(tmp->GetX() - xline) < 10 * width)) tmp->AddPoint(xline, yline);
         }
      }
      if (projey) delete projey;
   }
 
   TIter nextli(&Lines);
   while ((tmp = (KVSpiderLine*)nextli()))tmp->Sort(true);
 
   tmp = 0;
   for (int xx = startBin + width; xx <= xbmax - width / 2; xx += width) {
      projey = fLinearHisto->ProjectionY("ProjectionAfterLin", xx - width / 2, xx + width / 2);
      int nfound = fSpectrum.Search(projey, 0.05, "goff", 0.02);
#if ROOT_VERSION_CODE > ROOT_VERSION(5,99,01)
      Double_t* xpeaks = fSpectrum.GetPositionX();
      Double_t* ypeaks = fSpectrum.GetPositionY();
#else
      Float_t* xpeaks = fSpectrum.GetPositionX();
      Float_t* ypeaks = fSpectrum.GetPositionY();
#endif
      for (int p = 0; p < nfound; p++) {
         if (p >= nLines + 1) continue;
         if (ypeaks[p] < 5) continue;
         Double_t xline = fLinearHisto->GetXaxis()->GetBinCenter(xx);
         Double_t yline = xpeaks[p];
         KVSpiderLine* tmp = 0;
         TIter next(&Lines);
         while ((tmp = (KVSpiderLine*)next())) {
            if (TMath::Abs(tmp->GetY() - yline) < 0.05) break;
         }
         if (tmp) {
            if ((TMath::Abs(tmp->GetX() - xline) < 10 * width)) tmp->AddPoint(xline, yline);
         }
      }
      if (projey) delete projey;
   }
 
   fLines->AddAll(&Lines);
}
 
 
 
 
void KVZALineFinder::SortLines(TList* Lines)
{
   int nn = Lines->GetSize();
   if (!nn) return;
 
   Int_t zz = ((KVSpiderLine*)Lines->At(0))->GetZ();
 
   double* yy = new double[nn];
   Int_t*    ii = new int[nn];
   for (int i = 0; i < nn; i++) yy[i] = ((KVSpiderLine*)Lines->At(i))->GetY(0);
 
   KVNucleus nuc;
   TMath::Sort(nn, yy, ii, kFALSE);
   Int_t iMostProb = 0;
   for (int i = 0; i < nn; i++) {
      if (ii[i] == 0) iMostProb = i;
   }
 
   Int_t aMostProb =  fAList.at(zz - 1);
   //   if(zz<=fAList.size()) aMostProb = fAList.at(zz-1);
   //   else aMostProb = 2*zz+1;
 
   for (int i = 0; i < nn; i++) {
      Int_t aa = aMostProb - (iMostProb - i);
      nuc.SetZandA(zz, aa);
      if ((nuc.GetLifeTime() < 1.e-06)) aa -= 1;
      ((KVSpiderLine*)Lines->At(ii[i]))->SetA(aa);
   }
 
   return;
}
 
 
 
 
 
void KVZALineFinder::MakeGrid()
{
   fGeneratedGrid = new KVIDZAGrid;
   fGeneratedGrid->SetName(Form("%s_Up", fGrid->GetName()));
   fGeneratedGrid->SetVarX(fGrid->GetVarX());
   fGeneratedGrid->SetVarY(fGrid->GetVarY());
   fGeneratedGrid->AddIDTelescopes(fGrid->GetIDTelescopes());
 
   KVIDZALine* TheLine = 0;
   KVIDZALine* TheZLine = 0;
   KVSpiderLine* spline = 0;
   TIter next_line(fLines);
   while ((spline = (KVSpiderLine*)next_line())) { // generate KVLines from KVSpiderLines
      spline->Sort();
      if ((spline->GetN() > 5)) {
         TheLine = (KVIDZALine*)((KVIDZAGrid*)fGeneratedGrid)->NewLine("ID");
         TheLine->SetZ(spline->GetZ());
         TheLine->SetA(spline->GetA());
         for (int i = 0; i < spline->GetN(); i++) {
            Int_t ibin = fConvertHisto->FindBin(spline->GetX(i), spline->GetY(i));
            Double_t y = fConvertHisto->GetBinContent(ibin);
            Int_t di = 1;
            while (!y) {
               y = fConvertHisto->GetBinContent(ibin + di);
               di++;
            }
            if ((i > spline->GetN() - 3) && (std::abs(y - TheLine->GetY()[TheLine->GetN() - 1]) > 20)) continue;
            TheLine->SetPoint(i, spline->GetX(i), y);//spline->GetY(i));
         }
 
         TheZLine = (KVIDZALine*)fGrid->GetIdentifier(TheLine->GetZ(), 200);
         if (!TheZLine) continue;
         Double_t x0 = TheLine->GetX()[0];
         Double_t x0ref = TheZLine->GetX()[0];
         Double_t x1ref = TheZLine->GetX()[TheZLine->GetN() - 1];
         if ((x0 > 2.*x0ref) && (x0 < 0.15 * x1ref)) {
            Double_t dy = (TheLine->GetY()[0]) - (TheZLine->Eval(x0));
            for (int i = 0; i <= TheZLine->GetN(); i++) {
               Double_t x = TheZLine->GetX()[i];
               Double_t y = dy + TheZLine->GetY()[i];
               if (x >= x0) break;
               TheLine->SetPoint(TheLine->GetN(), x, y);
            }
            TheLine->Sort(&TGraph::CompareX, kTRUE);
         }
 
         Double_t x1 = TheLine->GetX()[TheLine->GetN() - 1];
         if ((x1 < x1ref)) {
            Double_t dy = (TheLine->GetY()[TheLine->GetN() - 1]) - (TheZLine->Eval(x1));
            for (int i = TheZLine->GetN() - 1; i > 0; i--) {
               Double_t x = TheZLine->GetX()[i];
               Double_t y = dy + TheZLine->GetY()[i];
               if (x <= x1) break;
               TheLine->SetPoint(TheLine->GetN(), x, y);
            }
            TheLine->Sort(&TGraph::CompareX, kTRUE);
         }
 
         fGeneratedGrid->Add("ID", TheLine);
         fLines->Remove(spline);
      }
   }
 
   //    return;
 
   //    TheLine = 0;
   //    spline = 0;
   //    TIter next(fLines);
   //    while((spline = (KVSpiderLine*)next())) // scan rejected lines
   //    {
   //        Int_t zl = spline->GetZ();
   //        if(zl<4) continue;
   //        Int_t aMostProb = 0;
   //        if(zl<=fAList.size()) aMostProb = fAList.at(zl-1);
   //        else aMostProb = 2*zl+1;
   //        int index = 0;
   //        KVIDZALine* oldLine = 0;
   //        oldLine=(fGeneratedGrid->GetZALine(spline->GetZ(),aMostProb,index));
   //        if(!oldLine)
   //        {
   //            continue;
   //        }
   //        TheLine = (KVIDZALine*)((KVIDZAGrid*)fGeneratedGrid)->NewLine("ID");
   //        TheLine->SetZ(spline->GetZ());
   //        TheLine->SetA(spline->GetA());
 
   //        Int_t ibin = fConvertHisto->FindBin(spline->GetX(),spline->GetY());
   //        Double_t y = fConvertHisto->GetBinContent(ibin);
   //        Int_t di = 1;
   //        while(!y)
   //        {
   //            y = fConvertHisto->GetBinContent(ibin+di);
   //            di++;
   //        }
   //        Double_t dy = y - oldLine->Eval(spline->GetX());
   //        for(int i=0; i<oldLine->GetN(); i++)
   //        {
   //            TheLine->SetPoint(i, (oldLine->GetX()[i]), (oldLine->GetY()[i])+dy);
   //        }
   //    }
 
 
}
 
 
 
 
 
void KVZALineFinder::DrawGrid()
{
   gIDGridEditor->StartViewer();
   gIDGridEditor->SetHisto(fLinearHisto);
   gIDGridEditor->SetGrid(fGeneratedGrid, kFALSE);
}
 
 
 
 
 
void KVZALineFinder::ProcessIdentification(Int_t zmin, Int_t zmax)
{
   Info("ProcessIdentification", "histo linearization using Z grid...");
   LinearizeHisto(fBinsByZ);
   Info("ProcessIdentification", "linearization done!");
 
   if (zmin < 0) zmin = ((KVIDentifier*)fGrid->GetIdentifiers()->First())->GetZ();
   if (zmax < 0) zmax = ((KVIDentifier*)fGrid->GetIdentifiers()->Last())->GetZ();
 
   int ww = 10;
   for (int z = zmin; z <= zmax; z++) {
      if (z > 4)  ww = 20;
      if (z > 6)  ww = 30;
      if (z > 10) ww = 40;
      if (z > 12) ww = 50;
      FindALine(z, ww);
      Increment(z - zmin);
      gSystem->ProcessEvents();
   }
   MakeGrid();
 
   Double_t zmGrid  = ((KVIDentifier*)fGrid->GetIdentifiers()->Last())->GetPID();
   if (zmax >= zmGrid) return;
 
   for (int z = zmax + 1; z <= zmGrid; z++) {
      FindZLine(z);
      Increment(z - zmin);
      gSystem->ProcessEvents();
   }
 
}