fixup: set corner for temperature BC

ORNL · Feb 14, 2024 · 100050a · 100050a
1 parent 9b68c26
commit 100050a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/examples/thermal_deformation.cpp b/examples/thermal_deformation.cpp
@@ -91,6 +91,7 @@ int main( int argc, char* argv[] )
 
         auto bc = createBoundaryCondition( CabanaPD::TempBCTag{}, 5000.0,
                                            exec_space{}, *particles, domain );
+        bc.setCorner( low_corner[1] );
 
         auto init_functor = KOKKOS_LAMBDA( const int pid )
         {

diff --git a/src/CabanaPD_Boundary.hpp b/src/CabanaPD_Boundary.hpp
@@ -177,13 +177,16 @@ struct BoundaryCondition<BCIndexSpace, TempBCTag>
 {
     double _value;
     BCIndexSpace _index_space;
+    double _low_corner = 0.0;
 
     BoundaryCondition( const double value, BCIndexSpace bc_index_space )
         : _value( value )
         , _index_space( bc_index_space )
     {
     }
 
+    void setCorner( const double low ) { _low_corner = low; }
+
     template <class ExecSpace, class Particles>
     void update( ExecSpace exec_space, Particles particles,
                  RegionBoundary plane )
@@ -197,13 +200,14 @@ struct BoundaryCondition<BCIndexSpace, TempBCTag>
         auto temp = particles.sliceTemperature();
         auto x = particles.sliceReferencePosition();
         auto value = _value;
+        auto low_corner = _low_corner;
         auto index_space = _index_space._view;
         Kokkos::RangePolicy<ExecSpace> policy( 0, index_space.size() );
         Kokkos::parallel_for(
             "CabanaPD::BC::apply", policy, KOKKOS_LAMBDA( const int b ) {
                 auto pid = index_space( b );
                 // This is specifically for the thermal deformation problem
-                temp( pid ) = value * x( pid, 1 ) * t;
+                temp( pid ) = value * ( x( pid, 1 ) - low_corner ) * t;
             } );
     }
 };