/**************************************************************************
 * This file is property of and copyright by the ALICE HLT Project        * 
 * All rights reserved.                                                   *
 *                                                                        *
 * Primary Authors: Oystein Djuvsland                                     *
 *                                                                        *
 * 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.                  *
 **************************************************************************/


/** 
 * @file   AliHLTPHOSESDMaker.cxx
 * @author Oystein Djuvsland
 * @date 
 * @brief  ESD maker for PHOS HLT 
 */

// see header file for class documentation
// or
// refer to README to build package
// or 
// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt

#include "AliHLTPHOSESDMaker.h"
#include "AliHLTPHOSCaloClusterContainerStruct.h"
#include "AliHLTPHOSBase.h"
#include "AliHLTPHOSCaloClusterDataStruct.h"
#include "AliESDEvent.h"
#include "AliESDCaloCluster.h"
#include "TClonesArray.h"

ClassImp(AliHLTPHOSESDMaker);

AliHLTPHOSESDMaker::AliHLTPHOSESDMaker() : 
  AliHLTPHOSBase(),
  fNCaloClusters(0),
  fCaloClustersPtr(0),
  fESDEventPtr(0),
  fCaloClusterContainerPtr(0)
{
  //See header file for documentation
  fCaloClustersPtr = new TClonesArray("AliESDCaloCluster", 0);
}

AliHLTPHOSESDMaker::~AliHLTPHOSESDMaker()
{
  //See header file for documentation
  if(fCaloClustersPtr)
    {
      //fCaloClustersPtr->Delete();
      //fCaloClustersPtr = 0;
    }
}

Int_t
AliHLTPHOSESDMaker::FillESDCaloClusters()
{
  //See header file for documentation

  AliESDCaloCluster *caloCluster = 0;
  AliHLTPHOSCaloClusterDataStruct* caloClusterStruct = 0;

  for(UInt_t i = 0; i < fCaloClusterContainerPtr->fNCaloClusters; i++)
    {
      caloCluster = (AliESDCaloCluster*)fCaloClustersPtr->At(i + fNCaloClusters);
      caloClusterStruct = &(fCaloClusterContainerPtr->fCaloClusterArray[i]);
      caloCluster->SetID(caloClusterStruct->fID);
      caloCluster->SetClusterType(caloClusterStruct->fClusterType);
      caloCluster->SetPosition((Float_t*)&caloClusterStruct->fGlobalPos[0]);
      caloCluster->SetE(caloClusterStruct->fEnergy);
      caloCluster->SetClusterDisp(caloClusterStruct->fDispersion);
      caloCluster->SetClusterChi2(caloClusterStruct->fFitQuality);
      caloCluster->SetPid((Float_t*)&caloClusterStruct->fPID[0]);
      caloCluster->SetM20(caloClusterStruct->fM20);
      caloCluster->SetM02(caloClusterStruct->fM02);
      // PT   caloCluster->SetM11(caloClusterStruct->fM11);
      caloCluster->SetNExMax(caloClusterStruct->fNExMax);
      caloCluster->SetEmcCpvDistance(caloClusterStruct->fEmcCpvDistance);
      caloCluster->SetDistanceToBadChannel(caloClusterStruct->fDistToBadChannel);
      caloCluster->SetNCells(caloClusterStruct->fNCells);
      caloCluster->SetCellsAbsId(caloClusterStruct->fCellsAbsId);
      caloCluster->SetCellsAmplitudeFraction(caloClusterStruct->fCellsAmpFraction);
      fNCaloClusters++;
    }

  return 0;
}

