[Haskell-cafe] how to make a KnownNat at runtime?
Johannes Waldmann
johannes.waldmann at htwk-leipzig.de
Tue Nov 14 14:01:52 UTC 2017
Dear Cafe,
I am using type-level numbers for
representing dimensions of vectors and matrices
(similar to https://github.com/ekmett/ersatz/blob/master/tests/Z001.hs )
Now when I indeed know all the dimensions at compile time,
(in the example code, line 20) everything works nicely.
But what do I do when the dimensions
will only become available at run time?
Here is a simplified example: print the null vector
where the size is read from the command line -
of course in my application "handle" does more,
but it essentially has the type given here.
I can extend the "case" in the "main" function
to include all the values I want to handle.
Looks ugly. Is there a better way?
- J.
{-# language KindSignatures, RankNTypes, LambdaCase,
TypeApplications, DataKinds, ScopedTypeVariables
#-}
import GHC.TypeLits
import Data.Proxy
import System.Environment
main :: IO ()
main = getArgs >>= \ case
[ "2" ] -> handle (Proxy :: Proxy 2)
[ "3" ] -> handle (Proxy :: Proxy 3)
handle :: forall (n::Nat) . KnownNat n => Proxy n -> IO ()
handle (_ :: Proxy n) = print (zero :: V n Int)
data V (n::Nat) a = V [a] deriving Show
zero :: forall (n::Nat) a . (KnownNat n, Num a) => V n a
zero = V $ replicate (fromIntegral $ natVal (Proxy :: Proxy n)) 0
More information about the Haskell-Cafe
mailing list