source: src/FunctionApproximation/FunctionArgument.hpp@ fc08c7

/*
 * FunctionArgument.hpp
 *
 *  Created on: 02.10.2012
 *      Author: heber
 */

#ifndef FUNCTIONARGUMENT_HPP_
#define FUNCTIONARGUMENT_HPP_

// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <utility>
#include <iosfwd>

#include "FunctionApproximation/Subgraph/SubgraphEdge.hpp"

/** This class encapsulates all information with respect to a single argument
 *  for a high-dimensional model function.
 *
 *  We restrict ourselves here to a function that dependent on a set of
 *  three-dimensional vectors, i.e. a set of positions in space. And for
 *  the moment to distances in between these sets.
 *
 */
struct argument_t : public SubgraphEdge
{
  //!> grant operator access to private parts
  friend std::ostream& operator<<(std::ostream &ost, const argument_t &arg);

  /** Default constructor for class argument_t.
   *
   */
  argument_t() :
    distance(0.),
    globalid(-1),
    bonded(false)
  {}

  /** Constructor for class argument_t.
   *
   * This constructors uses the index pair (0,1) as default.
   *
   * \param _distance distance argument
   * \param _bonded is this a distance between bonded (true) or nonbonded (false) atoms
   */
  argument_t(const double &_distance, const bool _bonded = false) :
    SubgraphEdge(
      indices_t( std::make_pair(0,1) ),
      types_t( std::make_pair(0,0) )),
    distance(_distance),
    globalid(-1),
    bonded(_bonded)
  {}

  /** Constructor for class argument_t.
   *
   * \param _indices pair of indices associated with the \a _distance
   * \param _distance distance argument
   * \param _bonded is this a distance between bonded (true) or nonbonded (false) atoms
   */
  argument_t(const indices_t &_indices, const double &_distance, const bool _bonded = false) :
    SubgraphEdge(
      indices,
      types_t( std::make_pair(0,0) )),
    distance(_distance),
    globalid(-1),
    bonded(_bonded)
  {}

  /** Constructor for class argument_t.
   *
   * \param _indices pair of indices associated with the \a _distance
   * \param _types pair of particle type
   * \param _distance distance argument
   * \param _bonded is this a distance between bonded (true) or nonbonded (false) atoms
   */
  argument_t(
      const indices_t &_indices,
      const types_t &_types,
      const double &_distance,
      const bool _bonded = false) :
    SubgraphEdge(_indices, _types),
    distance(_distance),
    globalid(-1),
    bonded(_bonded)
  {}

  /** Comparator with respect to the distance.
   *
   * \note We'll have this as static function to allow usage in e.g. STL's sort.
   *
   * \param one first argument
   * \param other other argument to compare to \a one to
   * \return true - first distance is less
   */
  static bool DistanceComparator(const argument_t &one, const argument_t &other)
  {
    return one.distance < other.distance;
  }

  /** Comparator with respect to the pair of types.
   *
   * \note We'll have this as static function to allow usage in e.g. STL's sort.
   *
   * \param one first argument
   * \param other other argument to compare to \a one to
   * \return true - first type is less or if equal, second type is less, else
   */
  bool static IndexComparator(const argument_t &one, const argument_t &other)
  {
    return (static_cast<const SubgraphEdge &>(one) < static_cast<const SubgraphEdge &>(other));
  }

  //!> distance
  double distance;
  //!> global id refers to some global index, e.g. the configuration id in training set
  size_t globalid;
  //!> states whether this argument is between bonded (true) or nonbonded (false) atoms
  bool bonded;
};

/** Print given \a arg to stream \a ost.
 *
 * \param ost output stream to print to
 * \param arg argument to print
 * \return output stream for concatenation
 */
std::ostream& operator<<(std::ostream &ost, const argument_t &arg);


#endif /* FUNCTIONARGUMENT_HPP_ */