Homec4science
Diffusion FENNECS: Finite Element Non-Neutral Electron Cloud Simulator (geom_testpardiso) Browse

fennecs/src/sort_mod.f90
065f4c6c57d8geom_testpardiso

Last Change
Raw File

sort_mod.f90
View Options

!------------------------------------------------------------------------------
! EPFL/Swiss Plasma Center
!------------------------------------------------------------------------------
!
! MODULE: sort
!
!> @author
!> Guillaume Le Bars EPFL/SPC
!
! DESCRIPTION:
!> Module containing particle sorting algorithms.
!------------------------------------------------------------------------------
MODULE sort
IMPLICIT NONE
CONTAINS
!---------------------------------------------------------------------------
!> @author
!> Guillaume Le Bars EPFL/SPC
!
! DESCRIPTION:
!>
!> @brief Sorts the particles according to their z position using quick sort algorithm
!
!> @param[in] leftlimit the particles variable needing to be allocated.
!> @param[in] right limit the maximum number of particles that will be stored in this variable
!---------------------------------------------------------------------------
RECURSIVE SUBROUTINE quicksortparts(leftlimit, rightlimit)
Use beam, ONLY: parts, exchange_parts
INTEGER,INTENT(IN):: leftlimit, rightlimit
DOUBLE PRECISION:: pivot
INTEGER::i, cnt, mid
IF(leftlimit .ge. rightlimit) RETURN ! Impossible indices, return
mid=(leftlimit+rightlimit)/2 ! Compute middle index
IF(parts%Z(mid).lt.parts%Z(leftlimit)) CALL exchange_parts(leftlimit,mid)
IF(parts%Z(rightlimit).lt.parts%Z(leftlimit)) CALL exchange_parts(leftlimit,rightlimit)
IF(parts%Z(mid).lt.parts%Z(rightlimit)) CALL exchange_parts(rightlimit,mid)
! Store the pivot point for comparison
pivot=parts%Z(rightlimit)
cnt=leftlimit
! Move all parts with Z smaller than pivot to the left of pivot
DO i=leftlimit, rightlimit
IF(parts%Z(i) .le. pivot) THEN
CALL exchange_parts(i,cnt)
cnt=cnt+1
END IF
END DO
! Quicksort the sub-arrays
CALL quicksortparts(leftlimit,cnt-2)
CALL quicksortparts(cnt,rightlimit)
END SUBROUTINE quicksortparts
!---------------------------------------------------------------------------
!> @author
!> Guillaume Le Bars EPFL/SPC
!
! DESCRIPTION:
!>
!> @brief Sorts the particles according to their index in the poisson solver grid.
!
!> @param[in] leftlimit the particles variable needing to be allocated.
!> @param[in] right limit the maximum number of particles that will be stored in this variable
!---------------------------------------------------------------------------
SUBROUTINE gridsort
USE beam
CALL creat_parts
END SUBROUTINE gridsort
END MODULE sort
c4science · Help