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

/////////////////////////////////////////////////////////////
//
// base class for AOD containers
//
/////////////////////////////////////////////////////////////

#include <TROOT.h>
#include <TParticle.h>
#include <TClass.h>
#include <TString.h>

#include "AliAOD.h"
#include "AliAODParticle.h"
#include "AliTrackPoints.h"

ClassImp(AliAOD)

AliAOD::AliAOD():
 fParticles(0x0),
 fIsRandomized(kFALSE),
 fPrimaryVertexX(0.0),
 fPrimaryVertexY(0.0),
 fPrimaryVertexZ(0.0),
 fParticleClass(0x0)
{
 //ctor
// Info("AliAOD()","Entered");
// SetOwner(kTRUE);
// Info("AliAOD()","Exited");
}
/**************************************************************************/

AliAOD::AliAOD(const AliAOD& in):
 TObject(in),
 fParticles((TClonesArray*)in.fParticles->Clone()),
  fIsRandomized(in.fIsRandomized),
  fPrimaryVertexX(fPrimaryVertexX),
  fPrimaryVertexY(in.fPrimaryVertexY),
  fPrimaryVertexZ(in.fPrimaryVertexZ),
  fParticleClass(in.fParticleClass)
{
//copy constructor
}
/**************************************************************************/

AliAOD& AliAOD::operator=(const AliAOD& in)
{
//assigment operator  

  if (this == &in ) return *this;
  
  delete fParticles;
  fParticles = (TClonesArray*)in.fParticles->Clone();
  fIsRandomized = in.fIsRandomized ;
  fPrimaryVertexX = in.fPrimaryVertexX ;
  fPrimaryVertexY = in.fPrimaryVertexY ;
  fPrimaryVertexZ = in.fPrimaryVertexZ ;
  fParticleClass = in.fParticleClass ; //althought it is pointer, this points to object in class list of gROOT
  return *this;
}
/**************************************************************************/

AliAOD::~AliAOD()
{
  //Destructor
  //fParticleClass does not belong to AliAOD -> Do not delete it
  delete fParticles;
  
}
/**************************************************************************/

void AliAOD::CopyData(AliAOD* aod)
{
 //Copys all data from aod, but leaves local type of particles
 if (aod == 0x0) return;
 if (aod == this) return;
 
 AliAOD& in = *this;
 
 fIsRandomized = in.fIsRandomized ;
 fPrimaryVertexX = in.fPrimaryVertexX ;
 fPrimaryVertexY = in.fPrimaryVertexY ;
 fPrimaryVertexZ = in.fPrimaryVertexZ ;
 fParticleClass = in.fParticleClass ; //althought it is pointer, this points to object in class list of gROOT

 
 if (in.fParticles == 0x0)
  {//if in obj has null fParticles we delete ours
    delete fParticles;
    fParticles = 0x0;
  }
 else
  { 
    if (fParticles)
     { //if ours particles were already created
       if (fParticles->GetClass() != in.fParticles->GetClass())
        {//if in obj has 
          delete fParticles;
          fParticles = (TClonesArray*)in.fParticles->Clone();
        }
       else
        {
         //it should be faster than cloning
          Int_t inentr = in.fParticles->GetEntriesFast();
          Int_t curentr = fParticles->GetEntriesFast();

          TClonesArray& arr = *fParticles;

          //we have to take care about different sizes of arrays
          if ( curentr < inentr )
           {
             for (Int_t i = 0; i < curentr; i++)
              {
                TObject& inobj = *(in.fParticles->At(i));
                TObject& obj = *(fParticles->At(i));
                obj = inobj;
              }
             
             TClass* partclass = GetParticleClass();
             if (partclass == 0x0)
              {
                Fatal("CopyData","Can not get particle class");
                return;//pro forma
              }
              
             for (Int_t i = curentr; i < inentr; i++)
              {
                TObject& inobj = *(in.fParticles->At(i));
                TObject& obj =  *((TObject*)(partclass->New(arr[i])));
                obj = inobj;
              }
           }
          else 
           {
             for (Int_t i = 0; i < inentr; i++)
              {
                TObject& inobj = *(in.fParticles->At(i));
                TObject& obj = *(fParticles->At(i));
                obj = inobj;
              }
             
             for (Int_t i = curentr ; i >= inentr ; i--)
              {
                fParticles->RemoveAt(i);
              }
           }
        } 
     }
    else
     {
       fParticles = (TClonesArray*)in.fParticles->Clone();
     } 
  } 
 
}
/**************************************************************************/

