correct the getting of EMCAL geometry

[u/mrichter/AliRoot.git] / EMCAL / AliEMCALPreprocessor.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */
/* History of cvs commits:
 *
 * $Log$
 * Revision 1.4  2007/01/24 16:57:14  gustavo
 * Calibratio file sources machines are now not hardcoded but retreived from shuttle
 *
 * Revision 1.3  2006/12/20 10:53:28  gustavo
 * Change const char * by TString, change AliInfos per AliPreprocessor::Log or AliDebug
 *
 * Revision 1.2  2006/12/12 17:16:09  gustavo
 * Detector name hardcoded in Preprocesor with new detector name notation (3 letters). New way to take reference histogram to avoid problems in case of low number of entries or no existing histogram. Change return 0 by return 1
 *
 * Revision 1.1  2006/12/07 16:32:16  gustavo
 * First shuttle code, online calibration histograms producer, EMCAL preprocessor
 * 
 *
*/
///////////////////////////////////////////////////////////////////////////////
// EMCAL Preprocessor class. It runs by Shuttle at the end of the run,
// calculates calibration coefficients and dead/bad channels
// to be posted in OCDB
//
// Author: Boris Polichtchouk, 4 October 2006
// Adapted for EMCAL by Gustavo Conesa Balbastre, October 2006
///////////////////////////////////////////////////////////////////////////////

//Root
#include "TFile.h"
#include "TH1.h"
#include "TMap.h"
#include "TRandom.h"
#include "TKey.h"
#include "TList.h"
#include "TString.h"
#include "TObjString.h"

//AliRoot
#include "AliEMCALPreprocessor.h"
#include "AliLog.h"
#include "AliCDBMetaData.h"
#include "AliEMCALCalibData.h"

ClassImp(AliEMCALPreprocessor)

//_______________________________________________________________________________________
AliEMCALPreprocessor::AliEMCALPreprocessor() :
AliPreprocessor("EMC",0)
{
  //default constructor
}

//_______________________________________________________________________________________
AliEMCALPreprocessor::AliEMCALPreprocessor(AliShuttleInterface* shuttle):
AliPreprocessor("EMC",shuttle)
{
  // Constructor
}

//_______________________________________________________________________________________
UInt_t AliEMCALPreprocessor::Process(TMap* /*valueSet*/)
{
  // process data retrieved by the Shuttle
  
  // The fileName with the histograms which have been produced by
  // AliEMCALCalibHistoProducer.
  // It is a responsibility of the SHUTTLE framework to form the fileName
  
  AliEMCALCalibData calibData;
  
  
  TList* list = GetFileSources(kDAQ, "AMPLITUDES");
  if (list)
    {
      AliInfo("The following sources produced files with the id AMPLITUDES");
      list->Print();
      TIter iter(list);
      TObjString *source;
      while ((source = dynamic_cast<TObjString *> (iter.Next()))) {
	//TObjString * str = dynamic_cast<TObjString*> (list->At(0));//Only one source?
	AliInfo(Form("found source %s", source->String().Data()));
	
	TString fileName = GetFile(kDAQ, "AMPLITUDES", source->GetName());
	
	//TString  fileName = GetFile(kDAQ, "AMPLITUDES", "GDC");
	Log(Form("Got filename: %s",fileName.Data()));
	
	TFile f(fileName);
	
	if(!f.IsOpen()) {
	  Log(Form("File %s is not opened, something goes wrong!",fileName.Data()));
	  return 0;
	}
	
	
	const Int_t nMod=12; // 1:5 modules
	const Int_t nCol=48; //1:56 columns in each module
	Int_t nRow=24; //1:64 rows in each module
	const Int_t nRowHalfSM = 12; //Supermodules 11 and 12 are half supermodules
	
	Double_t coeff;
	char hnam[80];
	TH1F* histo=0;
	
	
	//Get reference histogram
	TList * keylist = f.GetListOfKeys();
	Int_t nkeys   = f.GetNkeys();
	Bool_t ok = kFALSE;
	TKey  *key;
	TString refHistoName= "";
	Int_t ikey = 0;
	Int_t counter = 0;
	TH1F* hRef = new TH1F();
	
	//Check if the file contains any histogram
	if(nkeys< 2){
	  Log(Form("Not enough histograms for calibration, nhist = %d",nkeys));
	  return 1;
	}
	
	while(!ok){
	  ikey = gRandom->Integer(nkeys);
	  key = (TKey*)keylist->At(ikey);
	  refHistoName = key->GetName();
	  hRef = (TH1F*)f.Get(refHistoName);
	  counter++;
	  // Check if the reference has too little statistics and 
	  // if the histogram has the correct name (2 kinds, mod#col#row for 
	  // reference here, and mod#, see AliEMCALHistoProducer.
	  if(refHistoName.Contains("col") && hRef->GetEntries()>2 && hRef->GetMean()>0) 
	    ok=kTRUE;
	  if(!ok && counter >= nMod*nCol*nRow+nMod){
	    Log("No histogram with enough statistics for reference");
	    return 1;
	  }
	}
	
	Double_t refMean=hRef->GetMean();
	
	// Calculates relative calibration coefficients for all non-zero channels
	
	for(Int_t mod=0; mod<nMod; mod++) {
	  if(mod > 10) nRow = nRowHalfSM ;
	  for(Int_t col=0; col<nCol; col++) {
	    for(Int_t row=0; row<nRow; row++) {
	      sprintf(hnam,"mod%dcol%drow%d",mod,col,row);
	      histo = (TH1F*)f.Get(hnam);
	      //TODO: dead channels exclusion!
	      if(histo && histo->GetMean() > 0) {
		coeff = histo->GetMean()/refMean;
		calibData.SetADCchannel(mod+1,col+1,row+1,1./coeff);
		AliDebug(1,Form("mod %d col %d row %d  coeff %f\n",mod,col,row,coeff));
	      }
	      else
		calibData.SetADCchannel(mod+1,col+1,row+1,-111); 
	    }
	  }
	}
	f.Close();
      }//while
    }//If list
  else {
    Log("Sources not found for id AMPLITUDES");
    return 0;
  }

  AliCDBMetaData metaData;
  Int_t result = Store("Calib", "Data", &calibData, &metaData);
  

  
  return result;
  
}