2012-11-05 15 views
9

C'è una sintassi in Scala che consente agli estrattori di prendere un argomento di personalizzazione? Questo esempio è un po 'forzato. Supponiamo che abbia un albero di ricerca binario su numeri interi e che io debba corrispondere al nodo corrente se il suo valore è divisibile per qualche valore personalizzato.Scala Extractor with Argument

Utilizzando F # modelli attivi, posso fare il seguente:

type Tree = 
    | Node of int * Tree * Tree 
    | Empty 

let (|NodeDivisibleBy|_|) x t = 
    match t with 
    | Empty -> None 
    | Node(y, l, r) -> if y % x = 0 then Some((l, r)) else None 

let doit = function 
    | NodeDivisibleBy(2)(l, r) -> printfn "Matched two: %A %A" l r 
    | NodeDivisibleBy(3)(l, r) -> printfn "Matched three: %A %A" l r 
    | _ -> printfn "Nada" 

[<EntryPoint>] 
let main args = 
    let t10 = Node(10, Node(1, Empty, Empty), Empty) 
    let t15 = Node(15, Node(1, Empty, Empty), Empty) 

    doit t10 
    doit t15 

    0 

In Scala, posso fare qualcosa di simile, ma non proprio quello che voglio:

sealed trait Tree 
case object Empty extends Tree 
case class Node(v: Int, l: Tree, r: Tree) extends Tree 

object NodeDivisibleBy { 
    def apply(x: Int) = new { 
    def unapply(t: Tree) = t match { 
     case Empty => None 
     case Node(y, l, r) => if (y % x == 0) Some((l, r)) else None 
    } 
    } 
} 

def doit(t: Tree) { 
    // I would prefer to not need these two lines. 
    val NodeDivisibleBy2 = NodeDivisibleBy(2) 
    val NodeDivisibleBy3 = NodeDivisibleBy(3) 
    t match { 
    case NodeDivisibleBy2(l, r) => println("Matched two: " + l + " " + r) 
    case NodeDivisibleBy3(l, r) => println("Matched three: " + l + " " + r) 
    case _ => println("Nada") 
    } 
} 

val t10 = Node(10, Node(1, Empty, Empty), Empty) 
val t15 = Node(15, Node(1, Empty, Empty), Empty) 

doit(t10) 
doit(t15) 

sarebbe bello se ho potuto fare:

case NodeDivisibleBy(2)(l, r) => println("Matched two: " + l + " " + r) 
case NodeDivisibleBy(3)(l, r) => println("Matched three: " + l + " " + r) 

ma questo è un errore di compilazione:. '=>' previsto pero '(' trovato

Pensieri?

risposta

0

Come so la risposta è no.

Io uso anche il modo precedente per questo caso.

6

Da the spec:

SimplePattern ::= StableId ‘(’ [Patterns] ‘)’ 

An extractor pattern x(p1, ..., pn) where n ≥ 0 is of the same syntactic form as a constructor pattern. However, instead of a case class, the stable identifier x denotes an object which has a member method named unapply or unapplySeq that matches the pattern.

e:

A stable identifier is a path which ends in an identifier.

cioè, non un'espressione come NodeDivisibleBy(2).

Quindi no, questo non è possibile in modo semplice in Scala, e personalmente penso che vada bene: dover scrivere quanto segue (che tra l'altro probabilmente definirei nell'oggetto NodeDivisibleBy e importare):

val NodeDivisibleBy2 = NodeDivisibleBy(2) 
val NodeDivisibleBy3 = NodeDivisibleBy(3) 

è un piccolo prezzo da pagare per l'aumentata leggibilità di non dover decifrare espressioni arbitrarie nella clausola del caso.

4

Come ha notato Travis Brown, non è davvero possibile in scala.

Quello che faccio in questo scenario è solo separare la scomposizione dal test con una guardia e un alias.

val DivisibleBy = (n: Node, x: Int) => (n.v % x == 0) 

def doit(t: Tree) = t match { 
    case n @ Node(y, l, r) if DivisibleBy(n,2) => println("Matched two: " + l + " " + r) 
    case n @ Node(y, l, r) if DivisibleBy(n,3) => println("Matched three: " + l + " " + r) 
    case _ => println("Nada") 
} 

Definizione di una separata DivisibleBy è ovviamente completa eccessivo in questo caso semplice, ma può aiutare la leggibilità in scenari più complessi in maniera simile a come F # modelli attivi fanno.

Si potrebbe anche definire divisibilePer come metodo di nodo e avere:

case class Node(v: Int, l: Tree, r: Tree) extends Tree { 
    def divisibleBy(o:Int) = (v % o)==0 
} 

def doit(t: Tree) = t match { 
    case n @ Node(y, l, r) if n divisibleBy 2 => println("Matched two: " + l + " " + r) 
    case n @ Node(y, l, r) if n divisibleBy 3 => println("Matched three: " + l + " " + r) 
    case _ => println("Nada") 
} 

che credo sia più leggibile (se più verboso) rispetto alla F # versione

2

abbinare la classe caso, predicato e argomenti, quindi abbinare sul risultato come al solito.

case class Foo(i: Int) 

class Testable(val f: Foo, val ds: List[Int]) 

object Testable { 
    def apply(f: Foo, ds: List[Int]) = new Testable(f, ds) 
    def unapply(t: Testable): Option[(Foo, List[Int])] = { 
    val xs = t.ds filter (t.f.i % _ == 0) 
    if (xs.nonEmpty) Some((t.f, xs)) else None 
    } 
} 

object Test extends App { 
    val f = Foo(100) 

    Testable(f, List(3,5,20)) match { 
    case Testable(f, 3 :: Nil) => println(s"$f matched three") 
    case Testable(Foo(i), 5 :: Nil) if i < 50 
           => println(s"$f matched five") 
    case Testable(f, ds)  => println(s"$f matched ${ds mkString ","}") 
    case _      => println("Nothing") 
    } 
} 
+0

Questo è Scala 2.10 giusto? –

Problemi correlati