void AliAOD::SetParticleClassName(const char* classname)
{
//Sets type of particle that is going to be stored 
  if (gROOT == 0x0) Fatal("SetParticleClassName","ROOT System not initialized");
  TClass* pclass = gROOT->GetClass(classname);
  if ( pclass == 0x0 )
   {
     Error("SetParticleClass","Can not get TClass for class named %s",classname);
     return;
   }
  SetParticleClass(pclass);
}
/**************************************************************************/

void AliAOD::SetParticleClass(TClass* pclass)
{
//Sets type of particle that is going to be stored 

  if ( pclass == 0x0 )
   {
     Error("SetParticleClass","Parameter is NULL.");
     return;
   }
   
  if ( pclass->InheritsFrom("AliVAODParticle") == kFALSE )
   {
     Error("SetParticleClass","Class named %s does not inherit from AliVAODParticle",pclass->GetName());
     return;
   }
  if (pclass != fParticleClass)
   {
     fParticleClass = pclass;
     if (fParticleClass) delete fParticles;
     fParticles = new TClonesArray(fParticleClass);
   }
}
/**************************************************************************/
TClass* AliAOD::GetParticleClass()
{
//returns TClass of particle class
  if (fParticleClass) return fParticleClass;
  
  if (fParticles == 0x0) return 0x0;
  
  fParticleClass = fParticles->GetClass();
  return fParticleClass;
}

/**************************************************************************/

void  AliAOD::AddParticle(TParticle* part, Int_t idx)
{
  //Adds TParticle to event
  if (part == 0x0) 
   {
     Error("AddParticle(TParticle*,Int_t)","pointer to particle is NULL");
     return;
   }

  if (fParticles == 0x0) SetParticleClassName("AliAODParticle");
  AddParticle( new AliAODParticle(*part,idx) );
}
/**************************************************************************/

void  AliAOD::AddParticle(AliVAODParticle* particle)
{
 //add particle to AOD
 //MAKES ITS OWN COPY OF THE PARTICLE!!! (AOD is not going to keep and delete input pointer)
 
  if (fParticles == 0x0) SetParticleClassName("AliAODParticle");

  Int_t idx = fParticles->GetLast() + 1;
  TClonesArray& arr = *fParticles;

  TClass* partclass = GetParticleClass();
  if (partclass == 0x0)
   {
     Error("AddParticle(AliVAODParticle*)","Can not get particle class");
     return;
   }
  
  AliVAODParticle* pp = (AliVAODParticle*)partclass->New(arr[idx]);
  pp->operator=(*particle);
  
}
/**************************************************************************/

void  AliAOD::AddParticle(Int_t pdg, Int_t idx,
                          Double_t px, Double_t py, Double_t pz, Double_t etot,
                          Double_t vx, Double_t vy, Double_t vz, Double_t time)
{
  //adds particle to event (standard AOD class)

  if (fParticles == 0x0) SetParticleClassName("AliAODParticle");

  Int_t newpartidx = fParticles->GetLast() + 1;
  TClonesArray& arr = *fParticles;

  TClass* partclass = GetParticleClass();
  if (partclass == 0x0)
   {
     Error("AddParticle(Int_t,...)","Can not get particle class");
     return;
   }

  AliVAODParticle* p =  (AliVAODParticle*)partclass->New(arr[newpartidx]);
  
  p->SetPdgCode(pdg);
  p->SetUID(idx);
  p->SetMomentum(px,py,pz,etot);
  p->SetProductionVertex(vx,vy,vz,time);
  
}
/**************************************************************************/

