boundaryprojection.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_BOUNDARYPROJECTION_HH
#define DUNE_BOUNDARYPROJECTION_HH
 
//- system includes
#include <cmath>
#include <memory>
 
//- Dune includes
#include <dune/common/fvector.hh>
 
#include <dune/geometry/multilineargeometry.hh>
 
#include <dune/grid/common/boundarysegment.hh>
#include <dune/grid/common/datahandleif.hh>
#include <dune/grid/io/file/gmshreader.hh>
 
namespace Dune
{
  template <int dimworld>
  struct DuneBoundaryProjection;
 
  template <int dimworld>
  struct DuneBoundaryProjection
    : public BoundarySegmentBackupRestore< DuneBoundaryProjection< dimworld > >
  {
    typedef DuneBoundaryProjection< dimworld > ThisType;
    typedef BoundarySegmentBackupRestore< DuneBoundaryProjection< dimworld > > BaseType;
    typedef typename BaseType :: ObjectStreamType  ObjectStreamType;
 
    using BaseType :: restore;
    using BaseType :: registerFactory;
 
    typedef FieldVector< double, dimworld> CoordinateType;
    virtual ~DuneBoundaryProjection() {}
 
    virtual CoordinateType operator() (const CoordinateType& global) const = 0;
 
    virtual void backup( ObjectStreamType& buffer ) const
    {
      DUNE_THROW(NotImplemented,"DuneBoundaryProjection::backup not overloaded!");
    }
 
    template <class BufferImp>
    void toBuffer( BufferImp& buffer ) const
    {
      MessageBufferIF< BufferImp > buf( buffer );
      toBuffer( buf );
    }
 
    template <class BufferImp>
    void toBuffer( MessageBufferIF< BufferImp > & buffer ) const
    {
      ObjectStreamType str;
      // call virtual interface backup
      backup( str );
      std::string data = str.str();
      const size_t size = data.size();
      buffer.write( size );
      for( size_t i=0; i<size; ++i )
        buffer.write( data[ i ] );
    }
 
    template <class BufferImp>
    static std::unique_ptr< ThisType > restoreFromBuffer( BufferImp & buffer )
    {
      MessageBufferIF< BufferImp > buf( buffer );
      return restoreFromBuffer( buf );
    }
 
    template <class BufferImp>
    static std::unique_ptr< ThisType > restoreFromBuffer( MessageBufferIF< BufferImp > & buffer )
    {
      std::string data;
      size_t size = 0;
      buffer.read( size );
      data.resize( size );
      for( size_t i=0; i<size; ++i )
        buffer.read( data[ i ] );
 
      ObjectStreamType str;
      str.write( data.c_str(), size );
      return BaseType::restore( str );
    }
  };
 
  template < int dimworld >
  class BoundaryProjectionWrapper
    : public DuneBoundaryProjection< dimworld >
  {
  protected:
    typedef DuneBoundaryProjection< dimworld > BaseType;
    const BaseType& proj_;
  public:
    typedef typename BaseType :: CoordinateType CoordinateType;
 
    // constructor taking other projection
    BoundaryProjectionWrapper( const BaseType& proje )
      : proj_( proje )
    {}
 
    ~BoundaryProjectionWrapper () {}
 
    CoordinateType operator() (const CoordinateType& global) const
    {
      return proj_( global );
    }
  };
 
  // BoundarySegmentWrapper
  // ----------------------
 
  template< int dim, int dimworld >
  class BoundarySegmentWrapper
    : public DuneBoundaryProjection< dimworld >
  {
    typedef BoundarySegmentWrapper< dim, dimworld >  ThisType;
    typedef DuneBoundaryProjection< dimworld >       Base;
 
    typedef typename Base :: ObjectStreamType  ObjectStreamType;
 
    typedef MultiLinearGeometry<typename Base::CoordinateType::value_type,dim-1,dimworld> FaceMapping;
 
  public:
    typedef typename Base::CoordinateType CoordinateType;
    typedef Dune::BoundarySegment< dim, dimworld > BoundarySegment;
 
    BoundarySegmentWrapper ( const GeometryType &type,
                             const std::vector< CoordinateType > &vertices,
                             const std::shared_ptr< BoundarySegment > &boundarySegment )
      : faceMapping_( FaceMapping( type, vertices ) ),
        boundarySegment_( boundarySegment )
    {}
 
    BoundarySegmentWrapper( ObjectStreamType& buffer )
      : faceMapping_( readFaceMapping( buffer ) ),
        boundarySegment_( BoundarySegment::restore( buffer ).release() )
    {
    }
 
    CoordinateType operator() ( const CoordinateType &global ) const
    {
      return boundarySegment() ( faceMapping_.local( global ) );
    }
 
    const BoundarySegment &boundarySegment () const
    {
      return *boundarySegment_;
    }
 
    void backup( ObjectStreamType& buffer ) const
    {
      // write identifier key first
      buffer.write( (const char *) &key(), sizeof(int));
      // now all data
      GeometryType type = faceMapping_.type();
      buffer.write( (const char *) &type, sizeof(GeometryType) );
 
      int corners = faceMapping_.corners() ;
      buffer.write( (const char *) &corners, sizeof(int) );
 
      CoordinateType corner( 0 );
      for( int i=0; i<corners; ++i )
      {
        corner = faceMapping_.corner( i );
        buffer.write( (const char *) &corner[ 0 ], sizeof(double)*CoordinateType::dimension );
      }
 
      boundarySegment_->backup( buffer );
    }
 
    static void registerFactory()
    {
      if( key() < 0 )
      {
        key() = Base::template registerFactory< ThisType >();
      }
    }
 
  protected:
    static int& key()
    {
      static int k = -1;
      return k;
    }
 
    FaceMapping readFaceMapping( ObjectStreamType& buffer )
    {
      GeometryType type;
      buffer.read( (char *) &type, sizeof(GeometryType) );
      int corners = 0;
      buffer.read( (char *) &corners, sizeof(int) );
      std::vector< CoordinateType > vertices( corners, CoordinateType(0) );
      for( int i=0; i<corners; ++i )
      {
        buffer.read( (char *) &vertices[ i ][ 0 ], sizeof(double)*CoordinateType::dimension );
      }
      return FaceMapping( type, vertices );
    }
 
  private:
    FaceMapping faceMapping_;
    const std::shared_ptr< BoundarySegment > boundarySegment_;
  };
 
 
 
  //
  // Example of boundary projection projection to a circle
  //
  template <int dimworld>
  struct CircleBoundaryProjection : public DuneBoundaryProjection< dimworld >
  {
    typedef FieldVector< double, dimworld> CoordinateType;
 
    CircleBoundaryProjection(const double radius = std::sqrt( (double)dimworld ))
      : radius_( radius ) {}
 
    virtual ~CircleBoundaryProjection() {}
 
    virtual CoordinateType operator() (const CoordinateType& global) const
    {
      CoordinateType prj( global );
      // get adjustment factor
      const double factor = radius_  / global.two_norm();
      // adjust
      prj *= factor;
      return prj;
    }
 
  protected:
    const double radius_;
  };
 
} // end namespace
 
#endif // #ifndef DUNE_BOUNDARYPROJECTION_HH