/**************************************************************************
 * 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.                  *
 **************************************************************************/

/* $Id$ */

//_________________________________________________________________________
///*-- Author: Yves Schutz (SUBATECH)
//           : Aleksei Pavlinov (WSU); Jun 30, 2006 - ALICE numbering scheme
//           : Add decalibration and time calibration arrays: Jul 21, 2011 (GCB)
///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// class for EMCAL calibration                                               //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include <TMath.h>

#include "AliEMCALCalibData.h"

ClassImp(AliEMCALCalibData)

//________________________________________________________________
AliEMCALCalibData::AliEMCALCalibData()
{
  // Default constructor
  Reset();
}

//________________________________________________________________
AliEMCALCalibData::AliEMCALCalibData(const char* name)
{
  // Constructor
  TString namst = "Calib_";
  namst += name;
  SetName(namst.Data());
  SetTitle(namst.Data());
  Reset();
}

//________________________________________________________________
AliEMCALCalibData::AliEMCALCalibData(const AliEMCALCalibData& calibda) :
  TNamed(calibda)
{
  // copy constructor
  SetName(calibda.GetName());
  SetTitle(calibda.GetName());
  Reset();

  Int_t nSMod = AliEMCALGeoParams::fgkEMCALModules; //12
  Int_t nCol  = AliEMCALGeoParams::fgkEMCALCols;    //48
  Int_t nRow  = AliEMCALGeoParams::fgkEMCALRows;    //24
  Int_t nRow2 = AliEMCALGeoParams::fgkEMCALRows;    //12 - Modules 11 and 12 are half modules

  for(Int_t supermodule = 0; supermodule < nSMod; supermodule++) {
    
    if(supermodule >= 10)
      nRow = nRow2;
    
    for(Int_t column = 0; column<nCol; column++) {
      
      for(Int_t row = 0; row<nRow; row++) {
        
        fADCchannel[supermodule][column][row] = 
        calibda.GetADCchannel(supermodule,column,row);
        
        fADCchannelDecal[supermodule][column][row] = 
        calibda.GetADCchannelDecal(supermodule,column,row);
        
        fADCpedestal[supermodule][column][row] = 
        calibda.GetADCpedestal(supermodule,column,row);
        
        fTimeChannel[supermodule][column][row] = 
        calibda.GetTimeChannel(supermodule,column,row);
        
        fTimeChannelDecal[supermodule][column][row] = 
        calibda.GetTimeChannelDecal(supermodule,column,row);
        
        
      }
    }
  }
}


//________________________________________________________________
AliEMCALCalibData &AliEMCALCalibData::operator =(const AliEMCALCalibData& calibda)
{
  // assignment operator
  SetName(calibda.GetName());
  SetTitle(calibda.GetName());
  Reset();

  Int_t nSMod = AliEMCALGeoParams::fgkEMCALModules; //12
  Int_t nCol  = AliEMCALGeoParams::fgkEMCALCols;    //48
  Int_t nRow  = AliEMCALGeoParams::fgkEMCALRows;    //24
  Int_t nRow2 = AliEMCALGeoParams::fgkEMCALRows/2;  //12 - Modules 11 and 12 are half modules

  for(Int_t supermodule = 0; supermodule < nSMod; supermodule++) {
    
    if(supermodule >= 10)
      nRow = nRow2;
    
    for(Int_t column = 0; column<nCol; column++) {
      
      for(Int_t row = 0; row<nRow; row++) {
        
        fADCchannel[supermodule][column][row] = 
        calibda.GetADCchannel(supermodule,column,row);
        
        fADCchannelDecal[supermodule][column][row] = 
        calibda.GetADCchannelDecal(supermodule,column,row);
        
        fADCpedestal[supermodule][column][row] = 
        calibda.GetADCpedestal(supermodule,column,row);
        
        fTimeChannel[supermodule][column][row] = 
        calibda.GetTimeChannel(supermodule,column,row);

        fTimeChannelDecal[supermodule][column][row] = 
        calibda.GetTimeChannelDecal(supermodule,column,row);
        
      }
    }
  }
  return *this;
}

//________________________________________________________________
AliEMCALCalibData::~AliEMCALCalibData()
{
  // Destructor
}

