AliRsnCutManager:

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnAnalysisTask.cxx

//
// Class AliRsnAnalysisTask
//
// Virtual Class derivated from AliRsnVAnalysisTask which will be base class
// for all RSN SE tasks
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#include <Riostream.h>
#include <TList.h>
#include "AliESDEvent.h"
#include "AliMCEvent.h"
#include "AliAODEvent.h"

#include "AliRsnCutSet.h"
#include "AliRsnVATProcessInfo.h"
#include "AliRsnAnalysisTask.h"

ClassImp(AliRsnAnalysisTask)

//_____________________________________________________________________________
AliRsnAnalysisTask::AliRsnAnalysisTask(const char *name, Bool_t useKine) :
  AliRsnVAnalysisTask(name, useKine),
  fRsnAnalysisManager(),
  fEventCuts("eventCuts", AliRsnCut::kEvent),
  fOutList(0x0),
  fZeroEventPercentWarning(100),
  fUseZeroEventWarning(kTRUE)
{
//
// Default constructor.
// Defines another output slot for histograms/ntuples
//

  DefineOutput(2, TList::Class());
}

//_____________________________________________________________________________
AliRsnAnalysisTask::AliRsnAnalysisTask(const AliRsnAnalysisTask& copy) :
  AliRsnVAnalysisTask(copy),
  fRsnAnalysisManager(copy.fRsnAnalysisManager),
  fEventCuts(copy.fEventCuts),
  fOutList(0x0),
  fZeroEventPercentWarning(copy.fZeroEventPercentWarning),
  fUseZeroEventWarning(copy.fUseZeroEventWarning)
{
//
// Copy constructor.
//
}

//_____________________________________________________________________________
AliRsnAnalysisTask& AliRsnAnalysisTask::operator=(const AliRsnAnalysisTask& copy)
{
//
// Assigment operator.
//

  AliRsnVAnalysisTask::operator=(copy);
  
  fRsnAnalysisManager = copy.fRsnAnalysisManager;
  fEventCuts = copy.fEventCuts;
  if (fOutList) fOutList->Clear();
  fZeroEventPercentWarning = copy.fZeroEventPercentWarning;
  fUseZeroEventWarning = copy.fUseZeroEventWarning;
  
  return (*this);
}

//_____________________________________________________________________________
void AliRsnAnalysisTask::RsnUserCreateOutputObjects()
{
//
// Creation of output objects.
// These are created through the utility methods in the analysis manager,
// which asks all the AliRsnPair objects to initialize their output which
// is then linked to the TList data member of this, which will contain all the output.
//

  if (!fOutList) fOutList = new TList;
  fOutList->Clear();
  
  fRsnAnalysisManager.InitAllPairs(fOutList);

  PostData(2, fOutList);
}

//_____________________________________________________________________________
void AliRsnAnalysisTask::RsnUserExec(Option_t*)
{
//
// Execution of the analysis task.
// Recovers the input event and processes it with all included pair objects,
// using 'reconstructed' or 'MonteCarlo' functions depending on MC-only flag.
//
	if (IsMixing()) return;
  
	if (fMCOnly) 
    fRsnAnalysisManager.ProcessAllPairsMC();
  else 
    fRsnAnalysisManager.ProcessAllPairs();

	PostData(2, fOutList);
}

void AliRsnAnalysisTask::RsnUserExecMix(Option_t* /*opt*/)
{
  if (!IsMixing()) return;
  AliDebug(AliLog::kDebug,Form("RSN Mixing %lld %d [%lld,%lld] %d",fMixedEH->CurrentEntry(),fMixedEH->NumberMixed(),fMixedEH->CurrentEntryMain(),fMixedEH->CurrentEntryMix(),fMixedEH->CurrentBinIndex()));
  
  // the virtual class has already sorted tracks in the PID index
  // so we need here just to call the execution of analysis
  if (!fMCOnly) 
		fRsnAnalysisManager.ProcessAllPairs();
  else 
		fRsnAnalysisManager.ProcessAllPairsMC();
}

//_____________________________________________________________________________
void AliRsnAnalysisTask::RsnTerminate(Option_t*)
{
//
// Termination.
// Could be added some monitor histograms here.
//
}

//______________________________________________________________________________
Bool_t AliRsnAnalysisTask::EventProcess()
{
//
// Customized event pre-processing.
// First checks if the current event passes all cuts,
// and if it does, updates the informations and then
// call the operations which are already defined in the
// omonyme function in mother class
//

  // initially, an event is expected to be bad
  fTaskInfo.SetEventUsed(kFALSE);
  
  // check #1: number of tracks in event (reject empty events)
  Int_t    ntracks = fRsnEvent[0].GetMultiplicity();
  Double_t zeroEventPercent = 0.0;
  if (ntracks < 1) 
  {
    // if using the checker for amount of empty events, update it
    if (fUseZeroEventWarning)
    {
      TH1I *hist = (TH1I*)fInfoList->FindObject(fTaskInfo.GetEventHistogramName());
      if (hist)
      {
        if (hist->Integral() > 1) zeroEventPercent = (Double_t)hist->GetBinContent(1) / hist->Integral() * 100;
        if ((zeroEventPercent > fZeroEventPercentWarning) && (fEntry > 100))
          AliWarning(Form("%3.2f%% Events are with zero tracks (CurrentEvent=%d)!!!", zeroEventPercent, fEntry));
      }
    }
    
    // empty events are rejected by default
    fTaskInfo.SetEventUsed(kFALSE);
    AliDebug(AliLog::kDebug, "Empty event. Skipping...");
    return kFALSE;
  }

  // check the event cuts and update the info data accordingly
  // events not passing the cuts must be rejected
  if (!fEventCuts.IsSelected(&fRsnEvent[0]))
  {
    fTaskInfo.SetEventUsed(kFALSE);
    return kFALSE;
  }
  
  // if we reach this point, cuts were passed;
  // then additional operations can be done
  
  // find leading particle (without any PID/momentum restriction)
  fRsnEvent[0].SelectLeadingParticle(0);
  
  // final return value is positive
  // but call the mother class method which updates info object
  fTaskInfo.SetEventUsed(kTRUE);
  return AliRsnVAnalysisTask::EventProcess();
}