From d9c8e35429277bb00966bae2dd82ab00d41bf1c3 Mon Sep 17 00:00:00 2001
From: "Alberto F. Martin" <alberto.martin@monash.edu>
Date: Mon, 2 Sep 2024 15:41:27 +1000
Subject: [PATCH] Added missing BulkGhostPenaltyAssemblyMaps.jl source from
 previous commit

---
 BulkGhostPenaltyAssembleMaps.jl | 64 +++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 BulkGhostPenaltyAssembleMaps.jl

diff --git a/BulkGhostPenaltyAssembleMaps.jl b/BulkGhostPenaltyAssembleMaps.jl
new file mode 100644
index 0000000..3068b92
--- /dev/null
+++ b/BulkGhostPenaltyAssembleMaps.jl
@@ -0,0 +1,64 @@
+# TO-DO: Better name?
+struct BulkGhostPenaltyAssembleLhsMap{A} <: Gridap.Fields.Map
+    agg_cells_lhs_contribs::A
+end
+
+function _get_rank(::Type{Array{T,N}}) where {T,N}
+  N
+end
+
+function Gridap.Fields.return_cache(m::BulkGhostPenaltyAssembleLhsMap,cells)
+    cache_unassembled_lhs=array_cache(m.agg_cells_lhs_contribs)
+    T=eltype(m.agg_cells_lhs_contribs)
+    evaluate_result=Gridap.Arrays.CachedArray(eltype(T),_get_rank(T))
+    cache_unassembled_lhs,evaluate_result
+end
+  
+function Gridap.Fields.evaluate!(cache,m::BulkGhostPenaltyAssembleLhsMap,cells)
+    cache_unassembled_lhs,result=cache
+    contrib = getindex!(cache_unassembled_lhs,m.agg_cells_lhs_contribs,1)
+
+    Gridap.Arrays.setsize!(result,size(contrib))
+    result.array .= 0.0
+    for (i,cell) in enumerate(cells)
+        contrib = getindex!(cache_unassembled_lhs,m.agg_cells_lhs_contribs,cell)
+        result.array .+= contrib
+    end
+    result.array
+end
+
+struct BulkGhostPenaltyAssembleRhsMap{A,B} <: Gridap.Fields.Map 
+    agg_cells_local_dof_ids::A 
+    agg_cells_rhs_contribs::B
+end 
+
+function Gridap.Fields.return_cache(m::BulkGhostPenaltyAssembleRhsMap,aggregate_local_cells)
+  cache_agg_cells_local_dof_ids=array_cache(m.agg_cells_local_dof_ids)
+  cache_unassembled_rhs=array_cache(m.agg_cells_rhs_contribs)
+  evaluate_result=Gridap.Arrays.CachedArray(eltype(eltype(m.agg_cells_rhs_contribs)),2)
+  cache_agg_cells_local_dof_ids,cache_unassembled_rhs,evaluate_result
+end
+
+function Gridap.Fields.evaluate!(cache,m::BulkGhostPenaltyAssembleRhsMap,aggregate_local_cells)
+  cache_agg_cells_local_dof_ids,cache_unassembled_rhs,result=cache
+  contrib = getindex!(cache_unassembled_rhs,m.agg_cells_rhs_contribs,1)
+
+  max_local_dof_id=-1
+  for (i,cell) in enumerate(aggregate_local_cells)
+    current_cell_local_dof_ids = getindex!(cache_agg_cells_local_dof_ids,m.agg_cells_local_dof_ids,cell)
+    for local_dof in current_cell_local_dof_ids
+        max_local_dof_id=max(max_local_dof_id,local_dof)
+    end
+  end
+
+  Gridap.Arrays.setsize!(result,(size(contrib,1),max_local_dof_id))
+  result.array .= 0.0
+  for (i,cell) in enumerate(aggregate_local_cells)
+        current_cell_local_dof_ids = getindex!(cache_agg_cells_local_dof_ids,m.agg_cells_local_dof_ids,cell)
+        contrib = getindex!(cache_unassembled_rhs,m.agg_cells_rhs_contribs,cell)
+        for (j,local_dof) in enumerate(current_cell_local_dof_ids)
+           result.array[:,local_dof] += contrib[:,j]
+        end
+  end
+  result.array
+end
\ No newline at end of file