Fun4AllSpillSRawEventOutputManager.cxx

Go to the documentation of this file.

#include <cstdlib>
#include <string>
#include <iostream>
#include <iomanip>
#include <TSystem.h>
#include <TFile.h>
#include <TTree.h>
#include <TSQLServer.h>
#include <phool/phool.h>
#include <phool/getClass.h>
#include <phool/PHNode.h>
#include <phool/PHNodeIOManager.h>
#include <phool/PHNodeIterator.h>
#include <interface_main/SQEvent.h>
#include <interface_main/SQSpillMap.h>
#include <interface_main/SQHitVector.h>
#include <ktracker/SRawEvent.h>
#include <db_svc/DbSvc.h>
#include "UtilSRawEvent.h"
#include "Fun4AllSpillSRawEventOutputManager.h"
using namespace std;
 
Fun4AllSpillSRawEventOutputManager::Fun4AllSpillSRawEventOutputManager(const std::string &dir_base, const string &myname)
  : Fun4AllOutputManager(myname)
  , m_dir_base(dir_base)
  , m_tree_name("save")
  , m_branch_name("rawEvent")
  , m_file_name("")
  , m_run_id(0)
  , m_spill_id(0)
  , m_file(0)
  , m_tree(0)
  , m_sraw(0)
  , m_evt(0)
  , m_sp_map(0)
  , m_hit_vec(0)
  , m_trig_hit_vec(0)
  , m_db(0)
  , m_name_schema("")
  , m_name_table("")
  , m_use_sqlite(false)
{
  ;
}
 
Fun4AllSpillSRawEventOutputManager::~Fun4AllSpillSRawEventOutputManager()
{
  CloseFile();
  if (m_sraw) delete m_sraw;
  if (m_db) delete m_db;
}
 
void Fun4AllSpillSRawEventOutputManager::EnableDB(const std::string name_schema, const std::string name_table, const bool refresh_db)
{
  m_name_schema = name_schema;
  m_name_table  = name_table;
  m_db = new DbSvc(DbSvc::DB1);
  m_db->UseSchema(m_name_schema, true);
 
  if (refresh_db && m_db->HasTable(m_name_table)) m_db->DropTable(m_name_table);
  if (! m_db->HasTable(m_name_table)) {
    DbSvc::VarList list;
    list.Add("run_id"   , "INT", true);
    list.Add("spill_id" , "INT", true);
    list.Add("file_name", "VARCHAR(256)");
    list.Add("status"   , "INT");
    list.Add("utime_b"  , "INT");
    list.Add("utime_e"  , "INT");
    m_db->CreateTable(m_name_table, list);
  }
}
 
int Fun4AllSpillSRawEventOutputManager::Write(PHCompositeNode *startNode)
{
  if (! m_evt) {
    m_evt          = findNode::getClass<SQEvent    >(startNode, "SQEvent");
    m_sp_map       = findNode::getClass<SQSpillMap >(startNode, "SQSpillMap");
    m_hit_vec      = findNode::getClass<SQHitVector>(startNode, "SQHitVector");
    m_trig_hit_vec = findNode::getClass<SQHitVector>(startNode, "SQTriggerHitVector");
    if (!m_evt || !m_hit_vec || !m_trig_hit_vec) {
      cout << PHWHERE << "Cannot find the SQ data nodes.  Abort." << endl;
      exit(1);
    }
  }
  int run_id = m_evt->get_run_id();
  int sp_id  = m_evt->get_spill_id();
  if (m_run_id != run_id || m_spill_id != sp_id) { // New spill
    CloseFile();
    m_run_id   = run_id;
    m_spill_id = sp_id;
    OpenFile();
  }
  SQSpill* sp = m_sp_map ? m_sp_map->get(sp_id) : 0;
  UtilSRawEvent::SetEvent     (m_sraw, m_evt);
  UtilSRawEvent::SetSpill     (m_sraw, sp);
  UtilSRawEvent::SetHit       (m_sraw, m_hit_vec);
  UtilSRawEvent::SetTriggerHit(m_sraw, m_trig_hit_vec);
  m_tree->Fill();
  return 0;
}
 
