wads 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ced90f1073417f101cc5688139f63c6437814a7bf96249b54344ad3a72f5a0fc
-  data.tar.gz: '08fafccb767210d6f3653ee9ab6c33ca01962b4bd71503c54a3afcca83efcd9f'
+  metadata.gz: dffb479b6d7bdc47ff103db3f88b1acece5d0ec672361070aec604eb25aa6e9b
+  data.tar.gz: b9fd53c1345bc7f5ab18cd0b5ab2eb74cc0b3276cd538cdae6dad414523f9529
 SHA512:
-  metadata.gz: 2daa8f10e3cabab4c8824f5f070b5b9f5b124b5c90910b6bd9d0a75c8775e8ed1285ed4cb2a019a08320f27c735065789a2401e60b2098731d6e7d915073868d
-  data.tar.gz: 235bd6b70e43ef6f8714ea6406598a49747f5846495193d5193d089acf09b30ef80ff5dbf91733c6eb6e0e08468f34b192735262900b31b89d4caf71b41a1648
+  metadata.gz: dbeb4a1dd4f5918d047954b6336219e3dd9429db3de755f56f12f34cf60c4e187d92a967a3cbde88ad02dc35952257da492729f1da762f8b49035775462b4b1b
+  data.tar.gz: 6c28345cfc8658cd25c1243469ff0fa308aa8278597b48e42f9cdd71b1a4d7e285e3df989d19504ea2f3137a36e593ee6a46116a360d4ed92d627c2b9b334619

data/CHANGELOG.md

@@ -1,5 +1,8 @@
-## [Unreleased]
-
 ## [0.1.0] - 2021-08-23
 
 - Initial release
+
+## [0.1.1] - 2021-08-29
+
+- Added graph data structure and corresponding display widgets
+

data/README.md

@@ -25,3 +25,15 @@ cd ruby-wads
 
 This will run the sample NASDAQ stocks analysis that features the use of the 
 Stats class, and a table Gosu widget used to display the results of the analysis.
+
+You can also see the graph capabilities by running the Star Wars analysis example.
+This uses a data set that captures all character interactions within Episode 4.
+
+```
+./run-sample-app -j -g
+```
+
+## References
+
+The Star Wars data set is courtesy of:
+Gabasova, E. (2016). Star Wars social network. DOI: https://doi.org/10.5281/zenodo.1411479.

data/data/starwars-episode-4-interactions.json

