JGeometry.hh

Go to the documentation of this file.

#ifndef __JACOUSTICS__JGEOMETRY__
#define __JACOUSTICS__JGEOMETRY__

#include <ostream>
#include <iomanip>
#include <vector>
#include <map>
#include <algorithm>
#include <cmath>

#include "JLang/JException.hh"
#include "JLang/JPredicate.hh"
#include "JLang/JComparator.hh"
#include "JLang/JManip.hh"

#include "JGeometry3D/JPosition3D.hh"

#include "JTools/JHashMap.hh"

#include "JDetector/JLocation.hh"
#include "JDetector/JHydrophone.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorSupportkit.hh"

#include "JAcoustics/JAcousticsToolkit.hh"
#include "JAcoustics/JAcousticsSupportkit.hh"
#include "JAcoustics/JMechanics.hh"
#include "JAcoustics/JModel.hh"


/**
 * \file
 *
 * Acoustic geometries.
 * \author mdejong
 */
namespace JACOUSTICS {}
namespace JPP { using namespace JACOUSTICS; }

namespace JACOUSTICS {

  using JLANG::JValueOutOfRange;
  using JLANG::JNoValue;
  using JGEOMETRY3D::JVector3D;
  using JGEOMETRY3D::JPosition3D;
  using JTOOLS::JHashMap;
  using JDETECTOR::JLocation;
 

  /**
   * Auxiliary namespace to encapsulate different geometries.\n
   */
  namespace JGEOMETRY {

    /**
     * Floor geometry.
     */
    struct JFloor {
      /**
       * Default constructor.
       */
      JFloor() :
        height(0.0)
      {}


      /**
       * Constructor.\n
       * The height refers to the maximal distance from
       * the fixed reference point of the parent string.
       *
       * \param  height                   height
       */
      JFloor(const double height) :
        height(height)
      {}


      /**
       * Get height of this floor.\n
       * The height refers to the maximal distance from
       * the fixed reference point of the parent string.
       *
       * \return                          height
       */
      double getHeight() const
      {
        return height;
      }


      /**
       * Write floor parameters to output stream.
       *
       * \param  out                      output stream
       * \param  floor                    floor
       * \return                          output stream
       */
      friend inline std::ostream& operator<<(std::ostream& out, const JFloor& floor)
      {
        return out << FIXED(7,2) << floor.getHeight();
      }


    protected:
      double height;
    };


