JEditMechanics.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <map>

#include "JAcoustics/JMechanics.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 mechanics

  /**
   * Auxiliary class to apply mechanics modifications.
   */
  struct JModifier {
    /**
     * Apply modification to given mechanics.
     *
     * \param  mechanics   mechanics
     * \return             true if valid action; else false
     */
    bool apply(JMechanics& mechanics) const
    {
      if      (action == set_t)
        mechanics = this->mechanics;
      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.mechanics;
    }


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

      return out;
    }


    int                 id;
    std::string         action;
    JMechanics          mechanics;
  };
}

/**
 * \file
 *
 * Auxiliary program to modify mechanical model data.
 *
 * Syntax:
 * <pre>
 *     -M "<string number>     (set) a b"
 * </pre>
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef JContainer< map<int, JMechanics> >  container_type;

  string               file_name;
  vector<JModifier>    mod;
  int                  debug;

  try {

    JParser<> zap("Auxiliary program to modify mechanical model data.");

    zap['f'] = make_field(file_name,   "mechanics file");
    zap['M'] = make_field(mod,         "mechanics modifier")                 = JPARSER::initialised();
    zap['d'] = make_field(debug)                                             = 1;
    
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  container_type data;

  data.load(file_name.c_str());

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

    DEBUG("Modifier"                                      << ' '
          << "(" << FILL(2,'0') << i->id << FILL() << ")" << ' '
          << "action" << ' ' << i->mechanics << endl);

    JMechanics mechanics;

    container_type::iterator p = data.find(i->id);

    if (p != data.end()) {
      mechanics = p->second;
    }
    
    if (!i->apply(mechanics))
      ERROR("No valid action: " << *i << endl);
    else
      data[i->id] = mechanics;
  }

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