Составление частичных функций

У меня есть две PartialFunctions f и g. Они не имеют побочных эффектов и быстро выполняются. Как лучше всего скомпоновать их в другую частичную функцию h, такую, что h.isDefinedAt(x) тогда и только тогда, когда f.isDefinedAt(x) && g.isDefinedAt(f(x))?

Также нормально, если h является функцией, возвращающей Option, а не частичной функцией.

Я разочарован тем, что f andThen g не делает то, что я хочу:

scala> val f = Map("a"->1, "b"->2)
f: scala.collection.immutable.Map[String,Int] = Map(a -> 1, b -> 2)

scala> val g = Map(1->'c', 3->'d')
g: scala.collection.immutable.Map[Int,Char] = Map(1 -> c, 3 -> d)

scala> (f andThen g).isDefinedAt("b")
res3: Boolean = true

scala> (f andThen g).lift("b")
java.util.NoSuchElementException: key not found: 2
    at scala.collection.MapLike$class.default(MapLike.scala:228)

scala monads maybe partialfunction
person tba    schedule 12.04.2014    source источник
comment
возможный дубликат связывания PartialFunctions с andThen в Scala   -  person DNA    schedule 12.04.2014
comment
Я видел вопрос выше, но я надеюсь, что есть более элегантный способ, чем неявный класс 10 LOC   -  person tba    schedule 12.04.2014

Ответы (2)


arrow_upward
20
arrow_downward

Вот более короткий способ, чем связанный вопрос, взятый из этой темы. :

  val f = Map("a" -> 1, "b" -> 2)                 

  val g = Map(1 -> 'c', 3 -> 'd')                 

  def andThenPartial[A, B, C](pf1: PartialFunction[A, B], pf2: PartialFunction[B, C]): PartialFunction[A, C] = {
    Function.unlift(pf1.lift(_) flatMap pf2.lift)
  }                                               

  val h = andThenPartial(f, g)            //> h  : PartialFunction[String,Char]

  h.isDefinedAt("a")                      //> res2: Boolean = true
  h.isDefinedAt("b")                      //> res3: Boolean = false
  h.lift("a")                             //> res4: Option[Char] = Some(c)
  h.lift("b")                             //> res5: Option[Char] = None

Конечно, это также можно обернуть как неявный класс:

  implicit class ComposePartial[A, B](pf: PartialFunction[A, B]) {
    def andThenPartial[C](that: PartialFunction[B, C]): PartialFunction[A, C] =
      Function.unlift(pf.lift(_) flatMap that.lift)
  }

  val h2 = f andThenPartial g         //> h2  : PartialFunction[String,Char]

  h2.isDefinedAt("a")                 //> res6: Boolean = true
  h2.isDefinedAt("b")                 //> res7: Boolean = false
  h2.lift("a")                        //> res8: Option[Char] = Some(c)
  h2.lift("b")                        //> res9: Option[Char] = None
person DNA    schedule 12.04.2014

arrow_upward
-5
arrow_downward 

val h = f.mapValues(g)

Но это работает только для карт. Я не думаю, что есть короткий способ сделать это для любой частичной функции, вам просто нужно вручную создать новый объект PartialFunction.

редактировать: я вижу, что мой приведенный выше код не то, что вы хотели. Но, может быть, это лучше

val h = f.collect { case (k, v) if(g.contains(v)) => (k, g(v)) }
person SpiderPig    schedule 12.04.2014
comment
В вопросе f ​​и g — частичные функции, которые необходимо составить. В частности, он недоумевает, почему (f andThen g) определено для b, но при выполнении выдает и ошибку. Взгляните на: twitter.github.io/scala_school/< /а> - person Vinicius Miana; 16.04.2014