/b-gtk/fix-viz

# -*- coding: UTF-8 -*-

"""Revision history view.

"""

__copyright__ = "Copyright © 2005 Canonical Ltd."

__author__    = "Daniel Schierbeck <daniel.schierbeck@gmail.com>"

import sys

import string

import gtk

import gobject

import pango

import re

import treemodel

from linegraph import linegraph, same_branch

from graphcell import CellRendererGraph

from treemodel import TreeModel

class TreeView(gtk.ScrolledWindow):

    __gsignals__ = {

            'revisions-loaded': (gobject.SIGNAL_RUN_FIRST, 

                                 gobject.TYPE_NONE,

                                 ()),

            'revision-selected': (gobject.SIGNAL_RUN_FIRST,

                                  gobject.TYPE_NONE,

                                  ())

    }

    def __init__(self, branch, start, maxnum):

        """Create a new TreeView.

        :param branch: Branch object for branch to show.

        :param start: Revision id of top revision.

        :param maxnum: Maximum number of revisions to display, 

                       None for no limit.

        """

        gtk.ScrolledWindow.__init__(self)

        self.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)

        self.set_shadow_type(gtk.SHADOW_IN)

        self.construct_treeview()

        self.branch = branch

        self.branch.lock_read()

        gobject.idle_add(self.populate, start, maxnum)

        self.revision = None

        self.children = None

        self.parents  = None

        self.connect('destroy', lambda w: self.branch.unlock())

    def get_revision(self):

        """Return revision id of currently selected revision, or None."""

        return self.revision

    def set_revision(self, revid):

        """Change the currently selected revision.

        :param revid: Revision id of revision to display.

        """

        self.treeview.set_cursor(self.index[revid])

        self.treeview.grab_focus()

    def get_children(self):

        """Return the children of the currently selected revision.

        :return: list of revision ids.

        """

        return self.children

    def get_parents(self):

        """Return the parents of the currently selected revision.

        :return: list of revision ids.

        """

        return self.parents

    def back(self):

        """Signal handler for the Back button."""

        (path, col) = self.treeview.get_cursor()

        revision = self.model[path][treemodel.REVISION]

        parents = self.model[path][treemodel.PARENTS]

        if not len(parents):

            return

        for parent_id in parents:

            parent_index = self.index[parent_id]

            parent = self.model[parent_index][treemodel.REVISION]

            if same_branch(revision, parent):

                self.treeview.set_cursor(parent_index)

                break

        else:

            self.treeview.set_cursor(self.index[parents[0]])

        self.treeview.grab_focus()

    def forward(self):

        """Signal handler for the Forward button."""

        (path, col) = self.treeview.get_cursor()

        revision = self.model[path][treemodel.REVISION]

        children = self.model[path][treemodel.CHILDREN]

        if not len(children):

            return

        for child_id in children:

            child_index = self.index[child_id]

            child = self.model[child_index][treemodel.REVISION]

            if same_branch(child, revision):

                self.treeview.set_cursor(child_index)

                break

        else:

            self.treeview.set_cursor(self.index[children[0]])

        self.treeview.grab_focus()

    def populate(self, start, maxnum):

        """Fill the treeview with contents.

        :param start: Revision id of revision to start with.

        :param maxnum: Maximum number of revisions to display, or None 

                       for no limit.

        """

        (linegraphdata, index, columns_len) = linegraph(self.branch,

                                                        start,

                                                        maxnum)

        self.model = TreeModel(self.branch, linegraphdata)

        self.graph_cell.columns_len = columns_len

        width = self.graph_cell.get_size(self.treeview)[2]

        self.graph_column.set_fixed_width(width)

        self.graph_column.set_max_width(width)

        self.index = index

        self.treeview.set_model(self.model)

        self.treeview.set_cursor(0)

        self.emit('revisions-loaded')

        return False

    def show_diff(self, branch, revid, parentid):

        """Open a new window to show a diff between the given revisions."""

        from bzrlib.plugins.gtk.diff import DiffWindow

        window = DiffWindow(parent=self)

        (parent_tree, rev_tree) = branch.repository.revision_trees([parentid, 

                                                                   revid])

        description = revid + " - " + branch.nick

        window.set_diff(description, rev_tree, parent_tree)

        window.show()

    def construct_treeview(self):

        self.treeview = gtk.TreeView()

        self.treeview.set_rules_hint(True)

        self.treeview.set_search_column(4)

        self.treeview.get_selection().connect("changed",

                self._on_selection_changed)

        self.treeview.connect("row-activated", 

                self._on_revision_activated)

        self.treeview.connect("button-release-event", 

                self._on_revision_selected)

        self.treeview.set_property('fixed-height-mode', True)

        self.add(self.treeview)

        self.treeview.show()

        cell = gtk.CellRendererText()

        cell.set_property("width-chars", 15)

        cell.set_property("ellipsize", pango.ELLIPSIZE_END)

        column = gtk.TreeViewColumn("Revision No")

        column.set_resizable(True)

        column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)

        column.set_fixed_width(cell.get_size(self.treeview)[2])

        column.pack_start(cell, expand=True)

        column.add_attribute(cell, "text", treemodel.REVNO)

        self.treeview.append_column(column)

        self.graph_cell = CellRendererGraph()

        self.graph_column = gtk.TreeViewColumn()

        self.graph_column.set_resizable(True)

        self.graph_column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)

        self.graph_column.pack_start(self.graph_cell, expand=False)

        self.graph_column.add_attribute(self.graph_cell, "node", treemodel.NODE)

        self.graph_column.add_attribute(self.graph_cell, "in-lines", treemodel.LAST_LINES)

        self.graph_column.add_attribute(self.graph_cell, "out-lines", treemodel.LINES)

        self.treeview.append_column(self.graph_column)

        cell = gtk.CellRendererText()

        cell.set_property("width-chars", 65)

        cell.set_property("ellipsize", pango.ELLIPSIZE_END)

        column = gtk.TreeViewColumn("Message")

        column.set_resizable(True)

        column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)

        column.set_fixed_width(cell.get_size(self.treeview)[2])

        column.pack_start(cell, expand=True)

        column.add_attribute(cell, "text", treemodel.MESSAGE)

        self.treeview.append_column(column)

        cell = gtk.CellRendererText()

        cell.set_property("width-chars", 15)

        cell.set_property("ellipsize", pango.ELLIPSIZE_END)

        column = gtk.TreeViewColumn("Committer")

        column.set_resizable(True)

        column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)

        column.set_fixed_width(cell.get_size(self.treeview)[2])

        column.pack_start(cell, expand=True)

        column.add_attribute(cell, "text", treemodel.COMMITER)

        self.treeview.append_column(column)

    def _on_selection_changed(self, selection, *args):

        """callback for when the treeview changes."""

        (model, selected_rows) = selection.get_selected_rows()

        if len(selected_rows) > 0:

            iter = self.model.get_iter(selected_rows[0])

            self.revision = self.model.get_value(iter, treemodel.REVISION)

            self.parents = self.model.get_value(iter, treemodel.PARENTS)

            self.children = self.model.get_value(iter, treemodel.CHILDREN)

            self.emit('revision-selected')

    def _on_revision_selected(self, widget, event):

        from bzrlib.plugins.gtk.revisionmenu import RevisionPopupMenu

        if event.button == 3:

            menu = RevisionPopupMenu(self.branch.repository, 

                [self.get_revision().revision_id],

                self.branch)

            menu.popup(None, None, None, event.button, event.get_time())

    def _on_revision_activated(self, widget, path, col):

        # TODO: more than one parent

        """Callback for when a treeview row gets activated."""

        revision_id = self.model[path][treemodel.REVID]

        parents = self.model[path][treemodel.PARENTS]

        if len(parents) == 0:

            # Ignore revisions without parent

            return

        parent_id = parents[0]

        self.show_diff(self.branch, revision_id, parent_id)

        self.treeview.grab_focus()