fix for Coverity (B.Hippolyte)

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

//
// Class AliRsnVAnalysisTaskSE
//
// Virtual Class derivated from AliAnalysisTaskSE 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 "AliESDEvent.h"
#include "AliMCEvent.h"
#include "AliAODEvent.h"
#include "AliRsnEvent.h"
#include "AliRsnTarget.h"

#include "AliRsnVAnalysisTaskSE.h"

ClassImp(AliRsnVAnalysisTaskSE)

//_____________________________________________________________________________
AliRsnVAnalysisTaskSE::AliRsnVAnalysisTaskSE
(const char *name, Bool_t mcOnly) :
   AliAnalysisTaskSE(name),
   fLogType(AliLog::kInfo),
   fLogClassesString(""),
   fESDEvent(0x0),
   fMCEvent(0x0),
   fAODEventIn(0x0),
   fAODEventOut(0x0),
   fMCOnly(mcOnly),
   fRsnEvent(),
   fInfoList(0x0),
   fTaskInfo(name)
{
//
// Default constructor.
// Define the output slot for histograms.
//

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

//_____________________________________________________________________________
AliRsnVAnalysisTaskSE::AliRsnVAnalysisTaskSE(const AliRsnVAnalysisTaskSE& copy) :
   AliAnalysisTaskSE(copy),
   fLogType(copy.fLogType),
   fLogClassesString(copy.fLogClassesString),
   fESDEvent(copy.fESDEvent),
   fMCEvent(copy.fMCEvent),
   fAODEventIn(copy.fAODEventIn),
   fAODEventOut(copy.fAODEventOut),
   fMCOnly(copy.fMCOnly),
   fRsnEvent(),
   fInfoList(0x0),
   fTaskInfo(copy.fTaskInfo)
{
//
// Copy constructor.
// Defined for coding conventions compliance but never used.
//
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::LocalInit()
{
//
// Local initialization.
// Defines the debug message level and calls the mother class LocalInit().
//

   AliAnalysisTaskSE::LocalInit();
   SetDebugForAllClasses();
}

//_____________________________________________________________________________
Bool_t AliRsnVAnalysisTaskSE::UserNotify()
{
//
// Calls the mother class Notify()
//

   return AliAnalysisTaskSE::UserNotify();
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::ConnectInputData(Option_t *opt)
{
//
// Connect input data, which consist in initializing properly
// the pointer to the input event, which is dynamically casted
// to all available types, and this allows to know its type.
// Calls also the mother class omonyme method.
//

   AliAnalysisTaskSE::ConnectInputData(opt);

   // get AliESDEvent and, if successful
   // retrieve the corresponding MC if exists
   fESDEvent = dynamic_cast<AliESDEvent *>(fInputEvent);
   if (fESDEvent) {
      fMCEvent = (AliMCEvent*) MCEvent();
      AliInfo(Form("Input event is of type ESD   (%p)", fESDEvent));
      if (fMCEvent) AliInfo(Form("Input has an associated MC (%p)", fMCEvent));
   }

   // get AliAODEvent from input and, if successful
   // it will contain both the reconstructed and MC informations
   fAODEventIn = dynamic_cast<AliAODEvent *>(fInputEvent);
   if (fAODEventIn) {
      AliInfo(Form("Input event if of type native AOD (%p)", fAODEventIn));
   }

   // get AliAODEvent from output of previous task
   fAODEventOut = dynamic_cast<AliAODEvent *>(AODEvent());
   if (fAODEventOut) {
      AliInfo(Form("Input event if of type produced AOD from previous step (%p)", fAODEventOut));
   }
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::UserCreateOutputObjects()
{
//
// Creates and links to task all output objects.
// Does explicitly the initialization for the event info class,
// and then calls the customized function which must be overloaded
// in the applications of this base class.
//

   SetDebugForAllClasses();

   // set event info outputs
   fInfoList = new TList();
   fInfoList->SetOwner();
   fTaskInfo.GenerateInfoList(fInfoList);

   // create customized outputs
   RsnUserCreateOutputObjects();

   PostData(1, fInfoList);
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::UserExec(Option_t* opt)
{
//
// Prepares for execution, setting the correct pointers of the
// RSN package event interface, which will point to the not NULL
// objects obtained from dynamic-casts called in ConnectInputData().
//

   if (fMCOnly && fMCEvent) {
      fRsnEvent.SetRef(fMCEvent);
      fRsnEvent.SetRefMC(fMCEvent);
   } else if (fESDEvent) {
      fRsnEvent.SetRef(fESDEvent);
      fRsnEvent.SetRefMC(fMCEvent);
   } else if (fAODEventOut) {
      fRsnEvent.SetRef(fAODEventOut);
      fRsnEvent.SetRefMC(fAODEventOut);
   } else if (fAODEventIn) {
      fRsnEvent.SetRef(fAODEventIn);
      fRsnEvent.SetRefMC(fAODEventIn);
   } else {
      AliError("Unknown input event format. Skipping");
      return;
   }

   // since this class is for single-event analysis
   // both static pointers of AliRsnEvent class
   // will point to the same unique datamember
   AliRsnEvent::SetCurrentEvent1(&fRsnEvent, fEntry);
   AliRsnEvent::SetCurrentEvent2(&fRsnEvent, fEntry);
   AliRsnTarget::SwitchToFirst();

   // call event preprocessing...
   Bool_t preCheck = EventProcess();
   // ...then fill the information object and print informations...
   fTaskInfo.FillInfo();
   fTaskInfo.PrintInfo(fTaskInfo.GetNumerOfEventsProcessed());
   // ...and return if event did not pass selections
   if (!preCheck) {
      AliDebug(AliLog::kDebug, "Event preprocessing has failed. Skipping event");
      return;
   }


   // call customized implementation for execution
   RsnUserExec(opt);

   // post outputs for the info object
   // (eventually others are done in the derived classes)
   PostData(1, fInfoList);
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::Terminate(Option_t* opt)
{
//
// Termination routines.
// Stores all histograms (after checking they exist)
// and includes to the TList all task informations.
//

   AliAnalysisTask::Terminate();

   TList* list  = dynamic_cast<TList*>(GetOutputData(1));
   if (!list) {
      AliError(Form("At end of analysis, fOutList is %p", list));
      return;
   }

   RsnTerminate(opt);

   TH1I *hEventInfo = (TH1I*) list->FindObject(fTaskInfo.GetEventHistogramName());
   if (!hEventInfo) {
      AliError(Form("hEventInfo is %p", hEventInfo));
      return;
   }
   AliInfo(Form("=== %s ==================", GetName()));
   AliInfo(Form("Number Of Events Processed : %10lld", (Long64_t)hEventInfo->Integral()));
   AliInfo(Form("Number Of Events Accepted  : %10lld", (Long64_t)hEventInfo->GetBinContent(2)));
   AliInfo(Form("Number Of Events Skipped   : %10lld", (Long64_t)hEventInfo->GetBinContent(1)));
   AliInfo(Form("=== end %s ==============", GetName()));

   AliDebug(AliLog::kDebug + 2, "->");
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::RsnUserCreateOutputObjects()
{
//
// Define here all instructions to create output objects.
// This method will be called inside the "UserCreateOutputObjects"
// in the used task.
//

   AliWarning("Implement this in derived classes");
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::RsnUserExec(Option_t*)
{
//
//
//

   AliWarning("Implement this in derived classes");
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::RsnTerminate(Option_t*)
{
//
// Overload this to add additional termination operations
//

   AliWarning("Implement this in derived classes");
}

//_____________________________________________________________________________
Bool_t AliRsnVAnalysisTaskSE::EventProcess()
{
//
// Performs some pre-processing of current event,
// which is useful for all the operations which
// need to be done only once for each event.
//

   // in this case, return always a success
   return kTRUE;
}

//_____________________________________________________________________________
void AliRsnVAnalysisTaskSE::SetDebugForAllClasses()
{
//
// Set debug level for all classes for which it is required
//

   TObjArray  *array = fLogClassesString.Tokenize(":");
   TObjString *objStr;
   TString     str;
   Int_t       i, n = array->GetEntriesFast();

   for (i = 0; i < n; i++) {
      objStr = (TObjString*)array->At(i);
      str    = objStr->GetString();
      AliLog::SetClassDebugLevel(str.Data(), fLogType);
      AliInfo(Form("Setting Debug to %s", str.Data()));
   }
}