wads 0.1.0

data/lib/wads/app.rb

@@ -0,0 +1,199 @@
+require 'gosu'
+require_relative 'data_structures'
+require_relative 'widgets'
+require_relative 'version'
+
+include Wads
+
+class WadsSampleApp < Gosu::Window
+    
+    STOCKS_DATA_FILE = "./data/NASDAQ.csv"
+    LOTTERY_DATA_FILE = "./data/Pick4_12_21_2020.txt"
+
+    def initialize
+        super(800, 600)
+        self.caption = "Wads Sample App"
+        @font = Gosu::Font.new(32)
+        @title_font = Gosu::Font.new(38)
+        @version_font = Gosu::Font.new(22)
+        @small_font = Gosu::Font.new(22)
+        @banner_image = Gosu::Image.new("./media/Banner.png")
+        @display_widget = nil
+    end 
+
+    def parse_opts_and_run 
+        # Make help the default output if no args are specified
+        if ARGV.length == 0
+            ARGV[0] = "-h"
+        end
+
+        opts = SampleAppCommand.new.parse.run
+        if opts[:stocks]
+            stats = process_stock_data
+            if opts[:gui]
+                @display_widget = SampleStocksDisplay.new(@small_font, stats)
+                show
+            else 
+                stats.report(Date::DAYNAMES[1..5])
+            end
+        
+        elsif opts[:lottery] 
+            process_lottery_data
+        else
+            puts " "
+            puts "Select one of the following sample analysis options"
+            puts "-s   Run sample stocks analysis"
+            puts "-l   Run sample analysis of lottery numbers"
+            puts " "
+            exit
+        end
+
+    end
+
+    def update 
+        # TODO
+    end 
+    
+    def draw
+        draw_banner
+        @display_widget.draw
+    end 
+
+    def draw_banner 
+        @banner_image.draw(1,1,1,0.9,0.9)
+        @title_font.draw_text("Wads Sample App", 10, 20, 2, 1, 1, Gosu::Color::WHITE)
+        @version_font.draw_text("Version #{Wads::VERSION}", 13, 54, 2, 1, 1, Gosu::Color::WHITE)
+    end
+
+    def button_down id
+        close if id == Gosu::KbEscape
+        # Delegate button events to the primary display widget
+        result = @display_widget.button_down id, mouse_x, mouse_y
+        if result.close_widget
+            close
+        end
+    end
+
+    def process_stock_data 
+        # The data file comes from https://finance.yahoo.com
+        # The format of this file is as follows:
+        #
+        # Date,Open,High,Low,Close,Adj Close,Volume
+        # 2000-01-03,4186.189941,4192.189941,3989.709961,4131.149902,4131.149902,1510070000
+        # 2000-01-04,4020.000000,4073.250000,3898.229980,3901.689941,3901.689941,1511840000
+        # ...
+        # 2020-12-30,12906.509766,12924.929688,12857.759766,12870.000000,12870.000000,5292210000
+        # 2020-12-31,12877.089844,12902.070313,12821.230469,12888.280273,12888.280273,4771390000
+        
+        stats = Stats.new("NASDAQ")
+        previous_close = nil
+        
+        puts "Read the data file #{STOCKS_DATA_FILE}"
+        File.readlines(STOCKS_DATA_FILE).each do |line|
+            line = line.chomp  # remove the carriage return
+
+            # Ignore header and any empty lines, process numeric data lines
+            if line.length > 0 and line[0].match(/[0-9]/)
+                values = line.split(",")
+                date = Date.strptime(values[0], "%Y-%m-%d")
+                weekday = Date::DAYNAMES[date.wday]
+
+                open_value = values[1].to_f
+                close_value = values[4].to_f
+
+                if previous_close.nil?
+                    # Just use the first day to set the baseline
+                    previous_close = close_value
+                else 
+                    change_from_previous_close = close_value - previous_close
+                    change_intraday = close_value - open_value
+                    change_overnight = open_value - previous_close
+                    
+                    change_percent = change_from_previous_close / previous_close
+
+                    stats.add(weekday, change_percent)
+                    stats.add("#{weekday} prev close", change_from_previous_close)
+                    stats.add("#{weekday} intraday", change_intraday)
+                    stats.add("#{weekday} overnight", change_overnight)
+
+                    previous_close = close_value
+                end
+            end
+        end
+
+        stats
+    end
+
+    def process_lottery_data
+        puts "The lottery example has not been implemented yet"
+        puts "however the data is available in the data directory,"
+        puts "and the same pattern used in the stocks example can"
+        puts "be applied here."
+    end
+end
+
+class SampleStocksDisplay < Widget
+    attr_accessor :stats
+
+    def initialize(font, stats)
+        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)
+
+        @stats = stats
+        @data_table = SingleSelectTable.new(@x + 5, @y + 100,    # top left corner
+                                            770, 200,            # width, height
+            ["Day", "Min", "Avg", "StdDev", "Max", "p10", "p90"], # column headers
+                                            @font, COLOR_WHITE)   # font and text color
+        @data_table.selected_color = COLOR_LIGHT_GRAY
+        Date::DAYNAMES[1..5].each do |day|
+            min = format_percent(@stats.min(day))
+            avg = format_percent(@stats.average(day))
+            std = format_percent(@stats.std_dev(day))
+            max = format_percent(@stats.max(day))
+            p10 = format_percent(@stats.percentile(day, 0.1))
+            p90 = format_percent(@stats.percentile(day, 0.90))
+            @data_table.add_row([day, min, avg, std, max, p10, p90], COLOR_HEADER_BRIGHT_BLUE)
+        end
+        add_child(@data_table)
+        @selection_text = nil
+    end 
+
+    def format_percent(val)
+        "#{(val * 100).round(3)}%"
+    end
+
+    def sample_content
+        <<~HEREDOC
+          This sample stock analysis uses NASDAQ data from https://finance.yahoo.com looking
+          at closing data through the years 2000 to 2020. The percent gain or loss is broken
+          down per day, as shown in the table below.
+        HEREDOC
+    end
+
+    def render 
+        if @selection_text
+            @selection_text.draw 
+        end
+    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
+                    @selection_text = Text.new("You selected #{val}, a great day!",
+                                               x + 5, y + 400, @font)
+                end    
+            end 
+        end 
+        WidgetResult.new(false)
+    end 
+end

