fPostSlatId_db.C

#include "TROOT.h"
#include "TDataSet.h"
#include "TDatime.h"
#include "TString.h"
#include "TSystem.h"

#include <iostream>
#include <fstream>

void fPostSlatId_db(
                const char* opt   = "",
                const char* year  = "17sim",
                const char* input = "fPostSlatId.txt")
{
    // storeTime is beginning time for validity range in case of WRITING DB
    TString option(opt), yr(year), storeTime;
    int date, time; // time for READING DB
    
    std::cout <<"year = " <<year <<std::endl;
    if (yr.Contains("17sim"))
        {
            storeTime = "2016-12-10 00:00:00";
            date = 20161210;
            time = 0;
        }
    else if (yr.Contains("17ofl"))
        {
            storeTime = "2016-12-20 00:00:01";
            date = 20161220;
            time = 1;
        }
    else { std::cout << "Please specify valid year tag\n"; exit; }
    std::cout << "Opt ="         << opt << "\n";
    std::cout << "write = "      << option.Contains("writedb") << "\n";
    std::cout << "storetime = "  << storeTime << "\n";
    std::cout << "date, time = " << date <<" "<< time << "\n";
    
    gROOT->Macro("./loadlib.C");
    
    //-------------------------------------------
    
    const Int_t MAX_DB_INDEX = 241;
    fpostSlatId_st in[MAX_DB_INDEX];
    
    FILE *FP = fopen(input, "r");
    if (!FP) { printf("Could not open %s\n", input); exit; }
    printf("\nReading %s\n", input);
    
    char line[1000];
    int n = 0;
    while (fgets(line, 1000, FP) != NULL)
    {
        sscanf(line,"%d %d %d %d ",
                                &in[n].slatid, &in[n].quad, &in[n].layer, &in[n].slat);
        printf("slatId=%3d quad=%2d layer=%2d slat=%2d\n",
                                in[n].slatid, in[n].quad, in[n].layer, in[n].slat);
        n++;
    }
    printf("Found %d entries\n", n);
    
    //-------------------------------------------
    
    if (option.Contains("writedb"))
        {
            gSystem->Setenv("DB_ACCESS_MODE", "write");
            StDbManager*    mgr     = StDbManager::Instance();
            StDbConfigNode* node    = mgr->initConfig("Geometry_fps");
            StDbTable*      dbtable = node->addDbTable("fpostSlatId");
            mgr->setStoreTime(storeTime.Data());
            dbtable->SetTable((char*)&in, MAX_DB_INDEX);
            if (yr.Contains("sim")) dbtable->setFlavor("sim");
            mgr->storeDbTable(dbtable);
            std::cout << "INFO: table saved to database" << std::endl;
        }

        //-------------------------------------------

    if (option.Contains("readdb"))
        {
            std::cout << "INFO: Reading database" << std::endl;
            gSystem->Unsetenv("DB_ACCESS_MODE");
            //gSystem->Unsetenv("DB_SERVER_LOCAL_CONFIG");
            
            St_db_Maker* dbMk = new St_db_Maker("db", "MySQL:StarDb", "$STAR/StarDb");
            dbMk->SetDebug();
            dbMk->SetDateTime(date, time); // event or run start time, set to your liking
            if      (yr.Contains("ofl")) { dbMk->SetFlavor("ofl"); }
            else if (yr.Contains("sim")) { dbMk->SetFlavor("sim"); }
            dbMk->Init();
            dbMk->Make();
            
            TDataSet *DB = 0;
            DB = dbMk->GetDataBase("Geometry/fps/fpostSlatId");
            if (!DB) std::cout << "ERROR: no table found in db, or malformed local db config" << std::endl;
            
            St_fpostSlatId *dataset = 0;
            dataset = (St_fpostSlatId*) DB->Find("fpostSlatId");
            if (!dataset) { std::cout << "ERROR: dataset does not contain requested table" << std::endl; return; }
            Int_t rows = dataset->GetNRows();
            if (rows > 1) std::cout << "INFO: found INDEXED table with " << rows << " rows" << std::endl;
            
            TDatime val[2];
            dbMk->GetValidity((TTable*)dataset, val);
            std::cout << "Dataset validity range: [ " << val[0].GetDate() << "." << val[0].GetTime() << " - "
                      << val[1].GetDate() << "." << val[1].GetTime() << " ] " << std::endl;
            
            fpostSlatId_st *table = (fpostSlatId_st*)dataset->GetTable();
            
            for (Int_t i = 0; i < rows; i++)
                {
                    std::cout << Form("Row=%3d slatid=%3d quad=%2d layer=%2d slat=%2d\n",
                                      i, table[i].slatid, table[i].quad, table[i].layer, table[i].slat);
                }
        }
    return;
}//Main