scala - 根据布尔值在 Bifunctor 的“一侧”调用函数

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)
}

我怎样才能更简洁（和表达）表达？基本上，我试图在依赖于某个谓词的双函子（例如 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

Answer 2

编辑：固定返回(A,A)而不是A

也许我遗漏了一些东西，但这不是临时变量的用途吗？使用常规的 Scala 元组：

Some(bf).map(x => if (p) x.copy(_1 = f(x._1)) else x.copy(_2 = f(x._2))).get

或者

{ val (l,r) = bf; if (p) (f(l),r) else (l,f(r)) }

Answer 3

Apocalisp 解决方案的这种变化是否有效？

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

注意：我没有用 scalaz 对此进行测试。