scala - 根据 bool 值在Bifunctor的 "side"上调用函数

如果我有一个Bifunctor[A,A] bf实例，一个函数f : A => A和一个Boolean值p:

def calc[A, F[_,_]: Bifunctor](p: Boolean, bf: F[A, A], f: A => A): F[A, A] = {
  val BF = implicitly[Bifunctor[F]]
  BF.bimap(bf, (a : A) => if (p) f(a) else a, (a : A) => if (!p) f(a) else a)
}

我怎样才能更简洁(表达)些？基本上，我试图根据某些谓词在bifunctor的一侧调用函数(例如Tuple2)。如果谓词为真，则我想映射LHS和RHS(如果为假)

val t2 = (1, 2)
def add4 = (_ : Int) + 4
calc(true, t2, add4) //should be (5,2)
calc(false, t2, add4) //should be (1,6)

考虑到我想使用元组(而不是更普通的Bifunctor)，我似乎可以使用 arrows ，如下所示:

def calc[A](p: Boolean, bf: (A, A), f: A => A): (A, A)
  = (if (p) f.first[A] else f.second[A]) apply bf

最佳答案

没有那么好:

def calc[A, F[_,_]:Bifunctor](p: Boolean, bf: F[A, A], f: A => A): F[A, A] =
   (if (p) (bf :-> (_: A => A)) else ((_:A => A) <-: bf))(f)

更好一点:

def cond[A:Zero](b: Boolean, a: A) = if (b) a else mzero

def calc[A, F[_,_]:Bifunctor](p: Boolean, bf: F[A, A], f: Endo[A]): F[A, A] =
  cond(p, f) <-: bf :-> cond(!p, f)

一些Haskell，只为使语言羡慕不已:

calc p = if p then first else second

关于scala - 根据 bool 值在Bifunctor的 "side"上调用函数，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/4156262/