ParticleSetIterators.h

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
#pragma once
 
 
#include "ParticleUpdatePredicates.h"
#include "exaparty/CellMetaData.h"
#include "tarch/multicore/BooleanSemaphore.h"
#include "tarch/multicore/Lock.h"
 
 
namespace exaparty {
  namespace kernels {
 
    // template for convenience and code legibility.
    template <typename ParticleContainer>
    using PCParticle = typename std::remove_pointer<typename ParticleContainer::value_type>::type;
 
    template <typename F, typename Particle>
    concept ParticleUnaryOperatorOnVertex = requires(
      F                                           f,
      const peano4::datamanagement::VertexMarker& marker,
      Particle&                                   localParticle
    ) {
      { f(marker, localParticle) } -> std::same_as<void>;
    };
 
    template <typename F, typename Particle>
    concept ParticleUnaryOperatorOnCell = requires(
      F                                         f,
      const peano4::datamanagement::CellMarker& marker,
      Particle&                                 localParticle
    ) {
      { f(marker, localParticle) } -> std::same_as<void>;
    };
 
    template <typename F, typename LocalParticle, typename ActiveParticle>
    concept ParticleBinaryOperator = requires(
      F                                         f,
      const peano4::datamanagement::CellMarker& marker,
      LocalParticle&                            localParticle,
      ActiveParticle&                           activeParticle
    ) {
      { f(marker, localParticle, activeParticle) } -> std::same_as<void>;
    };
 
    namespace kernelrealisation {
      class GenericRealisation {};
      class CoalescedMemoryRealisation {};
      class CoalescedMemoryLocalActiveRealisation {};
      class CoalescedMemoryActiveLocalRealisation {};
      class EvaluatePredicateInPrologueRealisation {};
      class EvaluatePredicateInPrologueLocalActiveRealisation {};
      class EvaluatePredicateInPrologueActiveLocalRealisation {};
 
      class GenericOffloadRealisation {};
      class CoalescedMemoryOffloadRealisation {};
      class CoalescedMemoryLocalActiveOffloadRealisation {};
      class CoalescedMemoryActiveLocalOffloadRealisation {};
 
      extern GenericRealisation                     Generic;
      extern CoalescedMemoryRealisation             CoalescedMemory;
      extern CoalescedMemoryLocalActiveRealisation  CoalescedMemoryLocalActive;
      extern CoalescedMemoryActiveLocalRealisation  CoalescedMemoryActiveLocal;
      extern EvaluatePredicateInPrologueRealisation EvaluatePredicateInPrologue;
      extern EvaluatePredicateInPrologueLocalActiveRealisation
        EvaluatePredicateInPrologueLocalActive;
      extern EvaluatePredicateInPrologueActiveLocalRealisation
                                       EvaluatePredicateInPrologueActiveLocal;
 
      extern GenericOffloadRealisation GenericOffload;
      extern CoalescedMemoryOffloadRealisation CoalescedMemoryOffload;
      extern CoalescedMemoryLocalActiveOffloadRealisation CoalescedMemoryLocalActiveOffload;
      extern CoalescedMemoryActiveLocalOffloadRealisation CoalescedMemoryActiveLocalOffload;
    } // namespace kernelrealisation
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::VertexMarker& marker,
      auto&                                       assignedParticles,
      int  numberOfCoalescedAssignedParticles,
      ::exaparty::kernels::kernelrealisation::GenericRealisation realisation
    );
 
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::VertexMarker& marker,
      auto&                                       assignedParticles,
      int numberOfCoalescedAssignedParticles,
      ::exaparty::kernels::kernelrealisation::GenericOffloadRealisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::GenericRealisation                        realisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::GenericOffloadRealisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::VertexMarker& marker,
      auto&                                       assignedParticles,
      int                                         numberOfAssignedParticles,
      ::exaparty::kernels::kernelrealisation::CoalescedMemoryRealisation
        realisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::CellMarker& marker,
      auto&                                     localParticles,
      const std::vector<int>&                   numberOfLocalParticles,
      ::exaparty::kernels::kernelrealisation::CoalescedMemoryRealisation
        realisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::VertexMarker& marker,
      auto&                                       assignedParticles,
      int                                         numberOfAssignedParticles,
      ::exaparty::kernels::kernelrealisation::CoalescedMemoryOffloadRealisation
        realisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::CellMarker& marker,
      auto&                                     localParticles,
      const std::vector<int>&                   numberOfLocalParticles,
      ::exaparty::kernels::kernelrealisation::CoalescedMemoryOffloadRealisation
        realisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        CoalescedMemoryActiveLocalRealisation realisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        CoalescedMemoryActiveLocalOffloadRealisation realisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueActiveLocalRealisation realisation
    );
 
    template <auto PairOperator, auto LocalPredicate, auto PairPredicate>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueLocalActiveRealisation realisation
    );
 
    template <
      auto ParticlePairOp,
      auto LocalPredicateOp,
      auto ParticlePairPredicateOp>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        CoalescedMemoryActiveLocalRealisation realisation
    );
 
    template <
      auto ParticlePairOp,
      auto LocalPredicateOp,
      auto ParticlePairPredicateOp>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        CoalescedMemoryLocalActiveRealisation realisation
    );
 
    template <
      auto ParticlePairOp,
      auto LocalPredicateOp,
      auto ParticlePairPredicateOp>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        CoalescedMemoryLocalActiveOffloadRealisation realisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::VertexMarker& marker,
      auto&                                       assignedParticles,
      int numberOfCoalescedAssignedParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueRealisation
    );
 
    template <auto ParticleOp, auto PredicateOp>
    void forAllParticles(
      const peano4::datamanagement::CellMarker& marker,
      auto&                                     localParticles,
      const std::vector<int>&                   numberOfCoalescedLocalParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueRealisation
    );
 
    template <typename LocalParticleContainer, typename ActiveParticleContainer>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueActiveLocalRealisation realisation
    );
 
    template <typename LocalParticleContainer, typename ActiveParticleContainer>
    void forAllParticlePairs(
      const exaparty::CellMetaData& cellMetaData,
      auto&                       localParticles,
      auto&                       activeParticles,
      ::exaparty::kernels::kernelrealisation::
        EvaluatePredicateInPrologueLocalActiveRealisation realisation
    );
 
    namespace internal {
      extern tarch::multicore::BooleanSemaphore _multiscaleInteractionSemaphore;
 
 
      template <
        int SubChunkSize,
        typename LocalParticle,
        typename ActiveParticle>
      std::pair<std::bitset<SubChunkSize>, std::bitset<SubChunkSize>>
        identifyInteractingParticlesFixedLocalParticle(
          const peano4::datamanagement::CellMarker& marker,
          LocalParticle*                            localParticle,
          ActiveParticle* firstActiveParticleInSubchunk,
          int             thisSubchunkSize,
          UpdateParticlePairWithinCellPredicate<LocalParticle>
            particlePairPredicate
        );
 
      template <
        int SubChunkSize,
        typename LocalParticle,
        typename ActiveParticle>
      std::pair<std::bitset<SubChunkSize>, std::bitset<SubChunkSize>>
        identifyInteractingParticlesFixedActiveParticle(
          const peano4::datamanagement::CellMarker& marker,
          LocalParticle*  firstLocalParticleInSubchunk,
          int             thisSubchunkSize,
          ActiveParticle* activeParticle,
          UpdateParticlePairWithinCellPredicate<LocalParticle>
            particlePairPredicate
        );
    } // namespace internal
  }   // namespace kernels
} // namespace exaparty
 
 
#include "ParticleSetIterators.cpph"