@@ -0,0 +1,336 @@
+{
+  "nodes": [
+    {
+      "name": "C-3PO",
+      "value": 42,
+      "colour": "#FFD700"
+    },
+    {
+      "name": "LUKE",
+      "value": 88,
+      "colour": "#3881e5"
+    },
+    {
+      "name": "DARTH VADER",
+      "value": 25,
+      "colour": "#000000"
+    },
+    {
+      "name": "CAMIE",
+      "value": 3,
+      "colour": "#808080"
+    },
+    {
+      "name": "BIGGS",
+      "value": 18,
+      "colour": "#808080"
+    },
+    {
+      "name": "LEIA",
+      "value": 25,
+      "colour": "#DCDCDC"
+    },
+    {
+      "name": "BERU",
+      "value": 5,
+      "colour": "#808080"
+    },
+    {
+      "name": "OWEN",
+      "value": 6,
+      "colour": "#808080"
+    },
+    {
+      "name": "OBI-WAN",
+      "value": 22,
+      "colour": "#48D1CC"
+    },
+    {
+      "name": "MOTTI",
+      "value": 3,
+      "colour": "#808080"
+    },
+    {
+      "name": "TARKIN",
+      "value": 12,
+      "colour": "#808080"
+    },
+    {
+      "name": "HAN",
+      "value": 44,
+      "colour": "#ff9400"
+    },
+    {
+      "name": "GREEDO",
+      "value": 2,
+      "colour": "#808080"
+    },
+    {
+      "name": "JABBA",
+      "value": 2,
+      "colour": "#808080"
+    },
+    {
+      "name": "DODONNA",
+      "value": 3,
+      "colour": "#808080"
+    },
+    {
+      "name": "GOLD LEADER",
+      "value": 14,
+      "colour": "#808080"
+    },
+    {
+      "name": "WEDGE",
+      "value": 15,
+      "colour": "#808080"
+    },
+    {
+      "name": "RED LEADER",
+      "value": 33,
+      "colour": "#808080"
+    },
+    {
+      "name": "RED TEN",
+      "value": 8,
+      "colour": "#808080"
+    },
+    {
+      "name": "GOLD FIVE",
+      "value": 8,
+      "colour": "#808080"
+    }
+  ],
+  "links": [
+    {
+      "source": 3,
+      "target": 1,
+      "value": 2
+    },
+    {
+      "source": 4,
+      "target": 3,
+      "value": 2
+    },
+    {
+      "source": 4,
+      "target": 1,
+      "value": 4
+    },
+    {
+      "source": 2,
+      "target": 5,
+      "value": 1
+    },
+    {
+      "source": 6,
+      "target": 1,
+      "value": 3
+    },
+    {
+      "source": 6,
+      "target": 7,
+      "value": 3
+    },
+    {
+      "source": 6,
+      "target": 0,
+      "value": 2
+    },
+    {
+      "source": 1,
+      "target": 7,
+      "value": 3
+    },
+    {
+      "source": 0,
+      "target": 1,
+      "value": 18
+    },
+    {
+      "source": 0,
+      "target": 7,
+      "value": 2
+    },
+    {
+      "source": 0,
+      "target": 5,
+      "value": 6
+    },
+    {
+      "source": 5,
+      "target": 1,
+      "value": 17
+    },
+    {
+      "source": 6,
+      "target": 5,
+      "value": 1
+    },
+    {
+      "source": 1,
+      "target": 8,
+      "value": 19
+    },
+    {
+      "source": 0,
+      "target": 8,
+      "value": 6
+    },
+    {
+      "source": 5,
+      "target": 8,
+      "value": 1
+    },
+    {
+      "source": 9,
+      "target": 10,
+      "value": 2
+    },
+    {
+      "source": 2,
+      "target": 9,
+      "value": 1
+    },
+    {
+      "source": 2,
+      "target": 10,
+      "value": 7
+    },
+    {
+      "source": 11,
+      "target": 8,
+      "value": 9
+    },
+    {
+      "source": 11,
+      "target": 1,
+      "value": 26
+    },
+    {
+      "source": 12,
+      "target": 11,
+      "value": 1
+    },
+    {
+      "source": 11,
+      "target": 13,
+      "value": 1
+    },
+    {
+      "source": 0,
+      "target": 11,
+      "value": 6
+    },
+    {
+      "source": 5,
+      "target": 9,
+      "value": 1
+    },
+    {
+      "source": 5,
+      "target": 10,
+      "value": 1
+    },
+    {
+      "source": 11,
+      "target": 5,
+      "value": 13
+    },
+    {
+      "source": 2,
+      "target": 8,
+      "value": 1
+    },
+    {
+      "source": 14,
+      "target": 15,
+      "value": 1
+    },
+    {
+      "source": 14,
+      "target": 16,
+      "value": 1
+    },
+    {
+      "source": 14,
+      "target": 1,
+      "value": 1
+    },
+    {
+      "source": 15,
+      "target": 16,
+      "value": 1
+    },
+    {
+      "source": 15,
+      "target": 1,
+      "value": 1
+    },
+    {
+      "source": 1,
+      "target": 16,
+      "value": 2
+    },
+    {
+      "source": 4,
+      "target": 5,
+      "value": 1
+    },
+    {
+      "source": 5,
+      "target": 17,
+      "value": 1
+    },
+    {
+      "source": 1,
+      "target": 17,
+      "value": 3
+    },
+    {
+      "source": 4,
+      "target": 17,
+      "value": 3
+    },
+    {
+      "source": 4,
+      "target": 0,
+      "value": 1
+    },
+    {
+      "source": 0,
+      "target": 17,
+      "value": 1
+    },
+    {
+      "source": 17,
+      "target": 16,
+      "value": 3
+    },
+    {
+      "source": 15,
+      "target": 17,
+      "value": 1
+    },
+    {
+      "source": 4,
+      "target": 16,
+      "value": 2
+    },
+    {
+      "source": 17,
+      "target": 18,
+      "value": 1
+    },
+    {
+      "source": 4,
+      "target": 15,
+      "value": 1
+    },
+    {
+      "source": 1,
+      "target": 18,
+      "value": 1
+    }
+  ]
+}

