JEditTripod.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <vector>

#include "TRandom3.h"

#include "JDetector/JTripod.hh"

#include "JSupport/JMeta.hh"

#include "Jeep/JContainer.hh"
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"


namespace {
  
  using namespace JPP;

  /**
   * Wild card for tripod identifier.
   */
  static const int         WILDCARD    =  -1;

  static const std::string set_t       = "set";               //!< Set      position
  static const std::string add_t       = "add";               //!< Add      position
  static const std::string sub_t       = "sub";               //!< Subtract position

  static const std::string rand_t      = "rand";              //!< Random value(s)
  static const std::string randset_t   =  rand_t + set_t;     //!< Set      position
  static const std::string randadd_t   =  rand_t + add_t;     //!< Add      position
  static const std::string randsub_t   =  rand_t + sub_t;     //!< Subtract position

  /**
   * Auxiliary class to apply tripod modifications.
   */
  struct JModifier {
    /**
     * Apply modification to given tripod.
     *
     * \param  tripod      tripod
     * \return             true if valid action; else false
     */
    bool apply(JTripod& tripod) const
    {
      const JUTMPosition randpos(gRandom->Gaus(0.0, pos.getUTMEast()),
                                 gRandom->Gaus(0.0, pos.getUTMNorth()),
                                 gRandom->Gaus(0.0, pos.getUTMZ()));
      
      if      (action == set_t)                          // actions with fixed values
        tripod.setUTMPosition(pos);
      else if (action == add_t)
        tripod.add(pos);
      else if (action == sub_t)
        tripod.sub(pos);
      else if (action == randset_t)                      // actions with random values
        tripod.setUTMPosition(randpos);
      else if (action == randadd_t)
        tripod.add(randpos);
      else if (action == randsub_t)
        tripod.sub(randpos);
      else
        return false;

      return true;
    }


    /**
     * Read modifier from input.
     *
     * \param  in          input stream
     * \param  modifier    modifier
     * \return             input stream
     */
    friend inline std::istream& operator>>(std::istream& in, JModifier& modifier)
    {
      return in >> modifier.id >> modifier.action >> modifier.pos;
    }


    /**
     * Write modifier to output.
     *
     * \param  out         output stream
     * \param  modifier    modifier
     * \return             output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JModifier& modifier)
    {
      out << modifier.id;
      out << ' ';
      out << modifier.action;
      out << ' ';
      out << modifier.pos;

      return out;
    }


    int                 id;
    std::string         action;
    JUTMPosition        pos;
  };
}

/**
 * \file
 *
 * Auxiliary program to modify tripod configuration.
 *
 * Syntax:
 * <pre>
 *     -T "<tripod identifier>     (set|add|sub) east north z"
 * </pre>
 *
 * The options
 * <pre>
 *     -A "<tripod identifier> east north z"
 *     -D "<tripod identifier>"
 * </pre>
 * can be used to add and remove a tripod, respectively. 
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef JContainer< vector<JTripod> >  container_type;

  string               file_name;
  vector<JModifier>    mod;
  vector<JTripod>      add;
  vector<int>          del;
  bool                 rm;
  int                  debug;

  try {

    JParser<> zap("Auxiliary program to modify tripod configuration.");

    zap['f'] = make_field(file_name,   "tripod file");
    zap['T'] = make_field(mod,         "tripod modifier")                    = JPARSER::initialised();
    zap['A'] = make_field(add,         "add tripod")                         = JPARSER::initialised();
    zap['D'] = make_field(del,         "delete tripod")                      = JPARSER::initialised();
    zap['r'] = make_field(rm,          "remove previous comments");
    zap['d'] = make_field(debug)                                             = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  gRandom->SetSeed(0);

  container_type data;

  try {
    data.load(file_name.c_str());
  }
  catch(const exception&) {}

  if (rm) {
    data.comment.clear();
  }

  data.comment.add(JMeta(argc, argv));
  
  for (vector<JTripod>::const_iterator i = add.begin(); i != add.end(); ++i) {
    data.push_back(*i);
  }

  for (vector<JModifier>::const_iterator i = mod.begin(); i != mod.end(); ++i) {

    for (container_type::iterator target = data.begin(); target != data.end(); ++target) {

      if (target->getID() == i->id || i->id == WILDCARD) {

        DEBUG("Modifier"                                                << ' '
              << "(" << FILL(2,'0') << target->getID() << FILL() << ")" << ' '
              << "action" << ' ' << i->pos << endl);

        if (!i->apply(*target)) {
          ERROR("No valid action: " << *i << endl);
        }
      }
    }
  }

  for (vector<int>::const_iterator i = del.begin(); i != del.end(); ++i) {
    for (container_type::iterator target = data.begin(); target != data.end(); ) {
      if (target->getID() == *i)
        target = data.erase(target);
      else
        ++target;
    }
  }

  data.store(file_name.c_str());
}