{-# LANGUAGE PatternSynonyms #-}

-- |
-- Module      : Jikka.Core.Language.BuiltinPatterns
-- Description : provides pattern synonyms for builtin functions. / 組み込み関数のための pattern synonyms を提供します。
-- Copyright   : (c) Kimiyuki Onaka, 2020
-- License     : Apache License 2.0
-- Maintainer  : kimiyuki95@gmail.com
-- Stability   : experimental
-- Portability : portable
--
-- `Jikka.Core.Language.BuiltinPatterns` provides pattern synonyms for applications of `Builtin` functions.
-- For example, provide a pattern @Sum' e@ which is interpreted as @AppBuiltin Sum [e]@, or the same thing, @App (Lit (LitBuiltin Sum)) [e]@.
module Jikka.Core.Language.BuiltinPatterns where

import Jikka.Core.Language.Expr

-- arithmetical functions
pattern Negate' e = AppBuiltin Negate e

pattern Plus' e1 e2 = AppBuiltin2 Plus e1 e2

pattern Minus' e1 e2 = AppBuiltin2 Minus e1 e2

pattern Mult' e1 e2 = AppBuiltin2 Mult e1 e2

pattern FloorDiv' e1 e2 = AppBuiltin2 FloorDiv e1 e2

pattern FloorMod' e1 e2 = AppBuiltin2 FloorMod e1 e2

pattern CeilDiv' e1 e2 = AppBuiltin2 CeilDiv e1 e2

pattern CeilMod' e1 e2 = AppBuiltin2 CeilMod e1 e2

pattern Pow' e1 e2 = AppBuiltin2 Pow e1 e2

-- advanced arithmetical functions
pattern Abs' e = AppBuiltin Abs e

pattern Gcd' e1 e2 = AppBuiltin2 Gcd e1 e2

pattern Lcm' e1 e2 = AppBuiltin2 Lcm e1 e2

pattern Min2' t e1 e2 = AppBuiltin2 (Min2 t) e1 e2

pattern Max2' t e1 e2 = AppBuiltin2 (Max2 t) e1 e2

pattern Iterate' t n step base = AppBuiltin3 (Iterate t) n step base

-- logical functions
pattern Not' e = AppBuiltin Not e

pattern And' e1 e2 = AppBuiltin2 And e1 e2

pattern Or' e1 e2 = AppBuiltin2 Or e1 e2

pattern Implies' e1 e2 = AppBuiltin2 Implies e1 e2

pattern If' t e1 e2 e3 = AppBuiltin3 (If t) e1 e2 e3

-- bitwise functions
pattern BitNot' e = AppBuiltin BitNot e

pattern BitAnd' e1 e2 = AppBuiltin2 BitAnd e1 e2

pattern BitOr' e1 e2 = AppBuiltin2 BitOr e1 e2

pattern BitXor' e1 e2 = AppBuiltin2 BitXor e1 e2

pattern BitLeftShift' e1 e2 = AppBuiltin2 BitLeftShift e1 e2

pattern BitRightShift' e1 e2 = AppBuiltin2 BitRightShift e1 e2

-- matrix functions

pattern MatAp' h w e1 e2 = AppBuiltin2 (MatAp h w) e1 e2

pattern MatAdd' h w e1 e2 = AppBuiltin2 (MatAdd h w) e1 e2

pattern MatMul' h n w e1 e2 = AppBuiltin2 (MatMul h n w) e1 e2

pattern MatPow' n e1 e2 = AppBuiltin2 (MatPow n) e1 e2

pattern VecFloorMod' n e1 e2 = AppBuiltin2 (VecFloorMod n) e1 e2

pattern MatFloorMod' h w e1 e2 = AppBuiltin2 (MatFloorMod h w) e1 e2

-- modular functions
pattern ModNegate' e1 e2 = AppBuiltin2 ModNegate e1 e2

pattern ModPlus' e1 e2 e3 = AppBuiltin3 ModPlus e1 e2 e3

pattern ModMinus' e1 e2 e3 = AppBuiltin3 ModMinus e1 e2 e3

pattern ModMult' e1 e2 e3 = AppBuiltin3 ModMult e1 e2 e3

pattern ModInv' e1 e2 = AppBuiltin2 ModInv e1 e2

pattern ModPow' e1 e2 e3 = AppBuiltin3 ModPow e1 e2 e3

pattern ModMatAp' h w e1 e2 e3 = AppBuiltin3 (ModMatAp h w) e1 e2 e3

pattern ModMatAdd' h w e1 e2 e3 = AppBuiltin3 (ModMatAdd h w) e1 e2 e3

pattern ModMatMul' h n w e1 e2 e3 = AppBuiltin3 (ModMatMul h n w) e1 e2 e3

pattern ModMatPow' n e1 e2 e3 = AppBuiltin3 (ModMatPow n) e1 e2 e3

-- list functions
pattern Nil' t = Lit (LitNil t)

pattern Cons' t e1 e2 = AppBuiltin2 (Cons t) e1 e2

pattern Snoc' t e1 e2 = AppBuiltin2 (Snoc t) e1 e2

pattern Foldl' t1 t2 e1 e2 e3 = AppBuiltin3 (Foldl t1 t2) e1 e2 e3

pattern Scanl' t1 t2 e1 e2 e3 = AppBuiltin3 (Scanl t1 t2) e1 e2 e3

pattern Build' t e1 e2 e3 = AppBuiltin3 (Build t) e1 e2 e3

pattern Len' t e = AppBuiltin (Len t) e

pattern Map' t1 t2 f e = AppBuiltin2 (Map t1 t2) f e

pattern Filter' t f e = AppBuiltin2 (Filter t) f e

pattern At' t e1 e2 = AppBuiltin2 (At t) e1 e2

pattern SetAt' t e1 e2 e3 = AppBuiltin3 (SetAt t) e1 e2 e3

pattern Elem' t e1 e2 = AppBuiltin2 (Elem t) e1 e2

pattern Sum' e = AppBuiltin Sum e

pattern Product' e = AppBuiltin Product e

pattern ModSum' e1 e2 = AppBuiltin2 ModSum e1 e2

pattern ModProduct' e1 e2 = AppBuiltin2 ModProduct e1 e2

pattern Min1' t e = AppBuiltin (Min1 t) e

pattern Max1' t e = AppBuiltin (Max1 t) e

pattern ArgMin' t e = AppBuiltin (ArgMin t) e

pattern ArgMax' t e = AppBuiltin (ArgMax t) e

pattern All' e = AppBuiltin All e

pattern Any' e = AppBuiltin Any e

pattern Sorted' t e = AppBuiltin (Sorted t) e

pattern Reversed' t e = AppBuiltin (Reversed t) e

pattern Range1' e = AppBuiltin Range1 e

pattern Range2' e1 e2 = AppBuiltin2 Range2 e1 e2

pattern Range3' e1 e2 e3 = AppBuiltin3 Range3 e1 e2 e3

-- tuple functions
pattern Tuple' ts = Lit (LitBuiltin (Tuple ts))

pattern Proj' ts n e = AppBuiltin (Proj ts n) e

-- arithmetical relations
pattern LessThan' t e1 e2 = AppBuiltin2 (LessThan t) e1 e2

pattern LessEqual' t e1 e2 = AppBuiltin2 (LessEqual t) e1 e2

pattern GreaterThan' t e1 e2 = AppBuiltin2 (GreaterThan t) e1 e2

pattern GreaterEqual' t e1 e2 = AppBuiltin2 (GreaterEqual t) e1 e2

-- equality relations (polymorphic)
pattern Equal' t e1 e2 = AppBuiltin2 (Equal t) e1 e2

pattern NotEqual' t e1 e2 = AppBuiltin2 (NotEqual t) e1 e2

-- combinational functions
pattern Fact' e = AppBuiltin Fact e

pattern Choose' e1 e2 = AppBuiltin2 Choose e1 e2

pattern Permute' e1 e2 = AppBuiltin2 Permute e1 e2

pattern MultiChoose' e1 e2 = AppBuiltin2 MultiChoose e1 e2

-- data structures
pattern ConvexHullTrickInit' = Lit (LitBuiltin ConvexHullTrickInit)

pattern ConvexHullTrickGetMin' cht a = AppBuiltin2 ConvexHullTrickGetMin cht a

pattern ConvexHullTrickInsert' cht a b = AppBuiltin3 ConvexHullTrickInsert cht a b

pattern SegmentTreeInitList' semigrp a = AppBuiltin (SegmentTreeInitList semigrp) a

pattern SegmentTreeGetRange' semigrp segtree e1 e2 = AppBuiltin3 (SegmentTreeGetRange semigrp) segtree e1 e2

pattern SegmentTreeSetPoint' semigrp segtree e1 e2 = AppBuiltin3 (SegmentTreeSetPoint semigrp) segtree e1 e2

-- errors
pattern Bottom' t err = Lit (LitBottom t err)