/*
PMD DA for online calibration

contact: basanta@iitb.ac.in, Satyajit.Jena@cern.ch
Link: https://twiki.cern.ch/twiki/bin/view/ALICE/DA
Reference run:/afs/cern.ch/user/s/sjena/public/run83496.raw
Run Type: PHYSICS
DA Type: MON
Number of events needed: 1 million for PB+PB, 200 milion for p+p
Input Files: PMD/Calib/Mapping/Run0_999999999_v0_s0.root, PMD_PED.root, PMD_GAIN_CONFIGFILE, pmd_gain_tempfile.dat
Output Files: PMDGAINS.root, to be exported to the DAQ FES
Trigger types used: PHYSICS_EVENT

*/
extern "C" {
#include <daqDA.h>
}

#include "event.h"
#include "monitor.h"

#include <Riostream.h>
#include <stdio.h>
#include <stdlib.h>

//AliRoot
#include "AliRawReaderDate.h"
#include "AliPMDCalibPedestal.h"
#include "AliPMDCalibGain.h"
#include "AliLog.h"
#include "AliCDBManager.h"

//ROOT
#include "TFile.h"
#include "TH1F.h"
#include "TBenchmark.h"
#include "TTree.h"
#include "TROOT.h"
#include "TPluginManager.h"
#include "TSystem.h"