data/lib/wads/app.rb

@@ -1,4 +1,5 @@
 require 'gosu'
+require 'json'
 require_relative 'data_structures'
 require_relative 'widgets'
 require_relative 'version'
@@ -9,6 +10,7 @@ class WadsSampleApp < Gosu::Window
     
     STOCKS_DATA_FILE = "./data/NASDAQ.csv"
     LOTTERY_DATA_FILE = "./data/Pick4_12_21_2020.txt"
+    STAR_WARS_DATA_FILE = "./data/starwars-episode-4-interactions.json"
 
     def initialize
         super(800, 600)
@@ -19,6 +21,7 @@ class WadsSampleApp < Gosu::Window
         @small_font = Gosu::Font.new(22)
         @banner_image = Gosu::Image.new("./media/Banner.png")
         @display_widget = nil
+        @update_count = 0
     end 
 
     def parse_opts_and_run 
@@ -36,13 +39,18 @@ class WadsSampleApp < Gosu::Window
             else 
                 stats.report(Date::DAYNAMES[1..5])
             end
-        
+
+        elsif opts[:jedi] 
+            graph = process_star_wars_data
+            @display_widget = SampleStarWarsDisplay.new(@small_font, graph)
+            show
         elsif opts[:lottery] 
             process_lottery_data
         else
             puts " "
             puts "Select one of the following sample analysis options"
             puts "-s   Run sample stocks analysis"
+            puts "-j   Run sample Star Wars character analysis"
             puts "-l   Run sample analysis of lottery numbers"
             puts " "
             exit
@@ -51,7 +59,8 @@ class WadsSampleApp < Gosu::Window
     end
 
     def update 
-        # TODO
+        @display_widget.update(@update_count, mouse_x, mouse_y)
+        @update_count = @update_count + 1
     end 
     
     def draw
@@ -74,6 +83,10 @@ class WadsSampleApp < Gosu::Window
         end
     end
 
+    def button_up id
+        @display_widget.button_up id, mouse_x, mouse_y
+    end
+
     def process_stock_data 
         # The data file comes from https://finance.yahoo.com
         # The format of this file is as follows:
@@ -130,6 +143,41 @@ class WadsSampleApp < Gosu::Window
         puts "and the same pattern used in the stocks example can"
         puts "be applied here."
     end
+
+    def process_star_wars_data 
+        star_wars_json = File.read(STAR_WARS_DATA_FILE)
+        data_hash = JSON.parse(star_wars_json)
+        characters = data_hash['nodes']
+        interactions = data_hash['links']
+
+        # The interactions in the data set reference the characters by their
+        # zero based index, so we keep a reference in our graph by index.
+        # The character's value is the number of scenes in which they appear.
+        graph = Graph.new
+        characters.each do |character|
+            node_tags = {}
+            node_color_str = character['colour']
+            # This is a bit of a hack, but our background is black so black text
+            # will not show up. Change this to white
+            if node_color_str == "#000000"
+                node_color_str = "#FFFFFF"
+            end
+            # Convert hex string (ex. "#EE00AA") into int hex representation
+            # understood by Gosu color (ex. 0xFFEE00AA)
+            node_color = "0xFF#{node_color_str[1..-1]}".to_i(16)
+            node_tags['color'] = node_color
+            graph.add_node(Node.new(character['name'], character['value'], node_tags))
+        end
+        interactions.each do |interaction| 
+            character_one = graph.node_by_index(interaction['source'])
+            character_two = graph.node_by_index(interaction['target'])
+            number_of_scenes_together = interaction['value']
+            edge_tags = {}
+            edge_tags["scenes"] = number_of_scenes_together
+            character_one.add_output_edge(character_two, edge_tags)
+        end
+        graph
+    end
 end
 
 class SampleStocksDisplay < Widget
