/*************************************************************************
* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  * 
**************************************************************************/

#include "Riostream.h" //needed as include
#include "TProfile.h"
#include "TProfile2D.h"
#include "TList.h"


class AliAnalysisTaskSE;
#include "AliAnalysisManager.h"
#include "AliFlowEventSimple.h"


#include "AliAnalysisTaskMCEventPlane.h"
#include "AliFlowAnalysisWithMCEventPlane.h"
#include "AliFlowCommonHist.h"
#include "AliFlowCommonHistResults.h"

// AliAnalysisTaskMCEventPlane:
//
// analysis task for Monte Carlo Event Plane
//
// Author: Naomi van der Kolk (kolk@nikhef.nl)

using std::endl;
using std::cout;
ClassImp(AliAnalysisTaskMCEventPlane)

//________________________________________________________________________
AliAnalysisTaskMCEventPlane::AliAnalysisTaskMCEventPlane(const char *name) : 
  AliAnalysisTaskSE(name), 
  fEvent(NULL),
  fMc(NULL),
  fListHistos(NULL),
  fHarmonic(2),
  fEvaluateMixedHarmonics(kFALSE),
  fnBinsMult(10000),
  fMinMult(0.),  
  fMaxMult(10000.),    
  fNinCorrelator(2),
  fMinCorrelator(2),
  fXinPairAngle(0.5) 
{
  // Constructor
  cout<<"AliAnalysisTaskMCEventPlane::AliAnalysisTaskMCEventPlane(const char *name)"<<endl;

  // Define input and output slots here
  // Input slot #0 works with a TChain
  DefineInput(0, AliFlowEventSimple::Class());
  // Output slot #0 writes into a TList container
  DefineOutput(1, TList::Class()); 
}

//________________________________________________________________________
AliAnalysisTaskMCEventPlane::AliAnalysisTaskMCEventPlane() : 
  AliAnalysisTaskSE(),
  fEvent(NULL),
  fMc(NULL),
  fListHistos(NULL),
  fHarmonic(0),
  fEvaluateMixedHarmonics(kFALSE),
  fnBinsMult(0),
  fMinMult(0.0),  
  fMaxMult(0.0),     
  fNinCorrelator(0),
  fMinCorrelator(0),
  fXinPairAngle(0.0) 
{
  // Constructor
  cout<<"AliAnalysisTaskMCEventPlane::AliAnalysisTaskMCEventPlane()"<<endl;

}

//________________________________________________________________________
AliAnalysisTaskMCEventPlane::~AliAnalysisTaskMCEventPlane()
{

  //destructor

}

//________________________________________________________________________
void AliAnalysisTaskMCEventPlane::UserCreateOutputObjects() 
{
  // Called once
  cout<<"AliAnalysisTaskMCEventPlane::CreateOutputObjects()"<<endl;

  //Analyser
  fMc  = new AliFlowAnalysisWithMCEventPlane() ;
  
  fMc->SetHarmonic(fHarmonic);
  
  // Setters for mixed harmonics study:
  fMc->SetEvaluateMixedHarmonics(fEvaluateMixedHarmonics);
  fMc->SetNinCorrelator(fNinCorrelator);
  fMc->SetMinCorrelator(fMinCorrelator);
  fMc->SetXinPairAngle(fXinPairAngle);
  fMc->SetnBinsMult(fnBinsMult);
  fMc->SetMinMult(fMinMult);
  fMc->SetMaxMult(fMaxMult);
  
  // Initialized:    
  fMc-> Init();

  if (fMc->GetHistList()) {
    //fMc->GetHistList()->Print();
    fListHistos = fMc->GetHistList();
    //fListHistos->Print();
  }
  else {Printf("ERROR: Could not retrieve histogram list"); }

 PostData(1,fListHistos);

}

//________________________________________________________________________
void AliAnalysisTaskMCEventPlane::UserExec(Option_t *) 
{
  // Main loop
  // Called for each event

  fEvent = dynamic_cast<AliFlowEventSimple*>(GetInputData(0));
  if (fEvent){
    fMc->Make(fEvent);
  }
  else {
    cout << "Warning no input data!!!" << endl;
  }

  PostData(1,fListHistos); 
}      


//________________________________________________________________________
void AliAnalysisTaskMCEventPlane::Terminate(Option_t *) 
{
  // Called once at the end of the query
  AliFlowAnalysisWithMCEventPlane* fMcTerm = new AliFlowAnalysisWithMCEventPlane() ;

  //Get output data
  fListHistos = (TList*)GetOutputData(1);
  if (fListHistos) {
    fMcTerm->GetOutputHistograms(fListHistos);
    fMcTerm->Finish();
    PostData(1,fListHistos);
  } else 
    { 
      cout << "histogram list pointer is empty" << endl;
    }
}