THiCalibration.cxx

Go to the documentation of this file.

//  THiCalibration.cxx.
#include <iostream>
#include <sstream>

#include <TSQLRow.h>
#include <TMySQLServer.h>
#include <TMySQLResult.h>

#include "THiCalibration.h"

using std::cout;
using std::cerr;
using std::endl;
using std::stringstream;

THiCalibration::THiCalibration(UInt_t channels)
  : fDatabase("mysql://localhost:3306/Online"),
    fAdc(channels), fTdc(channels)
{
}

THiCalibration::~THiCalibration()
{
}

void THiCalibration::Clear(const Char_t *option)
{
  fAdc.Clear();
  fTdc.Clear();
  cout << "Clear: THiCalibration cleared" << endl;
}

void THiCalibration::Print(const Char_t *option) const
{
  cout << "THiCalibration:" << endl;
  cout << "ADC calibration: " << endl;
  fAdc.Print();
  cout << "TDC calibration: " << endl;
  fTdc.Print();
}

void THiCalibration::FindCalibration(const Char_t *detectorName)
{
  TMySQLResult *result = GetDatabaseInfo(detectorName);

  if (result) {
    UShort_t channels = fAdc.GetChannels();
    if (static_cast<UInt_t>(result->GetRowCount()) != channels)
      cerr << "FindCalibration: WARNING - rows in database ("
           << result->GetRowCount()    << ") do not match number " << endl
           << "of expected channels (" << channels    << ") for detector "
           << detectorName << " - using default calibration" << endl;
    else if (result->GetFieldCount() != 9)
      cerr << "FindCalibration: WARNING - columns in database ("
           << result->GetFieldCount() << ") do not match number "  << endl
           << "of expected columns (9) for detector " << detectorName
           << " - using default calibration" << endl;
    else {
      for (Int_t i = 0; i < result->GetRowCount(); i++) {
        stringstream *fields = new stringstream[result->GetFieldCount()];
        TSQLRow *row = result->Next();
        for (Int_t i = 0; i < result->GetFieldCount(); i++)
          fields[i] << row->GetField(i);
        UShort_t detectorChannel;
        UShort_t adcMap;
        Short_t adcOffset;
        Float_t  adcGain;
        UShort_t adcHistogramChannel;
        UShort_t tdcMap;
        Short_t tdcOffset;
        Float_t  tdcGain;
        UShort_t tdcHistogramChannel;
        fields[0] >> detectorChannel;
        fields[1] >> adcMap;
        fields[2] >> adcOffset;
        fields[3] >> adcGain;
        fields[4] >> adcHistogramChannel;
        fields[5] >> tdcMap;
        fields[6] >> tdcOffset;
        fields[7] >> tdcGain;
        fields[8] >> tdcHistogramChannel;
        fAdc.AddToChannel(adcMap, detectorChannel);
        fAdc.AddToMap(detectorChannel, adcMap);
        fAdc.AddToOffset(detectorChannel, adcOffset);
        fAdc.AddToGain(detectorChannel, adcGain);
        fAdc.AddToHistogramMap(detectorChannel, adcHistogramChannel);
        fTdc.AddToChannel(tdcMap, detectorChannel);
        fTdc.AddToMap(detectorChannel, tdcMap);
        fTdc.AddToOffset(detectorChannel, tdcOffset);
        fTdc.AddToGain(detectorChannel, tdcGain);
        fTdc.AddToHistogramMap(detectorChannel, tdcHistogramChannel);
        delete row;
        delete [] fields;
      }
      fAdc.SetBitMask();
      fTdc.SetBitMask();
    }
    delete result;
  } else
    cerr << "FindCalibration: ERROR - could not obtain database information"
         << endl;
}

TMySQLResult * THiCalibration::GetDatabaseInfo(const Char_t *detectorName)
{
  TMySQLResult *result = 0;
  if (! detectorName) {
    cerr << "GetDatabaseInfo: ERROR - detector name must be set before "
         << "attempting to set the detector calibration" << endl;
  } else {
    string sql("SELECT * FROM ");
    
    sql.append(detectorName);
    sql.append(" ORDER BY `adc map`;");
    TMySQLServer connect(fDatabase.c_str(), "dragon", "dragonTail");
    result = static_cast<TMySQLResult *>(connect.Query(sql.c_str()));
  }
  return result;
}

---