sdf_shape_2D.h

Go to the documentation of this file.

/*
  This file is part of MADNESS.
  
  Copyright (C) 2007,2010 Oak Ridge National Laboratory
  
  This program is free software; you can redistribute it 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 program 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 program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  
  For more information please contact:
  
  Robert J. Harrison
  Oak Ridge National Laboratory
  One Bethel Valley Road
  P.O. Box 2008, MS-6367
  
  email: harrisonrj@ornl.gov
  tel:   865-241-3937
  fax:   865-572-0680
  
  $Id$
*/

#ifndef MADNESS_MRA_SDF_SHAPE_2D_H__INCLUDED
#define MADNESS_MRA_SDF_SHAPE_2D_H__INCLUDED

#include <madness/mra/sdf_domainmask.h>

namespace madness {

    class SDFCircle : public SignedDFInterface<2> {
    protected:
        const double radius; 
        const coord_2d center; 

    public:
        SDFCircle(const double radius, const coord_2d &center) 
            : radius(radius)
            , center(center) 
        {}

        double sdf(const coord_2d& pt) const {
            double temp, r;
            int i;
            
            r = 0.0;
            for(i = 0; i < 2; ++i) {
                temp = pt[i] - center[i];
                r += temp * temp;
            }
            
            return sqrt(r) - radius;
        }

        coord_2d grad_sdf(const coord_2d& pt) const {
            double x = pt[0] - center[0];
            double y = pt[1] - center[1];
            double r = sqrt(x*x + y*y);
            coord_2d g;
            if(r < 1.0e-6) {
                g[0] = g[1] = 0.0;
            }
            else {
                g[0] = x/r;
                g[1] = y/r;
            }
            return g;
        }
    };

    class SDFRectangle : public SignedDFInterface<2> {
    protected:
        const coord_2d lengths;  
        const coord_2d center; 

    public:
        SDFRectangle(const coord_2d& length, const coord_2d& center) 
            : lengths(length*0.5), center(center) 
        {}

        double sdf(const coord_2d& pt) const {
            double diff, max;
            
            max = fabs(pt[0] - center[0]) - lengths[0];
            diff = fabs(pt[1] - center[1]) - lengths[1];
            if(diff > max)
                max = diff;
            
            return max;
        }

        coord_2d grad_sdf(const coord_2d& pt) const {
            MADNESS_EXCEPTION("gradient method is not yet implemented for this shape",0);
        }
    };

} // end of madness namespace

#endif // MADNESS_MRA_SDF_SHAPE_2D_H__INCLUDED