@@ -196,4 +244,72 @@ class SampleStocksDisplay < Widget
         end 
         WidgetResult.new(false)
     end 
+end
+
+class SampleStarWarsDisplay < Widget
+    attr_accessor :graph
+
+    def initialize(font, graph)
+        super(10, 100, COLOR_HEADER_BRIGHT_BLUE)
+        set_dimensions(780, 500)
+        set_font(font)
+        add_child(Document.new(sample_content, x + 5, y + 5, @width, @height, @font))
+        @exit_button = Button.new("Exit", 380, bottom_edge - 30, @font)
+        add_child(@exit_button)
+
+        @graph = graph
+        @data_table = SingleSelectTable.new(@x + 5, @y + 70,     # top left corner
+                                            770, 180,            # width, height
+                                            ["Character", "Number of Scenes"], # column headers
+                                            @font, COLOR_WHITE,  # font and text color
+                                            5)                   # max visible rows
+        @data_table.selected_color = COLOR_LIGHT_GRAY
+        @graph.node_list.each do |character|
+            @data_table.add_row([character.name, character.value], character.get_tag("color"))
+        end
+        add_child(@data_table)
+        @graph_display = GraphWidget.new(x + 5, y + 260, 770, 200, @font, @color, @graph) 
+    end 
+
+    def sample_content
+        <<~HEREDOC
+          This sample analysis shows the interactions between characters in the Star Wars
+          Episode 4: A New Hope. Click on a character to see more detail.
+        HEREDOC
+    end
+
+    def render 
+        @graph_display.draw 
+    end 
+
+    def button_down id, mouse_x, mouse_y
+        if id == Gosu::MsLeft
+            if @exit_button.contains_click(mouse_x, mouse_y)
+                return WidgetResult.new(true)
+            elsif @data_table.contains_click(mouse_x, mouse_y)
+                val = @data_table.set_selected_row(mouse_y, 0)
+                if val.nil?
+                    # nothing to do
+                else
+                    node = @graph.find_node(val)
+                    @graph_display.set_display(node, 2)
+                end  
+            elsif @graph_display.contains_click(mouse_x, mouse_y)
+                @graph_display.button_down id, mouse_x, mouse_y
+            end 
+        elsif id == Gosu::KbUp
+            @data_table.scroll_up
+        elsif id == Gosu::KbDown
+            @data_table.scroll_down
+        end 
+        WidgetResult.new(false)
+    end 
+
+    def button_up id, mouse_x, mouse_y
+        @graph_display.button_up id, mouse_x, mouse_y
+    end
+
+    def update update_count, mouse_x, mouse_y
+        @graph_display.update update_count, mouse_x, mouse_y
+    end
 end

data/lib/wads/data_structures.rb

@@ -4,7 +4,25 @@ module Wads
 
     SPACER = "  "
     VALUE_WIDTH = 10
-    NODE_UNKNOWN = "undefined"
+
+    DEG_0 = 0
+    DEG_45 = Math::PI * 0.25
+    DEG_90 = Math::PI * 0.5
+    DEG_135 = Math::PI * 0.75
+    DEG_180 = Math::PI 
+    DEG_225 = Math::PI * 1.25
+    DEG_270 = Math::PI * 1.5
+    DEG_315 = Math::PI * 1.75
+    DEG_360 = Math::PI * 2
+
+    DEG_22_5 = Math::PI * 0.125
+    DEG_67_5 = DEG_45 + DEG_22_5
+    DEG_112_5 = DEG_90 + DEG_22_5
+    DEG_157_5 = DEG_135 + DEG_22_5
+    DEG_202_5 = DEG_180 + DEG_22_5
+    DEG_247_5 = DEG_225 + DEG_22_5
+    DEG_292_5 = DEG_270 + DEG_22_5
+    DEG_337_5 = DEG_315 + DEG_22_5
 
     class HashOfHashes 
         attr_accessor :data
@@ -225,27 +243,198 @@ module Wads
     end
 
     class Node
-        attr_accessor :x
-        attr_accessor :y
         attr_accessor :name
