PhotoelectricXMLHelper.h

#ifndef PHOTOELECTRICXMLHELPER_H
#define PHOTOELECTRICXMLHELPER_H 1

// CLHEP
#include <CLHEP/Vector/TwoVector.h>

// GEAR
#include <gearxml/tinyxml.h>
#include <gear/TPCParameters.h>


// stl
#include <vector>
#include <map>
#include <set>

namespace marlintpc{
struct Photodot {
  CLHEP::Hep2Vector _position;
  double _diameter;
  int _module;
  int _id;
  gear::TiXmlElement* _element;
  bool operator== (Photodot a){return a._id == _id;}
};

struct Photoline {
  CLHEP::Hep2Vector _start;
  CLHEP::Hep2Vector _end;
  double _width;
  int _module;
  gear::TiXmlElement* _element;
  bool operator== (Photoline a){return a._module == _module;}
};

class PhotoelectricXMLFile {
private:
  gear::TiXmlDocument _photodotFile; 

  std::vector<Photodot> _photodots;

  std::vector<Photoline> _photolines;
  static bool Comparison(std::pair<double, std::vector<Photodot>::const_iterator> i, std::pair<double, std::vector<Photodot>::const_iterator> j);
  void AddPhotodotProjection(gear::TiXmlElement* element, const CLHEP::Hep2Vector centroid, const std::set<gear::GlobalPadIndex> &pads);
  bool FillPhotodotInformation();

  bool FillPhotolineInformation();

public:
  PhotoelectricXMLFile(std::string xmlFile);
  ~PhotoelectricXMLFile(){};
  std::vector<std::pair<double, std::vector<Photodot>::const_iterator> > GetSortedPhotodots(double center_x, double center_y);

  std::vector<std::vector<Photodot>::const_iterator> GetPhotodots();

  std::vector<std::vector<Photoline>::const_iterator> GetPhotolines();

  void SetTimeInfo(float start, float end);
  bool GetTimeInfo(float &start, float &end);
  void SaveFile(std::string fileName){_photodotFile.SaveFile(fileName);};
  gear::TiXmlElement* GetPhotoElectricElement();
  void AddPhotodotProjection(int photodot_id, const CLHEP::Hep2Vector centroid, const std::set<gear::GlobalPadIndex> &pads);
  void AddPhotodotProjection(const CLHEP::Hep2Vector centroid, const std::set<gear::GlobalPadIndex> &pads);
  std::map<int, std::set<gear::GlobalPadIndex> > GetPhotodotPads();
  std::map<int, std::set<gear::GlobalPadIndex> > GetPhotolinePads();
  void AddPhotolineProjection(int module_id, const std::set<gear::GlobalPadIndex> &pads);
};
}
#endif