ghc-9.13: The GHC API
Safe HaskellNone
LanguageGHC2021

GHC.Utils.Misc

Description

Highly random utility functions

Synopsis

Miscellaneous higher-order functions

applyWhen :: Bool -> (a -> a) -> a -> a Source #

Apply a function iff some condition is met.

nTimes :: Int -> (a -> a) -> a -> a Source #

Apply a function n times to a given value.

const2 :: a -> b -> c -> a Source #

General list processing

zipEqual :: HasDebugCallStack => String -> [a] -> [b] -> [(a, b)] Source #

zipWithEqual :: HasDebugCallStack => String -> (a -> b -> c) -> [a] -> [b] -> [c] Source #

zipWith3Equal :: HasDebugCallStack => String -> (a -> b -> c -> d) -> [a] -> [b] -> [c] -> [d] Source #

zipWith4Equal :: HasDebugCallStack => String -> (a -> b -> c -> d -> e) -> [a] -> [b] -> [c] -> [d] -> [e] Source #

stretchZipWith :: (a -> Bool) -> b -> (a -> b -> c) -> [a] -> [b] -> [c] Source #

stretchZipWith p z f xs ys stretches ys by inserting z in the places where p returns True

zipWithAndUnzip :: (a -> b -> (c, d)) -> [a] -> [b] -> ([c], [d]) Source #

zipAndUnzip :: [a] -> [b] -> ([a], [b]) Source #

This has the effect of making the two lists have equal length by dropping the tail of the longer one.

filterByList :: [Bool] -> [a] -> [a] Source #

filterByList takes a list of Bools and a list of some elements and filters out these elements for which the corresponding value in the list of Bools is False. This function does not check whether the lists have equal length.

filterByLists :: [Bool] -> [a] -> [a] -> [a] Source #

filterByLists takes a list of Bools and two lists as input, and outputs a new list consisting of elements from the last two input lists. For each Bool in the list, if it is True, then it takes an element from the former list. If it is False, it takes an element from the latter list. The elements taken correspond to the index of the Bool in its list. For example:

filterByLists [True, False, True, False] "abcd" "wxyz" = "axcz"

This function does not check whether the lists have equal length.

partitionByList :: [Bool] -> [a] -> ([a], [a]) Source #

partitionByList takes a list of Bools and a list of some elements and partitions the list according to the list of Bools. Elements corresponding to True go to the left; elements corresponding to False go to the right. For example, partitionByList [True, False, True] [1,2,3] == ([1,3], [2]) This function does not check whether the lists have equal length; when one list runs out, the function stops.

unzipWith :: (a -> b -> c) -> [(a, b)] -> [c] Source #

mapFst :: Functor f => (a -> c) -> f (a, b) -> f (c, b) Source #

mapSnd :: Functor f => (b -> c) -> f (a, b) -> f (a, c) Source #

chkAppend :: [a] -> [a] -> [a] Source #

mapAndUnzip :: (a -> (b, c)) -> [a] -> ([b], [c]) Source #

mapAndUnzip3 :: (a -> (b, c, d)) -> [a] -> ([b], [c], [d]) Source #

mapAndUnzip4 :: (a -> (b, c, d, e)) -> [a] -> ([b], [c], [d], [e]) Source #

filterOut :: (a -> Bool) -> [a] -> [a] Source #

Like filter, only it reverses the sense of the test

partitionWith :: (a -> Either b c) -> [a] -> ([b], [c]) Source #

Uses a function to determine which of two output lists an input element should join

partitionWithM :: Monad m => (a -> m (Either b c)) -> [a] -> m ([b], [c]) Source #

Monadic version of partitionWith

dropWhileEndLE :: (a -> Bool) -> [a] -> [a] Source #

spanEnd :: (a -> Bool) -> [a] -> ([a], [a]) Source #

spanEnd p l == reverse (span p (reverse l)). The first list returns actually comes after the second list (when you look at the input list).

last2 :: [a] -> Maybe (a, a) Source #

Get the last two elements in a list.

lastMaybe :: [a] -> Maybe a Source #

onJust :: b -> Maybe a -> (a -> b) -> b Source #

onJust x m f applies f to the value inside the Just or returns the default.

foldl1' :: HasCallStack => (a -> a -> a) -> [a] -> a #

foldl2 :: (acc -> a -> b -> acc) -> acc -> [a] -> [b] -> acc Source #

count :: (a -> Bool) -> [a] -> Int Source #

