PHG4DSTReader.h

Go to the documentation of this file.

// $Id: PHG4DSTReader.h,v 1.7 2015/02/27 23:42:23 jinhuang Exp $
 
#ifndef PHG4DSTREADER_H_
#define PHG4DSTREADER_H_
 
#include <fun4all/SubsysReco.h>
 
 
#ifndef __CINT__
 
#include <memory>
 
#endif
 
#include <iostream>
#include <string>
#include <set>
#include <vector>
 
class PHCompositeNode;
 
class PHG4Particle;
 
class TClonesArray;
class TTree;
 
 
class PHG4DSTReader : public SubsysReco
{
public:
  PHG4DSTReader(const std::string &filename);
  virtual
  ~PHG4DSTReader();
 
  int
  Init(PHCompositeNode *);
 
  int
  process_event(PHCompositeNode *);
 
  int
  End(PHCompositeNode *);
 
  void
  AddNode(const std::string &name)
  {
    _node_postfix.push_back(name);
  }
 
  void
  AddHit(const std::string &name)
  {
    _node_postfix.push_back(name);
  }
 
  void
  set_load_all_particle(bool b)
  {
    _load_all_particle = b;
  }
 
  void
  set_load_active_particle(bool b)
  {
    _load_active_particle = b;
  }
 
  void
  set_save_particle(bool b)
  {
    _save_particle = b;
  }
 
  void
  set_save_vertex(bool b)
  {
    _save_vertex = b;
  }
 
protected:
 
  std::vector<std::string> _node_postfix;
//  std::vector<std::string> _node_name;
  int nblocks;
 
#ifndef __CINT__
 
  typedef std::shared_ptr<TClonesArray> arr_ptr;
 
  struct record
  {
    unsigned int _cnt;
    std::string _name;
    arr_ptr _arr;
    TClonesArray * _arr_ptr;
 
    enum enu_type
    {
      typ_hit, typ_part, typ_vertex
    };
    enu_type _type;
  };
  typedef std::vector<record> records_t;
  records_t _records;
  /*
  typedef PHG4Particlev2 part_type;
  typedef PHG4HitEval hit_type;
  typedef PHG4VtxPointv1 vertex_type;
  typedef RawTowerv1 RawTower_type;
  typedef JetV1 PHPyJet_type;
  */
#endif
 
  int _event;
 
  std::string _out_file_name;
 
//  TFile * _file;
  TTree * _T;
 
  bool _save_particle;
 
  bool _load_all_particle;
 
  bool _load_active_particle;
 
  typedef std::set<int> PartSet_t;
  PartSet_t _particle_set;
  PartSet_t _vertex_set;
 
  bool _save_vertex;
 
#ifndef __CINT__
 
  void
  add_particle(record & rec, PHG4Particle * part);
 
#endif
 
  void
  build_tree();
};
 
#endif /* PHG4DSTREADER_H_ */