#include "TROOT.h"
#include "TObjString.h"
#include "TString.h"
#include "TFile.h"
#include "TTree.h"
#include "TSystem.h"
#include "htool.h"


#include <fstream>
#include <iostream>
using namespace std;

Int_t check(TString name,Int_t nevents,TString& msg)
{

    TFile* in = new TFile(name.Data(),"READ");
    if(!in){
	msg = Form("%s : ERROR : Could not open file",name.Data());
	cout<<msg<<endl;
	in->Close();
	delete in;
	return 2;
    }
    TTree* T = (TTree*)in->Get("T");
    if(!T){
	msg = Form("%s : ERROR : Could not find Tree",name.Data());
	cout<<msg<<endl;
	in->Close();
	delete in;
	return 3;
    }
    TKey* hades = in->FindKey("Hades");
    if(!hades){
	msg = Form("%s : ERROR : Could not find Hades object",name.Data());
	cout<<msg<<endl;
	in->Close();
	delete in;
	return 4;
    }
    if(nevents!=-1){
	Int_t n = T->GetEntries();
	if(n<nevents){
	    msg = Form("%s : ERROR : Not enough Events in Tree : %d (%d)",name.Data(),n,nevents);
	    cout<<name.Data()<<msg<<endl;
	    in->Close();
	    delete in;
	    return 5;
	}
    }
    in->Close();
    return 0;
}


Int_t testFiles(TString file,Int_t nevents,TString outFile = "")
{

    TString msg;
    if(file.Contains(","))
    { // comma separated list
        cout<<"input comma separate list"<<endl;
	ofstream out;
	if(outFile != ""){
	    out.open(outFile.Data());
	}
	TString tmp = file;
	TObjArray* a = tmp.Tokenize(",");
	if(a){
	    for(Int_t i = 0 ; i < a->GetEntries(); i++){
		TObjString* sn = (TObjString*) a->At(i);
		TString name = sn->GetString();
		Int_t ret = check(name,nevents,msg);
		if(ret == 0) msg = Form("Check : file \" %s \" is OK!",name.Data());
		if(outFile != "" && msg != "" )out<<msg<<endl;
	    }
	    delete a;
	}
	if(outFile != "") out.close();

    } else if(file.EndsWith(".list")) { // list : 1 file per line

	cout<<"input file list"<<endl;

	if(gSystem->AccessPathName(file.Data()) !=0 ){
	    cout<<"Error : file \""<<file.Data()<<"\"does not exist!"<<endl;
	    return 1;
	}

	TString name;
	Int_t total=0;
	ifstream input;
	input.open(file.Data());

	while(!input.eof()){
	    input>>name;
	    if(input.good()) { total++; }
	}

	input.close();

	TString outname = file;
	TString indir  = gSystem->DirName(file.Data());
	TString infile = gSystem->BaseName(file.Data());

	input.open(file.Data());

	ofstream out;
	if(outFile != ""){
	    out.open(outFile.Data());
	}

	Int_t ct = 0;
	while(!input.eof()){

	    input>>name;
	    ct++;
	    HTool::printProgress(ct,total, 1,"") ;
            Int_t ret = check(name,nevents,msg);
            if(ret == 0) msg = Form("Check : file \" %s \" is OK!",name.Data());
	    if(outFile != "" && msg != "" ) out<<msg<<endl;
	}

	if(outFile != "") out.close();

    } else { // single root file or reg exp

	TObjArray* a = HTool::glob(file);
	if(a){
	    Int_t n = a->GetEntries();
	    if(n == 0) {
		cout<<"Error : No file matching expression "<<file<<" found !"<<endl;
		return 1;
	    } else {

		ofstream out;
		if(outFile != ""){
		    out.open(outFile.Data());
		}


		for(Int_t i = 0; i < n; i ++){
		    TString name = ((TObjString*)a->At(i))->GetString();
		    Int_t ret = check(name,nevents,msg);
		    if(ret == 0) {
			msg = Form("Check : file \" %s \" is OK!",name.Data());
			if(n == 1) cout<<msg<<endl;
		    }
		    if(n == 1) return ret;
		    if(outFile != "" && msg != "" ) out<<msg<<endl;
		}

		if(outFile != "") out.close();
	    }
	    delete a;
	} else {
	    cout<<"Error : No file matching expression \""<<file<<"\" found !"<<endl;
            return 1;
	}
    }

    return 0;
}
int main(int argc, char **argv)
{
    TROOT AnalysisDST("AnalysisDST","test macro");


    TString nevents = "-1";
    switch (argc)
    {
    case 2:
	return testFiles(TString(argv[1]),nevents.Atoi(),""); // inputfile
	break;
    case 3:

        nevents=argv[2];
	return testFiles(TString(argv[1]),nevents.Atoi(),""); // inputfile + nevents
	break;
    case 4:

        nevents=argv[2];
	return testFiles(TString(argv[1]),nevents.Atoi(),argv[3]); // inputfile + nevents = outfile
	break;
    default:  // inputfile + outdir + nevents
	cout<<"usage : "<<argv[0]<<" file [ nevent ] [ outputfile ] "<<endl;
	cout<<" file       : file.root, for single file the return value will be 0 (OK) or > 0 (CORRUPT)"<<endl;
	cout<<"            : file.list (1file per line)"<<endl;
	cout<<"            : comma separated list : file1.root,file2,root,file3.root"<<endl;
	cout<<"            : 'file*.root' regular expression (do not forget quotes!)"<<endl;
	cout<<" nevent     : checks if the tree contains more than nevent events"<<endl;
	cout<<" outputfile : in case of comma separated list or file list the check result can be written to file"<<endl;
	cout<<"            : this option is not supported in single file mode"<<endl;
        return 1;
    }
}