在“一侧”调用函数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)
}

我怎样才能更简洁（和表达）地表达它？基本上，我试图在依赖于某些谓词的双函子（例如Tuple2）一侧调用函数。如果谓词为 true，我想映射 LHS，如果为 false，则映射 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)

---

Given that I want to use tuples (as opposed to the more general Bifunctor),I seem to be able to use arrows as follows:

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

这种变化是否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 对此进行测试。

Answer 3

编辑：修复为返回 (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)) }