QAveragePulseHandle.cc

Go to the documentation of this file.

#include "QAveragePulseHandle.hh"
#include "QAverageVector.hh"
#include "QDianaDb.hh"
 
using namespace Diana;
 
QError QAveragePulseHandle::FillFromDB()
{
  QError err = QERR_SUCCESS;  
  
  QAverageVector av;
  av.Clear();
 
  std::string bareOwner, extraLabel;
  GetLabel().GetBareOwnerAndExtraLabel(bareOwner,extraLabel);    
  if (extraLabel=="") extraLabel="None";
      
  const int dataset = GetDataset();
  const int run = GetRun();
 
  // set validity_kind
  std::string validity_kind;
  int validity_start = Q_INT_DEFAULT, validity_end = Q_INT_DEFAULT;
 
  if (dataset!=Q_INT_DEFAULT)
    //validity kind set to dataset
    {
      validity_kind = "'data_set'";
      validity_start = dataset;
      validity_end = dataset;
    }
  else if (run!=Q_INT_DEFAULT)
    //validity kind set to run
    {
      validity_kind = "'run'";
      validity_start = run;
      validity_end = run;
    }
  else {
    err = QERR_DB_INCONSISTENCY;
    err.SetDescription(__FILE__,__LINE__,"Must provide run (SetRun()), or DataSet (SetDataSet()))");
  }
  if(err != QERR_SUCCESS) return err;
 
  std::vector<int> sruns;
  std::vector<double> avgpulse;
 
  QDianaDb *db = QDianaDb::Get();
  
  // query average_pulse
  std::stringstream squery;
  squery<<"select average_pulse from average_pulses where ap_algo = '"<<bareOwner
        <<"' and ap_version = '"<<GetVersion()
        <<"' and ap_extralabel = '"<<extraLabel
        <<"' and channel = "<< GetChannel()
        <<" and validity_kind = "<<validity_kind
        <<" and validity_start = "<<validity_start
        <<" and validity_end = "<<validity_end;
  
  try {
    avgpulse = db->DoQueryVectorDouble(squery.str());
  }
  
  catch(const QError& cerr) {
    if(cerr!=QERR_SUCCESS) {
      err = cerr;
      return err;   
    }  
  }
  
  av.Resize(avgpulse.size());
  for(size_t i=0; i<avgpulse.size(); i++)
    {
      av[i] = avgpulse[i];
    }
 
  squery.str("");
  
  // query source_runs
  squery<<"select source_runs from average_pulses where ap_algo = '"<<bareOwner
        <<"' and ap_version= '"<<GetVersion()
        <<"' and ap_extralabel= '"<<extraLabel
        <<"' and channel= "<< GetChannel()
        <<" and validity_kind= "<<validity_kind
        <<" and validity_start= "<<validity_start
        <<" and validity_end= "<<validity_end;
  
  try {
    sruns = db->DoQueryVectorInt(squery.str());        
  }
  
  catch(const QError& cerr) {
    if(cerr!=QERR_SUCCESS) {
      err = cerr;
      return err;   
    }  
  }
  
  for(size_t i=0; i<sruns.size(); i++)
    {
      av.GetSourceRuns().push_back(sruns[i]);
    }
  
  squery.str("");
 
  // query numevents
  squery<<"select num_events from average_pulses where ap_algo = '"<<bareOwner
        <<"' and ap_version= '"<<GetVersion()
        <<"' and ap_extralabel= '"<<extraLabel
        <<"' and channel= "<< GetChannel()
        <<" and validity_kind= "<<validity_kind
        <<" and validity_start= "<<validity_start
        <<" and validity_end= "<<validity_end;
  
  try {
    av.SetNumEvents(db->DoQueryInt(squery.str()));
  }
  
  catch(const QError& cerr) {
    if(cerr!=QERR_SUCCESS) {
      err = cerr;
      return err;   
    }  
  }
  
  Set(av);
  
  return err;
}
 
 
QError QAveragePulseHandle::StoreOnDB() const
{ 
  QError err = QERR_SUCCESS;  
 
  const QAverageVector& av = Get();
      
  std::string bareOwner, extraLabel;
  GetLabel().GetBareOwnerAndExtraLabel(bareOwner,extraLabel);    
  if (extraLabel=="") extraLabel="None";
  
  const int dataset = GetDataset();
  const int run = GetRun();
 
  std::string validity_kind;
  int validity_start = Q_INT_DEFAULT, validity_end = Q_INT_DEFAULT;
 
  // set validity_kind  
  if (dataset!=Q_INT_DEFAULT)
    //validity kind set to dataset
    {
      validity_kind = "'data_set'";
      validity_start = dataset;
      validity_end = dataset;
    }
  else if (run!=Q_INT_DEFAULT)
    //validity kind set to run
    {
      validity_kind = "'run'";
      validity_start = run;
      validity_end = run;
    }
  else {
    err = QERR_DB_INCONSISTENCY;
    err.SetDescription(__FILE__,__LINE__,"Must provide run (SetRun()), or DataSet (SetDataSet()))");
  }
  if(err != QERR_SUCCESS) return err;
 
    // creates a string containing source runs list 
  std::stringstream list_run;
  list_run<< "'{";
 
  for(size_t i=0; i<av.GetSourceRuns().size(); i++)
    { 
      if(i==av.GetSourceRuns().size()-1) {
        list_run<<av.GetSourceRuns()[i]<<"}'";
      } else{
        list_run<<av.GetSourceRuns()[i]<<",";
      }
    }
  
 
  // creates a string containing all components of the qvector
 
  std::stringstream list_av ;
  list_av<<"'{";
 
  for(size_t i=0; i<av.Size(); i++)
    { 
      if(i==av.Size()-1) {
        list_av<<av[i]<<"}'";;
      } else{
        list_av<<av[i]<<",";
      }
    }
  
  
  // DO QUERY
 
  QVdt::QVdt_vector fields,values;
  fields.push_back((std::string)"ap_algo");       values.push_back("'"+bareOwner+"'");
  fields.push_back((std::string)"ap_version");    values.push_back("'"+GetVersion()+"'");
  fields.push_back((std::string)"ap_extralabel"); values.push_back("'"+extraLabel+"'");
  fields.push_back((std::string)"channel");       values.push_back(GetChannel());
  fields.push_back((std::string)"validity_kind"); values.push_back(validity_kind);
  fields.push_back((std::string)"validity_start");values.push_back(validity_start);
  fields.push_back((std::string)"validity_end");  values.push_back(validity_end);
  fields.push_back((std::string)"source_runs");   values.push_back(list_run.str());
  fields.push_back((std::string)"average_pulse"); values.push_back(list_av.str());
  fields.push_back((std::string)"num_events");    values.push_back(av.GetNumEvents());
 
  try {
        QDianaDb *db = QDianaDb::Get();
        db->Insert("average_pulses",fields,values);
  }
 
 
 
  catch(const QError& cerr) {
    err = cerr;
  }
 
 
 
  return err;
}