src/Tools/jEdit/src/jedit/TheoryView.scala
author immler@in.tum.de
Tue, 02 Jun 2009 22:22:03 +0200
changeset 34596 2b46d92e4642
parent 34574 d68352286c91
child 34597 a0c84b0edb9a
permissions -rw-r--r--
linearset works faster here

/*
 * jEdit text area as document text source
 *
 * @author Fabian Immler, TU Munich
 * @author Johannes Hölzl, TU Munich
 * @author Makarius
 */

package isabelle.jedit

import scala.actors.Actor

import isabelle.proofdocument.Text
import isabelle.prover.{Prover, Command}

import java.awt.Graphics2D
import java.awt.event.{ActionEvent, ActionListener}
import java.awt.Color
import javax.swing.Timer
import javax.swing.event.{CaretListener, CaretEvent}

import org.gjt.sp.jedit.buffer.{BufferListener, JEditBuffer}
import org.gjt.sp.jedit.textarea.{JEditTextArea, TextAreaExtension, TextAreaPainter}
import org.gjt.sp.jedit.syntax.SyntaxStyle


object TheoryView {

  val MAX_CHANGE_LENGTH = 1500
  
  def choose_color(state: Command): Color = {
    if (state == null) Color.red
    else
      state.status match {
        case Command.Status.UNPROCESSED => new Color(255, 228, 225)
        case Command.Status.FINISHED => new Color(234, 248, 255)
        case Command.Status.FAILED => new Color(255, 192, 192)
        case _ => Color.red
      }
  }
}


