JEllipse.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
 
#include "TROOT.h"
#include "TFile.h"
#include "TEllipse.h"
 
#include "Jeep/JContainer.hh"
#include "Jeep/JProperties.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * Auxiliary data structure for ellipse.
 */
struct JEllipse {
 
  /**
   * Read ellipse from input stream.
   *
   * \param  in         input stream
   * \param  ellipse    ellipse
   * \return            input stream
   */
  friend inline std::istream& operator>>(std::istream& in, JEllipse& ellipse)
  {
    if (in >> ellipse.x1 >> ellipse.y1 >> ellipse.r1) {
 
      if (in >> ellipse.r2) {
 
        if (in >> ellipse.phimin >> ellipse.phimax) {
          
        } else {
 
          ellipse.phimin =   0.0;
          ellipse.phimax = 360.0;
        }
        
      } else {
 
        ellipse.r2     = ellipse.r1;
        ellipse.phimin =   0.0;
        ellipse.phimax = 360.0;
      }
 
      in.clear();
    }
 
    return in;
  }
 
  /**
   * Write ellipse to output stream.
   *
   * \param  out        output stream
   * \param  ellipse    ellipse
   * \return            output stream
   */
  friend inline std::ostream& operator<<(std::ostream& out, const JEllipse& ellipse)
  {
    out << ellipse.x1     << ' '
        << ellipse.y1     << ' '
        << ellipse.r1     << ' '
        << ellipse.r2     << ' '
        << ellipse.phimin << ' '
        << ellipse.phimax;
 
    return out;
  }
  
  double x1;
  double y1;
  double r1;
  double r2;
  double phimin;
  double phimax;
};
 
 
/**
 * \file
 * Auxiliary program to create TEllipse.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  typedef JContainer< vector<JEllipse> >  JParameters_t;
 
  string         outputFile;
  JParameters_t  parameters;
  struct {
    Color_t        color        = kBlack;
    Style_t        style        = 0;            // hollow
  } fill;
  struct {
    Color_t        color        = kBlack;
    Style_t        style        = kSolid;
    Width_t        width        = 1;
  } line;
  int            debug;
 
  try {
 
    JProperties properties(JEquationParameters("=", "\n", "", "#"));
 
    properties.insert(zmake_property(fill.color));
    properties.insert(zmake_property(fill.style));
    properties.insert(zmake_property(line.color));
    properties.insert(zmake_property(line.style));
    properties.insert(zmake_property(line.width));
 
    JParser<> zap("Auxiliary program to create TEllipse");
 
    zap['o'] = make_field(outputFile);
    zap['p'] = make_field(parameters,   "x y r1 [r2 [phimin phimax]]");
    zap['@'] = make_field(properties,   "ellipse attributes")  =  JPARSER::initialised();
    zap['d'] = make_field(debug)        = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  TFile out(outputFile.c_str(), "recreate");
 
  for (size_t i = 0; i != parameters.size(); ++i) {
    
    TEllipse* p = new TEllipse(parameters[i].x1, parameters[i].y1, parameters[i].r1, parameters[i].r2, parameters[i].phimin, parameters[i].phimax);
 
    p->SetFillColor(fill.color);
    p->SetFillStyle(fill.style);
    p->SetLineColor(line.color);
    p->SetLineStyle(line.style);
    p->SetLineWidth(line.width);
 
    out.WriteTObject(p);
  }
 
  out.Write();
  out.Close();
}