JDBReader.hh

Go to the documentation of this file.

#ifndef __JDB_JDBREADER__
#define __JDB_JDBREADER__

#include <string>
#include <vector>

#include "dbclient/KM3NeTDBClient.h"

#include "JLang/JNullStream.hh"
#include "JLang/JEquationParameters.hh"
#include "JLang/JRedirectString.hh"
#include "JLang/JSingleton.hh"
#include "JLang/JLangToolkit.hh"
#include "JROOT/JRootClass.hh"
#include "JROOT/JRootDictionary.hh"
#include "JROOT/JRootStreamer.hh"

#include "JDB/JDBDictionary.hh"


/**
 * \author mdejong
 */
namespace JDATABASE {}
namespace JPP { using namespace JDATABASE; }

namespace JDATABASE {

  using JLANG::JSingleton;
  using KM3NeT::DB::ResultSet;
  using KM3NeT::DB::Selector;

  /**
   * Special characters.
   */
  static const std::string SPECIAL_CHARACTERS  =  "!@#$%^&*()-+=[]{};:'\"\\|,.<>/?";


  /**
   * Read ROOT object from selector.
   *
   * \param  selection     selection
   * \param  object        object
   * \return               true if read; else false
   */
  template<class T>
  inline bool operator>>(const std::vector<Selector>& selection, T& object)
  {
    using namespace std;
    using namespace JPP;

    JRootReader        reader(null, JEquationParameters(), JDBDictionary::getInstance());

    JRootReadableClass cls(object);

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

      string parameter = to_upper(i->Name);

      for (string::const_iterator i = SPECIAL_CHARACTERS.begin(); i != SPECIAL_CHARACTERS.end(); ++i) {
        for (string::size_type pos; (pos = parameter.find(*i)) != string::npos; ) {
          parameter.erase(pos);
        }
      }
     
      JRootReadableClass abc = cls.find(parameter.c_str());
    
      if (abc.is_valid()) {     

        JRedirectString redirect(reader, i->Value);
      
        reader.getObject(abc);
      }
    }

    return true;
  }


  /**
   * Auxiliary class to read ROOT object from database.
   *
   * The names of the table columns are one-to-one mapped to the names of the data members 
   * of the corresponding data structure using the ROOT dictionary.\n 
   * The method JDBReader::set can be used to <em>a priori</em> assign default values 
   * to those data members that are defined by the selector used in the query.\n
   * Note that the list of column names associated to the template parameter is considered to be permanent.\n
   * If not, the static method JDBReader::reset should be called before the next read operation.
   */
  template<class T>
  struct JDBReader :
    public JSingleton< JDBReader<T> >
  {
    /**
     * Default constructor.
     */
    JDBReader() :
      object(),
      reader(JLANG::null, JLANG::JEquationParameters(), JDBDictionary::getInstance())
    {}


    /**
     * Reset internal lookup table.
     */
    static void reset()
    {
      JDBReader<T>::getInstance().buffer.clear();

      JDBReader<T>::getInstance().value = T();
    }


    /**
     * Set default value corresponding to given selection.
     *
     * \param  selection     selection
     */
    static void set(const std::vector<Selector>& selection)
    {
      selection >> JDBReader<T>::getInstance().value;
    }


    /**
     * Read ROOT object from result set.
     *
     * \param  rs            result set
     * \return               object
     */
    const T& operator()(ResultSet& rs) 
    {
      using namespace std;
      using namespace JPP;

      object = value;

      if (buffer.empty()) {

        JRootReadableClass cls(object);

        for (size_t i = 0; i != rs.FieldCount(); ++i) {

          string parameter = rs.FieldName(i);

          for (string::const_iterator i = SPECIAL_CHARACTERS.begin(); i != SPECIAL_CHARACTERS.end(); ++i) {
            for (string::size_type pos; (pos = parameter.find(*i)) != string::npos; ) {
              parameter.erase(pos);
            }
          }
        
          buffer.push_back(cls.find(parameter.c_str()));
        }
      }

      for (unsigned int i = 0; i != rs.FieldCount(); ++i) {

        if (buffer[i].is_valid()) {     

          JRedirectString redirect(reader, rs.GetString(i));
      
          reader.getObject(buffer[i]);
        }
      }

      return object;
    }

  private:
    T                  object;                      //!< internal value
    T                  value;                       //!< default  value
    JROOT::JRootReader reader;

    std::vector<JROOT::JRootReadableClass> buffer;  // field count -> data member
  };


  /**
   * Read ROOT object from result set.
   *
   * \param  rs            result set
   * \param  object        object
   * \return               true if read; else false
   */
  template<class T>
  inline bool operator>>(ResultSet& rs, T& object)
  {
    JDBReader<T>& reader = JDBReader<T>::getInstance();

    if (rs.Next()) {

      object = reader(rs);

      return true;
    }

    return false;
  }


  /**
   * Read vector of ROOT objects from result set.
   *
   * \param  rs            result set
   * \param  buffer
   * \return               true if read; else false
   */
  template<class JElement_t, class JAllocator_t>
  inline bool operator>>(ResultSet& rs, std::vector<JElement_t, JAllocator_t>& buffer)
  {
    for (JElement_t object; rs >> object; ) {
      buffer.push_back(object);
    }

    return true;
  }
}

#endif