    /**
     * String geometry.
     *
     * This data structure provides for the implementation of the model dependent geometry of a detector string.\n
     * The position of a string corresponds to the top of the so-called T-bar which is mounted on the anchor.\n
     * A piezo sensor is mounted in each optical module.\n 
     * The position of the piezo sensor is determined by 
     * the floor number of the optical module and
     * the actual string parameters.\n
     * A hydrophone is optionally mounted on the anchor which corresponds to floor zero.\n
     * It has a fixed position which is relative to the reference position of the string.
     */
    struct JString :
      public JPosition3D,
      public JHashMap<int, JFloor>
    {
      using JPosition3D::getPosition;
      using JPosition3D::getDistance;


      /**
       * Default constructor.
       */
      JString() :
        has_hydrophone(false)
      {}


      /**
       * Constructor.
       *
       * The given position corresponds to the reference point of the string from
       * which the positions of the piezo sensors and hydrophone are calculated.
       *
       * \param  position                 position
       */
      JString(const JVector3D& position) :
        JPosition3D(position),
        has_hydrophone(false)
      {}


      /**
       * Constructor.
       *
       * The template parameter should correspond to a data type which implements the following policy methods.
       * <pre>
       *     int          getFloor();
       *     JPosition3D  getPosition();
       * </pre>
       *
       * The given position corresponds to the reference point of the string from
       * which the positions of the piezo sensors and hydrophone are calculated.
       *
       * Note that the position of the piezo is offset by JDETECTOR::getPiezoPosition and
       * that the position of the hydrophone should manually be set.
       *
       * \param  position                 position
       * \param  __begin                  begin of optical modules
       * \param  __end                    end   of optical modules
       * \param  mechanics                mechanical model parameters
       */
      template<class T>
      JString(const JVector3D& position,
              T __begin, 
              T __end,
              const JMechanics& mechanics) :
        JPosition3D(position),
        has_hydrophone(false),
        mechanics(mechanics)
      {
        for (T i = __begin; i != __end; ++i) {
          (*this)[i->getFloor()] = this->getDistance(i->getPosition() + JDETECTOR::getPiezoPosition());
        }
      }


      /**
       * Check if this string has given floor.
       *
       * \param  floor                    floor
       * \return                          true if floor present; else false
       */
      bool hasFloor(int floor) const
      {
        if (floor == 0)
          return has_hydrophone;
        else
          return this->has(floor);
      }


      /**
       * Get effective height.
       *
       * \param  height                   height
       * \return                          height
       */
      double getHeight(const double height) const
      {
        return this->mechanics.getHeight(height);
      }


      /**
       * Get effective height of given floor
       *
       * \param  floor                    floor
       * \return                          height
       */
      double getHeight(const int floor) const
      {
        if (floor == 0) {

          if (has_hydrophone) {
            return hydrophone.getZ();
          }

        } else if (this->has(floor)) {
        
          return getHeight(this->get(floor).getHeight());
        }
 
        THROW(JValueOutOfRange, "Invalid floor " << floor);
      }


      /**
       * Get position at given height according to actual string parameters.
       *
       * \param  parameters               parameters
       * \param  height                   height
       * \return                          position
       */
      JPosition3D getPosition(const JMODEL::JString& parameters,
                              const double           height) const
      {
        const double tx = parameters.tx;
        const double ty = parameters.ty;
        const double tz = sqrt(1.0 - tx*tx - ty*ty);

        return JPosition3D(this->getX()  +  tx * this->mechanics.getHeight(height),
                           this->getY()  +  ty * this->mechanics.getHeight(height),
                           this->getZ()  +  tz * height);
      }


      /**
       * Get position of given floor according to actual string parameters.
       *
       * \param  parameters               parameters
       * \param  floor                    floor
       * \return                          position
       */
      JPosition3D getPosition(const JMODEL::JString& parameters,
                              const int              floor) const
      {
        if (floor == 0) {

          if (has_hydrophone) {
            return getPosition() + hydrophone.getPosition();
          }

        } else if (this->has(floor)) {
         
          return getPosition(parameters, this->get(floor).getHeight());
        }

        THROW(JValueOutOfRange, "Invalid floor " << floor);
      }


      /**
       * Get position at given height according to default string parameters.
       *
       * \param  height                   height
       * \return                          position
       */
      JPosition3D getPosition(const double height) const
      {
        return getPosition(JMODEL::JString(), height);
      }


      /**
       * Get position of given floor according to default string parameters.
       *
       * \param  floor                    floor
       * \return                          position
       */
      JPosition3D getPosition(const int floor) const
      {
        return getPosition(JMODEL::JString(), floor);
      }


      /**
       * Get distance between given position and floor according to actual string parameters.
       *
       * \param  parameters               parameters
       * \param  position                 position
       * \param  floor                    floor
       * \return                          distance
       */
      double getDistance(const JMODEL::JString& parameters,
                         const JVector3D&       position,
                         const int              floor) const
      {
        return this->getPosition(parameters, floor).getDistance(position);
      }


      /**
       * Get model gradient of distance between given position and floor according to actual string parameters.
       *
       * \param  parameters               parameters
       * \param  position                 position
       * \param  floor                    floor
       * \return                          gradient
       */
      JMODEL::JString getGradient(const JMODEL::JString& parameters,
                                  const JVector3D&       position,
                                  const int              floor) const
      {
        if (floor == 0) {

          return JMODEL::JString();

        } else if (this->has(floor)) {

          const double      height = this->get(floor).getHeight();
          const JPosition3D pos    = this->getPosition(parameters, height);

          const double tx = parameters.tx;
          const double ty = parameters.ty;
          const double tz = sqrt(1.0 - tx*tx - ty*ty);

          const double dx = pos.getX() - position.getX();
          const double dy = pos.getY() - position.getY();
          const double dz = pos.getZ() - position.getZ();

          const double D  = sqrt(dx*dx + dy*dy + dz*dz);

          return JMODEL::JString(height * dx / D  -  (tx / tz) * dz / D,
                                 height * dy / D  -  (ty / tz) * dz / D);

        } else {

          THROW(JValueOutOfRange, "Invalid floor " << floor);
        }       
      }


      /**
       * Write string parameters to output stream.
       *
       * \param  out                      output stream
       * \param  string                   string
       * \return                          output stream
       */
      friend inline std::ostream& operator<<(std::ostream& out, const JString& string)
      {
        using namespace std;

        for (JString::const_iterator i = string.begin(); i != string.end(); ++i) {
          out << setw(2)    << i->first       << ' ' 
              << FIXED(7,3) << i->second      << " | " 
              << string.getPosition(i->first) << ' ' 
              << string.mechanics             << endl;
        }

        return out;
      }


      /**
       * Hydrophone.
       *
       * The position of the hydrophone is relative to the reference position of the string.
       */
      JPosition3D hydrophone;
      bool        has_hydrophone;

      /**
       * Mechanical data.
       */
      JMechanics mechanics;
    };


