Fun4AllSpillSRawEventOutputManager.h

Go to the documentation of this file.

#ifndef __FUN4ALL_SPILL_SRAW_EVENT_OUTPUT_MANAGER_H__
#define __FUN4ALL_SPILL_SRAW_EVENT_OUTPUT_MANAGER_H__
#include <fun4all/Fun4AllOutputManager.h>
#include <string>
#include <vector>
class TFile;
class TTree;
class PHCompositeNode;
class SRawEvent;
class SQEvent;
class SQSpillMap;
class SQHitVector;
class DbSvc;
 
class Fun4AllSpillSRawEventOutputManager: public Fun4AllOutputManager
{
  typedef enum {
    UNDEF  = 0,
    OPEN   = 1,
    UPDATE = 2,
    CLOSE  = 3
  } Status_t;
  std::string m_dir_base;
  std::string m_tree_name;
  std::string m_branch_name;
  std::string m_file_name;
  int m_run_id;
  int m_spill_id;
  TFile* m_file;
  TTree* m_tree;
  SRawEvent* m_sraw;
  
  SQEvent* m_evt;
  SQSpillMap* m_sp_map;
  SQHitVector* m_hit_vec;
  SQHitVector* m_trig_hit_vec;
 
  DbSvc* m_db;
  std::string m_name_schema;
  std::string m_name_table;
  bool m_use_sqlite;
 
 public:
  Fun4AllSpillSRawEventOutputManager(const std::string &dir_base, const std::string &myname = "SPILLSRAWEVENTOUT");
  virtual ~Fun4AllSpillSRawEventOutputManager();
 
  void SetTreeName  (const std::string name) { m_tree_name   = name; }
  void SetBranchName(const std::string name) { m_branch_name = name; }
  void EnableDB(const std::string name_schema="user_e1039_maindaq", const std::string name_table="sraw_file_status", const bool refresh_db=false);
  void EnableSQLiteDB(const bool use=true) { m_use_sqlite = use; }
 
  virtual int Write(PHCompositeNode *startNode);
 
 protected:
  void CloseFile();
  void OpenFile();
  void UpdateDBStatus(const int status);
};
 
#endif /* __FUN4ALL_SPILL_SRAW_EVENT_OUTPUT_MANAGER_H__ */