-        attr_accessor :type
-        attr_accessor :inputs
+        attr_accessor :value
+        attr_accessor :backlinks
         attr_accessor :outputs
         attr_accessor :visited
+        attr_accessor :tags
 
-        def initialize(name, type = NODE_UNKNOWN)
-            @name = name
-            @type = type
-            @inputs = [] 
+        def initialize(name, value = nil, tags = {})
+            @name = name 
+            @value = value
+            @backlinks = [] 
             @outputs = []
             @visited = false
+            @tags = tags
+        end 
+        
+        def add_child(name, value)
+            add_output(name, value)
         end 
 
-        #
-        # TODO Visitor pattern and solution for detecting cyclic graphs
-        #
-        # when you visit, reset all the visited flags
-        # set it to true when you visit the node
-        # first check though if visited already true, if so, you have a cycle
+        def add_output(name, value)
+            child_node = Node.new(name, value)
+            add_output_node(child_node) 
+        end
+
+        def add_output_node(child_node)
+            child_node.backlinks << self
+            @outputs << child_node
+            child_node
+        end
+
+        def add_output_edge(destination, tags = {})
+            edge = Edge.new(destination, tags)
+            destination.backlinks << self
+            @outputs << edge
+            edge
+        end
+
+        def add_tag(key, value)
+            @tags[key] = value 
+        end
+
+        def get_tag(key) 
+            @tags[key] 
+        end
+
+        def find_node(search_name)
+            if @name == search_name
+                return self
+            end
+            found_node_in_child = nil
+    
+            @outputs.each do |child|
+                if child.is_a? Edge 
+                    child = child.destination 
+                end
+                found_node_in_child = child.find_node(search_name)
+                if found_node_in_child
+                    return found_node_in_child
+                end 
+            end
+            nil
+        end
+    
+        def visit(&block)
+            node_queue = [self]
+            until node_queue.empty?
+                node = node_queue.shift
+                yield node
+                node.outputs.each do |c|
+                    if child.is_a? Edge 
+                        child = child.destination 
+                    end 
+                    node_queue << c
+                end
+            end
+        end
+
+        def to_display 
+            "#{@name}: #{value}   inputs: #{@backlinks.size}  outputs: #{@outputs.size}"
+        end 
+
+        def full_display 
+            puts to_display
+            @backlinks.each do |i|
+                puts "Input:  #{i.name}"
+            end
+            #@outputs.each do |o|
+            #    puts "Output: #{o.name}"
+            #end
+        end 
+    end
+
+    class Edge
+        attr_accessor :destination
+        attr_accessor :tags
+
+        def initialize(destination, tags = {})
+            @destination = destination
+            @tags = tags
+        end 
+
+        def add_tag(key, value)
+            @tags[key] = value 
+        end
+
+        def get_tag(key) 
+            @tags[key] 
+        end
+    end 
+
+    class Graph 
+        attr_accessor :node_list
+        attr_accessor :node_map
+
+        def initialize 
+            @node_list = []
+            @node_map = {}
+        end
+
+        def add_node(node) 
+            @node_list << node 
+            @node_map[node.name] = node 
+        end 
+
+        def add_edge(source, target, tags)
+            if source.is_a? String 
+                source = find_node(source)
+            end 
+            if target.is_a? String 
+                target = find_node(target)
+            end 
+        end
+
+        def find_node(name) 
+            @node_map[name]
+        end 
+
+        def node_by_index(index) 
+            @node_list[index] 
+        end 
+
+        def reset_visited 
+            @node_list.each do |node|
+                node.visited = false 
+            end 
+        end
+
+        def root_nodes 
+            list = []
+            @node_list.each do |node|
+                if node.backlinks.empty? 
+                    list << node 
+                end  
+            end 
+            list 
+        end 
+
+        def is_cycle(node)
+            reset_visited
+            node.visit do |n|
+                if n.visited 
+                    return true
+                else 
+                    n.visited = true
+                end
+            end
+            false
+        end
+
+        def fan_out(node, max_depth, current_depth = 1)
+            if current_depth > max_depth 
+                return {}
+            end
+            map = {}
+            map[node.name] = node
+            node.backlinks.each do |child|
+                map_from_child = fan_out(child, max_depth, current_depth + 1)
+                map_from_child.each do |key, value|
+                    map[key] = value 
+                end
+            end 
+            node.outputs.each do |child|
+                if child.is_a? Edge 
+                    child = child.destination 
+                end
+                map_from_child = fan_out(child, max_depth, current_depth + 1)
+                map_from_child.each do |key, value|
+                    map[key] = value 
+                end
+            end 
+            map
+        end
     end
 end

