mesh_kernel_manager.h

// -*-c++-*-
#ifndef STORMM_MESH_KERNEL_MANAGER_H
#define STORMM_MESH_KERNEL_MANAGER_H
 
#include <map>
#include <string>
#include "copyright.h"
#include "Constants/behavior.h"
#include "DataTypes/stormm_vector_types.h"
#include "Math/math_enumerators.h"
#include "Structure/structure_enumerators.h"
#include "Topology/atomgraph_enumerators.h"
#include "gpu_details.h"
#include "kernel_format.h"
 
namespace stormm {
namespace card {
 
using constants::PrecisionModel;
using stmath::Interpolant;
using structure::BoundaryCondition;
using structure::GridDetail;
using structure::MappingActivity;
using topology::UnitCellType;
  
class MeshKlManager : public KernelManager {
public:

  MeshKlManager(const GpuDetails &gpu_in = null_gpu);
  
  int getArchBlockMultiplier() const;

  int2 getMeshKernelDims(PrecisionModel prec, GridDetail picture, BoundaryCondition bounds,
                         MappingActivity process) const;
 
  int2 getMeshKernelDims(PrecisionModel prec, GridDetail picture, BoundaryCondition bounds,
                         UnitCellType unit_cell, Interpolant stencil_kind,
                         MappingActivity process) const;
 
private:

  int occ_mesh_block_multiplier_dp;
  int occ_mesh_block_multiplier_sp;
  int nbf_mesh_block_multiplier_dp;
  int nbf_mesh_block_multiplier_sp;

  void catalogMeshKernel(PrecisionModel prec, MappingActivity process, GridDetail picture,
                         BoundaryCondition bounds,
                         const std::string &kernel_name = std::string(""));
 
  void catalogMeshKernel(PrecisionModel prec, MappingActivity process, GridDetail picture,
                         BoundaryCondition bounds, UnitCellType unit_cell,
                         Interpolant stencil_kind,
                         const std::string &kernel_name = std::string(""));
};

int meshBlockMultiplier(PrecisionModel prec, GridDetail picture);

std::string meshKernelKey(PrecisionModel prec, GridDetail picture, BoundaryCondition bounds,
                          MappingActivity process);
 
std::string meshKernelKey(PrecisionModel prec, GridDetail picture, BoundaryCondition bounds,
                          UnitCellType unit_cell, Interpolant stencil_kind,
                          MappingActivity process);
 
} // card
} // stormm
 
#endif