JProductRouter.hh

Go to the documentation of this file.

#ifndef __JDB__JPRODUCTROUTER__
#define __JDB__JPRODUCTROUTER__
 
#include <ostream>
#include <map>
 
#include "JDB/JDetectorIntegration.hh"
#include "JDB/JPBS.hh"
#include "JDB/JUPI.hh"
#include "JLang/JComparable.hh"
#include "JLang/JVectorize.hh"
#include "JLang/JLangToolkit.hh"
 
 
/**
 * \author mdejong
 */
namespace JDATABASE {}
namespace JPP { using namespace JDATABASE; }
 
namespace JDATABASE {
 
  /**
   * Auxiliary data structure for location of product in detector.
   */
  struct JLocation_t :
    public JLANG::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)
    {}
 
 
    /**
     * 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;
 
      return out << setw(3) << object.string << ' ' << setw(2) << object.floor << ' ' << setw(2) << object.position;
    }
 
 
    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 :
    private std::map<JUPI, JLocation_t>
  {
    /**
     * Constructor.
     *
     * \param  detector      detector integration
     */
    JProductRouter(const JDetectorIntegration& detector) 
    {
      using namespace std;
      using namespace JPP;
 
      for (JDetectorIntegration::const_iterator i = detector.begin(); i != detector.end(); ++i) {
 
        if (i->has()) {
 
          {
            const JPBS pbs[] = { PBS::DOM, PBS::DETECTION_UNIT, PBS::DETECTOR };
 
            const JProductTrace trace(detector, i->content.getUPI(), make_array(pbs));
 
            if (trace.size() == getSize(pbs)) {
 
              this->insert(make_pair(i->content.getUPI(), JLocation_t(trace[2].position, trace[1].position, trace[0].position)));
 
              continue;
            }
          }
 
          {
            const JPBS pbs[] = { PBS::BASE, PBS::DETECTION_UNIT, PBS::DETECTOR };
 
            const JProductTrace trace(detector, i->content.getUPI(), make_array(pbs));
 
            if (trace.size() == getSize(pbs)) {
 
              this->insert(make_pair(i->content.getUPI(), JLocation_t(trace[2].position, trace[1].position, trace[0].position)));
 
              continue;
            }
          }
 
          {
            const JPBS pbs[] = { PBS::BASE_CONTAINER, PBS::DETECTION_UNIT, PBS::DETECTOR };
 
            const JProductTrace trace(detector, i->content.getUPI(), make_array(pbs));
 
            if (trace.size() == getSize(pbs)) {
 
              this->insert(make_pair(i->content.getUPI(), JLocation_t(trace[2].position, trace[1].position, trace[0].position)));
 
              continue;
            }
          }
        }
      }
    }
 
 
    /**
     * Get location of product with given %UPI.
     *
     * Note that only for products inside a string, a valid location is returned,
     * otherwise, the default location is returned.
     *
     * \param  upi           %UPI
     * \return               location
     */
    JLocation_t getLocation(const JUPI& upi) const
    {
      std::map<JUPI, JLocation_t>::const_iterator p = this->find(upi);
 
      if (p != this->end())
        return p->second;
      else
        return JLocation_t();
    }
  };
}
 
#endif