data/lib/wads/version.rb

@@ -1,3 +1,3 @@
 module Wads
-    VERSION = "0.1.0"
+    VERSION = "0.1.1"
 end

data/lib/wads/widgets.rb

@@ -1,3 +1,5 @@
+require_relative 'data_structures'
+
 module Wads
     COLOR_PEACH = Gosu::Color.argb(0xffe6b0aa)
     COLOR_LIGHT_PURPLE = Gosu::Color.argb(0xffd7bde2)
@@ -28,8 +30,8 @@ module Wads
 
     Z_ORDER_BACKGROUND = 2
     Z_ORDER_WIDGET_BORDER = 3
-    Z_ORDER_GRAPHIC_ELEMENTS = 4
-    Z_ORDER_SELECTION_BACKGROUND = 5
+    Z_ORDER_SELECTION_BACKGROUND = 4
+    Z_ORDER_GRAPHIC_ELEMENTS = 5
     Z_ORDER_PLOT_POINTS = 6
     Z_ORDER_OVERLAY_BACKGROUND = 7
     Z_ORDER_OVERLAY_ELEMENTS = 8
@@ -94,6 +96,10 @@ module Wads
             @x + ((right_edge - @x) / 2)
         end 
 
+        def center_y
+            @y + ((bottom_edge - @y) / 2)
+        end 
+
         def draw 
             if @visible 
                 render
@@ -109,8 +115,13 @@ module Wads
             end 
         end
 
-        def draw_background 
-            Gosu::draw_rect(@x + 1, @y + 1, @width - 1, @height - 1, @background_color, Z_ORDER_BACKGROUND) 
+        def draw_background(z_order = Z_ORDER_BACKGROUND)
+            Gosu::draw_rect(@x + 1, @y + 1, @width - 3, @height - 3, @background_color, z_order) 
+        end
+
+        def draw_shadow(color, z_order = Z_ORDER_WIDGET_BORDER)
+            Gosu::draw_line @x - 1, @y - 1, color, right_edge - 1, @y - 1, color, z_order
+            Gosu::draw_line @x - 1, @y - 1, color, @x - 1, bottom_edge - 1, color, z_order
         end
 
         def render 
@@ -119,19 +130,35 @@ module Wads
             # render is for specific drawing done by the widget
         end 
 
-        def draw_border(color = nil)
+        def draw_border(color = nil, zorder = Z_ORDER_WIDGET_BORDER)
             if color.nil? 
-                color = @color 
+                if @border_color
+                    color = @border_color 
+                else
+                    color = @color 
+                end
             end
-            Gosu::draw_line @x, @y, color, right_edge, @y, color, Z_ORDER_WIDGET_BORDER
-            Gosu::draw_line @x, @y, color, @x, bottom_edge, color, Z_ORDER_WIDGET_BORDER
-            Gosu::draw_line @x,bottom_edge, color, right_edge, bottom_edge, color, Z_ORDER_WIDGET_BORDER
-            Gosu::draw_line right_edge, @y, color, right_edge, bottom_edge, color, Z_ORDER_WIDGET_BORDER
+            Gosu::draw_line @x, @y, color, right_edge, @y, color, zorder
+            Gosu::draw_line @x, @y, color, @x, bottom_edge, color, zorder
+            Gosu::draw_line @x,bottom_edge, color, right_edge, bottom_edge, color, zorder
+            Gosu::draw_line right_edge, @y, color, right_edge, bottom_edge, color, zorder
         end
 
         def contains_click(mouse_x, mouse_y)
             mouse_x >= @x and mouse_x <= right_edge and mouse_y >= @y and mouse_y <= bottom_edge
         end
