Additional protection (Yu.Belikov)

[u/mrichter/AliRoot.git] / TRD / AliTRDclusterCorrection.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/
 
///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD clusterCorrection                                                    //
//  marian.ivanov@cern.ch                                                    //
//                                                                           //
/////////////////////////////////////////////////////////////////////////////// 

#include "AliTRDclusterCorrection.h"
#include "TFile.h"

ClassImp(AliTRDclusterCorrection)

AliTRDclusterCorrection * gTRDclusterCorrection=0;

AliTRDclusterCorrection::AliTRDclusterCorrection()
{
  //
  //default constructor for correction
  //
  for (Int_t iplane=0;iplane<6;iplane++)
    for (Int_t itime=0;itime<30;itime++)
      for(Int_t iangle=0;iangle<20;iangle++){	
	fCorrections[iplane][itime][iangle][0]=0;
	fCorrections[iplane][itime][iangle][1]=0;
      }
  fOffsetAngle =0;
}


void AliTRDclusterCorrection::SetCorrection(Int_t plane,Int_t timebin, Float_t angle, 
					    Float_t value, Float_t sigma)
{
  Int_t iangle = int( (angle-fOffsetAngle+1.)*10.+0.5);
  if (iangle<0) return;
  if (iangle>=20) return;
  fCorrections[plane][timebin][iangle][0] = value;
  fCorrections[plane][timebin][iangle][1] = sigma;
}

Float_t AliTRDclusterCorrection::GetCorrection(Int_t plane, Int_t timebin, Float_t angle)
{
  Int_t iangle = int( (angle-fOffsetAngle+1.)*10.+0.5);
  if (iangle<0) return 0.;
  if (iangle>=20) return 0.;
  return fCorrections[plane][timebin][iangle][0];
}

Float_t AliTRDclusterCorrection::GetSigma(Int_t plane, Int_t timebin, Float_t angle)
{
  Int_t iangle = int( (angle-fOffsetAngle+1.)*10.+0.5);
  if (iangle<0) return 1.;
  if (iangle>=20) return 1.;
  return fCorrections[plane][timebin][iangle][1];
}


AliTRDclusterCorrection *  AliTRDclusterCorrection::GetCorrection()
{
  if (gTRDclusterCorrection!=0) return gTRDclusterCorrection;
  //
  TFile * f  = new TFile("$ALICE_ROOT/TRD/TRDcorrection.root");
  if (!f){
    ////
    gTRDclusterCorrection = new AliTRDclusterCorrection();
    return gTRDclusterCorrection;
  }
  gTRDclusterCorrection = (AliTRDclusterCorrection*)f->Get("TRDcorrection");
  if (gTRDclusterCorrection==0)  gTRDclusterCorrection = new AliTRDclusterCorrection();
  return gTRDclusterCorrection;
  
}