JDB/JSelector.hh

Go to the documentation of this file.

#ifndef __JDB_JSELECTOR__
#define __JDB_JSELECTOR__
 
#include <istream>
#include <ostream>
#include <string>
#include <vector>
 
#include "JLang/JException.hh"
#include "JLang/JLangToolkit.hh"
#include "JMath/JMath.hh"
 
#include "dbclient/KM3NeTDBClient.h"
 
 
/**
 * \author mdejong
 */
 
namespace JDATABASE {};
namespace JPP { using namespace JDATABASE; }
 
namespace JDATABASE {
 
  using JMATH::JMath;
  using JLANG::JIndexOutOfRange;
  using KM3NeT::DB::Selector;
 
 
  /**
   * Available operands.
   */
  static const Selector::RelOp* const OPERAND[] = 
    {
      &Selector::RelOp::Equal,
      &Selector::RelOp::Different,
      &Selector::RelOp::Less,
      &Selector::RelOp::LessEqual,
      &Selector::RelOp::Greater,
      &Selector::RelOp::GreaterEqual
    };
 
 
  /**
   * Number of available operands.
   */
  static const int NUMBER_OF_OPERANDS = sizeof(OPERAND) / sizeof(OPERAND[0]);
 
 
  /**
   * Get operand.
   *
   * \param  index          index
   * \return                operand
   */
  inline const Selector::RelOp& getOperand(const int index)
  {
    if (index >= 0 && index <= NUMBER_OF_OPERANDS)
      return *OPERAND[index];
    else
      THROW(JIndexOutOfRange, "Invalid index " << index);
  }
 
 
  /**
   * Auxiliary class for specifying selection of database data.
   */
  class JSelector :
    public std::vector<Selector>,
    public JMath<JSelector>
  {
  public:
    /**
     * End-of-line.
     */
    static const char EOL = ';';    
 
 
    /**
     * Default constructor.
     */
    JSelector()
    {}
 
 
    /**
     * Constructor.
     *
     * \param  key            key
     * \param  value          value
     * \param  operand        operand
     */
    template<class T>
    JSelector(const std::string& key, const T value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      add(key, value, operand);
    }
 
 
    /**
     * Constructor.
     *
     * \param  data_member    data member
     * \param  value          value
     * \param  operand        operand
     */
    template<class JClass_t, class JType_t>
    JSelector(JType_t JClass_t::*data_member, const JType_t value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      add(data_member, value, operand);
    }
 
 
    /**
     * Constructor.
     *
     * \param  data_member    data member
     * \param  value          value
     * \param  operand        operand
     */
    template<class JClass_t>
    JSelector(std::string JClass_t::*data_member, const std::string value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      add(data_member, value, operand);
    }
 
 
    /**
     * Add selection.
     *
     * \param  selection      selection
     * \return                this selection
     */
    JSelector& add(const JSelector& selection)
    {
      using namespace std;
 
      copy(selection.begin(), selection.end(), back_inserter(*this));
 
      return *this;
    }
 
 
    /**
     * Add selection.
     *
     * \param  key            key
     * \param  value          value
     * \param  operand        operand
     * \return                this selection
     */
    template<class T>
    JSelector& add(const std::string& key, const T value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      using namespace std;
 
      ostringstream os;
 
      os << value;
 
      push_back(Selector(key.c_str(), os.str().c_str(), operand));
 
      return *this;
    }
 
 
    /**
     * Add selection.
     *
     * \param  data_member    data member
     * \param  value          value
     * \param  operand        operand
     * \return                this selection
     */
    template<class JClass_t, class JType_t>
    inline JSelector& add(JType_t JClass_t::*data_member, const JType_t value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      return add(getColumn(data_member), value, operand);
    }
 
 
    /**
     * Add selection.
     *
     * \param  data_member    data member
     * \param  value          value
     * \param  operand        operand
     * \return                this selection
     */
    template<class JClass_t>
    inline JSelector& add(std::string JClass_t::*data_member, const std::string& value, const Selector::RelOp& operand = Selector::RelOp::Equal) 
    {
      return add(getColumn(data_member), value, operand);
    }
 
 
    /**
     * Read selector from input stream.
     *
     * \param  in             input stream
     * \param  selector       selector
     * \return                input stream
     */
    friend inline std::istream& operator>>(std::istream& in, JSelector& selector) 
    {
      using namespace std;
      using namespace JPP;
 
      for (string buffer; getline(in, buffer, JSelector::EOL); ) {
 
        int operand = -1;
 
        for (int i = 0; i != NUMBER_OF_OPERANDS; ++i) {
 
          const string::size_type pos = buffer.find(getOperand(i).Render());
 
          if (pos != string::npos && (operand == -1 || getOperand(i).Render().length() > getOperand(operand).Render().length())) {
            operand = i;
          }
        }
 
        if (operand != -1) {
 
          const string::size_type pos = buffer.find(getOperand(operand).Render());
 
          const string key   = JPP::trim(buffer.substr(0, pos));
          const string value = JPP::trim(buffer.substr(pos + getOperand(operand).Render().length()));
 
          if (key  .find_first_of("=!<>") == string::npos &&
              value.find_first_of("=!<>") == string::npos) {
 
            selector.push_back(Selector(key.c_str(), value.c_str(), getOperand(operand)));
 
          } else {
 
            in.setstate(ios::badbit);
          }
 
        } else {
 
          in.setstate(ios::badbit);
        }
      }
 
      return in;
    }
    
    
    /**
     * Write selector to output stream.
     *
     * \param  out            output stream
     * \param  selector       selector
     * \return                output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JSelector& selector) 
    {
      for (const_iterator i = selector.begin(); i != selector.end(); ++i) {
        out << i->Name << i->RelationalOperator.Render()  << i->Value << JSelector::EOL << std::endl;
      }
 
      return out;
    }
  };
}
 
#endif