class TheoryView (text_area: JEditTextArea, document_actor: Actor)
    extends TextAreaExtension with BufferListener {

  def id() = Isabelle.plugin.id();
  
  private val buffer = text_area.getBuffer
  private val prover = Isabelle.prover_setup(buffer).get.prover
  buffer.addBufferListener(this)


  private var col: Text.Change = null

  private val col_timer = new Timer(300, new ActionListener() {
    override def actionPerformed(e: ActionEvent) = commit
  })

  col_timer.stop
  col_timer.setRepeats(true)


  private val phase_overview = new PhaseOverviewPanel(prover, text_area, to_current)


  /* activation */

  private val selected_state_controller = new CaretListener {
    override def caretUpdate(e: CaretEvent) = {
      val doc = prover.document
      val cmd = doc.find_command_at(e.getDot)
      if (cmd != null && doc.token_start(cmd.tokens.first) <= e.getDot &&
          Isabelle.prover_setup(buffer).get.selected_state != cmd)
        Isabelle.prover_setup(buffer).get.selected_state = cmd
    }
  }

  def activate() = {
    text_area.addCaretListener(selected_state_controller)
    text_area.addLeftOfScrollBar(phase_overview)
    text_area.getPainter.addExtension(TextAreaPainter.LINE_BACKGROUND_LAYER + 1, this)
    buffer.setTokenMarker(new DynamicTokenMarker(buffer, prover))
  }

  def deactivate() = {
    text_area.getPainter.removeExtension(this)
    text_area.removeLeftOfScrollBar(phase_overview)
    text_area.removeCaretListener(selected_state_controller)
  }


  /* painting */

  val repaint_delay = new isabelle.utils.Delay(100, () => repaint_all())
  
  val change_receiver = scala.actors.Actor.actor {
    scala.actors.Actor.loop {
      scala.actors.Actor.react {
        case _ => {
          Swing.now {
            repaint_delay.delay_or_ignore()
            phase_overview.repaint_delay.delay_or_ignore()
          }
        }
      }
    }
  }.start

  def _from_current(to_id: String, changes: List[Text.Change], pos: Int): Int =
    changes match {
      case Nil => pos
      case Text.Change(id, start, added, removed) :: rest_changes => {
        val shifted = if (start <= pos)
            if (pos < start + added.length) start
            else pos - added.length + removed
          else pos
        if (id == to_id) pos
        else _from_current(to_id, rest_changes, shifted)
      }
    }

  def _to_current(from_id: String, changes: List[Text.Change], pos: Int): Int =
    changes match {
      case Nil => pos
      case Text.Change(id, start, added, removed) :: rest_changes => {
        val shifted = _to_current(from_id, rest_changes, pos)
        if (id == from_id) pos
        else if (start <= shifted) {
          if (shifted < start + removed) start
          else shifted + added.length - removed
        } else shifted
      }
    }

  def to_current(from_id: String, pos : Int) =
    _to_current(from_id, if (col == null) changes else col :: changes, pos)
  def from_current(to_id: String, pos : Int) =
    _from_current(to_id, if (col == null) changes else col :: changes, pos)
  def to_current(document: isabelle.proofdocument.ProofDocument, pos: Int): Int =
    to_current(document.id, pos)
  def from_current(document: isabelle.proofdocument.ProofDocument, pos: Int): Int =
    from_current(document.id, pos)

  def repaint(cmd: Command) =
  {
    val document = prover.document
    if (text_area != null) {
      val start = text_area.getLineOfOffset(to_current(document.id, cmd.start(document)))
      val stop = text_area.getLineOfOffset(to_current(document.id, cmd.stop(document)) - 1)
      text_area.invalidateLineRange(start, stop)

      if (Isabelle.prover_setup(buffer).get.selected_state == cmd)
        Isabelle.prover_setup(buffer).get.selected_state = cmd  // update State view
    }
  }

  def repaint_all() =
  {
    if (text_area != null)
      text_area.invalidateLineRange(text_area.getFirstPhysicalLine, text_area.getLastPhysicalLine)
  }

  def encolor(gfx: Graphics2D,
    y: Int, height: Int, begin: Int, finish: Int, color: Color, fill: Boolean) =
  {
    val start = text_area.offsetToXY(begin)
    val stop =
      if (finish < buffer.getLength) text_area.offsetToXY(finish)
      else {
        val p = text_area.offsetToXY(finish - 1)
        val metrics = text_area.getPainter.getFontMetrics
        p.x = p.x + (metrics.charWidth(' ') max metrics.getMaxAdvance)
        p
      }

    if (start != null && stop != null) {
      gfx.setColor(color)
      if (fill) gfx.fillRect(start.x, y, stop.x - start.x, height)
      else gfx.drawRect(start.x, y, stop.x - start.x, height)
    }
  }


  /* TextAreaExtension methods */

  override def paintValidLine(gfx: Graphics2D,
    screen_line: Int, physical_line: Int, start: Int, end: Int, y: Int) =
  {
    val document = prover.document
    def from_current(pos: Int) = this.from_current(document.id, pos)
    def to_current(pos: Int) = this.to_current(document.id, pos)
    val saved_color = gfx.getColor

    val metrics = text_area.getPainter.getFontMetrics

    // encolor phase
    var e = document.find_command_at(from_current(start))
    while (e != null && e.start(document) < end) {
      val begin = start max to_current(e.start(document))
      val finish = end - 1 min to_current(e.stop(document))
      encolor(gfx, y, metrics.getHeight, begin, finish, TheoryView.choose_color(e), true)
      e = document.commands.next(e).getOrElse(null)
    }

    gfx.setColor(saved_color)
  }

  override def getToolTipText(x: Int, y: Int) = {
    val document = prover.document
    val offset = from_current(document.id, text_area.xyToOffset(x, y))
    val cmd = document.find_command_at(offset)
    if (cmd != null) {
      document.token_start(cmd.tokens.first)
      cmd.type_at(offset - cmd.start(document))
    } else null
  }

  /* BufferListener methods */

  private var changes: List[Text.Change] = Nil

  private def commit: Unit = synchronized {
    if (col != null) {
      def split_changes(c: Text.Change): List[Text.Change] = {
        val MCL = TheoryView.MAX_CHANGE_LENGTH
        if (c.added.length <= MCL) List(c)
        else Text.Change(c.id, c.start, c.added.substring(0, MCL), c.removed) ::
          split_changes(new Text.Change(id(), c.start + MCL, c.added.substring(MCL), c.removed))
      }
      val new_changes = split_changes(col)
      changes = new_changes.reverse ::: changes
      new_changes map (document_actor ! _)
    }
    col = null
    if (col_timer.isRunning())
      col_timer.stop()
  }

  private def delay_commit {
    if (col_timer.isRunning())
      col_timer.restart()
    else
      col_timer.start()
  }


  override def contentInserted(buffer: JEditBuffer,
    start_line: Int, offset: Int, num_lines: Int, length: Int) { }

  override def contentRemoved(buffer: JEditBuffer,
    start_line: Int, offset: Int, num_lines: Int, length: Int) { }

  override def preContentInserted(buffer: JEditBuffer,
    start_line: Int, offset: Int, num_lines: Int, length: Int) =
  {
    val text = buffer.getText(offset, length)
    if (col == null)
      col = new Text.Change(id(), offset, text, 0)
    else if (col.start <= offset && offset <= col.start + col.added.length)
      col = new Text.Change(col.id, col.start, col.added + text, col.removed)
    else {
      commit
      col = new Text.Change(id(), offset, text, 0)
    }
    delay_commit
  }

  override def preContentRemoved(buffer: JEditBuffer,
    start_line: Int, start: Int, num_lines: Int, removed: Int) =
  {
    if (col == null)
      col = new Text.Change(id(), start, "", removed)
    else if (col.start > start + removed || start > col.start + col.added.length) {
      commit
      col = new Text.Change(id(), start, "", removed)
    }
    else {
/*      val offset = start - col.start
      val diff = col.added.length - removed
      val (added, add_removed) =
        if (diff < offset)
          (offset max 0, diff - (offset max 0))
        else
          (diff - (offset min 0), offset min 0)
      col = new Text.Changed(start min col.start, added, col.removed - add_removed)*/
      commit
      col = new Text.Change(id(), start, "", removed)
    }
    delay_commit
  }

  override def bufferLoaded(buffer: JEditBuffer) { }
  override def foldHandlerChanged(buffer: JEditBuffer) { }
  override def foldLevelChanged(buffer: JEditBuffer, start_line: Int, end_line: Int) { }
  override def transactionComplete(buffer: JEditBuffer) { }
}