geometry.hpp

Go to the documentation of this file.

/* Copyright (c) 2005-2010 Taneli Kalvas. All rights reserved.
 *
 * You can redistribute this software and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option)
 * any later version.
 * 
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this library (file "COPYING" included in the package);
 * if not, write to the Free Software Foundation, Inc., 51 Franklin
 * Street, Fifth Floor, Boston, MA 02110-1301 USA
 * 
 * If you have questions about your rights to use or distribute this
 * software, please contact Berkeley Lab's Technology Transfer
 * Department at TTD@lbl.gov. Other questions, comments and bug
 * reports should be sent directly to the author via email at
 * taneli.kalvas@jyu.fi.
 * 
 * NOTICE. This software was developed under partial funding from the
 * U.S.  Department of Energy.  As such, the U.S. Government has been
 * granted for itself and others acting on its behalf a paid-up,
 * nonexclusive, irrevocable, worldwide license in the Software to
 * reproduce, prepare derivative works, and perform publicly and
 * display publicly.  Beginning five (5) years after the date
 * permission to assert copyright is obtained from the U.S. Department
 * of Energy, and subject to any subsequent five (5) year renewals,
 * the U.S. Government is granted for itself and others acting on its
 * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
 * the Software to reproduce, prepare derivative works, distribute
 * copies to the public, perform publicly and display publicly, and to
 * permit others to do so.
 */

#ifndef GEOMETRY_HPP
#define GEOMETRY_HPP 1


#include <stdint.h>
#include <vector>
#include <iostream>
#include "file.hpp"
#include "vec3d.hpp"
#include "solid.hpp"
#include "types.hpp"


struct Bound 
{
    bound_e         type;
    double          val;

    Bound( bound_e t, double v ) : type(t), val(v) {}

    Bound( std::istream &s ) {
        type = (bound_e)read_int32( s );
        val = read_double( s );
    }

    void save( std::ostream &fout ) const {
        write_int32( fout, type );
        write_double( fout, val );
    }
};


class Geometry 
{
    geom_mode_e                _geom_mode; 
    Int3D                      _size;      
    Vec3D                      _origo;     
    Vec3D                      _max;       
    double                     _h;         
    int32_t                    _n;         
    std::vector<const Solid*>  _sdata;     
    std::vector<Bound>         _bound;     
    bool                       _built;     
    signed char               *_smesh;     
    int32_t                    _brktc;     
    bool vac_or_neu( int32_t i, int32_t j, int32_t k );

public:

    Geometry( geom_mode_e geom_mode, Int3D size, Vec3D origo, double h );

    Geometry( std::istream &s );

    ~Geometry();

    geom_mode_e geom_mode( void ) const { return( _geom_mode ); }

    int32_t dim( void ) const;

    Int3D size( void ) const { return( _size ); }

    int32_t size( int i ) const { return( _size[i] ); }
   
    int32_t nodecount( void ) const { return( _size[0]*_size[1]*_size[2] ); }

    Vec3D origo( void ) const { return( _origo ); }

    double origo( int i ) const { return( _origo[i] ); }

    Vec3D max( void ) const { return( _max ); }

    double max( int i ) const { return( _max[i] ); }

    double h( void ) const { return( _h ); }

    void set_solid( int32_t n, const Solid *s );

    const Solid *get_solid( int32_t n ) const;

    void set_boundary( int32_t n, const Bound &b );

    Bound get_boundary( int32_t n ) const;

    std::vector<Bound> get_boundaries() const;

    void set_bracket_count( int32_t n );

    int32_t get_bracket_count( void ) const;

    int32_t inside( const Vec3D &x ) const;

    bool inside( int32_t n, const Vec3D &x ) const;

    double bracket_surface( int32_t n, const Vec3D &xin, const Vec3D &xout, Vec3D &xsurf ) const;

    bool built( void ) const { return( _built ); }

    void build_mesh( void );

    const signed char &mesh( int32_t i ) const { return( _smesh[i] ); }

    const signed char &mesh( int32_t i, int32_t j ) const {
        return( _smesh[i + j*_size[0]] ); 
    }

    const signed char &mesh( int32_t i, int32_t j, int32_t k ) const {
        return( _smesh[i + j*_size[0] + k*_size[0]*_size[1]] );
    }

    signed char &mesh( int32_t i ) { return( _smesh[i] ); }

    signed char &mesh( int32_t i, int32_t j ) {
        return( _smesh[i + j*_size[0]] );
    }

    signed char &mesh( int32_t i, int32_t j, int32_t k ) {
        return( _smesh[i + j*_size[0] + k*_size[0]*_size[1]] );
    }

    signed char mesh_check( int32_t i, int32_t j, int32_t k ) const;

    void save( std::ostream &s ) const;

    void debug_print( void ) const;
};


#endif