QAverageNoiseHandle.cc

Go to the documentation of this file.

#include "QAverageNoiseHandle.hh"
#include "QAverageVector.hh"
#include "QDianaDb.hh"
 
using namespace Diana;
 
QError QAverageNoiseHandle::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> avgnoise;
 
  QDianaDb *db = QDianaDb::Get();
 
  // query noise_power_spectrum
  std::stringstream squery;
  squery<<"select noise_power_spectrum  from noise_power_spectra where nps_algo = '"<<bareOwner
        <<"' and nps_version= '"<<GetVersion()
        <<"' and nps_extralabel= '"<<extraLabel
        <<"' and channel= "<< GetChannel()
        <<" and validity_kind= "<<validity_kind
        <<" and validity_start= "<<validity_start
        <<" and validity_end= "<<validity_end;
  
  try {
    avgnoise = db->DoQueryVectorDouble(squery.str());
  }
  
  catch(const QError& cerr) {
    if(cerr!=QERR_SUCCESS) {
      err = cerr;
      return err;   
    }      
  }
  
  for(size_t i=0; i<avgnoise.size(); i++)
    {
      av.Append(avgnoise[i]);
    }
 
  squery.str("");
  
  // query source_runs 
  squery<<"select source_runs from noise_power_spectra where nps_algo = '"<<bareOwner
        <<"' and nps_version= '"<<GetVersion()
        <<"' and nps_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 noise_power_spectra where nps_algo = '"<<bareOwner
        <<"' and nps_version= '"<<GetVersion()
        <<"' and nps_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 QAverageNoiseHandle::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;
  std::stringstream squery;
  fields.push_back((std::string)"nps_algo");                values.push_back("'"+bareOwner+"'");
  fields.push_back((std::string)"nps_version");             values.push_back("'"+GetVersion()+"'");
  fields.push_back((std::string)"nps_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)"noise_power_spectrum");    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("noise_power_spectra",fields,values);
  }
 
  catch(const QError& cerr) {
    err = cerr;
  }
 
    
  return err;
}