oscilloscope 0.1.1

data/bin/oscilloscope

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+
+require 'oscilloscope'

data/lib/analog_input.rb

@@ -0,0 +1,70 @@
+class AnalogInput
+  attr_reader :data, :pin, :value, :window
+  attr_accessor :visible
+
+  COLORS = [Gosu::Color::GREEN,Gosu::Color::CYAN,Gosu::Color::YELLOW,Gosu::Color::FUCHSIA,Gosu::Color::BLUE,Gosu::Color::WHITE,Gosu::Color::GRAY]
+
+  def self.[](p)
+    i=@@inputs.select {|x| x.pin == p }.first
+    return i || nil
+  end
+
+  def self.all
+    @@inputs ||= []
+    @@inputs
+  end
+
+  def initialize(args={})
+    @@inputs ||=[]
+
+    @window = args[:window]
+    @pin = args[:pin]
+    @data = []
+    @value = 0
+    @visible = args[:visible]||false
+    @font_large ||= Gosu::Font.new(30)
+    @font_small ||= Gosu::Font.new(20)
+    @yscale = @window.yscale
+    @height = @window.height
+
+    @@inputs << self
+  end
+
+  def callback(v)
+    @value = v
+  end
+
+  def update
+    @data.shift if @data.count > (@window.scope_width / @window.update_interval)
+    @data << {
+      voltage: @value/204.6,
+      value: @value,
+      time: Gosu::milliseconds
+      }
+  end
+
+  def draw
+    draw_line if visible
+    draw_label
+  end
+
+  def draw_label
+    margin = 60*self.pin
+    @font_large.draw("Analog #{self.pin}",@window.scope_width+5,10+margin,0,1.0,1.0, COLORS[self.pin])
+    @font_small.draw("#{sprintf('%.2fv',@data[-1][:voltage])} #{ self.visible ? @data[-1][:value] : "(hidden)"}",@window.scope_width+5,40+margin,0,1.0,1.0, COLORS[self.pin])
+    return true
+  end
+
+  def draw_line
+    (0..self.data.length - 2).each do |i|
+      x1=self.data[i][:time] - self.data[0][:time]
+      y1=@height-(self.data[i][:voltage]*@yscale)
+
+      x2=@data[i+1][:time] - @data[0][:time]
+      y2=@height-(self.data[i+1][:voltage]*@yscale)
+
+      @window.draw_line(x1,y1,COLORS[self.pin],x2,y2,COLORS[self.pin])
+    end
+  end
+
+end

data/lib/arduino_reader.rb

@@ -0,0 +1,56 @@
+
+class ArduinoReader
+  def initialize(window)
+    @window = window
+
+    @inputs = []
+    connect
+    pwm
+    setup_callbacks
+  end
+
+  private
+  
+  def pwm
+    # Setup PWM on pin 9 for 50% cycle to produce a square wave for testing.
+    @arduino.analog_write 9,127
+  end
+
+  def connect
+    # Use first arduino found
+    @arduino = ArduinoFirmata.connect nil, :nonblock_io => true
+  end
+
+  def setup_callbacks
+    @arduino.on :analog_read do |pin, value|
+      AnalogInput.new(window: @window, pin: pin) unless AnalogInput[pin]
+      AnalogInput[pin].callback(value)
+    end
+  end
+
+end
+
+class FakeArduinoReader < ArduinoReader
+
+  def connect
+    @inputs << AnalogInput.new(window: @window, pin: 0)
+    @inputs << AnalogInput.new(window: @window, pin: 1)
+  end
+
+  def pwm
+    return true
+  end
+
+  def setup_callbacks
+    Thread.new {
+      loop do
+        AnalogInput[0].callback(0)
+        AnalogInput[1].callback(rand(1023))
+        sleep 0.09
+        AnalogInput[0].callback(1023)
+        sleep 0.09
+      end
+    }
+  end
+
+end

data/lib/display.rb

