JProductRouter.hh

Go to the documentation of this file.

#ifndef __JDB__JPRODUCTROUTER__
#define __JDB__JPRODUCTROUTER__

#include <ostream>
#include <map>

#include "JDB/JDetectorIntegration_t.hh"
#include "JDB/JPBS_t.hh"
#include "JDB/JUPI_t.hh"
#include "JDB/JPBSSequence.hh"
#include "JLang/JComparable.hh"
#include "Jeep/JPrint.hh"


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

namespace JDATABASE {

  using JLANG::JComparable;


  /**
   * Auxiliary data structure for location of product in detector.
   */
  struct JLocation_t :
    public JComparable<JLocation_t>
  {
    /**
     * Default constructor.\n
     * The default corresponds to an invalid location.
     */
    JLocation_t() :
      string  (-1),
      floor   (-1),
      position(-1)
    {}


    /**
     * Constructor.
     *
     * \param  string        string
     * \param  floor         floor
     * \param  position      position
     */
    JLocation_t(const int string,
                const int floor,
                const int position) :
      string  (string),
      floor   (floor),
      position(position)
    {}


    /**
     * Check validity of location.
     *
     * \return               true if valid; else false
     */
    bool is_valid() const
    {
      return *this != JLocation_t();
    }


    /**
     * Less-than method.
     *
     * \param  location      location
     * \return               true if this location less than given location; else false
     */
    bool less(const JLocation_t& location) const
    {
      if (this->string == location.string) {

        if (this->floor == location.floor)
          return this->position < location.position;
        else
          return this->floor    < location.floor;

      } else {

        return this->string < location.string;
      }
    }


    /**
     * Write location to output stream.
     *
     * \param  out        output stream
     * \param  object     location
     * \return            output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JLocation_t& object)
    {
      using namespace std;
      using namespace JPP;

      return out << right
                 << FILL(4,'0') << object.string   << '.'
                 << FILL(2,'0') << object.floor    << '.'
                 << FILL(2,'0') << object.position << FILL() << left;
    }


    int string;           //!< position in detector
    int floor;            //!< position in string
    int position;         //!< position in floor
  };


  /**
   * Auxiliary class to map %UPI to location in detector.
   */
  struct JProductRouter :
    public std::map<JUPI_t, JLocation_t>
  {
    /**
     * Constructor.
     *
     * \param  detector      detector integration
     * \param  pbs           %PBS sequences
     */
    JProductRouter(const JDetectorIntegration_t& detector, const JPBSSequences& pbs)
    {
      for (JDetectorIntegration_t::const_iterator product = detector.begin(); product != detector.end(); ++product) {
        if (product->has()) {
          for (JPBSSequences::const_iterator i = pbs.begin(); i != pbs.end(); ++i) {
            if (insert(detector, *product, *i)) { 
              break;
            }
          }
        }
      }
    }


    /**
     * Get location of product with given %UPI.
     *
     * \param  upi           %UPI
     * \return               location
     */
    JLocation_t getLocation(const JUPI_t& upi) const
    {
      const_iterator p = this->find(upi);

      if (p != this->end())
        return p->second;
      else
        return JLocation_t();
    }

  private:
    /**
     * Insert product.
     *
     * \param  detector      detector integration
     * \param  product       product
     * \param  pbs           %PBS sequence
     * \return               true if inserted; else false
     */
    bool insert(const JDetectorIntegration_t& detector,
                const JProductIntegration_t&  product,
                const JPBSSequence&           pbs)
    {
      using namespace std;
      
      if (!pbs.empty()) {

        JPBSSequence::const_iterator i = pbs.begin();

        if (product.content.getPBS() == i->getPBS()) {

          JLocation_t location;

          for (JUPI_t upi = product.content.getUPI(); ++i != pbs.end(); ) {

            const JDetectorIntegration_t::range_type range = detector.find(upi);

            if (distance(range.first, range.second) == 1) {

              const JProductIntegration_t& ps = detector[range.first->second];

              if      (is_optical_module(ps.container.getPBS())  ||
                       is_base_module   (ps.container.getPBS())) { location.position = ps.position; }
              else if (is_string        (ps.container.getPBS())) { location.floor    = ps.position; }
              else if (is_detector      (ps.container.getPBS())) { location.string   = ps.position; }

              upi = ps.container.getUPI();

            } else {

              break;
            }
          }

          if (i == pbs.end()) {

            std::map<JUPI_t, JLocation_t>::insert(make_pair(product.content.getUPI(), location));

            return true;
          }
        }
      }

      return false;
    }
  };
}

#endif