countWhile :: (a -> Bool) -> [a] -> Int Source #

all2 :: (a -> b -> Bool) -> [a] -> [b] -> Bool Source #

any2 :: (a -> b -> Bool) -> [a] -> [b] -> Bool Source #

all2Prefix :: (a -> b -> Bool) -> [a] -> [b] -> Bool Source #

`all2Prefix p xs ys` is a fused version of `and $ zipWith2 p xs ys`. (It generates good code nonetheless.) So if one list is shorter than the other, p is assumed to be True for the suffix.

all3Prefix :: (a -> b -> c -> Bool) -> [a] -> [b] -> [c] -> Bool Source #

`all3Prefix p xs ys zs` is a fused version of `and $ zipWith3 p xs ys zs`. (It generates good code nonetheless.) So if one list is shorter than the others, p is assumed to be True for the suffix.

lengthExceeds :: [a] -> Int -> Bool Source #

(lengthExceeds xs n) = (length xs > n)

lengthIs :: [a] -> Int -> Bool Source #

(lengthIs xs n) = (length xs == n)

lengthIsNot :: [a] -> Int -> Bool Source #

(lengthIsNot xs n) = (length xs /= n)

lengthAtLeast :: [a] -> Int -> Bool Source #

(lengthAtLeast xs n) = (length xs >= n)

lengthAtMost :: [a] -> Int -> Bool Source #

(lengthAtMost xs n) = (length xs <= n)

lengthLessThan :: [a] -> Int -> Bool Source #

(lengthLessThan xs n) == (length xs < n)

atLength :: ([a] -> b) -> b -> [a] -> Int -> b Source #

atLength atLen atEnd ls n unravels list ls to position n. Precisely:

 atLength atLenPred atEndPred ls n
  | n < 0         = atLenPred ls
  | length ls < n = atEndPred (n - length ls)
  | otherwise     = atLenPred (drop n ls)

equalLength :: [a] -> [b] -> Bool Source #

True if length xs == length ys

compareLength :: [a] -> [b] -> Ordering Source #

leLength :: [a] -> [b] -> Bool Source #

True if length xs <= length ys

ltLength :: [a] -> [b] -> Bool Source #

True if length xs < length ys

only :: [a] -> a Source #

Utility function to go from a singleton list to it's element.

Wether or not the argument is a singleton list is only checked in debug builds.

expectOnly :: HasDebugCallStack => String -> [a] -> a Source #

Extract the single element of a list and panic with the given message if there are more elements or the list was empty. Like expectJust, but for lists.

singleton :: a -> [a] Source #

notNull :: Foldable f => f a -> Bool Source #

expectNonEmpty :: HasDebugCallStack => String -> [a] -> NonEmpty a Source #

Like expectJust msg . nonEmpty; a better alternative to fromList.

snocView :: [a] -> Maybe ([a], a) Source #

Split a list into its last element and the initial part of the list. snocView xs = Just (init xs, last xs) for non-empty lists. snocView xs = Nothing otherwise. Unless both parts of the result are guaranteed to be used prefer separate calls to last + init. If you are guaranteed to use both, this will be more efficient.

holes :: [a] -> [(a, [a])] Source #

Compute all the ways of removing a single element from a list.

holes [1,2,3] = [(1, [2,3]), (2, [1,3]), (3, [1,2])]

changeLast :: [a] -> a -> [a] Source #

Replace the last element of a list with another element.

whenNonEmpty :: Applicative m => [a] -> (NonEmpty a -> m ()) -> m () Source #

mergeListsBy :: (a -> a -> Ordering) -> [[a]] -> [a] Source #

Merge an unsorted list of sorted lists, for example:

mergeListsBy compare [ [2,5,15], [1,10,100] ] = [1,2,5,10,15,100]

\( O(n \log{} k) \)

isSortedBy :: (a -> a -> Ordering) -> [a] -> Bool Source #

mapMaybe' :: Foldable f => (a -> Maybe b) -> f a -> [b] Source #

Tuples

fstOf3 :: (a, b, c) -> a Source #

sndOf3 :: (a, b, c) -> b Source #

thdOf3 :: (a, b, c) -> c Source #

fstOf4 :: (a, b, c, d) -> a Source #

sndOf4 :: (a, b, c, d) -> b Source #

fst3 :: (a -> d) -> (a, b, c) -> (d, b, c) Source #

snd3 :: (b -> d) -> (a, b, c) -> (a, d, c) Source #