Int_t
AliHLTPHOSESDMaker::FillESDEvent()
{
  //See header file for documentation
  
  AliESDCaloCluster *caloCluster = 0;
  AliHLTPHOSCaloClusterDataStruct* caloClusterStruct = 0;
  for(UInt_t i = 0; i < fCaloClusterContainerPtr->fNCaloClusters; i++)
    {
      //      caloCluster = (AliESDCaloCluster*)fCaloClustersPtr->New(i + fNCaloClusters);
      caloCluster = (AliESDCaloCluster*)fCaloClustersPtr->New(i);
      caloClusterStruct = &(fCaloClusterContainerPtr->fCaloClusterArray[i]);

      caloCluster->SetID(caloClusterStruct->fID);
      caloCluster->SetClusterType(caloClusterStruct->fClusterType);
      caloCluster->SetPosition((Float_t*)&caloClusterStruct->fGlobalPos[0]);
      caloCluster->SetE(caloClusterStruct->fEnergy);
      caloCluster->SetClusterDisp(caloClusterStruct->fDispersion);
      caloCluster->SetClusterChi2(caloClusterStruct->fFitQuality);
      caloCluster->SetPid((Float_t*)&caloClusterStruct->fPID[0]);
      caloCluster->SetM20(caloClusterStruct->fM20);
      caloCluster->SetM02(caloClusterStruct->fM02);
      //     caloCluster->SetM11(caloClusterStruct->fM11);
      caloCluster->SetNExMax(caloClusterStruct->fNExMax);
      caloCluster->SetEmcCpvDistance(caloClusterStruct->fEmcCpvDistance);
      caloCluster->SetDistanceToBadChannel(caloClusterStruct->fDistToBadChannel);
      caloCluster->SetNCells(caloClusterStruct->fNCells);
      caloCluster->SetCellsAbsId(caloClusterStruct->fCellsAbsId);
      caloCluster->SetCellsAmplitudeFraction(caloClusterStruct->fCellsAmpFraction);
      fESDEventPtr->AddCaloCluster(caloCluster);
      fNCaloClusters++;  
    }
  
  return 0;
}
Int_t
AliHLTPHOSESDMaker::FillESDEvent(AliHLTPHOSCaloClusterContainerStruct* caloClusterContainerPtr)
{
  //See header file for documentation

  caloClusterContainerPtr--;
  caloClusterContainerPtr++;
  
  //  AliESDCaloCluster *caloCluster = 0;
  //  AliHLTPHOSCaloClusterDataStruct* caloClusterStruct = 0;
  
  for(UInt_t i = 0; i < 1/*caloClusterContainerPtr->fNCaloClusters*/; i++)
    {
      //      caloCluster = (AliESDCaloCluster*)fCaloClustersPtr->New(i + fNCaloClusters);
      //caloCluster = (AliESDCaloCluster*)fCaloClustersPtr->New(i + fNCaloClusters);
      /*      caloClusterStruct = &(caloClusterContainerPtr->fCaloClusterArray[i]);
      caloCluster->SetID(caloClusterStruct->fID);
      caloCluster->SetClusterType(caloClusterStruct->fClusterType);
      caloCluster->SetPosition((Float_t*)&caloClusterStruct->fGlobalPos[0]);
      caloCluster->SetE(caloClusterStruct->fEnergy);
      caloCluster->SetClusterDisp(caloClusterStruct->fDispersion);
      caloCluster->SetClusterChi2(caloClusterStruct->fFitQuality);
      caloCluster->SetPid((Float_t*)&caloClusterStruct->fPID[0]);
      caloCluster->SetM20(caloClusterStruct->fM20);
      caloCluster->SetM02(caloClusterStruct->fM02);
      caloCluster->SetNExMax(caloClusterStruct->fNExMax);
      caloCluster->SetEmcCpvDistance(caloClusterStruct->fEmcCpvDistance);
      caloCluster->SetDistanceToBadChannel(caloClusterStruct->fDistToBadChannel);
      caloCluster->SetNCells(caloClusterStruct->fNCells);
      caloCluster->SetCellsAbsId(caloClusterStruct->fCellsAbsId);
      caloCluster->SetCellsAmplitudeFraction(caloClusterStruct->fCellsAmpFraction);
      fESDEventPtr->AddCaloCluster(caloCluster);
       */
      fNCaloClusters++;  
    }
  
  return 0;
}

void 
AliHLTPHOSESDMaker::ResetESD()
{
  fNCaloClusters = 0;
  // fCaloClustersPtr->Delete();
}