//
// 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 <TH1.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.
//

   // initialize the list
   if (!fOutList) fOutList = new TList;
   fOutList->Clear();
   fOutList->SetOwner(kTRUE);

   // initialize all pairs
   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;

   fRsnAnalysisManager.ProcessAll(&fRsnEvent[0], 0x0, fMCOnly);

   PostData(2, fOutList);
}

//_____________________________________________________________________________
void AliRsnAnalysisTask::RsnUserExecMix(Option_t* /*opt*/)
{
//
// Execution of the analysis task with event mixing.
// 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;
   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
   fRsnAnalysisManager.ProcessAll(&fRsnEvent[0], &fRsnEvent[1], fMCOnly);
}

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

//______________________________________________________________________________
Bool_t AliRsnAnalysisTask::RsnEventProcess()
{
//
// 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);

   if (!AliRsnVAnalysisTask::RsnEventProcess()) return kFALSE;

   // check #1: number of tracks in event (reject empty events)
   Int_t    ntracks = fRsnEvent[0].GetMultiplicityFromTracks();
   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 kTRUE;
}