egs_voxelized_shape.h

Go to the documentation of this file.

/*
###############################################################################
#
#  EGSnrc egs++ voxelized shape headers
#  Copyright (C) 2015 National Research Council Canada
#
#  This file is part of EGSnrc.
#
#  EGSnrc is free software: you can redistribute it and/or modify it under
#  the terms of the GNU Affero General Public License as published by the
#  Free Software Foundation, either version 3 of the License, or (at your
#  option) any later version.
#
#  EGSnrc 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 Affero General Public License for
#  more details.
#
#  You should have received a copy of the GNU Affero General Public License
#  along with EGSnrc. If not, see <http://www.gnu.org/licenses/>.
#
###############################################################################
#
#  Author:          Iwan Kawrakow, 2009
#
#  Contributors:    Frederic Tessier
#
###############################################################################
*/


#ifndef EGS_VOXELIZED_SHAPE_
#define EGS_VOXELIZED_SHAPE_

#include "egs_shapes.h"
#include "egs_rndm.h"
#include "egs_math.h"
#include "egs_alias_table.h"

#ifdef WIN32

    #ifdef BUILD_VOXELIZED_SHAPE_DLL
        #define EGS_VOXELIZED_SHAPE_EXPORT __declspec(dllexport)
    #else
        #define EGS_VOXELIZED_SHAPE_EXPORT __declspec(dllimport)
    #endif
    #define EGS_VOXELIZED_SHAPE_LOCAL

#else

    #ifdef HAVE_VISIBILITY
        #define EGS_VOXELIZED_SHAPE_EXPORT __attribute__ ((visibility ("default")))
        #define EGS_VOXELIZED_SHAPE_LOCAL  __attribute__ ((visibility ("hidden")))
    #else
        #define EGS_VOXELIZED_SHAPE_EXPORT
        #define EGS_VOXELIZED_SHAPE_LOCAL
    #endif

#endif

class EGS_VOXELIZED_SHAPE_EXPORT EGS_VoxelizedShape : public EGS_BaseShape {

public:

    EGS_VoxelizedShape(int file_format, const char *fname,const string &Name="",
                       EGS_ObjectFactory *f=0);
    ~EGS_VoxelizedShape();
    void EGS_VoxelizedShapeFormat0(const char *fname,const string &Name="",
                                   EGS_ObjectFactory *f=0);
    EGS_Vector getPoint(EGS_RandomGenerator *rndm) {
        int bin = prob->sample(rndm);
        int voxel = type == 0 ? bin : map[bin];
        int iz = voxel/nxy;
        voxel -= iz*nxy;
        int iy = voxel/nx;
        int ix = voxel - iy*nx;
        EGS_Float eta_x = rndm->getUniform(),
                  eta_y = rndm->getUniform(),
                  eta_z = rndm->getUniform();
        return EGS_Vector(xpos[ix]*(1-eta_x) + xpos[ix+1]*eta_x,
                          ypos[iy]*(1-eta_y) + ypos[iy+1]*eta_y,
                          zpos[iz]*(1-eta_z) + zpos[iz+1]*eta_z);
    };

    bool isValid() const {
        return (type == 0 || type == 1);
    };

protected:

    EGS_SimpleAliasTable *prob;   
    EGS_Float  *xpos;             
    EGS_Float  *ypos;             
    EGS_Float  *zpos;             
    int        *map;              
    int        nx, ny, nz, nxy, nreg;
    int        type;
};

#endif