+
+        def update update_count, mouse_x, mouse_y
+            # empty base implementation
+        end
+
+        def button_down id, mouse_x, mouse_y
+            # empty base implementation
+        end
+
+        def button_up id, mouse_x, mouse_y
+            # empty base implementation
+        end
     end 
 
     class Text < Widget
@@ -204,7 +231,7 @@ module Wads
         end
 
         def render 
-            draw_border(COLOR_WHITE)
+            draw_border(@color)
             text_x = center_x - (@text_pixel_width / 2)
             @font.draw_text(@label, text_x, @y, Z_ORDER_TEXT, 1, 1, @text_color)
         end 
@@ -456,7 +483,7 @@ module Wads
         end 
 
         def scroll_down
-            if @current_row < @data_rows.size - 1
+            if @current_row + @max_visible_rows < @data_rows.size
                 @current_row = @current_row + @max_visible_rows 
             end 
         end 
@@ -546,7 +573,13 @@ module Wads
         def set_selected_row(mouse_y, column_number)
             row_number = determine_row_number(mouse_y)
             if not row_number.nil?
-                @selected_row = @current_row + row_number
+                new_selected_row = @current_row + row_number
+                if @selected_row 
+                    if @selected_row == new_selected_row
+                        return nil  # You can't select the same row already selected
+                    end 
+                end
+                @selected_row = new_selected_row
                 @data_rows[@selected_row][column_number]
             end
         end
@@ -702,8 +735,6 @@ module Wads
                 data_set.derive_values(@visible_range, @data_set_hash)
                 data_set.data_points.each do |point|
                     if is_on_screen(point) 
-                        #puts "Adding render point at x #{point.x}, #{Time.at(point.x)}"
-                        #puts "Visible range: #{Time.at(@visible_range.left_x)}  #{Time.at(@visible_range.right_x)}"
                         data_set.add_rendered_point PlotPoint.new(draw_x(point.x), draw_y(point.y), data_set.color, data_set.data_point_size)
                     end
                 end
@@ -824,4 +855,143 @@ module Wads
             [get_x_data_val(mouse_x), get_y_data_val(mouse_y)]
         end 
     end 