/* Main routine
      Arguments: 
      1- monitoring data source
*/
int main(int argc, char **argv) {
  
    /* magic line from Rene */
    gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
					  "*",
					  "TStreamerInfo",
					  "RIO",
					  "TStreamerInfo()");


    int status = 0;


    Int_t filestatus = -1, xvar = 5;
    Int_t totevt = -1, maxevt = -1;
    Int_t hotevtsize = -1;
    Bool_t hotfilestatus = false;

    // Reads the pedestal file and keep the values in memory for subtraction

    AliPMDCalibGain *calibgain = new AliPMDCalibGain();

    // Fetch the pedestal file - PMD_PED.root 

    status = daqDA_DB_getFile("PMD_PED.root","PMD_PED.root");

    if(!status)
      {
	printf("*** Pedestal file retrieved from DB *** \n");
      }
    else
      {
	printf("*** Pedestal file NOT retrieved from DB *** \n");
	return -1;
      }
    
    Int_t pstatus = calibgain->ExtractPedestal("PMD_PED.root");

    if(pstatus == -3) return -3;

    TTree *ic    = NULL;
    TTree *meanc = NULL;

    // Retrieve the PMD_GAIN_CONFIGFILE
    status = daqDA_DB_getFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");

    FILE *fp1 = NULL;

    fp1 = fopen("PMD_GAIN_CONFIGFILE","r");

    if (fp1 == NULL)
      {
	printf("*** PMD GAIN Configfile doesn't exist,Provide one ***\n");
	return -1;
      }
    else
      {
	fscanf(fp1,"%d %d %d %d %d\n",&filestatus, &xvar, &totevt, &maxevt, &hotevtsize);
	//printf("%d %d %d %d %d\n",filestatus, xvar, totevt, maxevt, hotevtsize);
      }
    fclose(fp1);

    
    if (filestatus == 1)
      {
	// Retrieve the Temporray ascii file from DB
	status = daqDA_DB_getFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");
	if(!status)
	  {
	    calibgain->ReadTempFile("pmd_gain_tempfile.dat");
	  }
	else
	  {
	    printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	  }
	// Retrieve the hot cell file from DB - PMD_HOT.root
	status = daqDA_DB_getFile("PMD_HOT.root","PMD_HOT.root");
	if(!status)
	  {
	    calibgain->ExtractHotChannel("PMD_HOT.root");
	  }
	else
	  {
	    printf("--- pmd_gain_tempfile.dat: not retrieved from DB ---\n");
	  }
      }


    // decoding the events
    

    if (argc!=2) {
	printf("Wrong number of arguments\n");
	return -1;
    }
    
    
    /* define data source : this is argument 1 */  
    status=monitorSetDataSource( argv[1] );
    if (status!=0) {
	printf("monitorSetDataSource() failed : %s\n",monitorDecodeError(status));
	return -1;
    }
    
    /* declare monitoring program */
    status=monitorDeclareMp( __FILE__ );
    if (status!=0) {
	printf("monitorDeclareMp() failed : %s\n",monitorDecodeError(status));
	return -1;
    }
    
    /* define wait event timeout - 1s max */
    monitorSetNowait();
    monitorSetNoWaitNetworkTimeout(1000);
    
    /* log start of process */
    printf("PMD GAIN DA - strted generating the gain of a cell\n");  
    
    /* init some counters */
    int nevents_physics=0;
    int nevents_total=0;
    
    struct eventHeaderStruct *event;
    eventTypeType eventT = 0;

    Int_t iev=0;

    // Get run number

    int runNr = 0;

    if (getenv("DATE_RUN_NUMBER")==0) {
      printf("DATE_RUN_NUMBER not properly set.\n");
      printf("Run Number set to Zero \n");
      //return -1;
    }
    else
      {
	runNr = atoi(getenv("DATE_RUN_NUMBER"));
      }

    if (gSystem->AccessPathName("localOCDB/PMD/Calib/Mapping",kFileExists))
      {
	if (gSystem->mkdir("localOCDB/PMD/Calib/Mapping",kTRUE) != 0)
	  {
	    printf("Failed to create directory: localOCDB/PMD/Calib/Mapping");
	    return -1;
	  }
      }
    status = daqDA_DB_getFile("PMD/Calib/Mapping/Run0_999999999_v0_s0.root","localOCDB/PMD/Calib/Mapping/Run0_999999999_v0_s0.root");
    if (status)
      {
	printf("Failed to get PMD-Mapping file (PMD/Calib/Mapping/Run0_999999999_v0_s0.root) from DAQdetDB, status=%d\n", status);
	return -1;
      }

    // Global initializations
    AliLog::SetGlobalLogLevel(AliLog::kError);
    AliCDBManager *man = AliCDBManager::Instance();
    man->SetDefaultStorage("local://localOCDB");
    man->SetRun(runNr);

    
    /* main loop (infinite) */
    for(;;) {
	
	/* check shutdown condition */
	if (daqDA_checkShutdown()) {break;}
	
	/* get next event (blocking call until timeout) */
	status=monitorGetEventDynamic((void **)&event);
	if (status==MON_ERR_EOF) {
	    printf ("End of File detected\n");
	    break; /* end of monitoring file has been reached */
	}
	
	if (status!=0) {
	    printf("monitorGetEventDynamic() failed : %s\n",monitorDecodeError(status));
	    break;
	}
	
	/* retry if got no event */
	if (event==NULL) {
	    continue;
	}
	
	iev++; 
	
	/* use event - here, just write event id to result file */
	nevents_total++;
	eventT=event->eventType;
	switch (event->eventType){
      
	    /* START OF RUN */
	    case START_OF_RUN:
		break;
		/* END START OF RUN */
		
		/* END OF RUN */
	    case END_OF_RUN:
		break;
		
	    case PHYSICS_EVENT:
		nevents_physics++;
		totevt++;
		//if(nevents_physics%100 == 0)printf("Physis Events = %d\n",nevents_physics);
		AliRawReader *rawReader = new AliRawReaderDate((void*)event);
		TObjArray *pmdddlcont = new TObjArray();
		calibgain->ProcessEvent(rawReader, pmdddlcont);

		if (totevt%hotevtsize == 0) hotfilestatus = true;
		delete pmdddlcont;
		pmdddlcont = 0x0;
		delete rawReader;
		rawReader = 0x0;
		
	}
       
	/* free resources */
	free(event);
	
    }

    /* exit when last event received, no need to wait for TERM signal */


    ic    = new TTree("ic","PMD Gain tree");
    meanc = new TTree("meanc","PMD Module mean tree");

    if (filestatus == 0)
      {
	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");

	TTree *hot = new TTree("hot","PMD Hot cell tree");
	
	calibgain->FindHotCell(hot,xvar);
	
	hot->Write();
	hotRun->Close();

	// store the hot cell root file in the DB

	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");

	// store the hot cell root file in the file exchange server

	printf("root file for hot cell is created and getting exported\n");
	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
      }

    if (hotfilestatus)
      {
	TFile *hotRun = new TFile ("PMD_HOT.root","RECREATE");

	TTree *hot = new TTree("hot","PMD Hot cell tree");
	
	calibgain->FindHotCell(hot,xvar);
	
	hot->Write();
	hotRun->Close();

	// store the hot cell root file in the DB

	status = daqDA_DB_storeFile("PMD_HOT.root","PMD_HOT.root");
	
	// store the hot cell root file in the file exchange server

	printf("root file for hot cell is created and getting exported\n");
	status = daqDA_FES_storeFile("PMD_HOT.root","PMD_HOT.root");
      }


    fp1 = fopen("PMD_GAIN_CONFIGFILE","w+");

    if (totevt < maxevt)
      {
	printf("-----------------------------------------------\n");
	printf("***  Required Number of Events not reached  ***\n");
	printf("***  Number of Events processed = %d        ***\n",totevt);
	printf("***  Writing the intermediate ASCII file    ***\n");
	printf("-----------------------------------------------\n");

	calibgain->WriteTempFile("pmd_gain_tempfile.dat");

	// Store the Intermediate ascii file in the DB
	status = daqDA_DB_storeFile("pmd_gain_tempfile.dat","pmd_gain_tempfile.dat");

	filestatus = 1;
	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
	fclose(fp1);

	// Store the configfile in the DB
	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");

      }
    else if (totevt >= maxevt)
      {
	printf("-----------------------------------------------\n");
	printf("***  Required Number of Events reached = %d ***\n",totevt);
	printf("***  Writing the PMDGAINS.root file           ***\n");
	printf("-----------------------------------------------\n");

	calibgain->Analyse(ic, meanc);

	TFile * gainRun = new TFile ("PMDGAINS.root","RECREATE"); 
	ic->Write();
	gainRun->Close();

	TFile * meanRun = new TFile ("PMD_MEAN_SM.root","RECREATE"); 
	meanc->Write();
	meanRun->Close();


	filestatus = 0;
	totevt     = 0;
	fprintf(fp1,"%d %d %d %d %d\n",filestatus,xvar,totevt,maxevt,hotevtsize);
	fclose(fp1);

	// Store the configfile in the DB
	status = daqDA_DB_storeFile("PMD_GAIN_CONFIGFILE","PMD_GAIN_CONFIGFILE");
      }
    
    delete ic;
    ic = 0;

    delete meanc;
    meanc = 0;
    
    delete calibgain;

    /* store the result file on FES */
 
    if (filestatus == 0)
      {
	printf("root file for cell gain is created and getting exported\n");
	status = daqDA_FES_storeFile("PMDGAINS.root","PMDGAINS.root");
	printf("root file for normalised means of different modules\n");
	status = daqDA_FES_storeFile("PMD_MEAN_SM.root","PMD_MEAN_SM.root");
      }

    if (hotfilestatus)
      {
      }

    
    if (status) {
      status = -2;
    }



    return status;
}