haskell - 如何在 'special' 文件中定义复合函数的 `.hs` 行为？

Question

组合在 Haskell 中是很正常的，但我只知道我可以定义组合函数的特殊行为，说

Prelude> (floor . sqrt) (10^55)
3162277660168379365112938496
Prelude> let (floor . sqrt) n | n < 2 = n | otherwise = head $ dropWhile (\x -> x^2 > n) $ iterate (\x -> (x + n `div` x) `div` 2) (n `div` 2)
Prelude> (floor . sqrt) (10^55)
3162277660168379331998893544

特殊定义函数的结果是正确的(因为第一个函数存在浮点错误)。

现在我想在 .hs 文件中执行相同的操作，例如

(floor . sqrt) n
    | n < 2 = n
    | otherwise = head $ dropWhile (\x -> x^2 > n)
                       $ iterate (\x -> (x + n `div` x) `div` 2) (n `div` 2)

main = do
    print $ (floor . sqrt) (10^55)

这次ghc对我大喊大叫

Ambiguous occurrence `.'
It could refer to either `Main..', defined at me.hs:1:8
                      or `Prelude..',
                         imported from `Prelude' at me.hs:1:1
                         (and originally defined in `GHC.Base')

那么是否可以在 .hs 文件中定义这样的函数？ (不过，在 main 中用 let 定义它是可以的)。

Answer 1

首先，您在 GHCi 中的示例并未定义 floor 和 sqrt 的特殊组合。相反，它定义了一个运算符 (.)，它接受三个名为 floor、sqrt 和 n 的参数并隐藏名为 (.) 的现有标准函数。

然后，您将新函数应用到标准库函数 floor 和 sqrt，它们将成为新 (.)< 中具有相同名称的参数 函数。

您得到的错误是因为顶级定义不会自动隐藏现有定义，而且无论如何这显然不是您真正想要做的。

现在，您当然可以定义一个全新的函数(希望有自己的名称，如 groovy 的答案)来执行您的专用 floor 。 sqrt 函数，但 Haskell 本身无法将其定义为现有函数的专门版本。

您可能想到并且可能使用 compiler pragma rewrite rules在 GHC 中自动用等效的改进版本替换特定表达式。不过，您应该小心，重写的表格给出的答案与其替换的答案相同。否则，您将面临一些令人困惑的调试 session 的风险。