/*************************************************************************
* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  * 
**************************************************************************/

#define ALIFLOWANALYSISWITHSCALARPRODUCT_CXX
 
#include "TFile.h"      
#include "TList.h"
#include "TMath.h"
#include "TString.h"
#include "TProfile.h"
#include "TVector2.h"
#include "TH1D.h"
#include "TH1F.h"
#include "TH2D.h"

#include "AliFlowCommonConstants.h"
#include "AliFlowEventSimple.h"
#include "AliFlowVector.h"
#include "AliFlowTrackSimple.h"
#include "AliFlowCommonHist.h"
#include "AliFlowCommonHistResults.h"
#include "AliFlowAnalysisWithScalarProduct.h"

//////////////////////////////////////////////////////////////////////////////
// AliFlowAnalysisWithScalarProduct:
// Description: Maker to analyze Flow from the Scalar Product method.
// authors: Naomi van del Kolk (kolk@nikhef.nl)
//          Ante Bilandzic (anteb@nikhef.nl)
// mods:    Carlos Perez (cperez@nikhef.nl)
//////////////////////////////////////////////////////////////////////////////

ClassImp(AliFlowAnalysisWithScalarProduct)

//-----------------------------------------------------------------------
AliFlowAnalysisWithScalarProduct::AliFlowAnalysisWithScalarProduct():
fDebug(0),
fUsePhiWeights(0),
fApplyCorrectionForNUA(0),
fHarmonic(2),
fNormalizationType(1),
fTotalQvector(3),
fWeightsList(NULL),
fHistList(NULL),
fHistProConfig(NULL),
fHistProQaQbNorm(NULL),
fHistSumOfWeights(NULL),
fHistProNUAq(NULL),
fHistProQNorm(NULL),
fHistProQaQb(NULL),
fHistProQaQbM(NULL),
fHistMaMb(NULL),
fHistQNormQaQbNorm(NULL),
fHistQaNormMa(NULL),
fHistQbNormMb(NULL),
fResolution(NULL),
fHistQaQb(NULL),
fHistQaQbCos(NULL),
fCommonHists(NULL),
fCommonHistsuQ(NULL),
fCommonHistsRes(NULL)
{
  //ctor
  for(int i=0; i!=2; ++i) {
    fPhiWeightsSub[i] = NULL;
    for(int j=0; j!=2; ++j) {
      fHistProUQ[i][j] = NULL;
      fHistProUQQaQb[i][j] = NULL;
      fHistProNUAu[i][j][0] = NULL;
      fHistProNUAu[i][j][1] = NULL;
      for(int k=0; k!=3; ++k)
        fHistSumOfWeightsu[i][j][k] = NULL;
    }
  }
}
//-----------------------------------------------------------------------
AliFlowAnalysisWithScalarProduct::~AliFlowAnalysisWithScalarProduct() 
{
  //destructor
  delete fWeightsList;
  delete fHistList;
}
//-----------------------------------------------------------------------
void AliFlowAnalysisWithScalarProduct::Init() {
  //Define all histograms
  printf("---Analysis with the Scalar Product Method--- Init\n");
  printf("--- fNormalizationType %d ---\n", fNormalizationType);
  //save old value and prevent histograms from being added to directory
  //to avoid name clashes in case multiple analaysis objects are used
  //in an analysis
  Bool_t oldHistAddStatus = TH1::AddDirectoryStatus();
  TH1::AddDirectory(kFALSE);
 
  fHistList = new TList();
  fHistList->SetName("cobjSP");
  fHistList->SetOwner();

  TList *uQRelated = new TList();
  uQRelated->SetName("uQ");
  uQRelated->SetOwner();

  TList *nuaRelated = new TList();
  nuaRelated->SetName("NUA");
  nuaRelated->SetOwner();

  TList *errorRelated = new TList();
  errorRelated->SetName("error");
  errorRelated->SetOwner();

  TList *tQARelated = new TList();
  tQARelated->SetName("QA");
  tQARelated->SetOwner();

  fCommonHists = new AliFlowCommonHist("AliFlowCommonHist_SP");
  (fCommonHists->GetHarmonic())->Fill(0.5,fHarmonic); // store harmonic 
  fHistList->Add(fCommonHists);

  fCommonHistsuQ = new AliFlowCommonHist("AliFlowCommonHist_uQ");
  (fCommonHistsuQ->GetHarmonic())->Fill(0.5,fHarmonic); // store harmonic 
  fHistList->Add(fCommonHistsuQ);

  fCommonHistsRes = new AliFlowCommonHistResults("AliFlowCommonHistResults_SP","",fHarmonic);
  fHistList->Add(fCommonHistsRes);

  fHistProConfig = new TProfile("FlowPro_Flags_SP","Flow_Flags_SP",4,0.5,4.5,"s");
  fHistProConfig->GetXaxis()->SetBinLabel(1,"fApplyCorrectionForNUA");
  fHistProConfig->GetXaxis()->SetBinLabel(2,"fNormalizationType");
  fHistProConfig->GetXaxis()->SetBinLabel(3,"fUsePhiWeights");
  fHistProConfig->GetXaxis()->SetBinLabel(4,"fHarmonic");
  fHistProConfig->Fill(1,fApplyCorrectionForNUA);
  fHistProConfig->Fill(2,fNormalizationType);
  fHistProConfig->Fill(3,fUsePhiWeights);
  fHistProConfig->Fill(4,fHarmonic);
  fHistList->Add(fHistProConfig);

  fHistProQaQbNorm = new TProfile("FlowPro_QaQbNorm_SP","FlowPro_QaQbNorm_SP", 1, 0.5, 1.5,"s");
  fHistProQaQbNorm->SetYTitle("<QaQb/Na/Nb>");
  errorRelated->Add(fHistProQaQbNorm);

  fHistProNUAq = new TProfile("FlowPro_NUAq_SP","FlowPro_NUAq_SP", 6, 0.5, 6.5,"s");
  fHistProNUAq->GetXaxis()->SetBinLabel( 1,"<<sin(#Phi_{a})>>");
  fHistProNUAq->GetXaxis()->SetBinLabel( 2,"<<cos(#Phi_{a})>>");
  fHistProNUAq->GetXaxis()->SetBinLabel( 3,"<<sin(#Phi_{b})>>");
  fHistProNUAq->GetXaxis()->SetBinLabel( 4,"<<cos(#Phi_{b})>>");
  fHistProNUAq->GetXaxis()->SetBinLabel( 5,"<<sin(#Phi_{t})>>");
  fHistProNUAq->GetXaxis()->SetBinLabel( 6,"<<cos(#Phi_{t})>>");
  nuaRelated->Add(fHistProNUAq);

  fHistSumOfWeights = new TH1D("Flow_SumOfWeights_SP","Flow_SumOfWeights_SP",2,0.5, 2.5);
  fHistSumOfWeights->GetXaxis()->SetBinLabel( 1,"#Sigma Na*Nb");
  fHistSumOfWeights->GetXaxis()->SetBinLabel( 2,"#Sigma (Na*Nb)^2");
  errorRelated->Add(fHistSumOfWeights);

  TString sPOI[2] = {"RP","POI"}; // backward compatibility
  TString sEta[2] = {"Pt","eta"}; // backward compatibility
  TString sTitle[2] = {"p_{T} [GeV]","#eta"};
  TString sWeights[3] = {"uQ","uQuQ","uQQaQb"};
  Int_t iNbins[2];
  Double_t dMin[2], dMax[2];
  iNbins[0] = AliFlowCommonConstants::GetMaster()->GetNbinsPt();
  iNbins[1] = AliFlowCommonConstants::GetMaster()->GetNbinsEta();
  dMin[0]   = AliFlowCommonConstants::GetMaster()->GetPtMin();
  dMin[1]   = AliFlowCommonConstants::GetMaster()->GetEtaMin();
  dMax[0]   = AliFlowCommonConstants::GetMaster()->GetPtMax();
  dMax[1]   = AliFlowCommonConstants::GetMaster()->GetEtaMax();
  for(Int_t iPOI=0; iPOI!=2; ++iPOI) for(Int_t iSpace=0; iSpace!=2; ++iSpace) {
    // uQ
    fHistProUQ[iPOI][iSpace] = new TProfile( Form( "FlowPro_UQ_%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                       Form( "FlowPro_UQ%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                       iNbins[iSpace], dMin[iSpace], dMax[iSpace], "s");
    fHistProUQ[iPOI][iSpace]->SetXTitle(sTitle[iSpace].Data());
    fHistProUQ[iPOI][iSpace]->SetYTitle("<uQ>");
    uQRelated->Add(fHistProUQ[iPOI][iSpace]);

    // NUAu
    fHistProNUAu[iPOI][iSpace][0] = new TProfile( Form("FlowPro_NUAu_%s%s_IM_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           Form("FlowPro_NUAu_%s%s_IM_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           iNbins[iSpace], dMin[iSpace], dMax[iSpace]);
    fHistProNUAu[iPOI][iSpace][0]->SetXTitle(sTitle[iSpace].Data());
    nuaRelated->Add(fHistProNUAu[iPOI][iSpace][0]);
    fHistProNUAu[iPOI][iSpace][1] = new TProfile( Form("FlowPro_NUAu_%s%s_RE_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           Form("FlowPro_NUAu_%s%s_RE_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           iNbins[iSpace], dMin[iSpace], dMax[iSpace]);
    fHistProNUAu[iPOI][iSpace][1]->SetXTitle(sTitle[iSpace].Data());
    nuaRelated->Add(fHistProNUAu[iPOI][iSpace][1]);

    // uQ QaQb
    fHistProUQQaQb[iPOI][iSpace] = new TProfile( Form("FlowPro_UQQaQb_%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           Form("FlowPro_UQQaQb_%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ),
                                           iNbins[iSpace], dMin[iSpace], dMax[iSpace]);
    fHistProUQQaQb[iPOI][iSpace]->SetXTitle(sTitle[iSpace].Data());
    fHistProUQQaQb[iPOI][iSpace]-> SetYTitle("<Qu QaQb>");
    errorRelated->Add(fHistProUQQaQb[iPOI][iSpace]);

    // uWeights
    for(Int_t i=0; i!=3; ++i) {
      fHistSumOfWeightsu[iPOI][iSpace][i] = new TH1D( Form("Flow_SumOfWeights_%s%s_%s_SP",sWeights[i].Data(),sPOI[iPOI].Data(),sEta[iSpace].Data()),
                                                      Form("Flow_SumOfWeights_%s%s_%s_SP",sWeights[i].Data(),sPOI[iPOI].Data(),sEta[iSpace].Data()),
                                                      iNbins[iSpace], dMin[iSpace], dMax[iSpace]);
      fHistSumOfWeightsu[iPOI][iSpace][i]->SetYTitle(sWeights[i].Data());
      fHistSumOfWeightsu[iPOI][iSpace][i]->SetXTitle(sTitle[iSpace].Data());
      errorRelated->Add(fHistSumOfWeightsu[iPOI][iSpace][i]);
    }
  }
  //weights
  if(fUsePhiWeights) {
    if(!fWeightsList) {
      printf( "WARNING: fWeightsList is NULL in the Scalar Product method.\n" );
      exit(0);
    }
    fPhiWeightsSub[0] = dynamic_cast<TH1F*>
                        (fWeightsList->FindObject("phi_weights_sub0"));
    if(!fPhiWeightsSub[0]) {
      printf( "WARNING: phi_weights_sub0 not found in the Scalar Product method.\n" );
      exit(0);  
    }
    nuaRelated->Add( fPhiWeightsSub[0] );
    fPhiWeightsSub[1] = dynamic_cast<TH1F*>
                      (fWeightsList->FindObject("phi_weights_sub1"));
    if(!fPhiWeightsSub[1]) {
      printf( "WARNING: phi_weights_sub1 not found in the Scalar Product method.\n" );
      exit(0);  
    }
    nuaRelated->Add( fPhiWeightsSub[1] );
  }


  fHistProQNorm = new TProfile("FlowPro_QNorm_SP","FlowPro_QNorm_SP",       1,0.5,1.5,"s");
  fHistProQNorm->SetYTitle("<|Qa+Qb|>");
  tQARelated->Add(fHistProQNorm);

  fHistProQaQb  = new TProfile("FlowPro_QaQb_SP","FlowPro_QaQb_SP",         1,0.5,1.5,"s");
  fHistProQaQb->SetYTitle("<QaQb>");
  tQARelated->Add(fHistProQaQb);

  fHistProQaQbM = new TProfile("FlowPro_QaQbvsM_SP","FlowPro_QaQbvsM_SP",1000,0.0,10000);
  fHistProQaQbM->SetYTitle("<QaQb>");
  fHistProQaQbM->SetXTitle("M");
  fHistProQaQbM->Sumw2();
  tQARelated->Add(fHistProQaQbM);

  fHistMaMb = new TH2D("Flow_MavsMb_SP","Flow_MavsMb_SP",100,0.,100.,100,0.,100.);
  fHistMaMb->SetYTitle("Ma");
  fHistMaMb->SetXTitle("Mb");
  tQARelated->Add(fHistMaMb);

  fHistQNormQaQbNorm = new TH2D("Flow_QNormvsQaQbNorm_SP","Flow_QNormvsQaQbNorm_SP",88,-1.1,1.1,22,0.,1.1);
  fHistQNormQaQbNorm->SetYTitle("|Q/Mq|");
  fHistQNormQaQbNorm->SetXTitle("QaQb/MaMb");
  tQARelated->Add(fHistQNormQaQbNorm);

  fHistQaNormMa = new TH2D("Flow_QaNormvsMa_SP","Flow_QaNormvsMa_SP",100,0.,100.,22,0.,1.1);
  fHistQaNormMa->SetYTitle("|Qa/Ma|");
  fHistQaNormMa->SetXTitle("Ma");
  tQARelated->Add(fHistQaNormMa);

  fHistQbNormMb = new TH2D("Flow_QbNormvsMb_SP","Flow_QbNormvsMb_SP",100,0.,100.,22,0.,1.1);
  fHistQbNormMb->SetYTitle("|Qb/Mb|");
  fHistQbNormMb->SetXTitle("Mb");
  tQARelated->Add(fHistQbNormMb);

  fResolution = new TH1D("Flow_resolution_SP","Flow_resolution_SP",100,-1.0,1.0);
  fResolution->SetYTitle("dN/d(Cos2(#phi_a - #phi_b))");
  fResolution->SetXTitle("Cos2(#phi_a - #phi_b)");
  tQARelated->Add(fResolution);

  fHistQaQb = new TH1D("Flow_QaQb_SP","Flow_QaQb_SP",200,-100.,100.);
  fHistQaQb->SetYTitle("dN/dQaQb");
  fHistQaQb->SetXTitle("dQaQb");
  tQARelated->Add(fHistQaQb);

  fHistQaQbCos = new TH1D("Flow_QaQbCos_SP","Flow_QaQbCos_SP",63,0.,TMath::Pi());
  fHistQaQbCos->SetYTitle("dN/d#phi");
  fHistQaQbCos->SetXTitle("#phi");
  tQARelated->Add(fHistQaQbCos);

  fHistList->Add(uQRelated);
  fHistList->Add(nuaRelated);
  fHistList->Add(errorRelated);
  fHistList->Add(tQARelated);

  TH1::AddDirectory(oldHistAddStatus);
}

//-----------------------------------------------------------------------
void AliFlowAnalysisWithScalarProduct::Make(AliFlowEventSimple* anEvent) {
  // Scalar Product method
  if (!anEvent) return; // for coverity

  // Get Q vectors for the subevents
  AliFlowVector* vQarray = new AliFlowVector[2];
  if (fUsePhiWeights)
    anEvent->Get2Qsub(vQarray,fHarmonic,fWeightsList,kTRUE);
  else
    anEvent->Get2Qsub(vQarray,fHarmonic);
  // Subevent a
  AliFlowVector vQa = vQarray[0];
  // Subevent b
  AliFlowVector vQb = vQarray[1];
  delete [] vQarray;

  Double_t dMa = vQa.GetMult();
  if( dMa < 2 ) return;
  Double_t dMb = vQb.GetMult();
  if( dMb < 2 ) return;
  //fill control histograms
  if (fUsePhiWeights) {
    fCommonHists->FillControlHistograms(anEvent,fWeightsList,kTRUE);
  } else {
    fCommonHists->FillControlHistograms(anEvent);
  }

  //Normalizing: weight the Q vectors for the subevents
  Double_t dNa = fNormalizationType ? dMa: vQa.Mod(); // SP corresponds to true
  Double_t dNb = fNormalizationType ? dMb: vQb.Mod(); // SP corresponds to true
  Double_t dWa = fNormalizationType ? dMa: 1; // SP corresponds to true
  Double_t dWb = fNormalizationType ? dMb: 1; // SP corresponds to true

  //Scalar product of the two subevents vectors
  Double_t dQaQb = (vQa*vQb);

  //printf("==============\n");
  //printf("vQa: { %f, %f }\n",vQa.X(),vQa.Y());
  //printf("QaQb/|Qa||Qb|: %f\n",dQaQb/vQa.Mod()/vQb.Mod());
  //printf("QaQb/|Ma||Mb|: %f\n",dQaQb/dMa/dMb);
  
  //      01    10     11   <===   fTotalQVector
  // Q ?= Qa or Qb or QaQb
  AliFlowVector vQm;
  vQm.Set(0.0,0.0);
  Double_t dNq=0;
  if( (fTotalQvector%2)>0 ) { // 01 or 11
    vQm += vQa;
    dNq += dMa;
  }
  if( fTotalQvector>1 ) { // 10 or 11
    vQm += vQb;
    dNq += dMb;
  }
  Double_t dWq = fNormalizationType ? dNq: 1; // SP corresponds to true
  dNq = fNormalizationType ? dNq: vQm.Mod(); // SP corresponds to true

  //fill some EP control histograms
  fHistProQaQbNorm->Fill(1.,dQaQb/dNa/dNb,dWa*dWb);  //Fill (QaQb/NaNb) with weight (WaWb).
  //needed for the error calculation:
  fHistSumOfWeights -> Fill(1.,dWa*dWb);
  fHistSumOfWeights -> Fill(2.,pow(dWa*dWb,2.));
  //needed for correcting non-uniform acceptance: 
  fHistProNUAq->Fill(1.,vQa.Y()/dNa,dWa); // to get <<sin(phi_a)>>
  fHistProNUAq->Fill(2.,vQa.X()/dNa,dWa); // to get <<cos(phi_a)>>
  fHistProNUAq->Fill(3.,vQb.Y()/dNb,dWb); // to get <<sin(phi_b)>>
  fHistProNUAq->Fill(4.,vQb.X()/dNb,dWb); // to get <<cos(phi_b)>>
  fHistProNUAq->Fill(5.,vQm.Y()/dNq,dWq);
  fHistProNUAq->Fill(6.,vQm.X()/dNq,dWq);

  //loop over the tracks of the event
  AliFlowTrackSimple*   pTrack = NULL; 
  Int_t iNumberOfTracks = anEvent->NumberOfTracks(); 
  for (Int_t i=0;i<iNumberOfTracks;i++) {
    pTrack = anEvent->GetTrack(i) ; 
    if (!pTrack) continue;
    Double_t dPhi = pTrack->Phi();
    Double_t dPt  = pTrack->Pt();
    Double_t dEta = pTrack->Eta();

    //calculate vU
    TVector2 vU;
    Double_t dUX = TMath::Cos(fHarmonic*dPhi);
    Double_t dUY = TMath::Sin(fHarmonic*dPhi);
    vU.Set(dUX,dUY);

    //      01    10     11   <===   fTotalQVector
    // Q ?= Qa or Qb or QaQb
    vQm.Set(0.0,0.0); //start the loop fresh
    Double_t dMq=0;
    if( (fTotalQvector%2)>0 ) { // 01 or 11
      vQm += vQa;
      dMq += dMa;
    }
    if( fTotalQvector>1 ) { // 10 or 11
      vQm += vQb;
      dMq += dMb;
    }

    //remove track if in subevent
    for(Int_t inSubEvent=0; inSubEvent!=2; ++inSubEvent) {
      if( !pTrack->InSubevent( inSubEvent ) )
        continue;
      if(inSubEvent==0)
        if( (fTotalQvector%2)!=1 )
          continue;
      if(inSubEvent==1)
        if( fTotalQvector>0 )
          continue;
      Double_t dW=1;
      Double_t dPhiCenter = dPhi;
      //if phi weights are used
      if(fUsePhiWeights && fPhiWeightsSub[inSubEvent]) {
        Int_t iNBinsPhiSub = fPhiWeightsSub[inSubEvent]->GetNbinsX();
        Int_t phiBin = 1+(Int_t)(TMath::Floor(dPhi*iNBinsPhiSub/TMath::TwoPi()));
        dW = fPhiWeightsSub[inSubEvent]->GetBinContent(phiBin);
        dPhiCenter = fPhiWeightsSub[inSubEvent]->GetBinCenter(phiBin);
      }
      Double_t dQmX = vQm.X() - dW*(pTrack->Weight())* TMath::Cos(fHarmonic*dPhiCenter);
      Double_t dQmY = vQm.Y() - dW*(pTrack->Weight())* TMath::Sin(fHarmonic*dPhiCenter);
      vQm.Set(dQmX,dQmY);
      dMq = dMq-dW*pTrack->Weight();
    }
    dNq = fNormalizationType ? dMq : vQm.Mod();
    dWq = fNormalizationType ? dMq : 1;

    //Filling QA (for compatibility with previous version)
    fHistProQaQb->Fill(1,vQa*vQb,1);
    fHistProQaQbM->Fill(anEvent->GetNumberOfRPs()+0.5,(vQa*vQb)/dMa/dMb,dMa*dMb);
    fHistQaQbCos->Fill(TMath::ACos((vQa*vQb)/vQa.Mod()/vQb.Mod()));
    fResolution->Fill( TMath::Cos( vQa.Phi()-vQb.Phi() ) );
    fHistQaQb->Fill(vQa*vQb);
    fHistMaMb->Fill(dMb,dMa);
    fHistProQNorm->Fill(1,vQm.Mod()/dMq,dMq);
    fHistQNormQaQbNorm->Fill((vQa*vQb)/dMa/dMb,vQm.Mod()/dMq);
    fHistQaNormMa->Fill(dMa,vQa.Mod()/dMa);
    fHistQbNormMb->Fill(dMb,vQb.Mod()/dMb);

    Double_t dUQ = vU*vQm;

    //fill the profile histograms
    for(Int_t iPOI=0; iPOI!=2; ++iPOI) {
      if( (iPOI==0)&&(!pTrack->InRPSelection()) )
        continue;
      if( (iPOI==1)&&(!pTrack->InPOISelection()) )
        continue;
      fHistProUQ[iPOI][0]->Fill(dPt ,dUQ/dNq,dWq); //Fill (uQ/Nq') with weight (Nq')
      fHistProUQ[iPOI][1]->Fill(dEta,dUQ/dNq,dWq); //Fill (uQ/Nq') with weight (Nq')
      //needed for the error calculation:
      fHistProUQQaQb[iPOI][0]-> Fill(dPt ,dUQ/dNq*dQaQb/dNa/dNb,dWq*dWa*dWb); //Fill [Qu/Nq']*[QaQb/NaNb] with weight (Nq')NaNb
      fHistProUQQaQb[iPOI][1]-> Fill(dEta,dUQ/dNq*dQaQb/dNa/dNb,dWq*dWa*dWb); //Fill [Qu/Nq']*[QaQb/NaNb] with weight (Nq')NaNb
      fHistSumOfWeightsu[iPOI][0][0]->Fill(dPt ,dWq);        // sum of Nq'     
      fHistSumOfWeightsu[iPOI][0][1]->Fill(dPt ,pow(dWq,2.));// sum of Nq'^2     
      fHistSumOfWeightsu[iPOI][0][2]->Fill(dPt ,dWq*dWa*dWb);// sum of Nq'*Na*Nb     
      fHistSumOfWeightsu[iPOI][1][0]->Fill(dEta,dWq);        // sum of Nq'     
      fHistSumOfWeightsu[iPOI][1][1]->Fill(dEta,pow(dWq,2.));// sum of Nq'^2     
      fHistSumOfWeightsu[iPOI][1][2]->Fill(dEta,dNq*dWa*dWb);// sum of Nq'*Na*Nb
      //NUA:
      fHistProNUAu[iPOI][0][0]->Fill(dPt,dUY,1.); //sin u
      fHistProNUAu[iPOI][0][1]->Fill(dPt,dUX,1.); //cos u
      fHistProNUAu[iPOI][1][0]->Fill(dEta,dUY,1.); //sin u
      fHistProNUAu[iPOI][1][1]->Fill(dEta,dUX,1.); //cos u
    }
  }//loop over tracks

}

//--------------------------------------------------------------------  
void AliFlowAnalysisWithScalarProduct::GetOutputHistograms(TList *outputListHistos){
  //get pointers to all output histograms (called before Finish())
  fHistList = outputListHistos;

  fCommonHists = (AliFlowCommonHist*) fHistList->FindObject("AliFlowCommonHist_SP");
  fCommonHistsuQ = (AliFlowCommonHist*) fHistList->FindObject("AliFlowCommonHist_uQ");
  fCommonHistsRes = (AliFlowCommonHistResults*) fHistList->FindObject("AliFlowCommonHistResults_SP");
  fHistProConfig = (TProfile*) fHistList->FindObject("FlowPro_Flags_SP");
  if(!fHistProConfig) printf("Error loading fHistProConfig\n");
  TList *uQ = (TList*) fHistList->FindObject("uQ");
  TList *nua = (TList*) fHistList->FindObject("NUA");
  TList *error = (TList*) fHistList->FindObject("error");

  fHistProQaQbNorm = (TProfile*) error->FindObject("FlowPro_QaQbNorm_SP");
  if(!fHistProQaQbNorm) printf("Error loading fHistProQaQbNorm\n");
  fHistProNUAq = (TProfile*) nua->FindObject("FlowPro_NUAq_SP");
  if(!fHistProNUAq) printf("Error loading fHistProNUAq\n");
  fHistSumOfWeights = (TH1D*) error->FindObject("Flow_SumOfWeights_SP");
  if(!fHistSumOfWeights) printf("Error loading fHistSumOfWeights\n");

  TString sPOI[2] = {"RP","POI"};
  TString sEta[2] = {"Pt","eta"};
  TString sWeights[3] = {"uQ","uQuQ","uQQaQb"};
  Int_t iNbins[2];
  Double_t dMin[2], dMax[2];
  
  iNbins[0] = AliFlowCommonConstants::GetMaster()->GetNbinsPt();
  iNbins[1] = AliFlowCommonConstants::GetMaster()->GetNbinsEta();
  dMin[0]   = AliFlowCommonConstants::GetMaster()->GetPtMin();
  dMin[1]   = AliFlowCommonConstants::GetMaster()->GetEtaMin();
  dMax[0]   = AliFlowCommonConstants::GetMaster()->GetPtMax();
  dMax[1]   = AliFlowCommonConstants::GetMaster()->GetEtaMax();
  for(Int_t iPOI=0; iPOI!=2; ++iPOI) for(Int_t iSpace=0; iSpace!=2; ++iSpace) {
    fHistProUQ[iPOI][iSpace] = (TProfile*) uQ->FindObject( Form( "FlowPro_UQ_%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ) );
    if(!fHistProUQ[iPOI][iSpace]) printf("Error loading fHistProUQ[%d][%d]\n",iPOI,iSpace);
    fHistProNUAu[iPOI][iSpace][0] = (TProfile*) nua->FindObject( Form("FlowPro_NUAu_%s%s_IM_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ) );
    if(!fHistProNUAu[iPOI][iSpace][0]) printf("Error loading fHistProNUAu[%d][%d][0]\n",iPOI,iSpace);
    fHistProNUAu[iPOI][iSpace][1] = (TProfile*) nua->FindObject( Form("FlowPro_NUAu_%s%s_RE_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ) );
    if(!fHistProNUAu[iPOI][iSpace][1]) printf("Error loading fHistProNUAu[%d][%d][1]\n",iPOI,iSpace);
    fHistProUQQaQb[iPOI][iSpace] = (TProfile*) error->FindObject( Form("FlowPro_UQQaQb_%s%s_SP", sEta[iSpace].Data(), sPOI[iPOI].Data() ) );
    for(Int_t i=0; i!=3; ++i){
      fHistSumOfWeightsu[iPOI][iSpace][i] = (TH1D*) error->FindObject( Form("Flow_SumOfWeights_%s%s_%s_SP",sWeights[i].Data(),sPOI[iPOI].Data(),sEta[iSpace].Data()) );
      if(!fHistSumOfWeightsu[iPOI][iSpace][i]) printf("Error loading fHistSumOfWeightsu[%d][%d][%d]\n",iPOI,iSpace,i);
    }
  }
  if(fHistProConfig) {
    fApplyCorrectionForNUA = (Int_t) fHistProConfig->GetBinContent(1);
    fNormalizationType  = (Int_t) fHistProConfig->GetBinContent(2);
    fUsePhiWeights = (Int_t) fHistProConfig->GetBinContent(3);
    fHarmonic = (Int_t) fHistProConfig->GetBinContent(4);
  }
}            

//--------------------------------------------------------------------            
void AliFlowAnalysisWithScalarProduct::Finish() {
  //calculate flow and fill the AliFlowCommonHistResults
  printf("AliFlowAnalysisWithScalarProduct::Finish()\n");
  
  // access harmonic:
  fApplyCorrectionForNUA = fHistProConfig->GetBinContent(1);
  fNormalizationType = fHistProConfig->GetBinContent(2);
  fHarmonic = fHistProConfig->GetBinContent(4);
  
  printf("*************************************\n");
  printf("*************************************\n");
  printf("      Integrated flow from           \n");
  printf("         Scalar Product              \n\n");
  if(!fNormalizationType)
    printf("          (BehaveAsEP)               \n\n");
  
  //Calculate reference flow
  //----------------------------------
  //weighted average over (QaQb/NaNb) with weight (NaNb)
  Double_t dEntriesQaQb = fHistProQaQbNorm->GetEntries();
  if( dEntriesQaQb < 1 )
    return;
  Double_t dQaQb  = fHistProQaQbNorm->GetBinContent(1);
  if( dQaQb < 0 )
    return;
  Double_t dSpreadQaQb = fHistProQaQbNorm->GetBinError(1);

  //NUA qq:
  Double_t dImQa = fHistProNUAq->GetBinContent(1);
  Double_t dReQa = fHistProNUAq->GetBinContent(2);
  Double_t dImQb = fHistProNUAq->GetBinContent(3);
  Double_t dReQb = fHistProNUAq->GetBinContent(4);
  if(fApplyCorrectionForNUA) 
    dQaQb = dQaQb - dImQa*dImQb - dReQa*dReQb; 
  printf("QaQb = %f +- %f\n", dQaQb, (dSpreadQaQb/TMath::Sqrt(dEntriesQaQb)) );
  Double_t dV = TMath::Sqrt(dQaQb);

  printf("ResSub = %f\n", dV );
  printf("fTotalQvector %d \n",fTotalQvector);
  if(!fNormalizationType) {
    if(fTotalQvector>2) {
      dV = ComputeResolution( TMath::Sqrt2()*FindXi(dV,1e-6) );
      printf("An estimate of the event plane resolution is: %f\n", dV );
    }
  }
  printf("ResTot = %f\n", dV );
  //statistical error of dQaQb: 
  //  statistical error = term1 * spread * term2:
  //  term1 = sqrt{sum_{i=1}^{N} w^2}/(sum_{i=1}^{N} w)
  //  term2 = 1/sqrt(1-term1^2) 
  Double_t dSumOfLinearWeights = fHistSumOfWeights->GetBinContent(1);
  Double_t dSumOfQuadraticWeights = fHistSumOfWeights->GetBinContent(2);
  Double_t dTerm1 = 0.;
  Double_t dTerm2 = 0.;
  if(dSumOfLinearWeights)
    dTerm1 = pow(dSumOfQuadraticWeights,0.5)/dSumOfLinearWeights;
  if(1.-pow(dTerm1,2.) > 0.)
    dTerm2 = 1./pow(1-pow(dTerm1,2.),0.5);
  Double_t dStatErrorQaQb = dTerm1 * dSpreadQaQb * dTerm2;
  Double_t dVerr = 0.;
  if(dQaQb > 0.)
    dVerr = (1./(2.*pow(dQaQb,0.5)))*dStatErrorQaQb;
  fCommonHistsRes->FillIntegratedFlow(dV,dVerr);
  printf("v%d(subevents) = %f +- %f\n",fHarmonic,dV,dVerr);

  Int_t iNbins[2];
  iNbins[0] = AliFlowCommonConstants::GetMaster()->GetNbinsPt();
  iNbins[1] = AliFlowCommonConstants::GetMaster()->GetNbinsEta();
   
  //Calculate differential flow and integrated flow (RP, POI)
  //---------------------------------------------------------
  //v as a function of eta for RP selection
  for(Int_t iRFPorPOI=0; iRFPorPOI != 2; ++iRFPorPOI)
  for(Int_t iPTorETA=0; iPTorETA != 2; ++iPTorETA)
  for(Int_t b=1; b != iNbins[iPTorETA]+1; ++b) {
    Double_t duQpro = fHistProUQ[iRFPorPOI][iPTorETA]->GetBinContent(b);
    if(fApplyCorrectionForNUA)
      duQpro = duQpro 
	- fHistProNUAu[iRFPorPOI][iPTorETA][1]->GetBinContent(b)*fHistProNUAq->GetBinContent(6)
	- fHistProNUAu[iRFPorPOI][iPTorETA][0]->GetBinContent(b)*fHistProNUAq->GetBinContent(5);
    Double_t dv2pro = -999.;
    if( TMath::Abs(dV!=0.) ) { dv2pro = duQpro/dV; }
    //calculate the statistical error
    Double_t dv2ProErr = CalculateStatisticalError(iRFPorPOI, iPTorETA, b, dStatErrorQaQb);
    if( (iRFPorPOI==0)&&(iPTorETA==0) )
      fCommonHistsRes->FillDifferentialFlowPtRP(  b, dv2pro, dv2ProErr);   
    if( (iRFPorPOI==0)&&(iPTorETA==1) )
      fCommonHistsRes->FillDifferentialFlowEtaRP( b, dv2pro, dv2ProErr);   
    if( (iRFPorPOI==1)&&(iPTorETA==0) )
      fCommonHistsRes->FillDifferentialFlowPtPOI( b, dv2pro, dv2ProErr);   
    if( (iRFPorPOI==1)&&(iPTorETA==1) )
      fCommonHistsRes->FillDifferentialFlowEtaPOI(b, dv2pro, dv2ProErr);
    //printf("POI %d | PT %d >>> %f +- %f\n",iRFPorPOI,iPTorETA,dv2pro,dv2ProErr);
  }
  
  printf("\n");
  printf("*************************************\n");
  printf("*************************************\n");
}

//-----------------------------------------------------------------------
void AliFlowAnalysisWithScalarProduct::WriteHistograms(TDirectoryFile *outputFileName) const {
 //store the final results in output .root file
 outputFileName->Add(fHistList);
 outputFileName->Write(outputFileName->GetName(), TObject::kSingleKey);
}

//--------------------------------------------------------------------            
Double_t AliFlowAnalysisWithScalarProduct::CalculateStatisticalError(Int_t iRFPorPOI, Int_t iPTorETA, Int_t b, Double_t aStatErrorQaQb) const {
  //calculate the statistical error for differential flow for bin b
  Double_t duQproSpread = fHistProUQ[iRFPorPOI][iPTorETA]->GetBinError(b);
  Double_t sumOfMq = fHistSumOfWeightsu[iRFPorPOI][iPTorETA][0]->GetBinContent(b);
  Double_t sumOfMqSquared = fHistSumOfWeightsu[iRFPorPOI][iPTorETA][1]->GetBinContent(b);
  Double_t dQaQb = fHistProQaQbNorm->GetBinContent(1);

  //non-isotropic terms:  
  if(fApplyCorrectionForNUA) {
    Double_t dImQa = fHistProNUAq->GetBinContent(1);  // <<sin(phi_a)>>
    Double_t dReQa = fHistProNUAq->GetBinContent(2);  // <<cos(phi_a)>>
    Double_t dImQb = fHistProNUAq->GetBinContent(3);  // <<sin(phi_b)>>
    Double_t dReQb = fHistProNUAq->GetBinContent(4);  // <<cos(phi_b)>>
    dQaQb = dQaQb - dImQa*dImQb - dReQa*dReQb; 
  }

  Double_t dTerm1 = 0.;
  Double_t dTerm2 = 0.;
  if(sumOfMq) {
    dTerm1 = (pow(sumOfMqSquared,0.5)/sumOfMq);
  } 
  if(1.-pow(dTerm1,2.)>0.) {
    dTerm2 = 1./pow(1.-pow(dTerm1,2.),0.5); 
  }
  Double_t duQproErr = dTerm1*duQproSpread*dTerm2;
  // covariances:
  Double_t dTerm1Cov = fHistSumOfWeightsu[iRFPorPOI][iPTorETA][2]->GetBinContent(b);
  Double_t dTerm2Cov = fHistSumOfWeights->GetBinContent(1);
  Double_t dTerm3Cov = sumOfMq;
  Double_t dWeightedCovariance = 0.;
  if(dTerm2Cov*dTerm3Cov>0.) {
    Double_t dDenominator = 1.-dTerm1Cov/(dTerm2Cov*dTerm3Cov);
    Double_t dPrefactor = dTerm1Cov/(dTerm2Cov*dTerm3Cov);
    if(dDenominator!=0) {
      Double_t dCovariance = ( fHistProUQQaQb[iRFPorPOI][iPTorETA]->GetBinContent(b)-dQaQb*fHistProUQ[iRFPorPOI][iPTorETA]->GetBinContent(b))/dDenominator;
      dWeightedCovariance = dCovariance*dPrefactor; 
    }
  }
  Double_t dv2ProErr = 0.; // final statitical error: 
  if(dQaQb>0.) {
    Double_t dv2ProErrorSquared = (1./4.)*pow(dQaQb,-3.)*
      (pow(fHistProUQ[iRFPorPOI][iPTorETA]->GetBinContent(b),2.)*pow(aStatErrorQaQb,2.)
       + 4.*pow(dQaQb,2.)*pow(duQproErr,2.)
       - 4.*dQaQb*fHistProUQ[iRFPorPOI][iPTorETA]->GetBinContent(b)*dWeightedCovariance);
    if(dv2ProErrorSquared>0.) dv2ProErr = pow(dv2ProErrorSquared,0.5);
  }
  return dv2ProErr;
}

Double_t AliFlowAnalysisWithScalarProduct::ComputeResolution( Double_t x ) const {
  // Computes resolution for Event Plane method
  if(x > 51.3) {
    printf("Warning: Estimation of total resolution might be WRONG. Please check!");
    return 0.99981;
  }
  Double_t a = x*x/4;
  Double_t b = TMath::Exp(-a)*TMath::BesselI0(a)+TMath::Exp(-a)*TMath::BesselI1(a);
  return TMath::Sqrt(TMath::PiOver2())/2*x*b;
}

Double_t AliFlowAnalysisWithScalarProduct::FindXi( Double_t res, Double_t prec ) const {
  // Computes x(res) for Event Plane method
  if(res > 0.99981) {
    printf("Warning: Resolution for subEvent is high. You reached the precision limit.");
    return 51.3;
  }
  int nSteps =0;
  Double_t xtmp=0, xmin=0, xmax=51.3, rtmp=0, delta=2*prec;
  while( delta > prec ) {
    xtmp = 0.5*(xmin+xmax);
    rtmp = ComputeResolution(xtmp);
    delta = TMath::Abs( res-rtmp );
    if(rtmp>res) xmax = xtmp;
    if(rtmp<res) xmin = xtmp;
    nSteps++;
  }
  return xtmp;
}