JGeometry3DTestkit.hh

Go to the documentation of this file.

#ifndef __JGEOMETRY3DTESTKIT__
#define __JGEOMETRY3DTESTKIT__
 
#include "TRandom3.h"
 
#include "JTools/JConstants.hh"
#include "JGeometry3D/JPosition3D.hh"
#include "JGeometry3D/JDirection3D.hh"
#include "JGeometry3D/JRotation3D.hh"
#include "JGeometry3D/JQuaternion3D.hh"
#include "JGeometry3D/JEulerMatrix3D.hh"
#include "JGeometry3D/JSphere3D.hh"
#include "JGeometry3D/JCylinder3D.hh"
#include "JMath/JRandom.hh"
 
/**
 * \author mdejong
 */
 
namespace JGEOMETRY3D {}
namespace JPP { using namespace JGEOMETRY3D; }
 
namespace JGEOMETRY3D {
 
  using JTOOLS::PI;
  using JMATH::getRandom;
 
 
  /**
   * Randomize position.
   *
   * \param  p             pointer to valid object
   */
  inline void randomize(JPosition3D* p)
  {
    new (p) JPosition3D(getRandom<double>(-1.0, +1.0),
                        getRandom<double>(-1.0, +1.0),
                        getRandom<double>(-1.0, +1.0));
  };
 
 
  /**
   * Randomize direction.
   *
   * \param  p             pointer to valid object
   */
  inline void randomize(JDirection3D* p)
  {
    double x, y, z;
 
    gRandom->Sphere(x, y, z, 1.0);
 
    new (p) JDirection3D(x, y, z);
  };
 
 
  /**
   * Randomize quaternion.
   *
   * \param  p             pointer to valid object
   */
  inline void randomize(JQuaternion3D* p)
  {
    double x, y, z;
 
    gRandom->Sphere(x, y, z, 1.0);
 
    const double alpha = getRandom<double>(-PI, +PI);
 
    new (p) JQuaternion3D(alpha, JVersor3D(x, y, z));
  }
 
 
  /**
   * Randomize Euler angle.
   *
   * \param  p             pointer to valid object
   */
  inline void randomize(JEulerAngle3D* p)
  {
    new (p) JEulerAngle3D(getRandom<double>(0.0, PI),
                          getRandom<double>(0.0, PI),
                          getRandom<double>(0.0, PI));
  }
 
 
  /**
   * Randomize Euler matrix.
   *
   * \param  p             pointer to valid object
   */
  inline void randomize(JEulerMatrix3D* p)
  {
    new (p) JEulerMatrix3D(getRandom<JEulerAngle3D>());
  }
 
 
  /**
   * Get random position.\n
   * The positions are uniformly distributed in given sphere.
   *
   * \param  sphere       sphere
   * \return              position
   */ 
  inline JVector3D getRandomPosition(const JSphere3D& sphere)
  {
    const double R3max = sphere.getRadius() * sphere.getRadius() * sphere.getRadius();
 
    const double R3  = gRandom->Uniform(0.0, R3max);
    const double ct  = gRandom->Uniform(-1.0, +1.0);
    const double phi = gRandom->Uniform(-PI, +PI);
    
    const double R   = cbrt(R3);
    const double st  = sqrt((1.0 + ct) * (1.0 - ct));
 
    return sphere.getPosition() + JVector3D(R * st * cos(phi),
                                            R * st * sin(phi),
                                            R * ct);
  }
 
 
  /**
   * Get random position.\n
   * The positions are uniformly distributed in given cylinder.
   *
   * \param  cylinder     cylinder
   * \return              position
   */ 
  inline JVector3D getRandomPosition(const JCylinder3D& cylinder)
  {
    const double R2max = cylinder.getRadius() * cylinder.getRadius();
 
    const double R2  = gRandom->Uniform(0.0, R2max);
    const double phi = gRandom->Uniform(-PI, +PI);
    const double z   = gRandom->Uniform(cylinder.getZmin(), cylinder.getZmax());
 
    const double R   = sqrt(R2);
 
    return JVector3D(cylinder.getX()  +  R * cos(phi),
                     cylinder.getY()  +  R * sin(phi),
                     z);
  }
}
 
#endif