PWGCF/Correlations/JCORRAN/AliJCard.cxx

   1 //====================================
   2 //last modified FK 6.NOV 2009
   3 //====================================
   4 //blah
   5 // blah
   6
   7 #include "AliJCard.h"
   8
   9 //ClassImp(AliJCard);
  10
  11 AliJCard::AliJCard():
  12     AliJBaseCard(),
  13     fhCorr(0),
  14     feventV3kv(0),
  15     fIndexVector(0),
  16     fpi0massbin(0)
  17 {
  18     //constructor
  19 }
  20
  21 AliJCard::AliJCard(const char *filename):
  22     AliJBaseCard(filename),
  23     fhCorr(0),
  24     feventV3kv(0),
  25     fIndexVector(0),
  26     fpi0massbin(0)
  27 {
  28     //constructor
  29     InitCard();
  30     MakeFastCorrTypeIndex();
  31 }
  32
  33
  34
  35 AliJCard::AliJCard(const AliJCard& obj) :
  36     AliJBaseCard(),
  37     fhCorr(obj.fhCorr),
  38     feventV3kv(obj.feventV3kv),
  39     fIndexVector(obj.fIndexVector),
  40     fpi0massbin(obj.fpi0massbin)
  41 {
  42     // copy constructor
  43     JUNUSED(obj);
  44 }
  45
  46
  47 AliJCard& AliJCard::operator=(const AliJCard& obj){
  48     // copy constructor
  49     JUNUSED(obj);
  50     return *this;
  51 }
  52
  53 AliJCard::~AliJCard(){
  54     // destructor
  55     if( fpi0massbin ){
  56         for( int i = 0; i < GetNoOfBins( kCentrType ); i++ ){
  57             delete [] fpi0massbin[i];
  58         }    // destructor
  59         delete [] fpi0massbin;
  60     }
  61     if( fhCorr ){
  62         delete fhCorr;
  63     }
  64 }
  65
  66 void AliJCard::MakeFastCorrTypeIndex(){
  67     // make fast findex array
  68     for( unsigned int i=0;i<fKeyWordVector.size();i++ ){
  69         int corrIndex = GetCorrType( fKeyWordVector[i] );
  70         if( corrIndex != kNoType )
  71             fIndexVector[corrIndex] = i;
  72     }
  73
  74 }
  75
  76 int AliJCard::IsLessThanUpperPairPtCut(double inPairPt){
  77     // pt cut
  78     int nB = GetN("UpperPairPtCut");
  79     if(inPairPt == -999) return nB;
  80     if(-999 < inPairPt && inPairPt <=0) return int(Get("UpperPairPtCut",-int(inPairPt)));
  81     int bin=-1, i=0;
  82     while(i<nB && inPairPt>Get("UpperPairPtCut",i)) i++;
  83     if(i<nB) bin=i;
  84     //cout<<" i="<<i<<" bin="<<bin<<" inval="<<inPairPt<<endl;;
  85     return bin;
  86 }
  87
  88
  89
  90 TString AliJCard::GetKeyWord(corrType ctype){
  91     // get keyword
  92     TString kw;
  93
  94     switch(ctype){
  95         case kTriggType:  kw = "TriggPtBorders"; break;
  96         case kAssocType:  kw = "AssocPtBorders"; break;
  97         case kXeType:     kw = "xEBorders"; break;
  98         case kLongType:   kw = "KlongBorders"; break;
  99         case kCentrType:  kw = "CentBinBorders"; break;
 100         case kZVertType:  kw = "zVertBins"; break;
 101         case kMassType:   kw = "PairInvariantMassBins"; break;
 102         case kEtaGapType: kw = "EtaGapThresholds"; break;
 103         case kDiJetType:  kw = "DiJetMassBorders"; break;
 104         case kRGapType:   kw = "RGapThresholds"; break;
 105                           //case kEPType:  kw = "EPBorders"; break;
 106         default : cout<<"ERROR: kNoType on input to AliJCard::GetKeyWord"<<endl; exit(1); break;
 107     }
 108     return kw;
 109 }
 110
 111 corrType AliJCard::GetCorrType( TString inStr ){
 112     // get corr type
 113     TString kw;
 114
 115     int i;
 116     TString s;
 117
 118     // go through all corrType to check if the parameter should be added
 119     // into fast findex array
 120     for( i = 0; i < kNcorrType; i++ ){
 121         if( i == kNoType )
 122             continue;
 123
 124         s = GetKeyWord( (corrType)i );
 125         if( ! strcmp( s.Data(), inStr.Data() ))
 126             return (corrType)i;
 127     }
 128
 129     return kNoType;
 130 }
 131
 132 int AliJCard::GetN(corrType ctype){
 133     //returns size of TVector
 134     return GetN(GetKeyWord(ctype));
 135 }
 136
 137 int AliJCard::GetNFast(corrType ctype){
 138     //returns size of TVector
 139     int findex = fIndexVector[ctype];
 140     if( findex > -1 ){
 141         return (int) fValuesVector[findex].GetNrows();
 142     }else{
 143         cout<<"ERROR: fValuesVector fast findex out of range "<< findex << " " << (GetKeyWord( ctype )).Data() << endl;
 144         exit(1);
 145     }
 146 }
 147
 148
 149 float AliJCard::Get(corrType ctype, int VectorComponent){
 150     //returns VectorComponent Component of  fValuesVecto`uor TVector for given keyword
 151     return Get(GetKeyWord(ctype), VectorComponent);
 152 }
 153
 154 float AliJCard::GetFast(corrType ctype, int VectorComponent){
 155     // fast get
 156
 157     if(0<=VectorComponent && VectorComponent<GetNFast(ctype) && fIndexVector[ctype] > -1 ){
 158         return fValuesVector[fIndexVector[ctype]][VectorComponent+1];
 159     }else{
 160         cout<<"ERROR: fValuesVector fast findex out of range "<< (GetKeyWord(ctype)).Data()<<endl;
 161         exit(1);
 162     }
 163 }
 164
 165
 166 void AliJCard::InitCard(){
 167     // Init card
 168     cout<<"Init of AliJCard"<<endl;
 169     // set the length of fIndexVector and disable all indices
 170     fIndexVector.resize( kNcorrType  );
 171     for( int i = 0; i < kNcorrType; i++ )
 172         fIndexVector[i] = -1;
 173 }
 174
 175 void AliJCard::FinishCard(){
 176     // Finish loading of card
 177     AliJBaseCard::FinishCard();
 178     // recompute fast idices
 179
 180     fpi0massbin = new Double_t*[GetNoOfBins( kCentrType )];
 181
 182     char centstr[10];
 183
 184     // read the pi0 mass bin parameters
 185     for( int i = 0; i < GetNoOfBins( kCentrType ); i++ ){
 186         fpi0massbin[i] = new Double_t[8];
 187
 188         for( int j = 0; j < 8; j++ ){
 189             sprintf( centstr, "cent%d", i );
 190             fpi0massbin[i][j] = Get( centstr, j );
 191         }
 192     }
 193
 194 }
 195
 196 void AliJCard::PrintOut(){
 197     // Print out contents of card
 198     AliJBaseCard::PrintOut();
 199     cout << "----- fast array ----------" << endl;
 200     for( unsigned int ii = 0; ii < fIndexVector.size(); ii++ ){
 201         if( ii == kNoType )
 202             continue;
 203         cout << (GetKeyWord( (corrType)ii )).Data() << " " << fIndexVector[ii];
 204         if( fIndexVector[ii] > -1 ){
 205             cout<<" (dim ="<<fValuesVector[fIndexVector[ii]].GetNrows()<<") ";//print size of TVector
 206             for(int j=1; j<=fValuesVector[fIndexVector[ii]].GetNrows(); j++){
 207                 cout<<fValuesVector[fIndexVector[ii]][j]<<" ";//TVector components
 208             }
 209         }
 210         else
 211             cout << " no link!";
 212
 213         cout << endl;
 214     }
 215 }
 216
 217 int AliJCard::GetBin(corrType ctype, float val){
 218     // get bin
 219
 220     if(ctype == kNoType) return 0;
 221
 222     TVector * v = &fValuesVector[fIndexVector[ctype]];
 223     int iBin2 = TMath::BinarySearch( v->GetNrows(), v->GetMatrixArray(), val );
 224     if( iBin2 >= v->GetNrows()-1 ) iBin2 = -1;
 225
 226     return iBin2;
 227 }
 228
 229 int AliJCard::GetBinFast(corrType ctype, float val){
 230     // fast get bin
 231
 232     if(ctype == kNoType) return 0;
 233
 234     for(int i=0; i<(GetNFast(ctype)-1); i++)
 235         if(GetFast(ctype,i)<=val && val<GetFast(ctype,i+1))
 236             return i;
 237
 238     return -1;
 239 }
 240
 241
 242
 243 bool AliJCard::IsGoodRun(int runID){
 244     // run quality
 245     bool isgood = true;
 246     for(int i=0; i<GetN("badRuns");i++) if(((int) Get("badRuns",i)) == runID) isgood = false;
 247
 248     return isgood;
 249 }
 250
 251 bool AliJCard::MbTrigger(int triggin) const {
 252     // MB trigger
 253     return (triggin & (1<<kMinBiasTriggerBitJCorran)) > 0; //masking on >0 masking off ==0 desired trigger
 254 }
 255
 256
 257 bool AliJCard::SimilarVertZ(float Z1, float Z2){
 258     // z vert
 259     static double v = Get("maxMixDZ");
 260     if( v < 0 ) return 1;
 261     return fabs(Z1-Z2) < v ;
 262 }
 263
 264 bool AliJCard::SimilarMultiplicity(float mult1, float mult2){
 265     // multi
 266     static double v = Get("maxMixDMult");
 267     if( v <0) return 1;
 268     return fabs(mult1-mult2) < v;
 269 }
 270
 271
 272 bool AliJCard::SimilarCentrality(float c1, float c2, int cbin){
 273     // centra
 274     static TVector *v = GetVector("maxDCent");
 275     if(v==NULL) return 1;
 276     return fabs(c1-c2) < (*v)[cbin+1]; // TODO
 277 }
 278
 279
 280 //--------- P H E N I X    C G L --------------
 281
 282 bool AliJCard::InPhiRange(float Phi){
 283     // phi
 284     bool isIn = false;
 285     for(int i=1; i<GetN("phiDCRange"); i+=2 )
 286         isIn = isIn || (Get("phiDCRange",i-1)<Phi && Phi<Get("phiDCRange",i));
 287     return isIn;
 288 }
 289
 290
 291
 292 bool  AliJCard::CheckTrackParamsInTPC(int NClustersTPC,float Chi2PerClusterTPC){
 293     // tpc pars
 294     bool isGoodTrack = true;
 295     if(NClustersTPC < Get("MinNClustersTPC"))           isGoodTrack = false;
 296     if(Chi2PerClusterTPC > Get("MaxChi2PerClusterTPC")) isGoodTrack = false;
 297     return isGoodTrack;
 298 }
 299
 300 bool  AliJCard::CheckMinNumTPCClustPt(int NClustersTPC, float fpt){
 301     // tpc pars
 302     float minNTPCCls =  Get("ParMinNClustTPCPt",0)*log(Get("ParMinNClustTPCPt",1)*fpt + Get("ParMinNClustTPCPt",2));
 303     if(NClustersTPC > minNTPCCls) return true; //track had enough clusters
 304     else return false;  //track did not have sufficient number of clusters
 305 }
 306
 307
 308 bool  AliJCard::CheckTrackImpact(float xyIm, float zIm, float fpt){
 309     // tpc pars
 310     bool isGoodTrack = true;
 311     if(TMath::Abs(xyIm) > Get("ParMaxDCAToVertexXYPtDep",0)+Get("ParMaxDCAToVertexXYPtDep",1)/pow(fpt,Get("ParMaxDCAToVertexXYPtDep",2))){
 312         isGoodTrack = false;
 313     }
 314     if(TMath::Abs(zIm) > Get("MaxDCAToVertexZ")) isGoodTrack = false;
 315
 316     return isGoodTrack;
 317 }
 318
 319
 320
 321 bool AliJCard::DeltaEtaCheck(const AliJBaseTrack *ftk1, const AliJBaseTrack *ftk2) {
 322     // deta
 323     double delta = fabs( log(tan(ftk1->Theta()/2.0)) -  log(tan(ftk2->Theta()/2.0)) );
 324     if(Get("etaCut")>0) return delta < Get("etaCut");
 325     if(Get("etaCut")<0) return delta > Get("etaCut");
 326
 327     return true;
 328 }
 329
 330
 331
 332
 333
 334
 335
 336
 337
 338
 339
 340
 341
 342
 343
 344
 345
 346
 347
 348
 349