data/lib/wads/data_structures.rb

@@ -0,0 +1,251 @@
+require 'date'
+
+module Wads
+
+    SPACER = "  "
+    VALUE_WIDTH = 10
+    NODE_UNKNOWN = "undefined"
+
+    class HashOfHashes 
+        attr_accessor :data
+
+        def initialize
+            @data = {}
+        end
+
+        def set(data_set_name, x, y)
+            data_set = @data[x]
+            if data_set.nil? 
+                data_set = {}
+                @data[x] = data_set
+            end
+            data_set[x] = y
+        end
+
+        def get(data_set_name, x)
+            data_set = @data[x]
+            if data_set.nil? 
+                return nil
+            end
+            data_set[x]
+        end
+    end 
+
+    class Stats
+        attr_accessor :name
+        attr_accessor :data
+
+        def initialize(name)
+            @name = name
+            @data = {}
+        end
+        
+        def add(key, value)
+            data_set = @data[key]
+            if data_set
+                data_set << value
+            else
+                data_set = []
+                data_set << value
+                @data[key] = data_set
+            end     
+        end
+        
+        def increment(key)
+            add(key, 1)
+        end
+        
+        def count(key)
+            data_set = @data[key]
+            if data_set
+                return data_set.size
+            else
+                return 0
+            end
+        end
+
+        def sum(key)
+            data_set = @data[key]
+            if data_set
+                return data_set.inject(0.to_f){|sum,x| sum + x }
+            else
+                return 0
+            end
+        end
+        
+        def average(key)
+            data_set = @data[key]
+            if data_set
+                return (sum(key) / count(key)).round(5)
+            else
+                return 0
+            end
+        end
+        
+        def min(key)
+            data_set = @data[key]
+            if data_set
+                return data_set.min
+            else
+                return 0
+            end
+        end
+
+        def max(key)
+            data_set = @data[key]
+            if data_set
+                return data_set.max
+            else
+                return 0
+            end
+        end
+
+        def sample_variance(key)
+            data_set = @data[key]
+            return 0 unless data_set
+            m = average(key)
+            s = data_set.inject(0.0){|accum, i| accum +(i-m)**2 }
+            s/(data_set.length - 1).to_f
+        end
+
+        def std_dev(key)
+            Math.sqrt(sample_variance(key))
+        end
+
+        def halfway(b, e)
+            d = e - b
+            m = b + (d / 2).to_i
+            m
+        end 
+
+        def percentile(key, pct)
+            data_set = @data[key]
+            if data_set
+                sorted_data_set = data_set.sort
+                pct_index = (data_set.length - 1).to_f * pct
+                mod = pct_index.modulo(1.0)
+                adj = pct_index
+                if mod > 0.9
+                    adj = pct_index.ceil
+                elsif mod < 0.1
+                    adj = pct_index.floor
+                else
+                    # We want halfway between the two indices
+                    low = pct_index.floor
+                    high = pct_index.ceil
+                    if low < 0
+                        low = 0
+                    end
+                    if high > data_set.size - 1
+                        high = data_set.size - 1
+                    end
+                    result = halfway(sorted_data_set[low], sorted_data_set[high])
+                    return result
+                end
+                
+                if adj < 0
+                    adj = 0
+                elsif adj > data_set.size - 1
+                    adj = data_set.size - 1
+                end
+                result = sorted_data_set[adj]
+                return result
+            else
+                return 0
+            end
+        end
+
+        def most_common(key)
+            value_counts = {}
+            data_set = @data[key]
+            data_set.each do |data|
+                c = value_counts[data]
+                if c.nil?
+                    value_counts[data] = 1
+                else
+                    value_counts[data] = value_counts[data] + 1
+                end
+            end
+            
+            largest_count = 0
+            largest_key = "none"
+            value_counts.keys.each do |key|
+                key_count = value_counts[key]
+                if key_count > largest_count
+                    largest_count = key_count
+                    largest_key = key
+                end
+            end
+            largest_key
+        end
+
+        def pad(str, size, left_align = false)
+            str = str.to_s
+            if left_align
+                str[0, size].ljust(size, ' ')
+            else
+                str[0, size].rjust(size, ' ')
+            end
+        end        
+
+        def display_counts
+            puts "#{pad(@name, 20)}   Value"
+            puts "#{'-' * 20}   #{'-' * 10}"
+            @data.keys.each do |key|
+                #data_set = @data[key]
+                puts "#{pad(key, 20)}   #{count(key)}"
+            end
+        end
+        
+        def keys 
+            @data.keys 
+        end 
+
+        def report(report_keys = keys)
+            puts "#{pad(@name, 10)}#{SPACER}#{pad('Count', 7)}#{SPACER}#{pad('Min', VALUE_WIDTH)}#{SPACER}#{pad('Avg', VALUE_WIDTH)}#{SPACER}#{pad('StdDev', VALUE_WIDTH)}#{SPACER}#{pad('Max', VALUE_WIDTH)}#{SPACER}| #{pad('p1', VALUE_WIDTH)}#{SPACER}#{pad('p10', VALUE_WIDTH)}#{SPACER}#{pad('p50', VALUE_WIDTH)}#{SPACER}#{pad('p90', VALUE_WIDTH)}#{SPACER}#{pad('p99', VALUE_WIDTH)}"
+            puts "#{'-' * 10}#{SPACER}#{'-' * 7}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}| #{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}#{SPACER}#{'-' * VALUE_WIDTH}"
+            if report_keys.nil?
+                report_keys = @data.keys
+            end
+            report_keys.each do |key|
+                data_set = @data[key]
+                m1 = min(key).round(5)
+                a = average(key).round(5)
+                sd = std_dev(key).round(5)
+                m2 = max(key).round(5)
+                p1 = percentile(key, 0.01).round(5)
+                p10 = percentile(key, 0.1).round(5)
+                p50 = percentile(key, 0.5).round(5)
+                p90 = percentile(key, 0.90).round(5)
+                p99 = percentile(key, 0.99).round(5)
+                puts "#{pad(key, 10)}#{SPACER}#{pad(count(key), 7)}#{SPACER}#{pad(m1, VALUE_WIDTH)}#{SPACER}#{pad(a, VALUE_WIDTH)}#{SPACER}#{pad(sd, VALUE_WIDTH)}#{SPACER}#{pad(m2, VALUE_WIDTH)}#{SPACER}| #{pad(p1, VALUE_WIDTH)}#{SPACER}#{pad(p10, VALUE_WIDTH)}#{SPACER}#{pad(p50, VALUE_WIDTH)}#{SPACER}#{pad(p90, VALUE_WIDTH)}#{SPACER}#{pad(p99, VALUE_WIDTH)}"
+            end
+            
+        end
+    end
+
+    class Node
+        attr_accessor :x
+        attr_accessor :y
+        attr_accessor :name
+        attr_accessor :type
+        attr_accessor :inputs
+        attr_accessor :outputs
+        attr_accessor :visited
+
+        def initialize(name, type = NODE_UNKNOWN)
+            @name = name
+            @type = type
+            @inputs = [] 
+            @outputs = []
+            @visited = false
+        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
+    end
+end