provide nproma in addition to block_size, since the first block_size …

…is not necessarally equal nproma
ecmwf · Aug 23, 2023 · 7893f78 · 7893f78
1 parent 9efc1ec
commit 7893f78
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/src/atlas/functionspace/detail/BlockStructuredColumnsInterface.cc b/src/atlas/functionspace/detail/BlockStructuredColumnsInterface.cc
@@ -175,6 +175,9 @@ idx_t atlas__fs__BStructuredColumns__block_begin(const detail::BlockStructuredCo
 idx_t atlas__fs__BStructuredColumns__block_size(const detail::BlockStructuredColumns* This, idx_t jblk) {
     return This->block_size(jblk);
 }
+idx_t atlas__fs__BStructuredColumns__nproma(const detail::BlockStructuredColumns* This) {
+    return This->nproma();
+}
 idx_t atlas__fs__BStructuredColumns__nblks(const detail::BlockStructuredColumns* This) {
     return This->nblks();
 }

diff --git a/src/atlas/functionspace/detail/BlockStructuredColumnsInterface.h b/src/atlas/functionspace/detail/BlockStructuredColumnsInterface.h
@@ -102,6 +102,7 @@ idx_t atlas__fs__BStructuredColumns__i_end_halo(const detail::BlockStructuredCol
 idx_t atlas__fs__BStructuredColumns__levels(const detail::BlockStructuredColumns* This);
 idx_t atlas__fs__BStructuredColumns__block_begin(const detail::BlockStructuredColumns* This, idx_t jblk);
 idx_t atlas__fs__BStructuredColumns__block_size(const detail::BlockStructuredColumns* This, idx_t jblk);
+idx_t atlas__fs__BStructuredColumns__nproma(const detail::BlockStructuredColumns* This);
 idx_t atlas__fs__BStructuredColumns__nblks(const detail::BlockStructuredColumns* This);
 
 field::FieldImpl* atlas__fs__BStructuredColumns__xy(const detail::BlockStructuredColumns* This);

diff --git a/src/atlas_f/functionspace/atlas_functionspace_BlockStructuredColumns_module.F90 b/src/atlas_f/functionspace/atlas_functionspace_BlockStructuredColumns_module.F90
@@ -89,6 +89,7 @@ module atlas_functionspace_BlockStructuredColumns_module
   procedure :: levels
   procedure :: block_begin
   procedure :: block_size
+  procedure :: nproma
   procedure :: nblks
 
   procedure :: xy
@@ -554,6 +555,13 @@ function block_size(this,j) result(i)
   i = atlas__fs__BStructuredColumns__block_size(this%CPTR_PGIBUG_A,j-1)
 end function
 
+function nproma(this) result(i)
+  use atlas_functionspace_BlockStructuredColumns_c_binding
+  integer(ATLAS_KIND_IDX) :: i
+  class(atlas_functionspace_BlockStructuredColumns), intent(in) :: this
+  i = atlas__fs__BStructuredColumns__nproma(this%CPTR_PGIBUG_A)
+end function
+
 function nblks(this) result(i)
   use atlas_functionspace_BlockStructuredColumns_c_binding
   integer(ATLAS_KIND_IDX) :: i