在Bifunctor的"side"上调用一个函数,该函数取决于boolean的值

Question

如果我有一个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(例如a 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),我似乎能够使用如下箭头:

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

Answer 1

并不是那么好：

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