geometry.hpp

#ifndef MCNP2CAD_GEOMETRY_H
#define MCNP2CAD_GEOMETRY_H

#include <vector>
#include <iosfwd>
#include <cmath>
#include <cstdlib>

#include "dataref.hpp"

// under visual studio, the following macro must be defined
// in order to get M_PI and friends from the cmath header.
#ifdef _MSC_VER
#define _USE_MATH_DEFINES
#endif 

#include <math.h>

class Vector3d{

public:

  double v[3];
  Vector3d(){
    v[2] = v[1] = v[0] = 0;
  }
  
  Vector3d( const double p[3] ){
    v[0] = p[0];
    v[1] = p[1];
    v[2] = p[2];
  }

  Vector3d( double x, double y, double z ){
    v[0] = x; 
    v[1] = y;
    v[2] = z;
  }

  Vector3d( const std::vector<double>& p, int idx = 0 ){
    v[0] = p.at(idx+0);
    v[1] = p.at(idx+1);
    v[2] = p.at(idx+2);
  }


  Vector3d( const Vector3d& other ) {
    v[0] = other.v[0];
    v[1] = other.v[1];
    v[2] = other.v[2];
  }
  
  double length() const{
    return sqrt( v[0]*v[0] + v[1]*v[1] + v[2]*v[2] );
  }

  Vector3d normalize() const {
    double length = this->length();
    return Vector3d( v[0]/length, v[1]/length, v[2]/length );
  }

  Vector3d& operator=( const Vector3d& other) {
    v[0] = other.v[0];
    v[1] = other.v[1];
    v[2] = other.v[2];
    return *this;
  }
  
  Vector3d operator-() const {
    return Vector3d(-v[0], -v[1], -v[2]);
  }

  Vector3d reverse() const { 
    return -(*this);
  }

  Vector3d scale( double d ) const {
    return Vector3d(v[0]*d, v[1]*d, v[2]*d);
  }

  Vector3d operator*( double d ) const {
    return scale(d);
  }

  Vector3d add( const Vector3d& alt ) const {
    return Vector3d( v[0]+alt.v[0], v[1]+alt.v[1], v[2]+alt.v[2] );
  }

  Vector3d operator+( const Vector3d& alt ) const {
    return add(alt);
  }

  double dot( const Vector3d& alt ) const {
    return v[0]*alt.v[0] + v[1]*alt.v[1] + v[2]*alt.v[2];
  }

  Vector3d cross( const Vector3d& alt ) const { 
    Vector3d c;
    c.v[0] = v[1]*alt.v[2] - v[2]*alt.v[1];
    c.v[1] = v[2]*alt.v[0] - v[0]*alt.v[2];
    c.v[2] = v[0]*alt.v[1] - v[1]*alt.v[0];
    return c;
  }

  Vector3d rotate_about( const Vector3d& v_p, double theta_p, bool degrees = true ) const {
    Vector3d v = v_p.normalize();
    double theta = theta_p;
    if( degrees ) { theta *= M_PI / 180.0; }
    double cos_t = cos( theta );
    Vector3d ret = scale(cos_t) + v.scale((1.0-cos_t)*(dot(v))) + cross(v).scale( sin(theta) );
    return ret;
  }

  /// project v onto this vector, return the result
  Vector3d projection( const Vector3d& v ) const {
    return scale( this->dot(v) / this->dot(*this) );
  }

};

std::ostream& operator<<( std::ostream& str, const Vector3d& v );

//std::string to_string( const Vector3d& v );

// determinant of 3x3 matrix (C-style matrix ordering)
double matrix_det( double mat[9] );

class Transform{

public:
  enum mat_format{ C_STYLE, FORTRAN_STYLE };

protected:
  Vector3d translation;
  bool has_rot;
  double theta; Vector3d axis;
  bool invert;