third3 :: (c -> d) -> (a, b, c) -> (a, b, d) Source #

uncurry3 :: (a -> b -> c -> d) -> (a, b, c) -> d Source #

List operations controlled by another list

takeList :: [b] -> [a] -> [a] Source #

dropList :: [b] -> [a] -> [a] Source #

splitAtList :: [b] -> [a] -> ([a], [a]) Source #

Given two lists xs and ys, return `splitAt (length xs) ys`.

dropTail :: Int -> [a] -> [a] Source #

drop from the end of a list

capitalise :: String -> String Source #

Convert a word to title case by capitalising the first letter

Sorting

sortWith :: Ord b => (a -> b) -> [a] -> [a] #

minWith :: Ord b => (a -> b) -> [a] -> a Source #

nubSort :: Ord a => [a] -> [a] Source #

ordNub :: Ord a => [a] -> [a] Source #

Remove duplicates but keep elements in order. O(n * log n)

ordNubOn :: Ord b => (a -> b) -> [a] -> [a] Source #

Remove duplicates but keep elements in order. O(n * log n)

Comparisons

(<&&>) :: Applicative f => f Bool -> f Bool -> f Bool infixr 3 Source #

(<||>) :: Applicative f => f Bool -> f Bool -> f Bool infixr 2 Source #

Edit distance

fuzzyLookup :: String -> [(String, a)] -> [a] Source #

Search for possible matches to the users input in the given list, returning a small number of ranked results

Transitive closures

transitiveClosure :: (a -> [a]) -> (a -> a -> Bool) -> [a] -> [a] Source #

Strictness

seqList :: [a] -> b -> b Source #

strictMap :: (a -> b) -> [a] -> [b] Source #

strictZipWith :: (a -> b -> c) -> [a] -> [b] -> [c] Source #

strictZipWith3 :: (a -> b -> c -> d) -> [a] -> [b] -> [c] -> [d] Source #

Module names

Integers

exactLog2 :: Integer -> Maybe Integer Source #

Determine the $log_2$ of exact powers of 2

Floating point

readSignificandExponentPair :: String -> (Integer, Integer) Source #

Parse a string into a significand and exponent. A trivial example might be: ghci> readSignificandExponentPair "1E2" (1,2) In a more complex case we might return a exponent different than that which the user wrote. This is needed in order to use a Integer significand. ghci> readSignificandExponentPair "-1.11E5" (-111,3)

readHexSignificandExponentPair :: String -> (Integer, Integer) Source #

Parse a string into a significand and exponent according to the "Hexadecimal Floats in Haskell" proposal. A trivial example might be: ghci> readHexSignificandExponentPair "0x1p+1" (1,1) Behaves similar to readSignificandExponentPair but the base is 16 and numbers are given in hexadecimal: ghci> readHexSignificandExponentPair "0xAp-4" (10,-4) ghci> readHexSignificandExponentPair "0x1.2p3" (18,-1)

IO-ish utilities

withAtomicRename :: MonadIO m => FilePath -> (FilePath -> m a) -> m a Source #

Filenames and paths

data Direction Source #

Constructors

Forwards 
Backwards 

Utils for defining Data instances

Utils for printing C code

Hashing

hashString :: String -> Int32 Source #

A sample hash function for Strings. We keep multiplying by the golden ratio and adding. The implementation is:

hashString = foldl' f golden
  where f m c = fromIntegral (ord c) * magic + hashInt32 m
        magic = 0xdeadbeef

Where hashInt32 works just as hashInt shown above.

Knuth argues that repeated multiplication by the golden ratio will minimize gaps in the hash space, and thus it's a good choice for combining together multiple keys to form one.

Here we know that individual characters c are often small, and this produces frequent collisions if we use ord c alone. A particular problem are the shorter low ASCII and ISO-8859-1 character strings. We pre-multiply by a magic twiddle factor to obtain a good distribution. In fact, given the following test:

testp :: Int32 -> Int
testp k = (n - ) . length . group . sort . map hs . take n $ ls
  where ls = [] : [c : l | l <- ls, c <- ['\0'..'\xff']]
        hs = foldl' f golden
        f m c = fromIntegral (ord c) * k + hashInt32 m
        n = 100000

We discover that testp magic = 0.

Call stacks

type HasCallStack = ?callStack :: CallStack #

type HasDebugCallStack = () Source #

A call stack constraint, but only when isDebugOn.