Add missing functions to Plutarch.Core.List

input-output-hk · Dec 16, 2024 · a1aa459 · a1aa459
1 parent 26cf63f
commit a1aa459
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/plutarch-onchain-lib/lib/Plutarch/Core/List.hs b/src/plutarch-onchain-lib/lib/Plutarch/Core/List.hs
@@ -4,14 +4,16 @@ module Plutarch.Core.List (
   pbuiltinListLengthFast,
   penforceNSpendRedeemers,
   phasNSetBits,
+  pisUniqueSet,
+  phasNUniqueElements,
   emptyByteArray
 ) where
 
 import           Plutarch                    (ClosedTerm, Config (NoTracing),
                                               PType, S, Term, perror, pfix,
                                               phoistAcyclic, plam, plet, pto,
                                               type (:-->), (#$), (#))
-import           Plutarch.Bitwise            (pcountSetBits)
+import           Plutarch.Bitwise            (pcountSetBits, pwriteBits)
 import           Plutarch.Bool               (PBool, PEq ((#==)),
                                               PPartialOrd ((#<), (#<=)), pif,
                                               pnot)
@@ -153,3 +155,13 @@ penforceNSpendRedeemers n rdmrs =
                 (pelimList (\x _ -> isNonSpend (pfstBuiltin # x)) (pconstant True) (ptail # redeemers))
                 perror
      in isLastSpend # (pdropFast # (n - 1) # pto rdmrs)
+
+pisUniqueSet :: Term s (PInteger :--> PBuiltinList PInteger :--> PBool)
+pisUniqueSet = phoistAcyclic $ plam $ \n xs ->
+  let flagUniqueBits = pwriteBits # emptyByteArray # xs # pconstant [True]
+  in (pcountSetBits # flagUniqueBits #== (pbuiltinListLengthFast # n # xs))
+
+phasNUniqueElements :: Term s (PInteger :--> PBuiltinList PInteger :--> PBool)
+phasNUniqueElements = phoistAcyclic $ plam $ \n xs ->
+  let flagUniqueBits = pwriteBits # emptyByteArray # xs # pconstant [True]
+  in (pcountSetBits # flagUniqueBits #== n)