add size benchmarking for scripts to agora-bench

agora-bench/Bench.hs

@@ -0,0 +1,33 @@
+module Bench (Benchmark (..), benchmarkSize) where
+
+import Codec.Serialise (serialise)
+import Data.ByteString.Lazy qualified as LBS
+import Data.ByteString.Short qualified as SBS
+import Data.Set (Set)
+import Data.Set qualified as Set
+import Data.Text (Text)
+import Plutus.V1.Ledger.Scripts qualified as Plutus
+
+--------------------------------------------------------------------------------
+
+-- | Represents the benchmark of a plutus script.
+data Benchmark = Benchmark
+  { name :: Text
+  -- ^ Human readable name describing script.
+  , size :: Int
+  -- ^ The on-chain size of a script.
+  }
+  deriving stock (Show, Eq, Ord)
+
+-- | Create a benchmark containing only the size of the script.
+benchmarkSize :: Text -> Plutus.Script -> Set Benchmark
+benchmarkSize name script =
+  Set.singleton $
+    Benchmark
+      { name = name
+      , size = scriptSize script
+      }
+
+-- | Compute the size of a script on-chain.
+scriptSize :: Plutus.Script -> Int
+scriptSize = SBS.length . SBS.toShort . LBS.toStrict . serialise

agora-bench/Main.hs

@@ -1,14 +1,42 @@
 module Main (main) where
 
+import Agora.AuthorityToken (authorityTokenPolicy)
+import Agora.Effect.TreasuryWithdrawal (treasuryWithdrawalValidator)
+import Agora.Governor (Governor (..))
+import Agora.Governor.Scripts (governorPolicy, governorValidator)
+import Agora.Proposal.Scripts (proposalPolicy, proposalValidator)
+import Agora.Stake.Scripts (stakePolicy, stakeValidator)
+import Agora.Treasury (treasuryValidator)
+import Bench
+import Data.Foldable (for_)
+import Plutus.V1.Ledger.Api (CurrencySymbol)
+import Sample.Shared
 import Prelude
 
 --------------------------------------------------------------------------------
 
---------------------------------------------------------------------------------
-
---------------------------------------------------------------------------------
-
---------------------------------------------------------------------------------
-
 main :: IO ()
-main = pure ()
+main = do
+  let benchmarks =
+        mconcat
+          [ -- GATs
+            benchmarkSize "authorityTokenPolicy" $ compile $ authorityTokenPolicy authorityToken
+          , -- Governor
+            benchmarkSize "governorValidator" $ compile $ governorValidator governor
+          , benchmarkSize "governorPolicy" $ compile $ governorPolicy governor
+          , -- Stake
+            benchmarkSize "stakeValidator" $ compile $ stakeValidator stake
+          , benchmarkSize "stakePolicy" $ compile $ stakePolicy governor.gtClassRef
+          , -- Proposal
+            benchmarkSize "proposalValidator" $ compile $ proposalValidator proposal
+          , benchmarkSize "proposalPolicy" $ compile $ proposalPolicy govAssetClass
+          , -- Treasury
+            benchmarkSize "treasuryValidator" $ compile $ treasuryValidator gatCS
+          , -- Effect validators
+            benchmarkSize "treasuryWithdrawalValidator" $ compile $ treasuryWithdrawalValidator gatCS
+          ]
+
+  for_ benchmarks print
+
+gatCS :: CurrencySymbol
+gatCS = "73475cb40a568e8da8a045ced110137e159f890ac4da883b6b17dc651b3a8049" -- arbitrary CS