PHG4DSTReader.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4DSTReader.h

// $Id: PHG4DSTReader.h,v 1.7 2015/02/27 23:42:23 jinhuang Exp $

#ifndef G4EVAL_PHG4DSTREADER_H
#define G4EVAL_PHG4DSTREADER_H

#include <fun4all/SubsysReco.h>

#include <boost/smart_ptr.hpp>

#include <set>
#include <string>
#include <vector>

class PHCompositeNode;

class PHG4Particle;

class TClonesArray;
class TTree;

class PHG4DSTReader : public SubsysReco
{
 public:
  PHG4DSTReader(const std::string &filename);
  ~PHG4DSTReader() override;

  int Init(PHCompositeNode *) override;

  int process_event(PHCompositeNode *) override;

  int End(PHCompositeNode *) override;

  void
  AddNode(const std::string &name)
  {
    _node_postfix.push_back(name);
  }

  void
  AddHit(const std::string &name)
  {
    _node_postfix.push_back(name);
  }

  void
  AddTower(const std::string &name)
  {
    _tower_postfix.push_back(name);
  }

  void
  AddJet(const std::string &name)
  {
    _jet_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;
  }

  double
  get_tower_zero_sup()
  {
    return _save_particle;
  }

  void
  set_tower_zero_sup(double b)
  {
    _tower_zero_sup = b;
  }

 protected:
  std::vector<std::string> _node_postfix;
  std::vector<std::string> _tower_postfix;
  std::vector<std::string> _jet_postfix;
  //  std::vector<std::string> _node_name;
  int nblocks = 0;

  typedef boost::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,
      typ_tower,
      typ_jets
    };
    enu_type _type;
  };
  typedef std::vector<record> records_t;
  records_t _records;

  int _event = 0;

  std::string _out_file_name;

  //  TFile * _file;
  TTree *_T = nullptr;

  bool _save_particle = true;

  bool _load_all_particle = true;

  bool _load_active_particle = true;

  typedef std::set<int> PartSet_t;
  PartSet_t _particle_set;
  PartSet_t _vertex_set;

  bool _save_vertex = true;

  double _tower_zero_sup = 0.;

  void
  add_particle(record &rec, PHG4Particle *part);

  void
  build_tree();
};

#endif /* G4EVAL_PHG4DSTREADER_H */