search

yarhrn / new-scaladoc

3 stars 1 forks source link

use collect for implementation of processPackage instead of manual re #11

Closed cvogt closed 9 years ago

cvogt commented 9 years ago 
  def processPackage(pack: Package): String = {
    val grouped: Map[String, Seq[DocElement]] = pack.elements.groupBy {
      case e: ClassDoc => "classes"
      case e: ObjectDoc => "objects"
      case e: TraitDoc => "traits"
      case e: DocElement => "nvm"
    }
    processObjects(grouped("objects").map(_.asInstanceOf[ObjectDoc]))

->

val objects = pack.elements.collect { case o: ObjectDoc => o }
processObjects(objects)