    /**
     * Detector geometry.
     */
    struct JDetector :
      public JHashMap<int, JString>
    {
      /**
       * Default constructor.
       */
      JDetector()
      {}


      /**
       * Constructor.
       *
       * If position of hydrophone is not available, it is assumed that there is no hydrophone.\n
       * If the position of hydrophone is manually set,
       * the corresponding parameter JString::has_hydrophone should be set to <tt>true</tt>.
       * 
       * Missing base modules (i.e.\ floor 0) in the detector can be added using application JDetectorDB.cc.
       *
       * \param  detector                 detector
       * \param  hydrophones              container with data of hydrophones
       */
      JDetector(const JDETECTOR::JDetector&                detector,
                const std::vector<JDETECTOR::JHydrophone>& hydrophones = std::vector<JDETECTOR::JHydrophone>())
      {
        using namespace std;
        using namespace JPP;

        map<int, vector<module_type> > buffer;                            // string -> modules

        for (JDETECTOR::JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {
          buffer[module->getString()].push_back(module_type(module->getLocation(), module->getPosition()));
        }

        for (map<int, vector<module_type> >::iterator i = buffer.begin(); i != buffer.end(); ++i) {

          sort(i->second.begin(), i->second.end(), make_comparator(&module_type::getFloor));

          if (i->second[0].getFloor() == 0) {

            vector<module_type>::iterator p = i->second.begin();

            ++p;

            (*this)[i->first] = JGEOMETRY::JString(i->second[0].getPosition(), p, i->second.end(), getMechanics[detector.getID()][i->first]);

            try {

              (*this)[i->first].hydrophone     = getPosition(hydrophones.begin(),
                                                             hydrophones.end(),
                                                             make_predicate(&JHydrophone::getString, i->first));

              (*this)[i->first].has_hydrophone = true;
            }
            catch(const exception&) {
              (*this)[i->first].has_hydrophone = false;
            }

          } else {

            THROW(JNoValue, "No floor 0 in string " << i->first << "; use e.g. JDetectorDB -W.");
          }
        }
      }


      /**
       * Check if this detector has given string.
       *
       * \param  string                   string
       * \return                          true if string present; else false
       */
      bool hasString(int string) const
      {
        return this->has(string);
      }


      /**
       * Check if this detector has given location.
       *
       * \param  location                 location
       * \return                          true if location present; else false
       */
      bool hasLocation(const JLocation& location) const
      {
        return this->hasString(location.getString()) && (*this)[location.getString()].hasFloor(location.getFloor());
      }


      /**
       * Write detector parameters to output stream.
       *
       * \param  out                      output stream
       * \param  detector                 detector
       * \return                          output stream
       */
      friend inline std::ostream& operator<<(std::ostream& out, const JDetector& detector)
      {
        using namespace std;

        for (JDetector::const_iterator i = detector.begin(); i != detector.end(); ++i) {
          out << setw(4) << i->first << endl << i->second;
        }

        return out;
      }


      /**
       * Auxiliary data structure for module location and position.
       */
      struct module_type :
        public JLocation,
        public JPosition3D
      {
        /**
         * Constructor.
         *
         * \param  location           module location
         * \param  position           module position
         */
        module_type(const JLocation&   location,
                    const JPosition3D& position) :
          JLocation  (location),
          JPosition3D(position)
        {}


        /**
         * Less-than operator.
         *
         * \param  first              first  module
         * \param  second             second module
         * \return                    true if floor of first module less than that of second; else false
         */
        friend inline bool operator<(const module_type& first, const module_type& second)
        {
          return first.getFloor() < second.getFloor();
        }
      };
    };
  }


  /**
   * Type definition of detector geometry.
   */
  typedef JGEOMETRY::JDetector  JGeometry;
}

#endif