Using GetRunType

[u/mrichter/AliRoot.git] / ZDC / AliZDCPreprocessor.cxx

// --- ROOT system
#include <TFile.h>
#include <TClonesArray.h>
#include <TList.h>
#include <TObjString.h>
#include <TTimeStamp.h>

#include "AliZDCPreprocessor.h"
#include "AliCDBManager.h"
#include "AliCDBEntry.h"
#include "AliCDBMetaData.h"
#include "AliDCSValue.h"
#include "AliAlignObj.h"
#include "AliAlignObjAngles.h"
#include "AliLog.h"
#include "AliZDCDataDCS.h"
#include "AliZDCCalibData.h"

//
// Class implementing ZDC pre-processor.
// It takes data from DCS and passes it to the class AliZDCDataDCS.
// The class is then written to the CDB.
//

ClassImp(AliZDCPreprocessor)

//______________________________________________________________________________________________
AliZDCPreprocessor::AliZDCPreprocessor(const char* detector, AliShuttleInterface* shuttle) :
  AliPreprocessor(detector, shuttle),
  fData(0)
{
  // constructor
}

//______________________________________________________________________________________________
AliZDCPreprocessor::~AliZDCPreprocessor()
{
  // destructor
}


//______________________________________________________________________________________________
void AliZDCPreprocessor::Initialize(Int_t run, UInt_t startTime,
	UInt_t endTime)
{
  // Creates AliZDCDataDCS object

  AliPreprocessor::Initialize(run, startTime, endTime);

	Log(Form("\n\tRun %d \n\tStartTime %s \n\tEndTime %s", run,
		TTimeStamp(startTime).AsString(),
		TTimeStamp(endTime).AsString()));

 	fRun = run;
        fStartTime = startTime;
        fEndTime = endTime;

	fData = new AliZDCDataDCS(fRun, fStartTime, fEndTime);
}

