#include "AliAnalysisTaskTransverseEnergy.h"
#include "AliAnalysisEtSelectionHandler.h"
#include "AliAnalysisManager.h"
#include "AliInputEventHandler.h"
#include "AliPhysicsSelectionTask.h"
#include "AliPhysicsSelection.h"
#include "AliCentrality.h"
#include "AliESDEvent.h"

//_________________________________________________________________________
//  Utility Class for transverse energy studies
//  Base class for Et tasks
//  - reconstruction and MonteCarlo output
//
//*-- Authors: Oystein Djuvsland (Bergen)
//_________________________________________________________________________//
#include "AliESDtrackCuts.h"
#include "AliLog.h"
#include "TH2F.h"
#include <iostream>


ClassImp(AliAnalysisTaskTransverseEnergy)

AliAnalysisTaskTransverseEnergy::AliAnalysisTaskTransverseEnergy(const char* name, Bool_t isMc) :
        AliAnalysisTaskSE(name)
	,fESDEvent(0)
        ,fMCConfigFile("ConfigEtMonteCarlo.C")
        ,fRecoConfigFile("ConfigEtReconstructed.C")
        ,fHistEtRecvsEtMC(0)
	,fHistEtRecOverEtMC(0)
	,fHistDiffEtRecEtMCOverEtMC(0)
        ,fEsdtrackCutsITSTPC(0)
        ,fEsdtrackCutsTPC(0)
        ,fEsdtrackCutsITS(0)
        ,fOutputList(0)
        ,fPhysSelTaskName("selctionTask")
        ,fCentSelTaskName("centralityTask")
	,fIsMc(isMc)
	,fUsingDefaultSelection(true)
	,fCurrentRunNum(-1)
	,fSelectionHandler(0)
{
  // Constructor
    LoadPhysicsSelection("physicsSelections.root");
}

AliAnalysisTaskTransverseEnergy::~AliAnalysisTaskTransverseEnergy()
{    // destructor
  delete fHistEtRecvsEtMC;
  delete fHistEtRecOverEtMC;
  delete fEsdtrackCutsITSTPC;
  delete fEsdtrackCutsTPC;
  delete fEsdtrackCutsITS;
  delete fOutputList;
  delete fSelectionHandler;
}

Int_t AliAnalysisTaskTransverseEnergy::CheckPhysicsSelection(Int_t runNumber)
{
  // Check if the physics selection is valid, if not load a new one
  
    if (runNumber == fCurrentRunNum || fIsMc)
    {
        return 0;
    }
    else
    {
        AliPhysicsSelection *selection = 0;
	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	if(!mgr){
	  AliError("Error: no analysis manager");
	  return -1;
	}
        AliPhysicsSelectionTask *physSelTask = dynamic_cast<AliPhysicsSelectionTask*>(mgr->GetTask("AliPhysicsSelectionTask"));
        if (physSelTask)
        {

            if ((selection = fSelectionHandler->GetPhysicsSelection(runNumber)))
            {
            }
            else if ((selection = fSelectionHandler->GetDefaultPhysicsSelection()))
            {
	      fUsingDefaultSelection = true;
            }
            else if ((selection = new AliPhysicsSelection()))
            {
	      fUsingDefaultSelection = true;
            }
            else
            {
                AliError("Something went very wrong, not able to load/create new physics selection");
                return -1;
            }
        }
        else
        {
            AliError("Could not get physics selection task from manager, undefined or no selection will be used");
            return -1;
        }
        AliInfo("Changing the physics selection");
        AliInputEventHandler* handler = dynamic_cast<AliInputEventHandler*> (mgr->GetInputEventHandler());
        // The physics selection task has a bit weird implementation, setting the the physics selection will not update the handler, so we do it manually
	if(!handler){
            AliError("Analysis manager does not exist!");
            return -1;
	}
        physSelTask->SetPhysicsSelection(selection);
        handler->SetEventSelection(selection);
        fCurrentRunNum = runNumber;

    }

    return 1;
}

Bool_t AliAnalysisTaskTransverseEnergy::IsPhysicsSelected() const
{
  // See header file for class documentation
    if(fIsMc) return true;
    if(!fUsingDefaultSelection) return (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & AliVEvent::kUserDefined);
    return (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & AliVEvent::kMB);
}

Int_t AliAnalysisTaskTransverseEnergy::LoadPhysicsSelection(TString name)
{	
  // See header file for class documentation
    if(fIsMc) return 0;
    fSelectionHandler = new AliAnalysisEtSelectionHandler(name);
    if (!fSelectionHandler) return -1;
    return 0;
}

AliCentrality* AliAnalysisTaskTransverseEnergy::GetCentralityObject()
{
  // See header file for class documentation
    if (fESDEvent)return fESDEvent->GetCentrality();
    else return 0;
}