Added CFEBDSOnTheFlyComputeDealii.t.cpp

src/basis/CFEBDSOnTheFlyComputeDealii.cpp

@@ -0,0 +1 @@
+#include "CFEBDSOnTheFlyComputeDealii.h"

src/basis/CFEBDSOnTheFlyComputeDealii.h

@@ -0,0 +1,242 @@
+/******************************************************************************
+ * 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 dftefeCFEBDSOnTheFlyComputeDealii_h
+#define dftefeCFEBDSOnTheFlyComputeDealii_h
+
+#include <utils/TypeConfig.h>
+#include <utils/MemorySpaceType.h>
+#include <utils/MemoryStorage.h>
+#include <basis/BasisDataStorage.h>
+#include <basis/FEBasisDataStorage.h>
+#include <basis/CFEBasisDofHandlerDealii.h>
+#include <basis/FEBasisDofHandler.h>
+#include <basis/LinearCellMappingDealii.h>
+#include <quadrature/QuadratureRuleGauss.h>
+#include <quadrature/QuadratureRuleGLL.h>
+#include <memory>
+#include <map>
+#include <vector>
+namespace dftefe
+{
+  namespace basis
+  {
+    /**
+     * @brief An abstract class to store and access data for a given basis,
+     * such as the basis function values on a quadrature grid, the overlap
+     * matrix of the basis, etc.
+     */
+    template <typename ValueTypeBasisCoeff,
+              typename ValueTypeBasisData,
+              dftefe::utils::MemorySpace memorySpace,
+              size_type                  dim>
+    class CFEBDSOnTheFlyComputeDealii
+      : public FEBasisDataStorage<ValueTypeBasisData, memorySpace>
+    {
+    public:
+      using QuadraturePointAttributes = quadrature::QuadraturePointAttributes;
+      using QuadratureRuleAttributes  = quadrature::QuadratureRuleAttributes;
+      using Storage =
+        typename BasisDataStorage<ValueTypeBasisData, memorySpace>::Storage;
+
+      CFEBDSOnTheFlyComputeDealii(
+        std::shared_ptr<const BasisDofHandler>      feBDH,
+        const quadrature::QuadratureRuleAttributes &quadratureRuleAttributes,
+        const BasisStorageAttributesBoolMap basisStorageAttributesBoolMap,
+        const size_type                     maxCellBlock,
+        linearAlgebra::LinAlgOpContext<memorySpace> &linAlgOpContext);
+
+      ~CFEBDSOnTheFlyComputeDealii() = default;
+
+      std::shared_ptr<const BasisDofHandler>
+      getBasisDofHandler() const override;
+
+      void
+      evaluateBasisData(
+        const quadrature::QuadratureRuleAttributes &quadratureRuleAttributes,
+        const BasisStorageAttributesBoolMap basisStorageAttributesBoolMap)
+        override;
+
+      void
+      evaluateBasisData(
+        const quadrature::QuadratureRuleAttributes &quadratureRuleAttributes,
+        std::shared_ptr<const quadrature::QuadratureRuleContainer>
+                                            quadratureRuleContainer,
+        const BasisStorageAttributesBoolMap basisStorageAttributesBoolMap)
+        override;
+
+      void
+      evaluateBasisData(
+        const quadrature::QuadratureRuleAttributes &quadratureRuleAttributes,
+        std::vector<std::shared_ptr<const quadrature::QuadratureRule>>
+                                            quadratureRuleVec,
+        const BasisStorageAttributesBoolMap basisStorageAttributesBoolMap)
+        override;
+
+      void
+      evaluateBasisData(
+        const quadrature::QuadratureRuleAttributes &quadratureRuleAttributes,
+        std::shared_ptr<const quadrature::QuadratureRule>
+          baseQuadratureRuleAdaptive,
+        std::vector<std::shared_ptr<const utils::ScalarSpatialFunctionReal>>
+          &                                 functions,
+        const std::vector<double> &         absoluteTolerances,
+        const std::vector<double> &         relativeTolerances,
+        const std::vector<double> &         integralThresholds,
+        const double                        smallestCellVolume,
+        const unsigned int                  maxRecursion,
+        const BasisStorageAttributesBoolMap basisStorageAttributesBoolMap)
+        override;
+
+
+      // void
+      // evaluateBasisData(
+      //   std::shared_ptr<const quadrature::QuadratureRuleContainer>
+      //                                       quadratureRuleContainer,
+      //   const QuadratureRuleAttributes &    quadratureRuleAttributes,
+      //   const BasisStorageAttributesBoolMap boolBasisStorageFlagsObj)
+      //   override;
+
+      void
+      deleteBasisData() override;
+
+
+
+      //      std::shared_ptr<const quadrature::QuadratureRuleContainer>
+      //      getCellQuadratureRuleContainer(std::shared_ptr<Storage>>
+      //        const QuadratureRuleAttributes &quadratureRuleAttributes) const
+      //        override;
+      // functions to get data for a basis function on a given quad point in a
+      // cell
+      Storage
+      getBasisData(const QuadraturePointAttributes &attributes,
+                   const size_type                  basisId) const override;
+      Storage
+      getBasisGradientData(const QuadraturePointAttributes &attributes,
+                           const size_type basisId) const override;
+      Storage
+      getBasisHessianData(const QuadraturePointAttributes &attributes,
+                          const size_type basisId) const override;
+
+      // functions to get data for a basis function on all quad points in a cell
+      Storage
+      getBasisDataInCell(const size_type cellId,
+                         const size_type basisId) const override;
+      Storage
+      getBasisGradientDataInCell(const size_type cellId,
+                                 const size_type basisId) const override;
+      Storage
+      getBasisHessianDataInCell(const size_type cellId,
+                                const size_type basisId) const override;
+
+      // functions to get data for all basis functions on all quad points in a
+      // cell
+      Storage
+      getBasisDataInCell(const size_type cellId) const override;
+      Storage
+      getBasisGradientDataInCell(const size_type cellId) const override;
+      Storage
+      getBasisHessianDataInCell(const size_type cellId) const override;
+
+      Storage
+      getJxWInCell(const size_type cellId) const override;
+
+      // functions to get data for all basis functions on all quad points in all
+      // cells
+      const Storage &
+      getBasisDataInAllCells() const override;
+      const Storage &
+      getBasisGradientDataInAllCells() const override;
+      const Storage &
+      getBasisHessianDataInAllCells() const override;
+
+      const Storage &
+      getJxWInAllCells() const override;
+
+      // get overlap of two basis functions in a cell
+      Storage
+      getBasisOverlap(const size_type cellId,
+                      const size_type basisId1,
+                      const size_type basisId2) const override;
+
+      // get overlap of all the basis functions in a cell
+      Storage
+      getBasisOverlapInCell(const size_type cellId) const override;
+
+      // get overlap of all the basis functions in all cells
+      const Storage &
+      getBasisOverlapInAllCells() const override;
+
+
+      // get the laplace operator in a cell
+      Storage
+      getBasisGradNiGradNjInCell(const size_type cellId) const override;
+
+      // get laplace operator in all cells
+      const Storage &
+      getBasisGradNiGradNjInAllCells() const override;
+
+
+      std::shared_ptr<const quadrature::QuadratureRuleContainer>
+      getQuadratureRuleContainer() const override;
+
+      // functions to get data for  basis functions on all quad points in range
+      // of cells
+      void
+      getBasisDataInCellRange(std::pair<size_type, size_type> cellRange,
+                              Storage &basisData) const override;
+      void
+      getBasisGradientDataInCellRange(
+        std::pair<size_type, size_type> cellRange,
+        Storage &                       basisGradientData) const override;
+
+    private:
+      bool d_evaluateBasisData;
+      std::shared_ptr<
+        const CFEBasisDofHandlerDealii<ValueTypeBasisCoeff, memorySpace, dim>>
+        d_feBDH;
+      std::shared_ptr<const quadrature::QuadratureRuleContainer>
+                                    d_quadratureRuleContainer;
+      QuadratureRuleAttributes      d_quadratureRuleAttributes;
+      BasisStorageAttributesBoolMap d_basisStorageAttributesBoolMap;
+      std::shared_ptr<Storage>      d_basisQuadStorage;
+      std::shared_ptr<Storage>      d_JxWStorage;
+      std::shared_ptr<Storage>      d_basisGradientParaCellQuadStorage;
+      std::shared_ptr<Storage>      d_basisJacobianInvQuadStorage;
+      std::shared_ptr<Storage>      d_basisHessianQuadStorage;
+      std::vector<size_type>        d_dofsInCell;
+      std::vector<size_type>        d_nQuadPointsIncell;
+      std::vector<size_type>        d_cellStartIdsBasisJacobianInvQuadStorage;
+      std::vector<size_type>        d_cellStartIdsBasisHessianQuadStorage;
+      std::vector<size_type>        d_cellStartIdsBasisQuadStorage;
+      size_type                     d_maxCellBlock;
+      Storage                       d_tmpGradientBlock;
+      linearAlgebra::LinAlgOpContext<memorySpace> &d_linAlgOpContext;
+
+    }; // end of CFEBDSOnTheFlyComputeDealii
+  }    // end of namespace basis
+} // end of namespace dftefe
+#include <basis/CFEBDSOnTheFlyComputeDealii.t.cpp>
+#endif // dftefeCFEBDSOnTheFlyComputeDealii_h