//________________________________________________________________
void AliEMCALCalibData::Reset()
{
  // Set all pedestals to 0 and all ADC channels widths to 1
  //memset(fADCchannel ,1,12*48*24*sizeof(Float_t));
  //memset(fADCpedestal,0,12*48*24*sizeof(Float_t));

  Int_t nSMod = AliEMCALGeoParams::fgkEMCALModules; //12
  Int_t nCol  = AliEMCALGeoParams::fgkEMCALCols;    //48
  Int_t nRow  = AliEMCALGeoParams::fgkEMCALRows;    //24
  Int_t nRow2 = AliEMCALGeoParams::fgkEMCALRows/2;  //12 - Modules 11 and 12 are half modules
  for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
    if(supermodule >= 10)
      nRow = nRow2;
    for (Int_t column=0; column < nCol; column++){
      
      for (Int_t row = 0; row < nRow; row++){
        
        fADCpedestal     [supermodule][column][row]=0.;
        
        fADCchannelDecal [supermodule][column][row]=1.;
        fADCchannel      [supermodule][column][row]=1.;
        
        fTimeChannel     [supermodule][column][row]=0.;
        fTimeChannelDecal[supermodule][column][row]=0.;

      }
    }
  } 
}

//________________________________________________________________
void  AliEMCALCalibData::Print(Option_t *option) const
{
  // Print tables of pedestals and ADC channels widths
  // options are: "gain", "ped", "decal", "time", "all"
  
  Int_t nSMod = AliEMCALGeoParams::fgkEMCALModules; //12
  Int_t nCol  = AliEMCALGeoParams::fgkEMCALCols;    //48
  Int_t nRow  = AliEMCALGeoParams::fgkEMCALRows;    //24
  Int_t nRow2 = AliEMCALGeoParams::fgkEMCALRows/2;  //12 - Modules 11 and 12 are half modules
  if (strstr(option,"ped") || strstr(option,"all")) {
    printf("\n	----	Pedestal values	----\n\n");
    for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
      if(supermodule >= 10)
        nRow = nRow2;
      printf("============== Supermodule %d\n",supermodule+1);
      for (Int_t column=0; column<nCol; column++){
        for (Int_t row=0; row<nRow; row++){
          printf(" %2.4f ",fADCpedestal[supermodule][column][row]);
        }
        printf("\n");
      }
    } 
  }
  
  if (strstr(option,"gain") || strstr(option,"all")) {
    printf("\n	----	ADC channel values	----\n\n");
    for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
      if(supermodule >= 10) 
        nRow = nRow2;
      printf("============== Supermodule %d\n",supermodule+1);
      for (Int_t column=0; column<nCol; column++){
        for (Int_t row=0; row<nRow; row++){
          printf(" %2.4f ",fADCchannel[supermodule][column][row]);
        }
        printf("\n");
      }
    }   
  }
  
  if (strstr(option,"adcdecal") || strstr(option,"all")) {
    printf("\n	----	ADC decalibration channel values	----\n\n");
    for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
      if(supermodule >= 10) 
        nRow = nRow2;
      printf("============== Supermodule %d\n",supermodule+1);
      for (Int_t column=0; column<nCol; column++){
        for (Int_t row=0; row<nRow; row++){
          printf(" %2.4f ",fADCchannelDecal[supermodule][column][row]);
        }
        printf("\n");
      }
    }   
  }
  
  if (strstr(option,"time") || strstr(option,"all")) {
    printf("\n	----	time channel values	----\n\n");
    for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
      if(supermodule >= 10) 
        nRow = nRow2;
      printf("============== Supermodule %d\n",supermodule+1);
      for (Int_t column=0; column<nCol; column++){
        for (Int_t row=0; row<nRow; row++){
          printf(" %2.4f ",fTimeChannel[supermodule][column][row]);
        }
        printf("\n");
      }
    }   
  }
  
  
  if (strstr(option,"time") || strstr(option,"all")) {
    printf("\n	----	time decalibration channel values	----\n\n");
    for (Int_t supermodule=0; supermodule<nSMod; supermodule++){
      if(supermodule >= 10) 
        nRow = nRow2;
      printf("============== Supermodule %d\n",supermodule+1);
      for (Int_t column=0; column<nCol; column++){
        for (Int_t row=0; row<nRow; row++){
          printf(" %2.4f ",fTimeChannelDecal[supermodule][column][row]);
        }
        printf("\n");
      }
    }   
  }
  
  
}

