UpdateParticle_SingleLevelInteraction.py

Go to the documentation of this file.

# This file is part of the ExaParty project. For conditions of distribution and
# use, please see the copyright notice at www.peano-framework.org
import peano4.api.toolbox.particles
 
 
from .TaskGraphKernelWrappers import (
    construct_touch_vertex_first_time_call,
    construct_touch_cell_first_time_call,
    construct_touch_vertex_last_time_call,
)
 
class UpdateParticle_SingleLevelInteraction(
    peano4.api.toolbox.particles.UpdateParticle_SingleLevelInteraction
):
    """!
    
    Wrapper around toolbox baseclass
    
    Close to trivial wrapper which really only adds a tarch::multicore::waitForAllTasks()
    iff the particles might move. The interesting stuff is actually happing in 
    TaskGarphKernelWrappers.py however.
    
    """
    def __init__(
        self,
        particle_set,
        particle_cell_update_kernel,
        touch_vertex_first_time_compute_particle_update_kernel,
        touch_vertex_last_time_compute_particle_update_kernel,
        prepare_traversal_kernel,
        unprepare_traversal_kernel,
        additional_includes,
        active_particle_set,
        alter_particle_position_or_interaction_radius,
        wait_for_all_tasks_to_finish_after_traversal,
    ):
        """!
 
        Construct
 
 
        @param wait_for_all_tasks_to_finish_after_traversal: Boolean
 
        @param alter_particle_position_or_interaction_radius: Boolean
          Set if the underlying AlgorithmStep's effect might change a
          particle's position or interaction radius.
          @see exaparty.api.particle.AlgorithmStep.AlgorithmStep.may_trigger_rerun()
 
        """
        if wait_for_all_tasks_to_finish_after_traversal:
            unprepare_traversal_kernel = (
                unprepare_traversal_kernel + "tarch::multicore::waitForAllTasks();"
            )
 
        super(UpdateParticle_SingleLevelInteraction, self).__init__(
            particle_set,
            particle_cell_update_kernel=construct_touch_cell_first_time_call(
                particle_set,particle_cell_update_kernel,False
            ),
            touch_vertex_first_time_compute_particle_update_kernel=construct_touch_vertex_first_time_call(
                particle_set,touch_vertex_first_time_compute_particle_update_kernel,False
            ),
            touch_vertex_last_time_compute_particle_update_kernel=construct_touch_vertex_last_time_call(
                particle_set,touch_vertex_last_time_compute_particle_update_kernel,False, alter_particle_position_or_interaction_radius
            ),
            prepare_traversal_kernel=prepare_traversal_kernel,
            unprepare_traversal_kernel=unprepare_traversal_kernel,
            additional_includes=additional_includes
            + """
#include "tarch/multicore/Task.h"
#include "exaparty/exaparty.h"                
#include "exaparty/TaskNumber.h"                
                 """,
            active_particle_set=active_particle_set,
        )
 
 
    def get_action_set_name(self):
        return __name__.replace(".py", "").replace(".", "_")