Генерировать функцию заданной арности в Haskell, используя номера типов - PullRequest
14 голосов
/ 03 мая 2011

Предположим, что я закодировал натуральные числа в типах Haskell, и у меня есть способ сложения и вычитания из них:

data Zero
data Succ n
-- ...

Я видел различные фрагменты кода, которые создают видимость функций с переменными числамиНапример, this , что позволяет следующее:

buildList "polyvariadic" "function" "wut?" :: [String]
-- ["polyvariadic","function","wut?"]

Что меня интересует, так это то, могу ли я построить из этого функцию, которая будет принимать только количество аргументов,соответствует экземпляру номера типа.То, что я пытаюсь сделать, выглядело бы примерно так:

one = Succ Zero
two = Succ one
three = Succ two

threeStrings :: String -> String -> String -> [String]
threeStrings = buildList three

threeStrings "asdf" "asdf" "asdf"
-- => ["asdf","asdf","asdf"]

threeStrings "asdf"
-- type checker is all HOLY CHRIST TYPE ERROR

threeStrings "asdf" "asdf" "asdf" "asdf"
-- type checker is all SWEET JESUS WHAT YOU ARE DOING

Я знаю, что это довольно глупо и что это, вероятно, пустая трата моего времени, но казалось, что это было бы весело длявыходные.

Ответы [ 4 ]

18 голосов
/ 03 мая 2011

OK.Да.Определенно, добавляя числовой тип вокруг рекурсивных экземпляров.

Во-первых, несколько шаблонов:

{-# LANGUAGE FunctionalDependencies #-}
{-# LANGUAGE MultiParamTypeClasses  #-}
{-# LANGUAGE EmptyDataDecls         #-}
{-# LANGUAGE FlexibleInstances      #-}
{-# LANGUAGE FlexibleContexts       #-}
{-# LANGUAGE ScopedTypeVariables    #-}

Ваш nat:

data Zero
data Succ n

Рекурсивный компоновщик для variadicфункции, теперь с аргументом n :

class BuildList n a r | r -> a where
    build' :: n -> [a] -> a -> r

Базовый случай: остановка при достижении Zero:

instance BuildList Zero a [a] where
    build' _ l x = reverse $ x:l

В противном случае, уменьшение на единицуи повторить:

instance BuildList n a r => BuildList (Succ n) a (a->r) where
    build' (_ :: Succ n) l x y = build' (undefined :: n) (x:l) y

Теперь мы хотим выполнить цикл только 3 раза, поэтому запишите это:

build :: BuildList (Succ (Succ Zero)) a r => a -> r
build x = build' (undefined :: Succ (Succ Zero)) [] x

Готово.

Тестирование:

> build "one" "two" "three" :: [[Char]]
["one","two","three"]

Ошибки меньше или больше:

*Main> build "one" "two" "three" "four" :: [[Char]]

<interactive>:1:1:
    No instance for (BuildList Zero [Char] ([Char] -> [[Char]]))

*Main> build "one" "two" :: [[Char]]

<interactive>:1:1:
    No instance for (BuildList (Succ Zero) [Char] [[Char]])
9 голосов
/ 06 мая 2011

Я вижу ваш многопараметрический параметр-переменная типа empty-datatpe с гибкой областью видимости и поднимаю вам версию на Haskell 98!Он использует HoleyMonoid, который доступен для взлома:

{-# LANGUAGE NoMonomorphismRestriction #-}

import Prelude hiding (id, (.))
import Control.Category
import Data.HoleyMonoid

suc n = later (:[]) . n

zero  = id
one   = suc zero
two   = suc one
three = suc two

buildList = run

Тестирование (не стесняйтесь опускать подписи любого типа):

> run three "one" "two" "three"
["one","two","three"]
4 голосов
/ 06 мая 2011

О, мой ... FlexibleContexts ???NoMonomorphismRestriction ???Да ладно, ребята, разве не для этого нужны TypeFamilies?

{-# LANGUAGE TypeFamilies #-}
data Zero = Zero
newtype Succ n = Succ n
zero = Zero
one = Succ zero
two = Succ one
three = Succ two
class BuildList n where
    type BL n
    buildListPrefix :: n -> ([String] -> [String]) -> BL n
instance BuildList Zero where
    type BL Zero = [String]
    buildListPrefix Zero h = h []
instance BuildList n => BuildList (Succ n) where
    type BL (Succ n) = String -> BL n
    buildListPrefix (Succ n) h s = buildListPrefix n (h . (s:))
buildList:: BuildList n => n -> BL n
buildList n = buildListPrefix n id
4 голосов
/ 06 мая 2011

Встраивание кода Мартина дает очень простое решение:

zero xs = xs
suc n xs x = n (xs ++ [x])
buildList n = n []
...