-- |
-- Module:     Data.Vector.Algorithms.Quicksort.Predefined.PIntParallelMedian3or5IO
-- Copyright:  (c) Sergey Vinokurov 2023
-- License:    Apache-2.0 (see LICENSE)
-- Maintainer: serg.foo@gmail.com

module Data.Vector.Algorithms.Quicksort.Predefined.PIntParallelMedian3or5IO
  ( sortPIntParallelMedian3or5IO
  ) where

import Control.Concurrent (getNumCapabilities)
import Control.Monad.ST
import Data.Int
import Data.Vector.Primitive qualified as P

import Data.Vector.Algorithms.Quicksort.Parameterised

import Data.Vector.Algorithms.Quicksort.Predefined.PIntSequentialMedian3IO ()

{-# NOINLINE sortPIntParallelMedian3or5IO #-}
sortPIntParallelMedian3or5IO :: P.MVector RealWorld Int64 -> IO ()
sortPIntParallelMedian3or5IO :: MVector RealWorld Int64 -> IO ()
sortPIntParallelMedian3or5IO MVector RealWorld Int64
xs = do
  Parallel
p <- Int -> IO Parallel
mkParallel forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< IO Int
getNumCapabilities
  forall p med x (m :: * -> *) a (v :: * -> * -> *).
(Fork2 p x m, Median med a m (PrimState m), PrimMonad m, Ord a,
 MVector v a) =>
p -> med -> v (PrimState m) a -> m ()
sortInplaceFM Parallel
p (forall {k} (a :: k). Median3or5 a
Median3or5 @Int64) MVector RealWorld Int64
xs
  Parallel -> IO ()
waitParallel Parallel
p