UniformPeriodicGrid.cpp

Go to the documentation of this file.

/* This file is part of MoDELib, the Mechanics Of Defects Evolution Library.
 *
 *
 * MODEL is distributed without any warranty under the
 * GNU General Public License (GPL) v2 <http://www.gnu.org/licenses/>.
 */
 
#ifndef model_UniformPeriodicGrid_cpp_
#define model_UniformPeriodicGrid_cpp_
 
#include <UniformPeriodicGrid.h>
#include <CTM.h>
 
 
namespace model
{
 
    template <int dim>
    UniformPeriodicGrid<dim>::UniformPeriodicGrid(const ArrayDimI& gridSize_in,const ArrayDimD& gridSpacing_in) :
    /* init */ gridSize(gridSize_in)
    /* init */,gridSpacing(gridSpacing_in)
    {
        
    }
 
    template <int dim>
    std::array<typename UniformPeriodicGrid<dim>::ArrayDimI,CTM::pow(2,dim)> UniformPeriodicGrid<dim>::cornerIdx(const std::pair<ArrayDimI,ArrayDimI>& idx)
    {
        std::array<ArrayDimI,CTM::pow(2,dim)> temp;
        for(int p=0;p<std::pow(2,dim);++p)
        {
            for(int i=0;i<dim;++i)
            {
                const bool isSecond((p%CTM::pow(2,dim-i))/CTM::pow(2,dim-1-i));
                temp[p](i)= (isSecond? idx.second(i) : idx.first(i));
            }
        }
        return temp;
    }
 
    template <int dim>
    std::array<typename UniformPeriodicGrid<dim>::ArrayDimI,CTM::pow(2,dim)> UniformPeriodicGrid<dim>::posToCornerIdx(const ArrayDimD& localPos) const
    {
        return cornerIdx(posToIdx(localPos));
    }
 
    template <int dim>
    std::array<typename UniformPeriodicGrid<dim>::ArrayDimI,CTM::pow(2,dim)> UniformPeriodicGrid<dim>::posToPeriodicCornerIdx(const ArrayDimD& localPos) const
    {
        return cornerIdx(posToPeriodicIdx(localPos));
    }
 
    template <int dim>
    typename UniformPeriodicGrid<dim>::ArrayDimD UniformPeriodicGrid<dim>::idxToPos(const ArrayDimI& idx) const
    {
        return idx.template cast<double>()*gridSpacing;
    }
 
    template <int dim>
    std::array<double,CTM::pow(2,dim)> UniformPeriodicGrid<dim>::posToWeights(const ArrayDimD& localPos) const
    {
        const auto cidx(posToCornerIdx(localPos));
        const double vol(gridSpacing.prod());
        std::array<double,CTM::pow(2,dim)> temp;
        for(int p=0;p<CTM::pow(2,dim);++p)
        {
            temp[p]=(gridSpacing-(idxToPos(cidx[p])-localPos).abs()).prod()/vol;
        }
        return temp;
    }
 
    template <int dim>
    std::pair<std::array<typename UniformPeriodicGrid<dim>::ArrayDimI,CTM::pow(2,dim)>,std::array<double,CTM::pow(2,dim)>> UniformPeriodicGrid<dim>::posToPeriodicCornerIdxAndWeights(const ArrayDimD& localPos) const
    {
        return std::make_pair(cornerIdx(posToPeriodicIdx(localPos)),posToWeights(localPos));
    }
 
    template <int dim>
    std::pair<typename UniformPeriodicGrid<dim>::ArrayDimI,typename UniformPeriodicGrid<dim>::ArrayDimI> UniformPeriodicGrid<dim>::posToIdx(const ArrayDimD& localPos) const
    {
        const ArrayDimI highIndex((localPos/gridSpacing).ceil().template cast<int>());
        return std::make_pair(highIndex-1,highIndex);
    }
 
    template <int dim>
    std::pair<typename UniformPeriodicGrid<dim>::ArrayDimI,typename UniformPeriodicGrid<dim>::ArrayDimI> UniformPeriodicGrid<dim>::posToPeriodicIdx(const ArrayDimD& localPos) const
    {
        return idxToPeriodicIdx(posToIdx(localPos));
    }
 
    template <int dim>
    typename UniformPeriodicGrid<dim>::ArrayDimI UniformPeriodicGrid<dim>::idxToPeriodicIdx(const ArrayDimI& gi) const
    {
        const ArrayDimD gd(gi.template cast<double>());
        return gi-(gd/gridSize.template cast<double>()).floor().template cast<int>()*gridSize;
    }
 
    template <int dim>
    std::pair<typename UniformPeriodicGrid<dim>::ArrayDimI,typename UniformPeriodicGrid<dim>::ArrayDimI> UniformPeriodicGrid<dim>::idxToPeriodicIdx(const std::pair<ArrayDimI,ArrayDimI>& idx) const
    {
        return std::make_pair(idxToPeriodicIdx(idx.first),idxToPeriodicIdx(idx.second));
    }
 
    template struct UniformPeriodicGrid<1>;
    template struct UniformPeriodicGrid<2>;
    template struct UniformPeriodicGrid<3>;
}
#endif