AMORE & QA fixes.

[u/mrichter/AliRoot.git] / MUON / AliMUONQAChecker.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$

#include "AliMUONQAChecker.h"

/// \class AliMUONQAChecker
///
/// Implementation of AliQACheckerBase for MCH and MTR
///
/// For the moment we only implement the checking of raw data QA for the tracker
/// by looking at the occupancy at the manu level.
/// We count the number of manus above a given occupancy threshold, and
/// depending on that number, the resulting QA flag is warning, error or fatal.
/// (there's no "info" type in this case).
///
/// \author Laurent Aphecetche, Subatech

#include "AliLog.h"
#include "AliMUONVTrackerData.h"
#include "AliMpManuIterator.h"
#include "AliQAv1.h"
#include <TDirectory.h>
#include <TH1.h>

/// \cond CLASSIMP
ClassImp(AliMUONQAChecker)
/// \endcond

//__________________________________________________________________
AliMUONQAChecker::AliMUONQAChecker() : 
    AliQACheckerBase("MUON","MUON Quality Assurance Data Maker") 
{
	/// ctor
}          

//__________________________________________________________________
AliMUONQAChecker::~AliMUONQAChecker() 
{
	/// dtor
}

//__________________________________________________________________
AliMUONQAChecker::AliMUONQAChecker(const AliMUONQAChecker& qac) : 
    AliQACheckerBase(qac.GetName(), qac.GetTitle()) 
{
	/// copy ctor 
}   

//______________________________________________________________________________
Double_t *
AliMUONQAChecker::Check(AliQAv1::ALITASK_t index, TObjArray ** list)
{
  /// Check objects in list
  
  if ( index == AliQAv1::kRAW ) 
  {
    return CheckRaws(list);
  }
  
  if ( index == AliQAv1::kREC)
  {
    return CheckRecPoints(list);
  }
  
  if ( index == AliQAv1::kESD )
  {
    return CheckESD(list);
  }
  
  AliWarning(Form("Checker for task %d not implement for the moment",index));
  return NULL;
}

//______________________________________________________________________________
TH1* 
AliMUONQAChecker::GetHisto(TObjArray* list, const char* hname, Int_t specie) const
{
  /// Get a given histo from the list
  TH1* h = static_cast<TH1*>(list->FindObject(Form("%s_%s",AliRecoParam::GetEventSpecieName(specie),hname)));
  if (!h)
  {
    AliError(Form("Did not find expected histo %s",hname));
  }
  return h;
}

//______________________________________________________________________________
Double_t *
AliMUONQAChecker::CheckRecPoints(TObjArray ** list)
{
  /// Check rec points
  /// Very binary check for the moment. 
  
  Double_t * rv = new Double_t[AliRecoParam::kNSpecies] ; 
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) 
    rv[specie] = 1.0 ; 
  
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
    TH1* h = GetHisto(list[specie],"hTrackerNumberOfClustersPerDE",specie);
  
    if ( !h ) rv[specie] =  0.75; // only a warning if histo not found, in order not to kill anything because QA failed...
  
    else if ( h->GetMean() == 0.0 ) rv[specie] =  MarkHisto(*h,0.0);
  }
  return rv;
}

//______________________________________________________________________________
Double_t 
AliMUONQAChecker::MarkHisto(TH1& histo, Double_t value) const
{
  /// Mark histo as originator of some QA error/warning
  
  if ( value != 1.0 )
  {
    histo.SetBit(AliQAv1::GetQABit());
  }
  
  return value;
}

//______________________________________________________________________________
Double_t *
AliMUONQAChecker::CheckESD(TObjArray ** list)
{
  /// Check ESD
  
  Double_t * rv = new Double_t[AliRecoParam::kNSpecies] ; 
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) 
    rv[specie] = 1.0 ; 
  
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
    
    TH1* h = GetHisto(list[specie],"hESDnTracks",specie);
  
    if (!h) rv[specie] = 0.75;
  
    else if ( h->GetMean() == 0.0 ) rv[specie] =  MarkHisto(*h,0.0); // no track -> fatal
  
    h = GetHisto(list[specie],"hESDMatchTrig",specie);
  
    if (!h) rv[specie] =  0.75;
  
    else if (h->GetMean() == 0.0 ) rv[specie] = MarkHisto(*h,0.25); // no trigger matching -> error
  }
  return rv;
}