@@ -0,0 +1,62 @@
+class MinMaxText
+  def initialize(w,min=0,max=5)
+    @window = w
+    @width = @window.scope_width
+    @height = @window.height
+    @font = Gosu::Font.new(12)
+
+    @min = min
+    @max = max
+
+  end
+
+  def draw
+    @font.draw("#{@max}v", @width-20, 10, 0, 1.0, 1.0, 0xff_ffff00)
+    @font.draw("#{@min}v", @width-20, @height-20, 0, 1.0, 1.0, 0xff_ffff00)
+  end
+end
+
+class CenterLine
+  def initialize(w)
+    @window = w
+    @interval = @window.update_interval
+    @height = @window.height
+    @width = @window.scope_width
+    @font = Gosu::Font.new(12)
+  end
+
+  def draw
+    line
+    time
+  end
+
+  private
+
+  def line
+    @window.draw_line(0,@height/2,Gosu::Color::RED, @width,@height/2,Gosu::Color::RED)
+  end
+
+  def time
+    (0..@width/@interval).each do |i|
+      t=(@interval*i)
+      @font.draw("#{t.to_i}", @width-8-(@width/@interval)*i,@height/2+10,0,1.0,1.0,Gosu::Color::GRAY)
+    end
+  end
+end
+
+class RightLine
+  def initialize(w)
+    @window = w
+    @width = @window.scope_width
+    @height = @window.height
+  end
+
+  def draw
+    line
+  end
+
+  private
+  def line
+    @window.draw_line(@width,0,Gosu::Color::RED, @width, @height,Gosu::Color::RED)
+  end
+end

data/lib/oscilloscope.rb

@@ -0,0 +1,10 @@
+require "gosu"
+require "arduino_firmata"
+
+require 'display.rb'
+require 'arduino_reader.rb'
+require 'analog_input.rb'
+require 'window.rb'
+
+window = ScopeWindow.new
+window.show

data/lib/window.rb

@@ -0,0 +1,46 @@
+class ScopeWindow < Gosu::Window
+  attr_reader :max_volts, :yscale, :clock, :scope_width
+  attr_accessor :display_pins
+  def initialize
+    super 1024+124,768
+
+    @scope_width = 1023
+    self.caption = "Oscilloscope"
+
+    @font = Gosu::Font.new(18)
+
+    @max_volts = 5
+    @yscale=(self.height/@max_volts)
+    @paused = false
+
+    @center_line = CenterLine.new(self)
+    @minmaxtext = MinMaxText.new(self)
+    @right_line = RightLine.new(self)
+    @ar = ArduinoReader.new(self)
+    # @ar = FakeArduinoReader.new(self)
+    AnalogInput[0].visible = true
+  end
+
+  def button_down(id)
+    close if id == Gosu::KbEscape or id == Gosu::KbQ
+    @paused = ! @paused if id == Gosu::KbP
+
+    [Gosu::Kb0,Gosu::Kb1,Gosu::Kb2,Gosu::Kb3,Gosu::Kb4,Gosu::Kb5].each_with_index {|b,i|
+        AnalogInput[i].visible = ! AnalogInput[i].visible if AnalogInput[i] if id == b
+    }
+
+  end
+
+  def update
+    return if @paused
+    AnalogInput.all.each {|i| i.update }
+  end
+
+  def draw
+    # Draw interface
+    @center_line.draw
+    @right_line.draw
+    @minmaxtext.draw
+    AnalogInput.all.each {|i| i.draw }
+  end
+end

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification
+name: oscilloscope
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+  prerelease: 
+platform: ruby
+authors:
+- James Paterni
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-11-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: gosu
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+- !ruby/object:Gem::Dependency
+  name: arduino_firmata
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.7
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.7
+description: A poor man oscilloscope using ruby and an arduino.
+email: james@ruby-code.com
+executables:
+- oscilloscope
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/analog_input.rb
+- lib/arduino_reader.rb
+- lib/display.rb
+- lib/oscilloscope.rb
+- lib/window.rb
+- bin/oscilloscope
+homepage: 
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Osilloscope
+test_files: []