#if !defined(__CINT__) || defined(__MAKECINT__)

#include <PamLevel2.h>

#include <TTree.h>
#include <TChain.h>
#include <TFile.h>
#include <TList.h>
#include <TROOT.h>
#include <TString.h>
#include <TBenchmark.h>
#include <TFriendElement.h>

#include <stdlib.h>
#include <iostream>
using namespace std;

#endif

//===================================
// global variables
//===================================
PamLevel2 *event = NULL;
TChain    *tree  = NULL;
TFile     *outfh = NULL;
TFile     *outft = NULL;
TTree     *otree = NULL;

bool fillTree   = false;
bool fillHistos = false;

//====================================
// Functions to be provided externally
//====================================
bool Select(PamLevel2*);
void CreateHistos();
void FillHistos(PamLevel2*);
void SaveHistos(TFile *);

//==========================================
//000000000000000000000000000000000000000000
//==========================================
bool Begin(){
	
    //------------------------
    //  load magnetic field
    //------------------------
    TString fieldpath = gSystem->Getenv("PAM_CALIB");
    if(fieldpath.IsNull()){
	cout << " **ERROR** : No PAMELA environment variables defined "<<endl;
	return 0;
    }
    fieldpath.Append("/trk-param/field_param-0/");
    event->LoadField(fieldpath.Data());


    //------------------------------
    //  create output histos/trees
    //------------------------------
    if(fillHistos) CreateHistos();
    if(fillTree)   event->CreateCloneTrees(tree);

    cout << "\nBegin() - done\n\n";
    return 1;
	
}
//==========================================
//000000000000000000000000000000000000000000
//==========================================

bool Process(int iev){

    //------------------------------
    //  create output histos/trees
    //------------------------------

    event->Clear();
    tree->GetEntry(iev);
    
    if( !Select(event) )return false;
//    cout <<event->TrkLevel2::ntrk()<<endl;


//    if(fillTree)event->FillCloneTrees();
//    if(fillHistos)FillHistos(event);

    return true;

}
//==========================================
//000000000000000000000000000000000000000000
//==========================================

bool Finish(){
    
    if(fillTree){
	outft->cd();
	event->WriteCloneTrees();
	outft->Close();
    }
    if(fillHistos){
	SaveHistos(outfh);
	outfh->Close();
    }
    
    cout << "Finish() - done\n";
    
    return 1;
}

//==========================================
//000000000000000000000000000000000000000000
//==========================================
Int_t Loop(TString ddir,TString list, ULong64_t nmax, TString options, TString outfile){
	
    if(options.Contains("fillTree"))fillTree=true;
    if(options.Contains("fillHisto"))fillHistos=true;

//  -------------------
//  create output files
//  -------------------
    TString outfile_t =0;
    TString outfile_h =0;
    if( outfile.IsNull() ){
	if(!list.IsNull())outfile = list(0,list.Last('.'));	    
	else outfile = "output";	    
    }else{	
	if(outfile.Contains(".root"))outfile = outfile(0,outfile.Last('.'));
    }
    if(fillTree){
	outfile_t = outfile;
	outfile_t.Append("-tree.root");
	outft = (TFile*)gROOT->FindObject(outfile_t); if (outft) outft->Close();
	outft = new TFile(outfile_t,"RECREATE");
	if(outft->IsZombie()){
	    cout << "Output file could not be created\n";
	    return 1;
	};
	cout << "Created output file: "<<outft->GetName()<<endl;
    }
    if(fillHistos){
	outfile_h = outfile;
	outfile_h.Append("-histo.root");
	outfh = (TFile*)gROOT->FindObject(outfile_h); if (outfh) outfh->Close();
	outfh = new TFile(outfile_h,"RECREATE");
	if(outfh->IsZombie()){
	    cout << "Output file could not be created\n";
	    return 1;
	};
	cout << "Created output file: "<<outfh->GetName()<<endl;
    }
    
//  --------------------
//  read input file/list
//  --------------------
    event = new PamLevel2();
    if(list.Contains(".root")){
	stringstream command;
	command.str("");
	command << " echo "<<list<<" > list-temp.txt";
	cout << command.str().c_str() << endl;
	gSystem->Exec(command.str().c_str());
	tree  = event->GetPamTree(event->GetListOfLevel2Files(ddir,"list-temp.txt"),options);
    }else{	 
	tree  = event->GetPamTree(event->GetListOfLevel2Files(ddir,list),options);
    };
    tree->SetCacheSize(0);

//  ---------------
//  initializations
//  ---------------
    if( !Begin() )return 0;;

//  -----------------
//  loop over events
//  -----------------
    ULong64_t nevents = tree->GetEntriesFast();	
    if(!nmax)nmax = numeric_limits<ULong64_t>::max();	
    if(nevents < nmax)nmax=nevents;
    
    cout << endl<<" Start loop over events:  "<< nmax<<endl;
    Int_t ntrk=0;
    Int_t sel=0;
    TBenchmark *benchmark = new TBenchmark();
    benchmark->Start("event-loop");
    
    TString current_file = "";
    for(ULong64_t iev=60000; iev<nmax; iev++){
	
//	tree->GetEntry(iev);
	if( Process(iev) )sel++;

	if(current_file.CompareTo(tree->GetFile()->GetName())){
	    current_file=tree->GetFile()->GetName();
	    cout << iev<< " -> "<< current_file << endl;
//	    cout << tree->GetFile()->GetName() << endl;
	};

	if(event->TrkLevel2::ntrk() >0)ntrk++;	
	if(!(iev%100))cout << iev << endl;
	
    };
    benchmark->Show("event-loop");
    cout << ntrk <<" tracks over "<<nmax<<" events ";
    if(nmax)cout<<"("<< 100*ntrk/nmax<<"%)"<<endl;
    cout << "Selected events: "<<sel<<endl;
    
//  --------------
//  close and exit
//  --------------
    Finish();

    return 0;
    
}

