CustomPolarObject.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2020 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2020 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2020 TotalEnergies
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOSX_MESH_SIMPLEGEOMETRICOBJECTS_CUSTOMPOLAROBJECT_HPP_
#define GEOSX_MESH_SIMPLEGEOMETRICOBJECTS_CUSTOMPOLAROBJECT_HPP_
 
#include "PlanarGeometricObject.hpp"
 
namespace geos
{
 
class CustomPolarObject : public PlanarGeometricObject
{
public:
 
 
  CustomPolarObject( const string & name,
                     Group * const parent );
 
  virtual ~CustomPolarObject() override;
 
 
 
  static string catalogName() { return "CustomPolarObject"; }
 
 
  bool isCoordInObject( real64 const ( &coord ) [3] ) const override final;
 
 
  virtual R1Tensor & getCenter() override final {return m_center;}
 
  virtual R1Tensor const & getCenter() const override final {return m_center;}
 
  void setCustomPolarObjectFunction( array1d< real64 > & coefficients )
  {
    //m_radius.m_coefficients = coefficients;
    m_coefficients = coefficients;
  }
 
  real64 getRadius( real64 angle ) const
  {
    real64 radius = 0;
    integer count = 0;
    for( auto coeff:m_coefficients )
    {
      radius = radius + coeff*cos( count*angle );
      count++;
    }
    return radius;
  }
 
protected:
 
  virtual void postProcessInput() override final;
 
private:
 
  R1Tensor m_center;
  array1d< real64 > m_coefficients;
  real64 m_tolerance;
 
 
  struct viewKeyStruct
  {
    static constexpr char const * centerString() { return "center"; }
    static constexpr char const * coefficientsString() { return "coefficients"; }
    static constexpr char const * toleranceString() { return "tolerance"; }
  };
 
 
};
} /* namespace geosx */
 
#endif /* GEOSX_MESH_SIMPLEGEOMETRICOBJECTS_CustomPolarObject_HPP_*/