fms_db_fmsgaincorr.C

#include <iostream.h> 
#include <fstream.h>

// max index dimensions
static const Int_t MAX = 2500;
static const Int_t MAXD = 14;
static const Int_t MAXCH= 578;

int getDetectorId(int ew, int nstb) {return (ew-1)*8+nstb-1;}
int getEW(int detid) {return detid/8 + 1;}
int getNSTB(int detid) {return detid%8 + 1;}

void fms_db_fmsgaincorr(char* option = "writetext 15sim", char* dataspec="allone", 
                        char* storetime="2007-11-09 12:00:00",
                        int readDate=20071210, int readTime=0, int isSim=0) {
  // that's begin time for validity range for WRITING TO DB 
  // your data will be available from storeTime till 2037    
  TString opt(option);
  TString data(dataspec);
  TString storeTime(storetime);  
  if(opt.Contains("run8") && opt.Contains("dAu200")){
      storeTime = "2007-11-09 12:00:00";
  }else if(opt.Contains("run8") && opt.Contains("pp200")){
      storeTime = "2008-01-28 12:00:00";
  }else if(opt.Contains("run9") && opt.Contains("pp200")){
      storeTime = "2009-01-16 00:00:00";
  }else if(opt.Contains("15sim")){
      storeTime = "2014-12-10 00:00:01"; readDate = 20141215; readTime = 0; isSim=1;
  }else if(opt.Contains("15ofl")){
      storeTime = "2014-12-20 00:00:00"; readDate = 20141225; readTime = 0;
  }      
  //else std::cout<<"Invalid year range"<<std::endl;
  
  std::cout << "Opt =" << option << "\n";
  std::cout << "testinput = " << opt.Contains("testinput")  << "\n";
  std::cout << "readtext  = " << opt.Contains("readtext")  << "\n";
  std::cout << "readdb    = " << opt.Contains("readdb")    << "\n";
  std::cout << "writedb   = " << opt.Contains("writedb")   << "\n";
  std::cout << "writetext = " << opt.Contains("writetext") << "\n";
  std::cout << "dataspec = "  << dataspec << endl; 
  std::cout << "storeTime = " << storeTime << endl; 
  std::cout << "readData = "  << readDate << " readTime="<<readTime<<endl; 
  std::cout << "isSim = "     << isSim << endl; 

  gROOT->Macro("LoadLogger.C");
  gSystem->Load("St_base.so");
  gSystem->Load("libStDb_Tables.so");
  gSystem->Load("StDbLib.so");

  // structure to fill up
  fmsGainCorrection_st corr[MAX];
  int idx[MAXD][MAXCH];
  memset(idx,-1,sizeof(idx));
  for(int i=0; i<MAX; i++){
      corr[i].detectorId=-1;
      corr[i].ch        =-1;
      corr[i].corr      =-1.0;
  }
  
  int detId[MAXD]  = { 0,  1, 2, 3,  4,  5,  6,  7,   8,   9,   10,  11,  12,  13};
  int detType[MAXD]= { 0,  0, 1, 1,  2,  2,  3,  3,   4,   4,    4,   4,   5,   5};
  int detNx[MAXD]  = { 7,  7, 7, 7, 48, 48,  1,  1,  17,  17,   12,  12,   9,   9};
  int detNy[MAXD]  = { 7,  7, 1, 1,  1,  1, 48, 48,  34,  34,   24,  24,  12,  12};
  int ew[MAXD]     = { 0,  0, 0, 0,  0,  0,  0,  0,   1,   1,    1,   1,   1,   1};
  int nstb[MAXD]   = { 1,  2, 5, 6,  3,  4,  7,  8,   1,   2,    3,   4,   1,   2};
  int maxch[MAXD]  = {49, 49, 7, 7, 48, 48, 48, 48, 578, 578,  288, 288, 108, 108};
  
  int index=0;
  for(int det=0; det<MAXD; det++){
      for(int ch=0; ch<maxch[det]; ch++){
          corr[index].detectorId=det;
          corr[index].ch        =ch+1;
          corr[index].corr      =0.0;
          idx[det][ch]          =index;   
          if(data.Contains("allone")) corr[index].corr=1.0;           
          index++;        
      }
  }
   
  if(opt.Contains("readtext")){
      FILE* fp;
      int rew,rnstb,rch;
      float rcorr;
      int nread = 0;
      cout << "Reading "<<dataspec<<"\n";
      if(fp=fopen(dataspec,"r")){
          while(fscanf(fp,"%d %d %d %f",&rew,&rnstb,&rch,&rcorr) != EOF){
              int detid=getDetectorId(rew,rnstb);
              int index=idx[detid][rch-1];
              printf("EW=%5d NSTB=%5d CH=%5d Gain=%6.4f | det=%2d idx=%4d\n",rew,rnstb,rch,rcorr,detid,index);    
              if(rew==1) continue; //hack! ignore east
              if(detid != corr[index].detectorId) printf("ERR DetId %d != %d\n",detid,corr[index].detectorId); 
              if(rch   != corr[index].ch)         printf("ERR Ch    %d != %d\n",rch,corr[index].ch);
              corr[index].corr=rcorr;
              nread++;
          }
      }
      fclose(fp);
      printf("read %d channels of gain corr\n",nread);
  }

  if(opt.Contains("readdb")){
    gSystem->Load("StChain");
    gSystem->Load("StBFChain");
    gSystem->Load("StUtilities");
    gSystem->Load("StIOMaker");
    gSystem->Load("StarClassLibrary");
    gSystem->Load("St_Tables");
    gSystem->Load("StDbLib");
    gSystem->Load("StDbBroker");
    gSystem->Load("St_db_Maker");

    St_db_Maker *dbMk=new St_db_Maker("db", "MySQL:StarDb", "$STAR/StarDb");
    dbMk->SetDebug();
    dbMk->SetDateTime(readDate,readTime);
    dbMk->SetFlavor("ofl"); // for offline calibrations/mapping
    // dbMk->SetFlavor("simu"); // for simulations
    dbMk->Init();
    dbMk->Make();

    // this is done inside ::Make method
    TDataSet *DB = 0;
    // "dbMk->" will NOT be needed. 
    //  if done inside your FmsDbMaker. Simply use DB = GetInputDb("Calibrations/fms/mapping")
    DB = dbMk->GetInputDB("Calibrations/fms");
    if (!DB) { std::cout << "ERROR: no db maker?" << std::endl; }

    // fetch ROOT descriptor of db table
    St_fmsGainCorrection *dbppmap = 0;
    dbppmap = (St_fmsGainCorrection*) DB->Find("fmsGainCorrection");
    // fetch data and place it to appropriate structure
    int ngain = 0;
    if (dbppmap) {
      std::cout << "Reading fmsGainCorrection table\n";
      fmsGainCorrection_st *pptable = dbppmap->GetTable();
      for(int iew=0; iew<2; iew++){
        for(int mod=0; mod<4; mod++){
          for(int ch0=0; ch0<maxch[mod+iew*4]; ch0++){
            printf("%3d%8d%8d%8.4f\n",int(pptable[ngain].detectorId/8+1),int(pptable[ngain].detectorId%8+1),
                   int(pptable[ngain].ch),float(pptable[ngain].corr) );
            ngain++;
          }
        }
      }
      memcpy(corr,pptable,sizeof(corr));
    }
    else {
      std::cout << "WARNING: No data in fmsGain table (wrong timestamp?). Nothing to return, then.\n";
    }
  }

  if(opt.Contains("writetext")){
    int ii = data.Length();
    data.Insert(ii-4,"_out");
    char* file=data.Data();
    FILE* fp;
    cout << "Writing "<<file<<"\n";
    int ngain = 0;
    if(fp=fopen(file,"w")){
        for(int i=0; i<MAX; i++){
            if(corr[i].corr>=0.0){
                int detid=corr[i].detectorId;
                int iew=getEW(detid);
                int instb=getNSTB(detid);
                fprintf(fp,"%d %d %d %5.3f\n",iew,instb,
                        corr[i].ch,corr[i].corr);
                ngain++;
            }
        }
    }
    fclose(fp);
    printf("Write %d channels\n",ngain);
  }

  if(opt.Contains("writedb")) {
    gSystem->Setenv("DB_ACCESS_MODE","write");
    //putenv("DB_ACCESS_MODE=write");
    //char* env = getenv("DB_ACCESS_MODE");
    //cout <<  "Setting DB_ACCESS_MODE " << env << endl;
    StDbManager* mgr = StDbManager::Instance();
    StDbConfigNode* node = mgr->initConfig("Calibrations_fms");
    StDbTable* table = node->addDbTable("fmsGainCorrection");
    mgr->setStoreTime(storeTime.Data());
    // store data in the table
    table->SetTable((char*)&corr,index+1);
    // set store time
    // store table in dBase
    if(isSim) table->setFlavor("sim");
    mgr->storeDbTable(table);
    //StDbTable* table = node->findTable("fmsGain");
     //node->removeTable(table);
    gSystem->Unsetenv("DB_ACCESS_MODE");
    //unsetenv("DB_ACCESS_MODE");
    std::cout << "Done with database upload with nrow=" << index+1 << endl;
  }
}