InterpolateFromCellToLocalPoints.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (c) 2017-2025  The Regents of the University of Michigan and DFT-FE
// authors.
//
// This file is part of the DFT-FE code.
//
// The DFT-FE code is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the DFT-FE distribution.
//
// ---------------------------------------------------------------------
//
 
/*
 * @author Vishal Subramanian, Bikash Kanungo
 */
 
#ifndef DFTFE_EXE_INTERPOLATEFROMCELLTOLOCALPOINTS_H
#define DFTFE_EXE_INTERPOLATEFROMCELLTOLOCALPOINTS_H
 
 
#include "BLASWrapper.h"
#include "Cell.h"
#include "dftUtils.h"
#include "FECell.h"
 
 
namespace dftfe
{
  template <dftfe::utils::MemorySpace memorySpace>
  class InterpolateFromCellToLocalPoints
  {
  public:
    InterpolateFromCellToLocalPoints(
      const std::shared_ptr<const dftfe::utils::FECell<3>> &srcCell,
      dftfe::uInt                                           numNodes,
      bool                                                  memOpt);
 
    void
    setRealCoordinatesOfLocalPoints(dftfe::uInt          numPoints,
                                    std::vector<double> &coordinates);
 
    void
    interpolate(
      const std::shared_ptr<dftfe::linearAlgebra::BLASWrapper<memorySpace>>
                              &BLASWrapperPtr,
      dftfe::uInt              numberOfVectors,
      const dataTypes::number *parentNodalMemSpacePtr,
      dataTypes::number       *outputMemSpacePtr);
 
    void
    interpolate(const std::shared_ptr<dftfe::linearAlgebra::BLASWrapper<
                  dftfe::utils::MemorySpace::HOST>>  &BLASWrapperPtr,
                dftfe::uInt                           numberOfVectors,
                const std::vector<dataTypes::number> &parentNodalHost,
                std::vector<dataTypes::number>       &outputHost);
 
  private:
    std::shared_ptr<const dftfe::utils::FECell<3>> d_srcCell;
    dftfe::uInt                                    d_numNodes, d_numPoints;
    dftfe::utils::MemoryStorage<dataTypes::number, memorySpace>
                                   d_shapeValuesMemSpace;
    std::vector<dataTypes::number> d_shapeValuesHost;
 
    const std::shared_ptr<dftfe::linearAlgebra::BLASWrapper<memorySpace>>
      d_BLASWrapperPtr;
 
    std::vector<double> d_paramCoordinates;
    bool                d_memOpt;
  };
 
} // namespace dftfe
#endif // DFTFE_EXE_INTERPOLATEFROMCELLTOLOCALPOINTS_H