HarpoRunConfig.cxx

Go to the documentation of this file.

// 
// File HarpoRunConfig.cxx
//
#include "HarpoRunConfig.h"
#include "HarpoDebug.h"
#include <TSQLServer.h>
#include <TSQLResult.h>
#include <TSQLRow.h>
#include <TSystem.h>
#include <strings.h>
#include <stdlib.h>
#include <iostream>



ClassImp(dbField) 

ClassImp(HarpoRunConfig) 


Bool_t HarpoRunConfig::GetRunConfig(Long_t nrun, const TString *DbUrl)
{
  const char *tables[] = { 
    "runconfig",
    "NSBeam",
    NULL
  };

  const char *dburl_default = "mysql://llrharpo.in2p3.fr/harpo";
  if(gHarpoDebug>0)
    printf("GetRunConfig %ld\n",nrun);

  TSQLServer *db = NULL;
  if ( DbUrl == NULL ) {
    const char *dburl = gSystem->Getenv("HARPO_DB_URL");
    if ( dburl == NULL || strlen(dburl) == 0 )
      dburl = dburl_default;

    db = TSQLServer::Connect(dburl,"reader", "ro");
  } else if ( DbUrl->IsNull() ) {
    std::cout << " NO database" << std::endl;
    return false;
  } else {
    db = TSQLServer::Connect(DbUrl->Data(),"reader", "ro");
  } 
  if ( db == NULL ) { 
    std::cout << " Error acess a database" << std::endl;
    return rec_found;
  }
  //   printf("Server info: %s\n", db->ServerInfo());
  
  TSQLRow *row;
  TSQLResult *res;
   
  // // list databases available on server
  // printf("\nList all databases on server %s\n", db->GetHost());
  // res = db->GetDataBases();
  // while ((row = res->Next())) {
  //    printf("%s\n", row->GetField(0));
  //    delete row;
  // }
  // delete res;

  // // list tables in database "test" (the permission tables)
  // printf("\nList all tables in database \"harpo\" on server %s\n",
  //        db->GetHost());
  // res = db->GetTables("harpo");

  // while ((row = res->Next())) {
  //    printf("%s\n", row->GetField(0));
  //    delete row;
  // }
  // delete res;

  // //////   
  // // list columns in table "runcatalog" in database "harpo"
  // //////   

  // printf("\nList all columns in table \"runcatalog\" in database \"harpo\" on server %s\n",
  //        db->GetHost());

  int itbl = 0;
  do {
    res = db->GetColumns("harpo", tables[itbl]);
    //   printf("No of field in GetColomns %d\n",res->GetFieldCount());
    while ((row = res->Next())) {
      // !!!!!hire 6 filed by program crached after 4 !!!!!!!!!!!!
      // for(int i=0;i<4;i++)
      //   { 
      //         std::cout << " f " << i << " v " << row->GetField(i);
      //   }
      //std::cout << std::endl;
      if ( fields.find(row->GetField(0)) == fields.end() ) {
        dbField *pf = new dbField;
        pf->ftype = row->GetField(1);
        fields[row->GetField(0)] = *pf;
      } else {
        if(gHarpoDebug>0)
          std::cout << "key " << row->GetField(0) << " exist" << std::endl;  
      }
      delete row;
    }
    delete res;
    itbl++;
  } while (tables[itbl] != NULL ); 
  // query database and print results
  char sql[4096];
  sprintf(sql,"select * from %s t0 left join %s t1 on t0.run = t1.run where t0.run = %ld;",tables[0],tables[1],nrun);
   
  if ( db->SelectDataBase("harpo") != 0 ) return rec_found;
  res = db->Query(sql);

  int nrows = res->GetRowCount();

  //  printf("\nGot %d rows in result\n", nrows);
  if (nrows != 1) return rec_found; // Bad query

  rec_found = true;
  run = nrun;
  int nfields = res->GetFieldCount();

  // std::cout << "N Fields in runconfig " << nfields 
  //         <<  "found "<< rec_found<< std::endl;

  row = res->Next();
   
  for (int j = 0; j < nfields; j++) {

    if (row->GetFieldLength(j) > 0 )
      fields[res->GetFieldName(j)].fval = row->GetField(j);

    // case 0: /* Long_t */ run = atol(row->GetField(j)); break;
    // case 1: /* char */ /* mysql date */ date = strdup(row->GetField(j)); break;
    // case 7: /* Double_t */ Temperature = atof(row->GetField(j)); break;
  }
  delete row;
  delete res;
   
  //close DB
  delete db;

  return rec_found;
}

void HarpoRunConfig::list()
{
  if ( !rec_found ) {
    printf("No run info or DB read error\n");
  } else {
    for (dbFields::iterator it=fields.begin(); it!=fields.end(); ++it) {
      std::cout << it->first << " ("<< it->second.ftype  << ")" << std::endl;
    }
  }
}

void HarpoRunConfig::print()
{
  if ( !rec_found ) {
    printf("No run info or DB read error\n");
  } else {
    for (dbFields::iterator it=fields.begin(); it!=fields.end(); ++it) {
      std::cout << it->first << " = " << it->second.fval  << std::endl;
    }
  }

}

//Look if filed exist
Bool_t HarpoRunConfig::Lookup(const TString field)
{
  return fields.find(field) != fields.end();
}

Bool_t HarpoRunConfig::Lookup(const char * field)
{
  return Lookup(TString(field));
}

Bool_t HarpoRunConfig::GetVal(const TString field, TString &val)
{
  dbFields::iterator it = fields.find(field);
  if ( it != fields.end() ) {
    val = it->second.fval;
    return true;
  }
  return false;
}

Bool_t HarpoRunConfig::GetVal(const char * field, TString &val)
{
  return GetVal(TString(field),val);
}

Bool_t HarpoRunConfig::GetType(const TString field, TString &type)
{
  dbFields::iterator it = fields.find(field);
  if ( it != fields.end() ) {
    type = it->second.ftype;
    return true;
  }
  return false;
}

Bool_t HarpoRunConfig::GetType(const char * field, TString &type)
{
  return GetType(TString(field),type);
}


// Update DB record base on class data
Bool_t HarpoRunConfig::UpdateRunConfig()
{
  //Bool_t res );
  printf("UpdateRunConfig NO IMPLEMENTED\n");
  return kFALSE;
}