//________________________________________________________________
Float_t AliEMCALCalibData::GetADCchannel(Int_t supermodule, Int_t column, Int_t row) const
{
  // Set ADC channel witdth values
  // All indexes start from 0!
  // Supermodule, column,raw should follow the ALICE convention:
  // supermodule 0:11, column 0:47, row 0:23

  return fADCchannel[supermodule][column][row];
}

//________________________________________________________________
Float_t AliEMCALCalibData::GetADCchannelDecal(Int_t supermodule, Int_t column, Int_t row) const
{
  // Set ADC channel decalibration witdth values
  // All indexes start from 0!
  // Supermodule, column,raw should follow the ALICE convention:
  // supermodule 0:11, column 0:47, row 0:23
  
  return fADCchannelDecal[supermodule][column][row];
}

//________________________________________________________________
Float_t AliEMCALCalibData::GetADCpedestal(Int_t supermodule, Int_t column, Int_t row) const
{
  // Get ADC pedestal values
 return fADCpedestal[supermodule][column][row];
}

//________________________________________________________________
Float_t AliEMCALCalibData::GetTimeChannel(Int_t supermodule, Int_t column, Int_t row) const
{
  // Set channel time witdth values
  return fTimeChannel[supermodule][column][row];
}

//________________________________________________________________
Float_t AliEMCALCalibData::GetTimeChannelDecal(Int_t supermodule, Int_t column, Int_t row) const
{
  // Set channel time witdth values
  return fTimeChannelDecal[supermodule][column][row];
}

//________________________________________________________________
void AliEMCALCalibData::SetADCchannel(Int_t supermodule, Int_t column, Int_t row, Float_t value)
{ 
  // Set ADC channel width values
  fADCchannel[supermodule][column][row] = value;
}

//________________________________________________________________
void AliEMCALCalibData::SetADCchannelDecal(Int_t supermodule, Int_t column, Int_t row, Float_t value)
{ 
  // Set ADC channel width values
  fADCchannelDecal[supermodule][column][row] = value;
}

//________________________________________________________________
void AliEMCALCalibData::SetADCpedestal(Int_t supermodule, Int_t column, Int_t row, Float_t value)
{
  // Set ADC pedestal values
  fADCpedestal[supermodule][column][row] = value;
}

//________________________________________________________________
void AliEMCALCalibData::SetTimeChannel(Int_t supermodule, Int_t column, Int_t row, Float_t value)
{
  // Set ADC pedestal values
  fTimeChannel[supermodule][column][row] = value;
}

//________________________________________________________________
void AliEMCALCalibData::SetTimeChannelDecal(Int_t supermodule, Int_t column, Int_t row, Float_t value)
{
  // Set ADC pedestal values
  fTimeChannelDecal[supermodule][column][row] = value;
}


//________________________________________________________________
void AliEMCALCalibData::Fill(const AliEMCALCalibData *cd1, const AliEMCALCalibData *cd2, Bool_t print)
{
  // Fill for (relative) recalibration to undo calibration 
  // from 1 and apply calibration from 2.

  Int_t nSMod = AliEMCALGeoParams::fgkEMCALModules; //12
  Int_t nCol  = AliEMCALGeoParams::fgkEMCALCols;    //48
  Int_t nRow  = AliEMCALGeoParams::fgkEMCALRows;    //24
  Int_t nRow2 = AliEMCALGeoParams::fgkEMCALRows/2;  //12 - Modules 11 and 12 are half modules
  for (Int_t supermodule=0; supermodule<nSMod; supermodule++) {
    if(supermodule >= 10)
      nRow = nRow2;
    for (Int_t column=0; column<nCol; column++) {
      for (Int_t row=0; row<nRow; row++){
        Double_t adc1 = cd1->GetADCchannel(supermodule, column, row);
        Double_t ped1 = cd1->GetADCpedestal(supermodule, column, row);
        Double_t adc2 = cd2->GetADCchannel(supermodule, column, row);
        Double_t ped2 = cd2->GetADCpedestal(supermodule, column, row);
        Double_t adc = adc2/adc1;
        Double_t ped = ped2-adc*ped1;
        if (print)
          printf("%d %d %d: %f %f %f %f -> %f %f\n",supermodule,column,row,adc1,ped1,adc2,ped2,adc,ped);
        fADCchannel [supermodule][column][row]=adc;
        fADCpedestal[supermodule][column][row]=ped;
      }
    }
  } 
}