void Fun4AllSpillSRawEventOutputManager::CloseFile()
{
  if (Verbosity() > 0) cout << "Fun4AllSpillSRawEventOutputManager::CloseFile(): run " << m_run_id << ", spill " << m_spill_id << endl;
  if (! m_file) return;
  m_file->Write();
  m_file->Close();
  delete m_file;
  m_file = 0;
  UpdateDBStatus(CLOSE);
 
  if (m_db) { // MySQL DB
    int utime = time(0);
    ostringstream oss;
    oss << "update " << m_name_table << " set status = " << CLOSE << ", utime_e = " << utime
        << " where run_id = " << m_run_id << " and spill_id = " << m_spill_id;
    if (! m_db->Con()->Exec(oss.str().c_str())) {
      cerr << "!!ERROR!!  Fun4AllSpillSRawEventOutputManager::CloseFile():  " << oss.str() << endl;
      return;
    }
  }
}
 
void Fun4AllSpillSRawEventOutputManager::OpenFile()
{
  if (Verbosity() > 0) cout << "Fun4AllSpillSRawEventOutputManager::OpenFile(): run " << m_run_id << ", spill " << m_spill_id << endl;
  ostringstream oss;
  oss << m_dir_base << "/sraw/run_" << setfill('0') << setw(6) << m_run_id;
  gSystem->mkdir(oss.str().c_str(), true);
  oss << "/run_" << setw(6) << m_run_id << "_spill_" << setw(9) << m_spill_id << "_sraw.root";
  m_file_name = oss.str();
  if (Verbosity() > 9) cout << "  " << m_file_name << endl;
  m_file = new TFile(m_file_name.c_str(), "RECREATE");
  if (!m_file->IsOpen()) {
    cout << PHWHERE << "Could not open " << m_file_name << ".  Abort." << endl;
    exit(1);
  }
  if (!m_sraw) m_sraw = new SRawEvent();
  m_tree = new TTree(m_tree_name.c_str(), "");
  m_tree->Branch(m_branch_name.c_str(), &m_sraw);
  UpdateDBStatus(OPEN);
 
  if (m_db) { // MySQL DB
    int utime = time(0);
    ostringstream oss;
    oss << "insert into " << m_name_table
        << " values(" << m_run_id << ", " << m_spill_id << ", '" << m_file_name << "', 1, " << utime << ", 0)"
        << " on duplicate key update file_name = '" << m_file_name << "', status = " << OPEN << ", utime_b = " << utime << ", utime_e = 0";
    if (! m_db->Con()->Exec(oss.str().c_str())) {
      cerr << "!!ERROR!!  Fun4AllSpillSRawEventOutputManager::OpenFile(): " << oss.str() << endl;
      return;
    }
  }
}
 
 
void Fun4AllSpillSRawEventOutputManager::UpdateDBStatus(const int status)
{
  if (! m_use_sqlite) return;
  
  const char* table_name = "data_status";
  ostringstream oss;
  oss << m_dir_base << "/data_status.db";
  DbSvc db(DbSvc::LITE, oss.str());
 
  oss.str("");
  oss << "insert or ignore into " << table_name << " (run_id, spill_id) values (" << m_run_id << ", " << m_spill_id << ")";
  if (! db.Con()->Exec(oss.str().c_str())) {
    cerr << "!!ERROR!!  Fun4AllSpillSRawEventOutputManager::UpdateDBStatus()." << endl;
    return;
  }
  oss.str("");
  oss << "update " << table_name << " set utime_deco = strftime('%s', 'now'), status_deco = " << status << " where run_id = " << m_run_id << " and spill_id = " << m_spill_id;
  if (! db.Con()->Exec(oss.str().c_str())) {
    cerr << "!!ERROR!!  Fun4AllSpillSRawEventOutputManager::UpdateDBStatus()." << endl;
    return;
  }
}