JEditTransmitter.cc

Go to the documentation of this file.

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

#include "JDetector/JTransmitter.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;

  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 rot_t       = "rot";               //!< Rotate around z-axis by given value [rad]

  /**
   * Auxiliary class to apply transmitter modifications.
   */
  struct JModifier {
    /**
     * Apply modification to given transmitter.
     *
     * \param  transmitter transmitter
     * \return             true if valid action; else false
     */
    bool apply(JTransmitter& transmitter) const
    {
      switch (data.size()) {

      case 1:
        return  apply(transmitter, action, data[0]);                                        // orientation calibration

      case 3:
        return  apply(transmitter, action, JVector3D(data[0], data[1], data[2]));           // 3D position calibration

      default:
        return false;
      }
    }


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

      if (in >> modifier.id >> modifier.action) {

        modifier.data.clear();

        for (double x; in >> x; ) {
          modifier.data.push_back(x);
        }

        in.clear(ios_base::eofbit);
      }

      return in;
    }


    /**
     * 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;

      for (std::vector<double>::const_iterator i = modifier.data.begin(); i != modifier.data.end(); ++i) {
        out << ' ' << *i;
      }

      return out;
    }


    int                 id;
    std::string         action;
    std::vector<double> data;


  private:
    /**
     * Apply orientation calibration to given transmitter.
     *
     * \param  transmitter transmitter
     * \param  action      action
     * \param  value       value
     * \return             true if valid action; else false
     */
    static bool apply(JTransmitter& transmitter, const std::string& action, const double value)
    {
      if        (action == rot_t)
        transmitter.rotate(JRotation3Z(value));
      else
        return false;

      return true;
    }


    /**
     * Apply position calibration to given transmitter.
     *
     * \param  transmitter transmitter
     * \param  action      action
     * \param  pos         pos
     * \return             true if valid action; else false
     */
    static bool apply(JTransmitter& transmitter, const std::string& action, const JVector3D& pos)
    {
      if      (action == set_t)
        transmitter.setPosition(pos);
      else if (action == add_t)
        transmitter.add(pos);
      else if (action == sub_t)
        transmitter.sub(pos);
      else
        return false;

      return true;
    }
  };
}

/**
 * \file
 *
 * Auxiliary program to modify transmitter configuration.
 *
 * Syntax:
 * <pre>
 *     -S "<string number>         (set|add|sub) x y z"
 *     -S "<string number>         (rot) phi"
 * </pre>
 *
 * For option <tt>rot</tt>,
 * the angle <tt>phi</tt> refers to an anti-clockwise rotation around the z-axis.\n
 * The rotation angle is defined in radians.
 *
 * The options
 * <pre>
 *     -A "<identifier> <string number> <floor> x y z"
 *     -r "<string number>"
 * </pre>
 * can be used to add and remove a transmitter, respectively. 
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef JContainer< vector<JTransmitter> >  container_type;

  string               inputFile;
  string               outputFile;
  vector<JModifier>    mod;
  vector<JTransmitter> add;
  set<int>             keep;
  set<int>             rm;
  bool                 squash;
  int                  debug;

  try {

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

    zap['f'] = make_field(inputFile,   "transmitter input file")             = "";
    zap['o'] = make_field(outputFile,  "transmitter output file");
    zap['S'] = make_field(mod,         "transmitter modifier")               = JPARSER::initialised();
    zap['A'] = make_field(add,         "add transmitter")                    = JPARSER::initialised();
    zap['k'] = make_field(keep,        "keep transmitter[s]")                = JPARSER::initialised();
    zap['r'] = make_field(rm,          "remove transmitter[s]")              = JPARSER::initialised();
    zap['q'] = make_field(squash,      "squash meta data");
    zap['d'] = make_field(debug,       "debug level")                        = 2;

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

  container_type data;

  if (inputFile != "") {
    try {
      data.load(inputFile.c_str());
    }
    catch(const exception& error) {
      //FATAL(error.what() << endl);
    }
  }

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

  data.comment.add(JMeta(argc, argv));
  
  for (vector<JTransmitter>::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->getString() == i->id) {

        DEBUG("Modifier"                                                                                                << ' '
              << "(" << FILL(4,'0') << target->getString() << "," << FILL(2,'0') << target->getFloor() << FILL() << ")" << ' '
              << "action" << ' ' << i->action << JEEPZ() << i->data << endl);

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

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

    if ((keep.empty() || keep.count(i->getString()) != 0)  &&
        (rm  .empty() || rm  .count(i->getString()) == 0))
      ++i;
    else
      i = data.erase(i);
  }

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