能否让接受Fixed.Vector的函数根据输入长度返回Boxed.Vec2或Vec3？

Haskell里没有和C++ if constexpr完全等价的语法，但可以通过类型类或GADT实现“根据类型自动选择函数实现”的逻辑，完全不需要Template Haskell，这也是更符合Haskell风格的做法。

方案一：类型类（自动推导分支） #

通过定义关联类型的类型类，让类型系统根据输入参数的结构自动匹配对应长度的worksFor函数。

首先启用必要的GHC扩展：

{-# LANGUAGE TypeFamilies, MultiParamTypeClasses, FlexibleInstances #-}
import qualified Data.Vector.Fixed as B

定义类型类，关联向量长度、输入参数类型和实现：

class VecArity n where
  -- 关联类型：指定对应长度向量所需的输入参数类型
  type Inputs n a :: *
  -- 统一调用入口，自动适配不同长度
  worksForN :: Inputs n a -> [B.Vec n a]

-- 2维向量的实例：对应worksFor2
instance VecArity 2 where
  type Inputs 2 a = ([a], [a])
  worksForN (xs, ys) = worksFor2 xs ys  -- 复用已实现的worksFor2

-- 3维向量的实例：对应worksFor3
instance VecArity 3 where
  type Inputs 3 a = ([a], [a], [a])
  worksForN (xs, ys, zs) = worksFor3 xs ys zs  -- 复用已实现的worksFor3

定义统一的bar函数：

-- bar处理逻辑，自动适配2/3维向量
bar :: (VecArity n, Show a) => Inputs n a -> IO ()
bar inputs = mapM_ print (worksForN inputs)

调用示例：

-- 自动匹配2维实现
main = bar ([1,2,3], [4,5,6])

-- 自动匹配3维实现
main = bar ([1,2], [3,4], [5,6])

类型系统会根据输入元组的长度自动推导向量长度n，选择对应的worksForN实现，解决类型推导错误的问题。

方案二：GADT（值层面显式分支） #

如果更倾向于类似if constexpr的“值层面分支”风格，可以用GADT携带类型信息，通过模式匹配选择实现：

启用GADT扩展：

{-# LANGUAGE GADTs #-}
import qualified Data.Vector.Fixed as B

定义GADT标记向量长度：

data VecLen n where
  Len2 :: VecLen 2
  Len3 :: VecLen 3

结合之前的VecArity类型类，实现统一的分支逻辑：

works :: VecArity n => VecLen n -> Inputs n a -> [B.Vec n a]
works Len2 = worksForN
works Len3 = worksForN

-- 带显式长度参数的bar函数
bar :: (VecArity n, Show a) => VecLen n -> Inputs n a -> IO ()
bar len inputs = mapM_ print (works len inputs)

调用示例：

main = bar Len2 ([1,2], [3,4])
main = bar Len3 ([1,2], [3,4], [5,6])

GADT确保了模式匹配的分支和类型严格对应，类型系统会检查输入参数是否与指定长度匹配，避免类型错误。

关于Template Haskell #

如果只是处理2、3两种长度，完全不需要Template Haskell。只有当需要支持大量不同长度（比如2到100）、避免重复手写实例时，TH才会带来便利——但对于少量场景，手写类型类实例更清晰、易维护。

内容的提问来源于stack exchange，提问作者Enlico