PWG2/FLOW/AliFlowAnalysisWithScalarProduct.cxx

   1 /*************************************************************************
   2 * Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15
  16 #define AliFlowAnalysisWithScalarProduct_cxx
  17
  18 #include "Riostream.h"  //needed as include
  19 //#include "TFile.h"      //needed as include
  20 #include "TList.h"
  21 #include "TMath.h"
  22 #include "TProfile.h"
  23 #include "TVector2.h"
  24
  25 class TH1F;
  26
  27 #include "AliFlowCommonConstants.h"    //needed as include
  28 #include "AliFlowEventSimple.h"
  29 #include "AliFlowTrackSimple.h"
  30 #include "AliFlowCommonHist.h"
  31 //#include "AliFlowCommonHistResults.h"
  32 #include "AliFlowAnalysisWithScalarProduct.h"
  33
  34 class AliFlowVector;
  35
  36 // AliFlowAnalysisWithScalarProduct:
  37 // Description:
  38 // Maker to analyze Flow with the Scalar product method.
  39 //
  40 // author: N. van der Kolk (kolk@nikhef.nl)
  41
  42 ClassImp(AliFlowAnalysisWithScalarProduct)
  43
  44   //-----------------------------------------------------------------------
  45
  46  AliFlowAnalysisWithScalarProduct::AliFlowAnalysisWithScalarProduct():
  47    fEventNumber(0),
  48    fDebug(kFALSE),
  49    fHistList(NULL),
  50    fHistProUQ(NULL),
  51    fCommonHists(NULL)
  52 {
  53   // Constructor.
  54   fHistList = new TList();
  55 }
  56  //-----------------------------------------------------------------------
  57
  58
  59  AliFlowAnalysisWithScalarProduct::~AliFlowAnalysisWithScalarProduct()
  60  {
  61    //destructor
  62    delete fHistList;
  63  }
  64
  65
  66 //-----------------------------------------------------------------------
  67 void AliFlowAnalysisWithScalarProduct::Init() {
  68
  69   //Define all histograms
  70   cout<<"---Analysis with the Scalar Product Method--- Init"<<endl;
  71
  72   Int_t iNbinsPt = AliFlowCommonConstants::GetNbinsPt();
  73   Double_t  dPtMin = AliFlowCommonConstants::GetPtMin();
  74   Double_t  dPtMax = AliFlowCommonConstants::GetPtMax();
  75
  76   fHistProUQ = new TProfile("Flow_UQ_SP","Flow_UQ_SP",iNbinsPt,dPtMin,dPtMax);
  77   fHistProUQ->SetXTitle("p_t (GeV)");
  78   fHistProUQ->SetYTitle("<uQ>");
  79   fHistList->Add(fHistProUQ);
  80
  81   fCommonHists = new AliFlowCommonHist("AliFlowCommonHistSP");
  82   fHistList->Add(fCommonHists);
  83
  84   //fCommonHistsRes = new AliFlowCommonHistResults("SP");
  85
  86   fEventNumber = 0;  //set number of events to zero
  87 }
  88
  89 //-----------------------------------------------------------------------
  90
  91 void AliFlowAnalysisWithScalarProduct::Make(AliFlowEventSimple* anEvent) {
  92
  93   //Fill histogram
  94   if (anEvent) {
  95
  96     //fill control histograms
  97     fCommonHists->FillControlHistograms(anEvent);
  98
  99     //get the Q vector from the FlowEvent
 100     AliFlowVector vQ = anEvent->GetQ();
 101
 102     //loop over the tracks of the event
 103     AliFlowTrackSimple*   pTrack = NULL;
 104     Int_t iNumberOfTracks = anEvent->NumberOfTracks();
 105     for (Int_t i=0;i<iNumberOfTracks;i++)
 106       {
 107         pTrack = anEvent->GetTrack(i) ;
 108         if (pTrack){
 109           if (pTrack->UseForDifferentialFlow()) {
 110           Double_t dPhi = pTrack->Phi();
 111
 112           //calculate vU
 113           TVector2 vU;
 114           Double_t dUX = TMath::Cos(2*dPhi);
 115           Double_t dUY = TMath::Sin(2*dPhi);
 116           vU.Set(dUX,dUY);
 117           Double_t dModulus = vU.Mod();
 118           if (dModulus!=0.) vU.Set(dUX/dModulus,dUY/dModulus);  // make length 1
 119           else cerr<<"dModulus is zero!"<<endl;
 120
 121           TVector2 vQm = vQ;
 122           //subtrackt particle from the flowvector if used to define it
 123           if (pTrack->UseForIntegratedFlow()) {
 124             Double_t dQmX = vQm.X() - dUX;
 125             Double_t dQmY = vQm.Y() - dUY;
 126             vQm.Set(dQmX,dQmY);
 127           }
 128
 129           //dUQ = scalar product of vU and vQm
 130           Double_t dUQ = vU * vQm;
 131           Double_t dPt = pTrack->Pt();
 132           //fill the profile histogram
 133           fHistProUQ->Fill(dPt,dUQ);
 134           }
 135         }//track selected
 136       }//loop over tracks
 137
 138     fEventNumber++;
 139     cout<<"@@@@@ "<<fEventNumber<<" events processed"<<endl;
 140   }
 141 }
 142
 143   //--------------------------------------------------------------------
 144 void AliFlowAnalysisWithScalarProduct::Finish() {
 145
 146   //*************make histograms etc.
 147   if (fDebug) cout<<"AliFlowAnalysisWithScalarProduct::Terminate()"<<endl;
 148
 149   fHistProUQ->Draw();
 150
 151   cout<<".....finished"<<endl;
 152  }
 153
 154