void AliAOD::SwapParticles(Int_t i, Int_t j)
{
//swaps particles positions; used by AliHBTEvent::Blend
  if ( (i<0) || (i>=GetNumberOfParticles()) ) return;
  if ( (j<0) || (j>=GetNumberOfParticles()) ) return;
  

  TClass* partclass = GetParticleClass();
  if (partclass == 0x0)
   {
     Error("SwapParticles","Can not get particle class");
     return;
   }
    
  AliVAODParticle* tmpobj = (AliVAODParticle*)partclass->New();
  AliVAODParticle& tmp = *tmpobj;
  AliVAODParticle& first = *(GetParticle(i));
  AliVAODParticle& second = *(GetParticle(j));
  
  tmp = first;
  first = second;
  second = tmp;
  delete tmpobj;
}
/**************************************************************************/

void  AliAOD::Reset()
{
  //deletes all particles from the event
   if (fParticles) fParticles->Clear("C");
   
   fIsRandomized = kFALSE;
}
/**************************************************************************/

void AliAOD::GetPrimaryVertex(Double_t&x, Double_t&y, Double_t&z)
{
//returns positions of the primary vertex
  x = fPrimaryVertexX;
  y = fPrimaryVertexY;
  z = fPrimaryVertexZ;
}
/**************************************************************************/

void AliAOD::SetPrimaryVertex(Double_t x, Double_t y, Double_t z)
{
//Sets positions of the primary vertex 
  fPrimaryVertexX = x;
  fPrimaryVertexY = y;
  fPrimaryVertexZ = z;
}
/**************************************************************************/

Int_t AliAOD::GetNumberOfCharged(Double_t etamin, Double_t etamax) const
{
  //reurns number of charged particles within given pseudorapidity range
  Int_t n = 0;
  Int_t npart = GetNumberOfParticles();
  for (Int_t i = 0; i < npart; i++)
   {
     AliVAODParticle* p = GetParticle(i);
     Double_t eta = p->Eta();
     if ( (eta < etamin) || (eta > etamax) ) continue;
     if (p->Charge() != 0.0) n++;
   }
  return n;
}
/**************************************************************************/

void AliAOD::Move(Double_t x, Double_t y, Double_t z)
{
 //moves all spacial coordinates about this vector
 // vertex
 // track points
 // and whatever will be added to AOD and AOD particles that is a space coordinate

  fPrimaryVertexX += x;
  fPrimaryVertexY += y;
  fPrimaryVertexZ += z;

  Int_t npart = GetNumberOfParticles();
  for (Int_t i = 0; i < npart; i++)
   {
     AliVAODParticle* p = GetParticle(i);
     AliTrackPoints* tp  = p->GetTPCTrackPoints();
     if (tp) tp->Move(x,y,z);
     tp  = p->GetITSTrackPoints();
     if (tp) tp->Move(x,y,z);
   }
}
/**************************************************************************/

void AliAOD::Print(const Option_t* /*option*/) const
{
  //Prints AOD
  TString ts;
  TString msg("\n");
  msg+="Particle Class: ";
  if (fParticleClass)
   {
     msg+=fParticleClass->GetName();
   }
  else
   {
     msg+="Not specified yet";
   } 
  msg += "\n";
  msg += "Vertex position X: ";
  msg += fPrimaryVertexX;
  msg += " Y:" ;
  msg += fPrimaryVertexY;
  msg += " Z:";
  msg += fPrimaryVertexZ;
  msg += "\n";
  
  msg += "Randomized: ";
  msg += fIsRandomized;
  msg += "\n";
  
  Info("Print","%s",msg.Data());
  
  Int_t npart = GetNumberOfParticles();
  Info("Print","Npart: %d",npart);
  for (Int_t i = 0; i < npart; i++)
   {
     Info("Print","Getting particle %d",i);
     AliVAODParticle* p = GetParticle(i);
     Info("Print","Printing particle %d, address %#x",i,p);
     p->Dump();
     p->Print();
     Info("Print","particle %d printed",i);
   }
}

void AliAOD::SetOwner(Bool_t /*owner*/)
{
//Sets the ownership of particles: if particles should be also deleted if AOD is deleted/reseted
//Since fParticles is Clones and not Object Array, it is always the owner and this method does not have sense
 
 MayNotUse("SetOwner");
 //if fParticles->SetOwner(owner);
 
}