/**************************************************************************
 * 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                                                    //
//  Author:                                                                  //
//    Marian Ivanov (marian.ivanov@cern.ch)                                  //
//                                                                           //
/////////////////////////////////////////////////////////////////////////////// 

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

ClassImp(AliTRDclusterCorrection)

AliTRDclusterCorrection *gAliTRDclusterCorrection = 0;

//_____________________________________________________________________________
AliTRDclusterCorrection::AliTRDclusterCorrection()
  :TObject()
  ,fOffsetAngle(0)
{
  //
  // Default constructor for AliTRDclusterCorrection
  //

  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.0;
	fCorrections[iplane][itime][iangle][1] = 0.0;
      }
    }
  }

}

//_____________________________________________________________________________
AliTRDclusterCorrection::~AliTRDclusterCorrection()
{
  //
  // Destructor
  //

}

//_____________________________________________________________________________
void AliTRDclusterCorrection::SetCorrection(Int_t plane,Int_t timebin, Float_t angle
					  , Float_t value, Float_t sigma)
{
  //
  // Set the correction factors
  //

  Int_t iangle = Int_t((angle - fOffsetAngle + 1.0) * 10.0 + 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) const
{
  //
  // Get the correction factors
  //

  Int_t iangle = Int_t((angle - fOffsetAngle + 1.0) * 10.0 + 0.5);
  if (iangle <   0) return 0.0;
  if (iangle >= 20) return 0.0;

  return fCorrections[plane][timebin][iangle][0];

}

//_____________________________________________________________________________
Float_t AliTRDclusterCorrection::GetSigma(Int_t plane, Int_t timebin, Float_t angle) const
{
  //
  // Returns the sigma
  //

  Int_t iangle = Int_t((angle - fOffsetAngle + 1.0) * 10.0 + 0.5);
  if (iangle <   0) return 1.0;
  if (iangle >= 20) return 1.0;

  return fCorrections[plane][timebin][iangle][1];

}

//_____________________________________________________________________________
AliTRDclusterCorrection *AliTRDclusterCorrection::GetCorrection()
{
  //
  // Return an instance of AliTRDclusterCorrection and sets the global
  // pointer gAliTRDclusterCorrection (Is this needed somewhere ????)
  //

  if (gAliTRDclusterCorrection != 0) {
    return gAliTRDclusterCorrection;
  }

  TFile *fileIn = new TFile("$ALICE_ROOT/TRD/TRDcorrection.root");
  if (!fileIn){
    gAliTRDclusterCorrection = new AliTRDclusterCorrection();
    return gAliTRDclusterCorrection;
  }

  gAliTRDclusterCorrection = (AliTRDclusterCorrection *) 
                             fileIn->Get("TRDcorrection");
  if (gAliTRDclusterCorrection == 0) {  
    gAliTRDclusterCorrection = new AliTRDclusterCorrection();
  }

  return gAliTRDclusterCorrection;
  
}