src/Tools/jEdit/src/prover/MarkupNode.scala
author immler@in.tum.de
Wed, 29 Apr 2009 16:08:22 +0200
changeset 34557 647a2430d1cd
parent 34555 7c001369956a
child 34558 668fae39d86d
permissions -rw-r--r--
immutable markup-nodes; more seperate nodes in command; restructured handling of markups in prover

/*
 * Document markup nodes, with connection to Swing tree model
 *
 * @author Fabian Immler, TU Munich
 */

package isabelle.prover

import isabelle.proofdocument.ProofDocument

import sidekick.IAsset
import javax.swing._
import javax.swing.text.Position
import javax.swing.tree._

object MarkupNode {

  def markup2default_node(node: MarkupNode, cmd: Command, doc: ProofDocument) : DefaultMutableTreeNode = {

    implicit def int2pos(offset: Int): Position =
      new Position { def getOffset = offset; override def toString = offset.toString}

    object RelativeAsset extends IAsset {
      override def getIcon : Icon = null
      override def getShortString : String = node.content
      override def getLongString : String = node.desc
      override def getName : String = node.id
      override def setName (name : String) = ()
      override def setStart(start : Position) = ()
      override def getStart : Position = node.abs_start(doc)
      override def setEnd(end : Position) = ()
      override def getEnd : Position = node.abs_stop(doc)
      override def toString = node.id + ": " + node.content + "[" + getStart + " - " + getEnd + "]"
    }

    new DefaultMutableTreeNode(RelativeAsset)
  }
}

class MarkupNode (val base : Command, val start : Int, val stop : Int,
                  val children: List[MarkupNode],
                  val id : String, val content : String, val desc : String) {

  def swing_node(doc: ProofDocument) : DefaultMutableTreeNode = {
    val node = MarkupNode.markup2default_node (this, base, doc)
    children.map(node add _.swing_node(doc))
    node
  }

  def abs_start(doc: ProofDocument) = base.start(doc) + start
  def abs_stop(doc: ProofDocument) = base.start(doc) + stop

  def set_children(newchildren: List[MarkupNode]): MarkupNode =
    new MarkupNode(base, start, stop, newchildren, id, content, desc)

  def add(child : MarkupNode) = set_children ((child :: children) sort ((a, b) => a.start < b.start))

  def remove(nodes : List[MarkupNode]) = set_children(children diff nodes)

  def dfs : List[MarkupNode] = {
    var all = Nil : List[MarkupNode]
    for (child <- children)
      all = child.dfs ::: all
    all = this :: all
    all
  }

  def leafs: List[MarkupNode] = {
    if (children == Nil) return List(this)
    else return children flatMap (_.leafs)
  }

  def flatten: List[MarkupNode] = {
    var next_x = start
    if(children.length == 0) List(this)
    else {
      val filled_gaps = for {
        child <- children
        markups = if (next_x < child.start) {
          new MarkupNode(base, next_x, child.start, Nil, id, content, "") :: child.flatten
        } else child.flatten
        update = (next_x = child.stop)
        markup <- markups
      } yield markup
      if (next_x < stop) filled_gaps + new MarkupNode(base, next_x, stop, Nil, id, content, "")
      else filled_gaps
    }
  }

  def +(new_child : MarkupNode) : MarkupNode = {
    if (new_child fitting_into this) {
      val new_children = children.map(c => if((new_child fitting_into c)) c + new_child else c)
      if (new_children == children) {
        // new_child did not fit into children of this -> insert new_child between this and its children
        val (fitting, nonfitting) = children span(_ fitting_into new_child)
        this remove fitting add ((new_child /: fitting) (_ add _))
      }
      else this set_children new_children
    } else {
      error("ignored nonfitting markup " + new_child.id + new_child.content + new_child.desc
                         + "(" +new_child.start + ", "+ new_child.stop + ")")
    }
  }

  // does this fit into node?
  def fitting_into(node : MarkupNode) = node.start <= this.start && node.stop >= this.stop

  override def toString = "([" + start + " - " + stop + "] " + id + "( " + content + "): " + desc
}