Adding AOD task for FMDEventPlaneFinder

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / AliCentralMCMultiplicityTask.cxx

//====================================================================
// 
// Base class for classes that calculate the multiplicity in the
// central region event-by-event
// 
// Inputs: 
//   - AliESDEvent 
//
// Outputs: 
//   - AliAODCentralMult 
// 
// Histograms 
//   
// Corrections used 
#include "AliCentralMCMultiplicityTask.h"
#include "AliForwardCorrectionManager.h"
#include "AliForwardUtil.h"
#include "AliLog.h"
#include "AliAODHandler.h"
#include "AliInputEventHandler.h"
#include "AliESDInputHandler.h"
#include "AliAnalysisManager.h"
#include "AliESDEvent.h"
#include "AliMultiplicity.h"
#include "AliFMDEventInspector.h"
#include <AliMCEvent.h>
#include <AliTrackReference.h>
#include <AliStack.h>
#include <TROOT.h>
#include <TH1D.h>
#include <TH2D.h>
#include <TH3D.h>
#include <TFile.h>
#include <TError.h>
#include <iostream>
#include <iomanip>

//====================================================================
AliCentralMCMultiplicityTask::AliCentralMCMultiplicityTask(const char* name) 
  : AliCentralMultiplicityTask(name),
    fTrackDensity(name),
    fAODMCCentral(kTRUE)
{
  // 
  // Constructor 
  //   
}
//____________________________________________________________________
AliCentralMCMultiplicityTask::AliCentralMCMultiplicityTask() 
  : AliCentralMultiplicityTask(),
    fTrackDensity(),
    fAODMCCentral(kTRUE)
{
  // 
  // Constructor 
  // 
}
//____________________________________________________________________
AliCentralMCMultiplicityTask::AliCentralMCMultiplicityTask(const AliCentralMCMultiplicityTask& o)
  : AliCentralMultiplicityTask(o),
    fTrackDensity(o.fTrackDensity),
    fAODMCCentral(o.fAODMCCentral)
{
  //
  // Copy constructor 
  // 
}
//____________________________________________________________________
AliCentralMCMultiplicityTask&
AliCentralMCMultiplicityTask::operator=(const AliCentralMCMultiplicityTask& o)
{
  // 
  // Assignment operator 
  //
  AliCentralMultiplicityTask::operator=(o);
  fAODMCCentral     = o.fAODMCCentral;
  fTrackDensity     = o.fTrackDensity;
  return *this;
}
//____________________________________________________________________
void AliCentralMCMultiplicityTask::UserCreateOutputObjects() 
{
  // 
  // Create output objects 
  // 
  //
  AliCentralMultiplicityTask::UserCreateOutputObjects();

  AliAnalysisManager* am = AliAnalysisManager::GetAnalysisManager();
  AliAODHandler*      ah = 
    dynamic_cast<AliAODHandler*>(am->GetOutputEventHandler());
  if (!ah) AliFatal("No AOD output handler set in analysis manager");
  
  
  TObject* obj = &fAODMCCentral;
  ah->AddBranch("AliAODCentralMult", &obj);

  
  fTrackDensity.DefineOutput(fList);

}
//____________________________________________________________________
void AliCentralMCMultiplicityTask::UserExec(Option_t* option) 
{
  // 
  // Process each event 
  // 
  // Parameters:
  //    option Not used
  //  
  fAODMCCentral.Clear("");
  
  // Call base class 
  AliCentralMultiplicityTask::UserExec(option);

  // check if we need this event 
  AliAnalysisManager* am = AliAnalysisManager::GetAnalysisManager();
  AliAODHandler*      ah = 
    dynamic_cast<AliAODHandler*>(am->GetOutputEventHandler());
  if (!ah)  
    AliFatal("No AOD output handler set in analysis manager");

  // if base class did not want this event, then neither to we 
  if (!ah->GetFillAOD() || fIvz <= 0) return;
  
  const AliMCEvent*  mcEvent = MCEvent();
  TH2D&              hist    = fAODMCCentral.GetHistogram();

  Double_t vz = GetManager().GetSecMap()->GetVertexAxis().GetBinCenter(fIvz);

  fTrackDensity.Calculate(*mcEvent, vz, hist, NULL);

  CorrectData(hist, fIvz);

}

//____________________________________________________________________
void AliCentralMCMultiplicityTask::Terminate(Option_t* option) 
{
  // 
  // End of job
  // 
  // Parameters:
  //    option Not used 
  //
  AliCentralMultiplicityTask::Terminate(option);
}
//____________________________________________________________________
void
AliCentralMCMultiplicityTask::Print(Option_t* option) const
{
  // 
  // Print information 
  // 
  // Parameters:
  //    option Not used
  //
  AliCentralMultiplicityTask::Print(option);
  gROOT->IncreaseDirLevel();
  fTrackDensity.Print(option);
  gROOT->DecreaseDirLevel();
}
//
// EOF
//