sll_m_pic_utilities.F90

Go to the documentation of this file.

!**************************************************************
!  Copyright INRIA
!  Authors :
!     CALVI project team
!
!  This code SeLaLib (for Semi-Lagrangian-Library)
!  is a parallel library for simulating the plasma turbulence
!  in a tokamak.
!
!  This software is governed by the CeCILL-B license
!  under French law and abiding by the rules of distribution
!  of free software.  You can  use, modify and redistribute
!  the software under the terms of the CeCILL-B license as
!  circulated by CEA, CNRS and INRIA at the following URL
!  "http://www.cecill.info".
!**************************************************************
 
module sll_m_pic_utilities
!+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#include "sll_assert.h"
#include "sll_memory.h"
#include "sll_working_precision.h"
#include "sll_accumulators.h"
 
   use sll_m_accumulators, only: &
      sll_t_charge_accumulator_2d, &
      sll_t_charge_accumulator_2d_cs, &
      sll_t_charge_accumulator_2d_cs_ptr, &
      sll_t_charge_accumulator_2d_ptr
 
   use sll_m_particle_group_2d, only: &
      sll_t_particle_group_2d
 
   use sll_m_particle_group_4d, only: &
      sll_t_particle_group_4d
 
   use sll_m_particle_representations, only: &
      sll_t_particle_2d, &
      sll_t_particle_4d
 
#ifdef _OPENMP
   use omp_lib, only: &
      omp_get_thread_num
 
#endif
   implicit none
 
   public :: &
      sll_s_first_charge_accumulation_2d, &
      sll_s_first_charge_accumulation_2d_cs, &
      sll_s_first_gc_charge_accumulation_2d, &
      sll_s_first_gc_charge_accumulation_2d_cs
 
   private
!+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#ifdef _OPENMP
!   logical :: openmp_st
#endif
!!    !$ openmp_st = OMP_IN_PARALLEL()
!!    print*, 'USE of omp', openmp_st
!!    !$omp end parallel
 
contains
 
   subroutine sll_s_first_charge_accumulation_2d(p_group, q_accumulator)
      type(sll_t_particle_group_4d), pointer                       :: p_group
      type(sll_t_charge_accumulator_2d_ptr), dimension(:), pointer :: q_accumulator
      type(sll_t_particle_4d), dimension(:), pointer :: p
      type(sll_t_charge_accumulator_2d), pointer :: q_accum
      sll_int64  :: i
      sll_int64  :: num_particles
      sll_real32 :: tmp1
      sll_real32 :: tmp2
      sll_int32  :: thread_id
 
      sll_assert(associated(p_group) .and. associated(q_accumulator))
      num_particles = int(p_group%number_particles, i64)
      p => p_group%p_list
 
!$omp parallel PRIVATE(thread_id, tmp1, tmp2, q_accum)
#ifdef _OPENMP
      thread_id = omp_get_thread_num()
#else
      thread_id = 0
#endif
      q_accum => q_accumulator(thread_id + 1)%q
!$omp do
      do i = 1, num_particles
         sll_accumulate_particle_charge(q_accum, p(i), tmp1, tmp2)
      end do
!$omp end do
!$omp end parallel
 
   end subroutine sll_s_first_charge_accumulation_2d
 
   subroutine sll_s_first_charge_accumulation_2d_cs(p_group, q_accumulator)
! ----  Remember : _CS is for use of Cubic Splines  ----
! ------------------------------------------------------
      type(sll_t_particle_group_4d), pointer         :: p_group
      type(sll_t_charge_accumulator_2d_cs_ptr), dimension(:), pointer  :: q_accumulator
      type(sll_t_particle_4d), dimension(:), pointer :: p
      type(sll_t_charge_accumulator_2d_cs), pointer :: q_accum
      sll_int64  :: i
      sll_int64  :: num_particles
      sll_real32 :: tmp(1:4, 1:2), temp
      sll_int32  :: thread_id
 
      sll_assert(associated(p_group) .and. associated(q_accumulator))
      num_particles = int(p_group%number_particles, i64)
      p => p_group%p_list
 
!$omp parallel default(SHARED) PRIVATE(thread_id, tmp, temp, q_accum)
#ifdef _OPENMP
      thread_id = omp_get_thread_num()
#else
      thread_id = 0
#endif
      q_accum => q_accumulator(thread_id + 1)%q
!$omp do
      do i = 1, num_particles
         sll_accumulate_particle_charge_cs(q_accum, p(i), tmp, temp)
      end do
!$omp end do
!$omp end parallel
 
   end subroutine sll_s_first_charge_accumulation_2d_cs
 
!!$ - - - -  for the GUIDING CENTER model   - - - -
   subroutine sll_s_first_gc_charge_accumulation_2d(p_group, q_accumulator)
      type(sll_t_particle_group_2d), pointer                       :: p_group
      type(sll_t_charge_accumulator_2d_ptr), dimension(:), pointer :: q_accumulator
      type(sll_t_particle_2d), dimension(:), pointer :: p
      type(sll_t_charge_accumulator_2d), pointer :: q_accum
      sll_int64  :: i
      sll_int64  :: num_particles
      sll_real32 :: tmp1
      sll_real32 :: tmp2
      sll_int32  :: thread_id
 
      sll_assert(associated(p_group) .and. associated(q_accumulator))
      num_particles = int(p_group%number_particles, i64)
      p => p_group%p_list
 
!$omp parallel default(SHARED) PRIVATE(thread_id, tmp1, tmp2, q_accum)
#ifdef _OPENMP
      thread_id = omp_get_thread_num()
#else
      thread_id = 0
#endif
      q_accum => q_accumulator(thread_id + 1)%q
!$omp do
      do i = 1, num_particles
         sll_accumulate_particle_charge(q_accum, p(i), tmp1, tmp2)
      end do
!$omp end do
!$omp end parallel
 
   end subroutine sll_s_first_gc_charge_accumulation_2d
 
   subroutine sll_s_first_gc_charge_accumulation_2d_cs(p_group, q_accumulator)
      type(sll_t_particle_group_2d), pointer         :: p_group
      type(sll_t_charge_accumulator_2d_cs_ptr), dimension(:), pointer  :: q_accumulator
      type(sll_t_particle_2d), dimension(:), pointer :: p
      type(sll_t_charge_accumulator_2d_cs), pointer :: q_accum
      sll_int64  :: i
      sll_int64  :: num_particles
      sll_real32 :: tmp(1:4, 1:2), temp
      sll_int32  :: thread_id
 
      sll_assert(associated(p_group) .and. associated(q_accumulator))
      num_particles = int(p_group%number_particles, i64)
      p => p_group%p_list
 
!$omp parallel default(SHARED) PRIVATE(thread_id, tmp, temp, q_accum)
#ifdef _OPENMP
      thread_id = omp_get_thread_num()
#else
      thread_id = 0
#endif
      q_accum => q_accumulator(thread_id + 1)%q
!$omp do
      do i = 1, num_particles
         sll_accumulate_particle_charge_cs(q_accum, p(i), tmp, temp)
      end do
!$omp end do
!$omp end parallel
 
   end subroutine sll_s_first_gc_charge_accumulation_2d_cs
 
end module sll_m_pic_utilities