/////////////////////////////////////////////////////////////////

#if !defined(__CINT__)

// input parameters
Bool_t    DEBUG;
TString   DIR;
TString   LIST;
ULong64_t MAXEV;
TString   OUTFILE;

void usage(){
    cout << "USAGE:"<<endl;
    cout << "-processDir  [DIR]   -  Level2/Level1 data directory \n";
    cout << "-processList [LIST]  -  list (txt) of names of files to be analysed \n";
    cout << "-outputFile  [PATH]  -  name of the output file \n";
    cout << "-NumEvents   [XXX]   -  number of events to be analysed \n";
    cout << "--debug, -g          -  debug mode \n";
    cout << "--help, -h           -  print this help \n";
}
//
int HandleInputPar(int argc, char **argv){

    if(argc>1){
		
	if(!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help") ){
	    usage();
	    return(1);
	};
// -----------------------
// Read input parameters
// -----------------------  
/*		DEBUG   = false;
		DIR     = gSystem->WorkingDirectory();
		LIST    = "";
		OUTFILE = "TrkLevel1Report_output.root";
		MAXEV = numeric_limits<ULong64_t>::max();
*/
		

	for (int i = 1; i < argc; i++){		
	    // -----------------------------------------------------//
	    if (!strcmp(argv[i], "-processDir")){
		if (++i >= argc) throw -1;
		DIR = argv[i];
		cout << "processDir "<<DIR<<endl;
		continue;
	    }
	    // -----------------------------------------------------//
	    else if (!strcmp(argv[i], "-processList")){
		if (++i >= argc) throw -1;
		LIST = argv[i];
		cout << "processList "<<LIST<<endl;
		continue;
	    }  
	    // -----------------------------------------------------//
	    else if (!strcmp(argv[i], "-outputFile")){
		if (++i >= argc) throw -1;
		OUTFILE = argv[i];
		cout << "outputFile "<<OUTFILE<<endl;
		continue;
	    }  
	    // -----------------------------------------------------//
	    else if (!strcmp(argv[i], "-NumEvents")){
		if (++i >= argc) throw -1;
		MAXEV = atoi(argv[i]);
		cout << "NumEvents "<<MAXEV<<endl;
		continue;	
	    }
	    // -----------------------------------------------------//
	    else if (!strcmp(argv[i], "--debug") || !strcmp(argv[i], "-g")){
		DEBUG = true;
		continue;	
	    }
	    // -----------------------------------------------------//
	    else{
		cout << "Unidentified input parameter. Ignored."<< endl;
	    };
	};
    }else{
	usage();
	return(1);
    };
// -----------------------
// Check input parameters
// -----------------------  
	
		
    return(0);
	
};
// 

int main(int argc, char **argv)
{
	
    if( HandleInputPar(argc,argv) )return(1);
	
    Loop(DIR,LIST,MAXEV,"-ALL+TRK1+TRK2+CAL1+CAL2+TOF+AC fillHisto fillTree",OUTFILE);
    
    cout << "Back to main - end"<<endl;
    
    return 0;
	
};

#endif