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

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  This class creates and fills the monitor histograms for V0s              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


#include "AliMonitorV0s.h"
#include "AliMonitorHisto.h"
#include "AliESDEvent.h"
#include "AliESDv0.h"
#include "AliLog.h"
#include <TFolder.h>
#include <TPDGCode.h>
#include <TMath.h>


ClassImp(AliMonitorV0s) 


//_____________________________________________________________________________
AliMonitorV0s::AliMonitorV0s():
  AliMonitor(),
  fRadius(NULL),
  fMassK0(NULL),
  fMassLambda(NULL),
  fMassAntiLambda(NULL)
{
// create a monitor object for V0s

}

//_____________________________________________________________________________
void AliMonitorV0s::CreateHistos(TFolder* folder)
{
// create the V0s monitor histograms

  fFolder = folder->AddFolder("V0s", "V0s");

  fRadius = CreateHisto1("Radius", "radius of V0 vertices", 
			 90, 0., 3., "r_{xy} [cm]", 
			 "#Delta N/N", AliMonitorHisto::kNormEvents);

  fMassK0 = CreateHisto1("MassK0", "invariant mass of K^{0} candidates", 
			 50, 0.4, 0.6, "m_{#pi^{+}#pi^{-}} [GeV/c^{2}]", 
			 "#Delta N/N", AliMonitorHisto::kNormEvents);

  fMassLambda = CreateHisto1("MassLambda", 
			     "invariant mass of #Lambda candidates", 
			     50, 1.0, 1.2, "m_{p#pi^{-}} [GeV/c^{2}]", 
			     "#Delta N/N", AliMonitorHisto::kNormEvents);

  fMassAntiLambda = CreateHisto1("MassAntiLambda", 
				 "invariant mass of #bar{#Lambda} candidates", 
				 50, 1.0, 1.2, 
				 "m_{#bar{p}#pi^{+}} [GeV/c^{2}]", 
				 "#Delta N/N", AliMonitorHisto::kNormEvents);
}


//_____________________________________________________________________________
void AliMonitorV0s::FillHistos(AliRunLoader* /*runLoader*/, 
			       AliRawReader*, AliESDEvent* esd)
{
// fill the V0s monitor histogrms

  for (Int_t i = 0; i < esd->GetNumberOfV0s(); i++) {
    AliESDv0* v0 = esd->GetV0(i);
    if (!v0) continue;
    Double_t x, y, z;
    v0->GetXYZ(x, y, z);
    fRadius->Fill(TMath::Sqrt(x*x + y*y));
    v0->ChangeMassHypothesis(kK0Short); 
    fMassK0->Fill(v0->GetEffMass());
    v0->ChangeMassHypothesis(kLambda0); 
    fMassLambda->Fill(v0->GetEffMass());
    v0->ChangeMassHypothesis(kLambda0Bar); 
    fMassAntiLambda->Fill(v0->GetEffMass());
  }
}