  void set_rots_from_matrix( double raw_matrix[9], enum mat_format  );

public:
  Transform():translation(),has_rot(false),invert(false){}
  Transform( const Vector3d& v ):translation(v),has_rot(false),invert(false){}
  Transform( const std::vector< double >& inputs, bool degree_format_p = false, enum mat_format = FORTRAN_STYLE );
  Transform( double rot[9], const Vector3d& trans,  enum mat_format = C_STYLE );

  const Vector3d& getTranslation() const { return translation; }
  void modify_translation( const Vector3d& translation_addition );
  void modify_translation( const Vector3d& translation_addition ) const;
  bool hasRot() const{ return has_rot; }
  bool hasInversion() const{ return invert; }
  double getTheta() const { return theta; }
  const Vector3d& getAxis() const { return axis; }
  void print( std::ostream& str ) const;

  Transform reverse() const;

};

// generate a full transformation matrix
std::vector<double> get_matrix(const Transform& t);

Transform combine(const Transform& a, const Transform& b);

std::ostream& operator<<(std::ostream& str, const Transform& t );

/** A universe and a transformation */
class FillNode {

protected:
  int universe;
  DataRef<Transform>* tr;

public:
  FillNode():
    universe(0), tr(new NullRef<Transform>())
  {}

  FillNode( int universe_p ):
    universe(universe_p), tr(new NullRef<Transform>())
  {}

  FillNode( int universe_p, DataRef<Transform>* tr_p ):
    universe(universe_p), tr(tr_p)
  {}

  FillNode( const FillNode& node_p ):
    universe(node_p.universe), tr(node_p.tr->clone())
  {}

  FillNode& operator=( const FillNode& node_p ){
    if( this != &node_p ){
      universe = node_p.universe;
      tr = node_p.tr->clone();
    }
    return *this;
  }

  ~FillNode(){
    delete tr;
  }
  
  int getFillingUniverse() const { return universe; }
  
  bool hasTransform() const{ return tr->hasData();}
  const Transform& getTransform() const { return tr->getData(); }

  void setTransform( DataRef<Transform>* tr_p ){
    delete tr;
    tr = tr_p;
  }
};

typedef std::pair<int,int> irange;

/** A fill is a 3-dimensional grid of fill nodes */
class Fill{

protected:
  std::vector<FillNode> nodes;
  bool has_grid;
  irange xrange, yrange, zrange;


 size_t indicesToSerialIndex( int x, int y, int z ) const ;

public:
  Fill():
    nodes(1,FillNode()),has_grid(false)
  {}

  Fill( const FillNode& origin_p ):
    nodes(1,origin_p), has_grid(false)
  {}

  Fill( irange x, irange y, irange z, std::vector<FillNode> nodes_p ):
    nodes(nodes_p), has_grid(true), xrange(x), yrange(y), zrange(z)
  {}

  const FillNode& getOriginNode() const;  
  const FillNode& getNode( int x, int y, int z ) const;

  friend class Lattice;
};


class Lattice{

protected:
  int num_finite_dims;
  Vector3d v1, v2, v3;

  DataRef<Fill> *fill;

public:
  Lattice() : fill(new NullRef<Fill>()){}
  Lattice( int dims, const Vector3d& v1_p, const Vector3d& v2_p, const Vector3d& v3_p, const FillNode& singleton_fill );
  Lattice( int dims, const Vector3d& v1_p, const Vector3d& v2_p, const Vector3d& v3_p, const Fill& full_fill );

  Lattice( const Lattice& l );
  Lattice& operator=( const Lattice& l );
  ~Lattice(){ delete fill; }
  

  int numFiniteDirections() const { return num_finite_dims; }
  Transform getTxForNode( int x, int y, int z ) const ;

  const FillNode& getFillForNode( int x, int y, int z ) const ;

  bool isFixedSize() const { return fill->getData().has_grid; }  
  irange getXRange() const { return fill->getData().xrange; }
  irange getYRange() const { return fill->getData().yrange; }
  irange getZRange() const { return fill->getData().zrange; }
  

};


#endif /* MCNP2CAD_GEOMETRY_H */