//______________________________________________________________________________
Double_t *
AliMUONQAChecker::CheckRaws(TObjArray ** list)
{
  /// Check raws

  Double_t * rv = new Double_t[AliRecoParam::kNSpecies] ; 
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) 
    rv[specie] = 1.0 ; 
 
  Bool_t IsAnyTrackerDataPresent = kFALSE;
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
    TIter next(list[specie]);
    TObject* object;
    AliMUONVTrackerData* data(0x0);
  
    while ( (object=next()) && !data )
      {
        if (object->InheritsFrom("AliMUONVTrackerData"))
          {
            data = static_cast<AliMUONVTrackerData*>(object);
          }
      }

    if ( !data ) 
      {
        AliWarning(Form("Did not find TrackerData for specie %s !",AliRecoParam::GetEventSpecieName(specie)));
        continue;
      }
  
    IsAnyTrackerDataPresent = kTRUE;
    AliMpManuIterator it;
    Int_t detElemId;
    Int_t manuId;
    Int_t n50(0); // number of manus with occupancy above 0.5
    Int_t n75(0); // number of manus with occupancy above 0.75
    Int_t n(0); // number of manus with some occupancy
  
    while ( it.Next(detElemId,manuId) )
      {
        Float_t occ = data->Manu(detElemId,manuId,2);
        if (occ > 0 ) ++n;
        if (occ >= 0.5 ) ++n50;
        if (occ >= 0.75 ) ++n75;    
      }

    AliDebug(1,Form("n %d n50 %d n75 %d",n,n50,n75));
  
    if ( n == 0 ) 
      {
        AliError("Oups. Got zero occupancy in all manus ?!");
        rv[specie] =  0.0;
      }

    if ( n75 ) 
      {
        AliError(Form("Got %d manus with occupancy above 0.75",n75));
        rv[specie] =  0.1;
      }
    
    if ( n50 ) 
      {
        AliWarning(Form("Got %d manus with occupancy above 0.5",n50));
        rv[specie] =  0.9;
      }
  }
  
  if ( !IsAnyTrackerDataPresent ) 
    {
      AliError("Did not find any TrackerData in the list !");
      return NULL;
    }
  
  return rv;
}

//______________________________________________________________________________
void AliMUONQAChecker::Init(const AliQAv1::DETECTORINDEX_t det) 
{
  /// intialises QA and QA checker settings
  AliQAv1::Instance(det) ; 
  Float_t hiValue[AliQAv1::kNBIT] ; 
  Float_t lowValue[AliQAv1::kNBIT] ;
  lowValue[AliQAv1::kINFO]      = 0.999   ; 
  hiValue[AliQAv1::kINFO]       = 1.0 ; 
  hiValue[AliQAv1::kWARNING]    = 0.99 ; 
  lowValue[AliQAv1::kWARNING]   = 0.5 ; 
  lowValue[AliQAv1::kERROR]     = 0.0   ; 
  hiValue[AliQAv1::kERROR]      = 0.5 ; 
  lowValue[AliQAv1::kFATAL]     = -1.0   ; 
  hiValue[AliQAv1::kFATAL]      = 0.0 ; 
  SetHiLo(&hiValue[0], &lowValue[0]) ; 
}

//______________________________________________________________________________
void 
AliMUONQAChecker::SetQA(AliQAv1::ALITASK_t index, Double_t * value) const
{
  /// sets the QA according the return value of the Check

  AliQAv1 * qa = AliQAv1::Instance(index);
  
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
    qa->UnSet(AliQAv1::kFATAL, specie);
    qa->UnSet(AliQAv1::kWARNING, specie);
    qa->UnSet(AliQAv1::kERROR, specie);
    qa->UnSet(AliQAv1::kINFO, specie);

    if ( ! value ) { // No checker is implemented, set all QA to Fatal
      qa->Set(AliQAv1::kFATAL, specie) ; 
    } else {
      if ( value[specie] >= fLowTestValue[AliQAv1::kFATAL] && value[specie] < fUpTestValue[AliQAv1::kFATAL] ) 
        qa->Set(AliQAv1::kFATAL, specie) ; 
      else if ( value[specie] > fLowTestValue[AliQAv1::kERROR] && value[specie] <= fUpTestValue[AliQAv1::kERROR]  )
        qa->Set(AliQAv1::kERROR, specie) ; 
      else if ( value[specie] > fLowTestValue[AliQAv1::kWARNING] && value[specie] <= fUpTestValue[AliQAv1::kWARNING]  )
        qa->Set(AliQAv1::kWARNING, specie) ;
      else if ( value[specie] > fLowTestValue[AliQAv1::kINFO] && value[specie] <= fUpTestValue[AliQAv1::kINFO] ) 
        qa->Set(AliQAv1::kINFO, specie) ; 	
    }
  }
}