SphericalDataMixed.h

Go to the documentation of this file.

/******************************************************************************
 * Copyright (c) 2021.                                                        *
 * The Regents of the University of Michigan and DFT-EFE developers.          *
 *                                                                            *
 * This file is part of the DFT-EFE code.                                     *
 *                                                                            *
 * DFT-EFE is free software: you can redistribute it and/or modify            *
 *   it under the terms of the Lesser GNU General Public License as           *
 *   published by the Free Software Foundation, either version 3 of           *
 *   the License, or (at your option) any later version.                      *
 *                                                                            *
 * DFT-EFE is distributed in the hope that it will be useful, but             *
 *   WITHOUT ANY WARRANTY; without even the implied warranty                  *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                     *
 *   See the Lesser GNU General Public License for more details.              *
 *                                                                            *
 * You should have received a copy of the GNU Lesser General Public           *
 *   License at the top level of DFT-EFE distribution.  If not, see           *
 *   <https://www.gnu.org/licenses/>.                                         *
 ******************************************************************************/
 
/*
 * @author Avirup Sircar
 */
 
#ifndef dftefeSphericalDataMixed_h
#define dftefeSphericalDataMixed_h
 
#include <utils/TypeConfig.h>
#include <vector>
#include <utils/Point.h>
#include <atoms/SphericalData.h>
#include <utils/Spline.h>
#include <utils/ScalarSpatialFunction.h>
#include <memory>
#include <utils/Point.h>
#include <atoms/Defaults.h>
#include <atoms/SphericalHarmonicFunctions.h>
#include <utils/Exceptions.h>
 
namespace dftefe
{
  namespace atoms
  {
    class SphericalDataMixed : public SphericalData
    {
    public:
      SphericalDataMixed(
        const std::vector<int>                  qNumbers,
        const std::vector<double>               radialPoints,
        const std::vector<double>               radialValues,
        utils::Spline::bd_type                  left,
        double                                  leftValue,
        utils::Spline::bd_type                  right,
        double                                  rightValue,
        const utils::ScalarSpatialFunctionReal &funcAfterRadialGrid,
        const SphericalHarmonicFunctions &      sphericalHarmonicFunc,
        const double polarAngleTolerance = SphericalDataDefaults::POL_ANG_TOL,
        const size_type dim              = SphericalDataDefaults::DEFAULT_DIM);
 
      ~SphericalDataMixed() = default;
 
      void
      initSpline(utils::Spline::bd_type left,
                 double                 leftValue,
                 utils::Spline::bd_type right,
                 double                 rightValue);
 
      std::vector<double>
      getValue(const std::vector<utils::Point> &point,
               const utils::Point &             origin) override;
 
      std::vector<double>
      getGradientValue(const std::vector<utils::Point> &point,
                       const utils::Point &             origin) override;
 
      std::vector<double>
      getHessianValue(const std::vector<utils::Point> &point,
                      const utils::Point &             origin) override;
 
      double
      getValue(const utils::Point &point, const utils::Point &origin) override;
 
      std::vector<double>
      getGradientValue(const utils::Point &point,
                       const utils::Point &origin) override;
 
      std::vector<double>
      getHessianValue(const utils::Point &point,
                      const utils::Point &origin) override;
 
      std::vector<double>
      getRadialValue(const std::vector<double> &r) override;
 
      std::vector<double>
      getAngularValue(const std::vector<double> &r,
                      const std::vector<double> &theta,
                      const std::vector<double> &phi) override;
 
      std::vector<double>
      getRadialDerivative(const std::vector<double> &r) override;
 
      std::vector<std::vector<double>>
      getAngularDerivative(const std::vector<double> &r,
                           const std::vector<double> &theta,
                           const std::vector<double> &phi) override;
 
      std::vector<int>
      getQNumbers() const override;
 
      double
      getCutoff() const override;
 
      double
      getSmoothness() const override;
 
      void
      getValue(const size_type numPoints,
               const double *  points,
               const double *  origin,
               double *        out) override
      {
        utils::throwException(
          false, "getValue not implemented for SphericalDataMixed.");
      }
 
      void
      getGradientValue(const size_type numPoints,
                       const double *  points,
                       const double *  origin,
                       double *        out) override
      {
        utils::throwException(
          false, "getGradientValue not implemented for SphericalDataMixed.");
      }
 
      void
      getHessianValue(const size_type numPoints,
                      const double *  points,
                      const double *  origin,
                      double *        out) override
      {
        utils::throwException(
          false, "getHessianValue not implemented for SphericalDataMixed.");
      }
 
#ifdef DFTEFE_WITH_DEVICE
      void
      getValueDevice(
        const size_type       numPoints,
        const double *        points,
        const double *        origin,
        double *              out,
        utils::deviceStream_t streamId = utils::defaultStream) override;
 
      void
      getGradientValueDevice(
        const size_type       numPoints,
        const double *        points,
        const double *        origin,
        double *              out,
        utils::deviceStream_t streamId = utils::defaultStream) override
      {
        utils::throwException(false,
                              "getGradientValueDevice not implemented for "
                              "SphericalDataMixed.");
      }
 
      void
      getHessianValueDevice(
        const size_type       numPoints,
        const double *        points,
        const double *        origin,
        double *              out,
        utils::deviceStream_t streamId = utils::defaultStream) override
      {
        utils::throwException(false,
                              "getHessianValueDevice not implemented for "
                              "SphericalDataMixed.");
      }
#endif
 
    private:
      std::vector<int>                     d_qNumbers;
      std::vector<double>                  d_radialPoints;
      std::vector<double>                  d_radialValues;
      std::shared_ptr<const utils::Spline> d_spline;
      double                               d_polarAngleTolerance;
      double                               d_cutoffTolerance;
      double                               d_radiusTolerance;
      size_type                            d_dim;
 
      const SphericalHarmonicFunctions &      d_sphericalHarmonicFunc;
      const utils::ScalarSpatialFunctionReal &d_funcAfterRadialGrid;
    };
 
  } // end of namespace atoms
} // end of namespace dftefe
#endif // dftefeSphericalDataMixed_h