#include "TString.h"
#include "TStopwatch.h"
#include "hades.h"
#include "hloop.h"
#include "hdst.h"
#include "htaskset.h"
#include "hrich700digiparcreator.h"

Int_t run_rich_digi_par_event(
                              TString dstFile = "/lustre/cbm/users/slebedev/hades/data/digipar/2020.rich_params.dst.e.1.root",
			      TString histFile = "/lustre/cbm/users/slebedev/hades/data/digipar/2020.hist.1.root",
			      Int_t nEvents = 10000
			     )
{
    TString inFiles(dstFile);

    TString asciiParFile = "";
    TString rootParFile = "/cvmfs/hades.gsi.de/param/sim/mar19/allParam_mar19_sim_run_16000_gen2_22102019.root";
    TString paramSource = "root"; // root, ascii, oracle
    TString paramRelease = "MAR19_v2"; // 27012016

    TStopwatch timer;
    // create loop obejct and hades
    HLoop loop(kTRUE);

    Int_t refId = -1; // apr12sim_mediumfieldalign_auau;
    TString beamtime = "mar19";

    HDst::setupSpectrometer(beamtime,NULL,"start,rich,mdc,tof,rpc,shower,wall");
    HDst::setupParameterSources(paramSource,asciiParFile,rootParFile,paramRelease);

    Bool_t ret = kFALSE;
    if(inFiles.Contains(",")){
        ret = loop.addMultFiles(inFiles);// file1,file2,file3
    } else{
        ret = loop.addFiles(inFiles); // myroot*.root
    }
    if(ret == kFALSE) {
    	cout<<"READBACK: ERROR : cannot find inputfiles : "<<inFiles.Data()<<endl;
    	return 1;
    }

    if(!loop.setInput("-*,+HGeantKine,+HRichHitSim")) {
       cout << "READBACK: ERROR : cannot read input !" << endl;
       exit(1);
    }
    loop.printCategories();
    loop.printChain();

    HTaskSet *masterTaskSet = gHades->getTaskSet("all");
    HRich700DigiParCreator *digiParCreator = new HRich700DigiParCreator();
    digiParCreator->setOutputFilePath("");
    digiParCreator->setOutputHistFilePath(string(histFile.Data()));
    masterTaskSet->add(digiParCreator);

    Int_t entries = loop.getEntries();
    if(nEvents < entries && nEvents >= 0 ) entries = nEvents;

    for (Int_t i = 0; i < entries; i++) {
        Int_t nbytes =  loop.nextEvent(i);
        if(nbytes <= 0) {
            cout<<nbytes<<endl;
            break;
        }
    }

    if(gHades)gHades->finalizeTasks();

    cout << "Calculation time:" << timer.RealTime() << endl;

    delete gHades;

    return 0;
}