//______________________________________________________________________________________________
UInt_t AliZDCPreprocessor::Process(TMap* dcsAliasMap)
{
  // *************** From DCS ******************
  // Fills data into a AliZDCDataDCS object
  if(!dcsAliasMap) return 0;

  // The processing of the DCS input data is forwarded to AliZDCDataDCS
  Float_t DCSValues[26];
  fData->ProcessData(*dcsAliasMap, DCSValues);
  //dcsAliasMap->Print("");
  //
  // --- Writing ZDC table positions into alignment object
  TClonesArray *array = new TClonesArray("AliAlignObjAngles",10);
  TClonesArray &alobj = *array;
  AliAlignObjAngles a;
  Double_t dx=0., dz=0., dpsi=0., dtheta=0., dphi=0.;
  // Vertical table position in mm from DCS
  Double_t dyZN1 = (Double_t) (DCSValues[0]/10.);
  Double_t dyZP1 = (Double_t) (DCSValues[1]/10.);
  Double_t dyZN2 = (Double_t) (DCSValues[2]/10.);
  Double_t dyZP2 = (Double_t) (DCSValues[3]/10.);
  const char *ZDCn1="ZDC/NeutronZDC1";
  const char *ZDCp1="ZDC/ProtonZDC1";
  const char *ZDCn2="ZDC/NeutronZDC2";
  const char *ZDCp2="ZDC/ProtonZDC2";
  UShort_t iIndex=0;
  AliAlignObj::ELayerID iLayer = AliAlignObj::kInvalidLayer;
  UShort_t volid = AliAlignObj::LayerToVolUID(iLayer,iIndex);
  //
  new(alobj[0]) AliAlignObjAngles(ZDCn1, volid, dx, dyZN1, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[1]) AliAlignObjAngles(ZDCp1, volid, dx, dyZP1, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[2]) AliAlignObjAngles(ZDCn2, volid, dx, dyZN2, dz, dpsi, dtheta, dphi, kTRUE);
  new(alobj[3]) AliAlignObjAngles(ZDCp2, volid, dx, dyZP2, dz, dpsi, dtheta, dphi, kTRUE);
  // save in CDB storage
  AliCDBMetaData md;
  md.SetResponsible("Chiara Oppedisano");
  md.SetComment("Alignment object for ZDC");
  UInt_t resultAl = 0;
  resultAl = Store("Align","Data", array, &md, 0, 0);
  
  // --- Writing ZDC PTMs HV values into calibration object
  AliZDCCalibData *calibdata = new AliZDCCalibData("ZDC");
  for(Int_t j=0; j<22; j++)
     calibdata->SetPMTHVVal(j,DCSValues[j+4]);

  
  // *************** From DAQ ******************
  // [a] PEDESTALS
TString runType = GetRunType();
if(runType = "PEDESTALS") {
  TList* daqSources = GetFileSources(kDAQ, "PEDESTALS");
  if(!daqSources){
    Log(Form("No source for PEDESTALS run %d !", fRun));
    return 0;
  }
  Log("\t List of sources for PEDESTALS");
  daqSources->Print();
  //
  TIter iter(daqSources);
  TObjString* source = 0;
  Int_t i=0;
  while((source = dynamic_cast<TObjString*> (iter.Next()))){
       Log(Form("\n\t Getting file #%d\n",++i));
       TString stringPedFileName = GetFile(kDAQ, "PEDESTALS", source->GetName());
       if(stringPedFileName.Length() <= 0){
          Log(Form("No PEDESTAL file from source %s!", source->GetName()));
	  return 0;
       }
       const char* PedFileName = stringPedFileName.Data();
       const Int_t NZDCch = 44;
       if(PedFileName){
         FILE *file;
         if((file = fopen(PedFileName,"r")) == NULL){
           printf("Cannot open file %s \n",PedFileName);
	   return 0;
         }
         Log(Form("File %s connected to analyze pedestal events", PedFileName));
         Float_t PedVal[(3*NZDCch)][2];
         for(Int_t i=0; i<(3*NZDCch); i++){
            for(Int_t j=0; j<2; j++){
               fscanf(file,"%f",&PedVal[i][j]);
	       //if(j==1) printf("PedVal[%d] -> %f, %f \n",i,PedVal[i][0],PedVal[i][1]);
            }
            if(i<NZDCch){
              calibdata->SetMeanPed(i,PedVal[i][0]);
              calibdata->SetMeanPedWidth(i,PedVal[i][1]);
            }
            else if(i>=NZDCch && i<(2*NZDCch)){
              calibdata->SetOOTPed(i-NZDCch,PedVal[i][0]);
              calibdata->SetOOTPedWidth(i-NZDCch,PedVal[i][1]);
            }
            else if(i>=(2*NZDCch) && i<(3*NZDCch)){
              calibdata->SetPedCorrCoeff(i-(2*NZDCch),PedVal[i][0],PedVal[i][1]);
            }
         }
       }
       else{
          Log(Form("File %s not found", PedFileName));
          return 0;
       }
       //
      //calibdata->Print("");
  }
  delete daqSources; daqSources = 0;
}
  // [a] EMD EVENTS
else if (runType = "EMD") {
  TList* daqSources = GetFileSources(kDAQ, "EMDCALIB");
  if(!daqSources){
    AliError(Form("No sources for EMDCALIB run %d !", fRun));
    return 0;
  }
  Log("\t List of sources for EMDCALIB");
  daqSources->Print();
  //
  TIter iter2(daqSources);
  TObjString* source = 0;
  Int_t j=0;
  while((source = dynamic_cast<TObjString*> (iter2.Next()))){
       Log(Form("\n\t Getting file #%d\n",++j));
       TString stringEMDFileName = GetFile(kDAQ, "EMDCALIB", source->GetName());
       if(stringEMDFileName.Length() <= 0){
         Log(Form("No EMDCALIB file from source %s!", source->GetName()));
	 return 0;
       }
       const char* EMDFileName = stringEMDFileName.Data();
       if(EMDFileName){
    	 FILE *file;
    	 if((file = fopen(EMDFileName,"r")) == NULL){
    	   printf("Cannot open file %s \n",EMDFileName);
	   return 0;
    	 }
    	 Log(Form("File %s connected to analyze EM dissociation events", EMDFileName));
    	 Float_t EMDFitVal[2];
    	 for(Int_t j=0; j<2; j++){	    
    	   fscanf(file,"%f",&EMDFitVal[j]);
         }
         calibdata->SetEnCalib(EMDFitVal);
       }
       else{
         Log(Form("File %s not found", EMDFileName));
         return 0;
       }
       //calibdata->Print("");
  }
}  
  // note that the parameters are returned as character strings!
  const char* nEvents = GetRunParameter("totalEvents");
  if(nEvents) Log(Form("Number of events for run %d: %s",fRun, nEvents));
  else Log(Form("Number of events not put in logbook!"));

  // Storing the final CDB file
  AliCDBMetaData metaData;
  metaData.SetBeamPeriod(0);
  metaData.SetResponsible("Chiara");
  metaData.SetComment("Filling AliZDCCalibData object");

  UInt_t resultCal = 0;
  resultCal = Store("Calib","Data",calibdata, &metaData, 0, 1);
 
  UInt_t result = 0;
  if(resultAl!=0 && resultCal!=0){
    if(resultAl==1 && resultCal==1) result = 1;
    else result = 2;
  }
  
  return result;
  
}