+
+    class NodeWidget < Button
+        attr_accessor :data_node
+
+        def initialize(node, x, y, font, border_color = COLOR_DARK_GRAY, text_color = COLOR_HEADER_BRIGHT_BLUE) 
+            super(node.name, x, y, font, nil, border_color, text_color) 
+            set_background(COLOR_BLACK)
+            @data_node = node
+        end
+
+        def render 
+            super 
+            draw_background(Z_ORDER_OVERLAY_BACKGROUND)
+            draw_shadow(COLOR_GRAY)
+        end
+    end 
+
+    class GraphWidget < Widget
+        attr_accessor :graph
+        attr_accessor :center_node 
+        attr_accessor :depth 
+        attr_accessor :selected_node
+        attr_accessor :selected_node_x_offset
+        attr_accessor :selected_node_y_offset
+
+        def initialize(x, y, width, height, font, color, graph) 
+            super x, y, color 
+            set_font(font)
+            set_dimensions(width, height)
+            set_border(color)
+            @graph = graph 
+        end 
+
+        def update update_count, mouse_x, mouse_y
+            if contains_click(mouse_x, mouse_y) and @selected_node 
+                @selected_node.x = mouse_x - @selected_node_x_offset
+                @selected_node.y = mouse_y - @selected_node_y_offset
+            end
+        end
+
+        def button_down id, mouse_x, mouse_y
+            if id == Gosu::MsLeft
+                # check to see if any node was selected
+                if @rendered_nodes
+                    @rendered_nodes.values.each do |rn|
+                        if rn.contains_click(mouse_x, mouse_y)
+                            @selected_node = rn 
+                            @selected_node_x_offset = mouse_x - rn.x 
+                            @selected_node_y_offset = mouse_y - rn.y
+                        end
+                    end
+                end
+            end
+            WidgetResult.new(false)
+        end
+
+        def button_up id, mouse_x, mouse_y
+            if id == Gosu::MsLeft
+                if @selected_node 
+                    @selected_node = nil 
+                end 
+            end
+        end
+
+        def set_display(center_node, max_depth)
+            # Determine the list of nodes to draw
+            @depth = depth 
+            @visible_data_nodes = @graph.fan_out(center_node, max_depth)
+
+            # Convert the data nodes to rendered nodes
+            # Start by putting the center node in the center, then draw others around it
+            @rendered_nodes = {}
+            @rendered_nodes[center_node.name] = NodeWidget.new(center_node, center_x, center_y, @font,
+                    center_node.get_tag("color"), center_node.get_tag("color"))
+
+            # Spread out the other nodes around the center node
+            # going in a circle
+            number_of_visible_nodes = @visible_data_nodes.size 
+            radians_between_nodes = DEG_360 / number_of_visible_nodes.to_f
+            current_radians = 0.05
+
+            @visible_data_nodes.each do |node_name, data_node|
+                if node_name == center_node.name 
+                    # skip, we already got this one
+                else 
+                    # Use radians to spread the other nodes around the center node
+                    # For now, we will randomly place them
+                    node_x = center_x + ((80 + rand(200)) * Math.cos(current_radians))
+                    node_y = center_y - ((40 + rand(100)) * Math.sin(current_radians))
+                    if node_x < @x 
+                        node_x = @x + 1
+                    elsif node_x > right_edge - 20
+                        node_x = right_edge - 20
+                    end 
+                    if node_y < @y 
+                        node_y = @y + 1
+                    elsif node_y > bottom_edge - 26 
+                        node_y = bottom_edge - 26
+                    end
+                    current_radians = current_radians + radians_between_nodes
+
+                    # Note we can link between data nodes and rendered nodes using the node name
+                    # We have a map of each
+                    @rendered_nodes[data_node.name] = NodeWidget.new(
+                                                        data_node,
+                                                        node_x,
+                                                        node_y,
+                                                        @font,
+                                                        data_node.get_tag("color"),
+                                                        data_node.get_tag("color"))
+                end
+            end
+        end
+
+        def render 
+            if @rendered_nodes
+                @rendered_nodes.values.each do |vn|
+                    vn.draw 
+                end 
+                # Draw the connections between nodes 
+                @visible_data_nodes.values.each do |data_node|
+                    data_node.outputs.each do |connected_data_node|
+                        if connected_data_node.is_a? Edge 
+                            connected_data_node = connected_data_node.destination 
+                        end
+                        rendered_node = @rendered_nodes[data_node.name]
+                        connected_rendered_node = @rendered_nodes[connected_data_node.name]
+                        if connected_rendered_node.nil?
+                            # Don't draw if it is not currently visible
+                        else
+                            Gosu::draw_line rendered_node.center_x, rendered_node.center_y, rendered_node.color,
+                                    connected_rendered_node.center_x, connected_rendered_node.center_y, connected_rendered_node.color,
+                                    Z_ORDER_GRAPHIC_ELEMENTS
+                        end
+                    end
+                end 
+            end
+        end 
+    end
 end

data/sample_app.rb

@@ -10,8 +10,8 @@ class SampleAppCommand
         command ""
 
         example <<~EOS
-        Run the sample app to analyze stock market data
-          $ ./run-sample-app
+        Run the sample app to analyze stock market data in gui mode
+          $ ./run-sample-app -s -g
         EOS
 
     end
@@ -34,6 +34,12 @@ class SampleAppCommand
         desc "Run sample analysis of lottery numbers"
     end 
 
+    flag :jedi do 
+        short "-j"
+        long "--jedi"
+        desc "Run sample analysis of interactions between Star Wars"
+    end
+
     flag :gui do 
         short "-g"
         long "--gui"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: wads
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - dbroemme
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-08-26 00:00:00.000000000 Z
+date: 2021-08-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gosu
@@ -69,6 +69,7 @@ files:
 - bin/setup
 - data/NASDAQ.csv
 - data/Pick4_12_21_2020.txt
+- data/starwars-episode-4-interactions.json
 - lib/wads.rb
 - lib/wads/app.rb
 - lib/wads/data_structures.rb