roby 0.7

data/lib/roby/log/gui/notifications.rb

@@ -0,0 +1,26 @@
+require 'roby/log/gui/basic_display_ui'
+require 'roby/log/notifications'
+
+module Ui
+    class NotificationsConfig < Ui::BasicDisplayConfig
+	attr_reader :streams_model
+
+	def setupUi(streams_model, widget)
+	    super(widget)
+
+	    display = Roby::Log::NotificationsDisplay.new
+	    @streams_model = FilteredDataStreamListModel.new(stream, display, 'roby-events', streams_model.streams)
+	    Qt::Object.connect(stream, SIGNAL("currentIndexChanged(int)"), @streams_model, SLOT("selectedStream()"))
+	    @streams_model.source_model = streams_model
+	    stream.model		= @streams_model
+	    display
+	end
+
+	def self.setup_optparse(opt, replay)
+	    opt.on("--notifications") do
+		replay.add_display('Notifications')
+	    end
+	end
+    end
+end
+

data/lib/roby/log/gui/relations.rb

@@ -0,0 +1,248 @@
+require 'Qt4'
+require 'roby'
+require 'roby/log/relations'
+require 'roby/log/gui/relations_ui'
+
+module Ui
+    # Manage relations using a two-level tree structure. The relation
+    # categories (tasks and events) have a model index of (row, column, nil)
+    # while the relations have a model index of (row, column, category), where
+    # category is TASK_ROOT_INDEX for task relations and EVENT_ROOT_INDEX for
+    # event relations
+    class RelationConfigModel < Qt::AbstractItemModel
+	COL_NAME    = 0
+	COL_COLOR   = 1
+
+	TASK_ROOT_INDEX  = 0
+	EVENT_ROOT_INDEX = 1
+	CATEGORIES = ['Task structure']
+
+	def event_root_index; createIndex(TASK_ROOT_INDEX, 0, -1) end
+	def task_root_index; createIndex(EVENT_ROOT_INDEX, 0, -1) end
+
+	attr_reader :relations
+	attr_reader :display
+	def initialize(display)
+	    super()
+
+	    Roby.load_all_relations
+	    @current_color = 0
+	    @display   = display
+	    @relations = []
+
+	    relations[TASK_ROOT_INDEX]  = Roby::TaskStructure.enum_for(:each_relation).to_a
+	end
+
+	def index(row, column, parent)
+	    if parent.valid? && parent.internalPointer == -1
+		createIndex(row, column, parent.row)
+	    elsif row < relations.size
+		createIndex(row, column, -1)
+	    else
+		Qt::ModelIndex.new
+	    end
+	end
+	def parent(index)
+	    category = index.internalPointer
+	    if !index.valid? || category == -1 then Qt::ModelIndex.new
+	    else createIndex(category, 0, -1) end
+	end
+	def columnCount(parent); 2 end
+	def hasChildren(parent); !parent.valid? || parent.internalPointer == -1 end
+	def rowCount(parent)
+	    if !parent.valid? then relations.size
+	    else relations[parent.row].size end
+	end
+	def headerData(section, orientation, role)
+	    return Qt::Variant.new unless role == Qt::DisplayRole && orientation == Qt::Horizontal
+	    value = if section == 0 then "Relation"
+		    else "Color" end
+	    Qt::Variant.new(value)
+	end
+	def data(index, role)
+	    return Qt::Variant.new unless index.valid?
+
+	    category = index.internalPointer
+	    value = if category == -1
+			if index.column == COL_NAME && role == Qt::DisplayRole
+			    CATEGORIES[index.row]
+			end
+		    else
+			relation = relations[category][index.row]
+			if index.column == COL_NAME && role == Qt::CheckStateRole
+			    if    display.relation_enabled?(relation) then Qt::Checked.to_i
+			    elsif display.layout_relation?(relation)  then Qt::PartiallyChecked.to_i
+			    else Qt::Unchecked.to_i
+			    end
+			elsif index.column == COL_NAME && role == Qt::DisplayRole
+			    relation.name.gsub(/.*Structure::/, '')
+			elsif index.column == COL_COLOR && role == Qt::DisplayRole
+			    display.relation_color(relation)
+			end
+		    end
+
+	    if value then Qt::Variant.new(value)
+	    else Qt::Variant.new
+	    end
+	end
+	def setData(index, value, role)
+	    category = index.internalPointer
+	    relation = relations[category][index.row]
+	    if role == Qt::CheckStateRole
+		case value.to_i
+		when Qt::Checked.to_i
+		    display.enable_relation(relation)
+		when Qt::PartiallyChecked.to_i
+		    display.layout_relation(relation)
+		else
+		    display.ignore_relation(relation)
+		end
+	    else
+		display.update_relation_color(relation, value.to_string)
+	    end
+	    display.update
+	    emit dataChanged(index, index)
+	end
+	def flags(index)
+	    if !index.valid? || index.internalPointer == -1 then Qt::ItemIsEnabled 
+	    else 
+		flags = Qt::ItemIsSelectable | Qt::ItemIsTristate | Qt::ItemIsEnabled | Qt::ItemIsUserCheckable
+		if index.column == 1
+		    flags = Qt::ItemIsEditable | flags
+		end
+		flags
+	    end
+	end
+    end
+    class RelationDelegate < Qt::ItemDelegate
+	MAX_WIDTH = 50
+	def createEditor(parent, option, index)
+	    color = index.model.data(index, Qt::DisplayRole).to_string
+	    new_color = Qt::ColorDialog.get_color(Qt::Color.new(color))
+	    index.model.setData(index, Qt::Variant.new(new_color.name), Qt::DisplayRole)
+
+	    nil
+	end
+	def paint(painter, option, index)
+	    if index.column == 1 && index.internalPointer >= 0
+		color = index.model.data(index, Qt::DisplayRole).to_string
+		rect = option.rect
+		rect.adjust(1, 1, -1, -1)
+		if rect.width > MAX_WIDTH
+		    rect.width = MAX_WIDTH
+		end
+		painter.fill_rect rect, Qt::Brush.new(Qt::Color.new(color))
+	    else
+		super
+	    end
+	end
+    end
+
+    class LayoutMethodModel < Qt::AbstractListModel
+	METHODS    = ["Auto", "dot [rankdir=LR]", "dot [rankdir=TB]", "circo", "neato [overlap=false]", "neato [overlap=false,mode=hier]", "twopi", "fdp"]
+	attr_reader :display, :combo
+	def initialize(display, combo)
+	    super()
+	    @display, @combo = display, combo
+	end
+
+	def rowCount(parent)
+	    return 0 if parent.valid?
+	    return METHODS.size
+	end
+	def data(index, role)
+	    return Qt::Variant.new unless role == Qt::DisplayRole && index.valid?
+	    Qt::Variant.new(METHODS[index.row])
+	end
+	def layout_method(index)
+	end
+
+	def flags(index)
+	    Qt::ItemIsSelectable | Qt::ItemIsEnabled
+	end
+
+	def selected
+	    index = combo.current_index
+	    display.layout_method = if index == 0 then nil
+				    else METHODS[index]
+				    end
+	end
+	slots 'selected()'
+    end
+
+    class RelationsConfig
+	attr_reader :current_color
+	attr_reader :relation_color
+	attr_reader :relation_item
+	attr_reader :model
+	attr_reader :delegate
+
+        def initialize(*args)
+            super()
+            if !system("dot", "-V")
+                raise "the 'dot' tool is unavailable"
+            end
+        end
+
+	def setupUi(streams_model, widget)
+	    super(widget)
+
+	    display   = Roby::Log::RelationsDisplay.new
+	    @model    = RelationConfigModel.new(display)
+	    @delegate = RelationDelegate.new
+	    relations.set_item_delegate @delegate
+	    relations.header.resize_mode = Qt::HeaderView::Stretch
+	    relations.model = @model
+	    relations.set_expanded model.task_root_index, true
+	    relations.set_expanded model.event_root_index, true
+
+	    @streams_model = FilteredDataStreamListModel.new(stream, display, 'roby-events', streams_model.streams)
+	    Qt::Object.connect(stream, SIGNAL("currentIndexChanged(int)"), @streams_model, SLOT("selectedStream()"))
+	    @streams_model.source_model = streams_model
+	    stream.model		= @streams_model
+
+	    @layout_model = LayoutMethodModel.new(display, layout_method)
+	    Qt::Object.connect(layout_method, SIGNAL("currentIndexChanged(int)"), @layout_model, SLOT("selected()"))
+	    layout_method.model = @layout_model
+
+	    display
+	end
+
+	def self.setup_optparse(opt, replay)
+	    opt.on("--relations=REL1,REL2", Array, "create a relation display with the given relations") do |relations|
+		if relations.include?("all")
+		    relations = Roby::TaskStructure.relations
+		else
+		    relations.map! do |relname|
+			rel = Roby::TaskStructure.relations.find { |rel| rel.name =~ /#{relname}/ }
+			unless rel
+			    STDERR.puts "Unknown relation #{relname}. Available relations are:"
+			    STDERR.puts "  Tasks: " + Roby::TaskStructure.enum_for(:each_relation).map { |r| r.name.gsub(/.*Structure::/, '') }.join(", ")
+			    exit(1)
+			end
+
+			rel
+		    end
+		end
+
+		replay.initial_setup << lambda do |gui|
+		    if relation_display = gui.add_display('Relations')
+                        relations.each do |rel|
+                            relation_display.enable_relation(rel)
+                        end
+                    end
+		end
+	    end
+	end
+    end
+end
+
+if $0 == __FILE__
+    a = Qt::Application.new(ARGV)
+    u = Ui::RelationsConfig.new
+    w = Qt::Widget.new
+    u.setupUi(w)
+    w.show
+    a.exec
+end
+

data/lib/roby/log/gui/relations.ui

@@ -0,0 +1,123 @@
+<ui version="4.0" >
+ <class>RelationsConfig</class>
+ <widget class="QWidget" name="RelationsConfig" >
+  <property name="geometry" >
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>696</width>
+    <height>668</height>
+   </rect>
+  </property>
+  <property name="windowTitle" >
+   <string>RelationsConfig</string>
+  </property>
+  <layout class="QVBoxLayout" >
+   <property name="margin" >
+    <number>9</number>
+   </property>
+   <property name="spacing" >
+    <number>6</number>
+   </property>
+   <item>
+    <layout class="QHBoxLayout" >
+     <property name="margin" >
+      <number>0</number>
+     </property>
+     <property name="spacing" >
+      <number>6</number>
+     </property>
+     <item>
+      <spacer>
+       <property name="orientation" >
+        <enum>Qt::Horizontal</enum>
+       </property>
+       <property name="sizeHint" >
+        <size>
+         <width>40</width>
+         <height>20</height>
+        </size>
+       </property>
+      </spacer>
+     </item>
+     <item>
+      <widget class="QPushButton" name="close" >
+       <property name="text" >
+        <string>Close</string>
+       </property>
+      </widget>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <layout class="QGridLayout" >
+     <property name="margin" >
+      <number>0</number>
+     </property>
+     <property name="spacing" >
+      <number>6</number>
+     </property>
+     <item row="1" column="2" >
+      <widget class="QComboBox" name="layout_method" >
+       <property name="sizePolicy" >
+        <sizepolicy>
+         <hsizetype>3</hsizetype>
+         <vsizetype>0</vsizetype>
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+      </widget>
+     </item>
+     <item row="0" column="0" >
+      <widget class="QLabel" name="label" >
+       <property name="sizePolicy" >
+        <sizepolicy>
+         <hsizetype>1</hsizetype>
+         <vsizetype>5</vsizetype>
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="text" >
+        <string>Data stream</string>
+       </property>
+      </widget>
+     </item>
+     <item row="0" column="1" colspan="2" >
+      <widget class="QComboBox" name="stream" >
+       <property name="sizePolicy" >
+        <sizepolicy>
+         <hsizetype>3</hsizetype>
+         <vsizetype>0</vsizetype>
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+      </widget>
+     </item>
+     <item row="1" column="0" colspan="2" >
+      <widget class="QLabel" name="label_2" >
+       <property name="sizePolicy" >
+        <sizepolicy>
+         <hsizetype>1</hsizetype>
+         <vsizetype>5</vsizetype>
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="text" >
+        <string>Layout method</string>
+       </property>
+      </widget>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <widget class="QTreeView" name="relations" />
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>

data/lib/roby/log/gui/relations_view.rb

@@ -0,0 +1,185 @@
+require 'roby/log/gui/relations_view_ui'
+
+class Ui::RelationsView
+    def scene; graphics.scene end
+    attr_reader :display
+    attr_reader :prefixActions
+
+    def config_path
+	"#{display.decoder.name}-events.yml"
+    end
+
+    def load_config
+	if File.readable?(config_path)
+	    STDERR.puts "Loading config from #{config_path}"
+	    config_data = File.open(config_path) do |io|
+		YAML.load(io)
+	    end
+
+	    display.removed_prefixes.clear
+	    if removed_prefixes_config = config_data['prefixes']
+		removed_prefixes_config.each do |name, enabled|
+		    display.removed_prefixes[name] = enabled
+		end
+	    end
+	else
+	    STDERR.puts "No such config file #{config_path}"
+	end
+    end
+    def save_config
+	STDERR.puts "Saving config into #{display.decoder.name}-events.yml"
+	config = { 'prefixes' => Hash.new }
+	display.removed_prefixes.each do |name, enabled|
+	    config['prefixes'][name] = enabled
+	end
+	config['show_ownership'] = display.show_ownership
+	config['show_arguments'] = display.show_arguments
+	config['hide_finalized'] = display.hide_finalized
+
+	File.open(config_path, 'w') do |io|
+	    YAML.dump(config, io)
+	end
+    end
+
+    def update_prefix_menu
+	@prefixActions ||= []
+	prefixActions.each do |action|
+	    menuRemovedPrefixes.remove_action(action)
+	end
+
+	prefixActions.clear
+	display.removed_prefixes.each do |prefix, bool|
+	    action = Qt::Action.new prefix, menuRemovedPrefixes
+	    prefixActions << action
+	    action.checkable = true
+	    action.checked = bool
+	    action.connect(SIGNAL(:triggered)) do 
+		display.removed_prefixes[prefix] = action.checked?
+		display.update
+	    end
+	    menuRemovedPrefixes.add_action action
+	end
+    end
+
+    ZOOM_STEP = 0.25
+    def setupUi(relations_display)
+	@display   = relations_display
+	super(relations_display.main)
+
+	actionOwnership.connect(SIGNAL(:triggered)) do
+	    display.show_ownership = actionOwnership.checked?
+	    display.update
+	end
+
+	actionHideFinalized.connect(SIGNAL(:triggered)) do
+	    display.hide_finalized = actionHideFinalized.checked?
+	    display.update
+	end
+
+	#############################################################
+	# Build the removed_prefixes menu
+	actionPrefixAdd.connect(SIGNAL(:triggered)) do
+	    new_prefix = Qt::InputDialog.get_text display.main, "New prefix", "New prefix to remove"
+	    if !new_prefix.nil?
+		display.removed_prefixes[new_prefix] = true
+		save_config
+		update_prefix_menu
+		display.update
+	    end
+	end
+	update_prefix_menu
+	
+	#############################################################
+	# Handle the other toolbar's buttons
+	graphics.singleton_class.class_eval do
+	    define_method(:contextMenuEvent) do |event|
+		item = itemAt(event.pos)
+		if item
+		    unless obj = relations_display.object_of(item)
+			return super(event)
+		    end
+		end
+
+		return unless obj.kind_of?(Roby::LoggedTask)
+
+		menu = Qt::Menu.new
+		hide_this     = menu.add_action("Hide")
+		hide_children = menu.add_action("Hide children")
+		show_children = menu.add_action("Show children")
+		return unless action = menu.exec(event.globalPos)
+
+		case action.text
+		when "Hide"
+		    relations_display.set_visibility(obj, false)
+		when "Hide children"
+		    for child in Roby::TaskStructure.children_of(obj, relations_display.enabled_relations)
+			relations_display.set_visibility(child, false)
+		    end
+		when "Show children"
+		    for child in Roby::TaskStructure.children_of(obj, relations_display.enabled_relations)
+			relations_display.set_visibility(child, true)
+		    end
+		end
+
+		relations_display.update
+	    end
+	end
+
+	actionShowAll.connect(SIGNAL(:triggered)) do
+	    display.graphics.keys.each do |obj|
+		display.set_visibility(obj, true) if obj.kind_of?(Roby::Task::DRoby) || (obj.kind_of?(Roby::EventGenerator::DRoby) && !obj.respond_to?(:task))
+	    end
+	    display.update
+	end
+	actionRedraw.connect(SIGNAL(:triggered)) do
+	    display.update
+	end
+
+	actionZoom.connect(SIGNAL(:triggered)) do 
+	    scale = graphics.matrix.m11
+	    if scale + ZOOM_STEP > 1
+		scale = 1 - ZOOM_STEP
+	    end
+	    graphics.resetMatrix
+	    graphics.scale scale + ZOOM_STEP, scale + ZOOM_STEP
+	end
+	actionUnzoom.connect(SIGNAL(:triggered)) do
+	    scale = graphics.matrix.m11
+	    graphics.resetMatrix
+	    graphics.scale scale - ZOOM_STEP, scale - ZOOM_STEP
+	end
+	actionFit.connect(SIGNAL(:triggered)) do
+	    graphics.fitInView(graphics.scene.items_bounding_rect, Qt::KeepAspectRatio)
+	end
+
+	actionKeepSignals.connect(SIGNAL(:triggered)) do 
+	    display.keep_signals = actionKeepSignals.checked?
+	end
+
+	actionPrint.connect(SIGNAL(:triggered)) do
+	    return unless scene
+	    printer = Qt::Printer.new;
+	    if Qt::PrintDialog.new(printer).exec() == Qt::Dialog::Accepted
+		painter = Qt::Painter.new(printer);
+		painter.setRenderHint(Qt::Painter::Antialiasing);
+		scene.render(painter);
+	    end
+	end
+
+	actionSVGExport.connect(SIGNAL(:triggered)) do
+	    return unless scene
+
+	    if path = Qt::FileDialog.get_save_file_name(nil, "SVG Export")
+		svg = Qt::SvgGenerator.new
+		svg.file_name = path
+		svg.size = Qt::Size.new(Integer(scene.width * 0.8), Integer(scene.height * 0.8))
+		painter = Qt::Painter.new
+		painter.begin(svg)
+		scene.render(painter)
+		painter.end
+	    end
+	end
+	actionSVGExport.enabled = defined?(Qt::SvgGenerator)
+    end
+end
+