trixter 1.0.0

data/bin/trixter

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+require 'trixter'
+
+port = ARGV[0]
+if port == nil
+  port = '/dev/tty.usbserial'
+end
+trix = Trixter.new(port)
+trix.run
+trix.join

data/lib/trixter.rb

@@ -0,0 +1 @@
+require 'trixter/trixter'

data/lib/trixter/difficulties.rb

@@ -0,0 +1,20 @@
+# These rates could be dynamically calculated
+Difficulties = [
+  "6a013d5b7974",
+  "6a114d6b89d4",
+  "6a215d7b99f4",
+  "6a316d8ba914",
+  "6a417d9bb974",
+  "6a518dabc9d4",
+  "6a619dbbd9f4",
+  "6a71adcbe994",
+  "6a81bddbf974",
+  "6a91cdeb0ad7",
+  "6aa1ddfb1af7",
+  "6ab1ed0c2a10",
+  "6ac1fd1c3a70",
+  "6ad10e2c4ad3",
+  "6ae11e3c5af3",
+  "6af12e4c6a93",
+  "6aff3c5a788b"
+]

data/lib/trixter/event.rb

@@ -0,0 +1,83 @@
+require 'time'
+
+class Event
+  attr_reader :time
+
+  def initialize(bytes)
+    @bytes = bytes
+    @time = Time.now.to_f
+  end
+
+  def crankPosition
+    return @bytes[6..7]
+  end
+
+  def rightBrakePressure
+    return @bytes[8..9]
+  end
+
+  def leftBrakePressure
+    return @bytes[10..11]
+  end
+
+  def leftGearUp
+    return @bytes[16] == '7'
+  end
+
+  def leftGearDown
+    return @bytes[18] == '7'
+  end
+
+  def leftControlLeft
+    return @bytes[16] == 'e'
+  end
+
+  def leftControlRight
+    return @bytes[16] == 'b'
+  end
+
+  def leftControlDown
+    return @bytes[17] == 'b'
+  end
+
+  def leftControlUp
+    return @bytes[17] == 'e'
+  end
+
+  def rightControlLeft
+    return @bytes[17] == 'd'
+  end
+
+  def rightControlRight
+    return @bytes[17] == '7'
+  end
+
+  def rightControlUp
+    return @bytes[16] == 'd'
+  end
+
+  def rightGearUp
+    return @bytes[18] == 'd'
+  end
+  
+  def rightGearDown
+    return @bytes[18] == 'b'
+  end
+
+  def seated
+    return @bytes[19] == '7'
+  end
+
+  def flywheelRotationTime
+    return @bytes[24..27]
+  end
+end
+
+class CrankChange
+  attr_accessor :ticks, :time, :elapsed
+  def initialize(ticks, time, elapsed)
+    @ticks = ticks
+    @time = time
+    @elapsed = elapsed
+  end
+end

data/lib/trixter/handler.rb

@@ -0,0 +1,177 @@
+# This handler, by default, does no more than log
+class EventHandler
+  def initialize
+    @current_event = nil
+    @crank_changes = []
+  end
+
+  def handleEvent(event)
+    # We throw out initial state
+    if @current_event == nil
+      @current_event = event
+      return
+    end
+
+    # Process boolean-on events (we only care when they are 'on')
+    if event.leftGearUp and not @current_event.leftGearUp
+      leftGearUp
+    end
+    if event.leftGearDown and not @current_event.leftGearDown
+      leftGearDown
+    end
+    if event.rightGearUp and not @current_event.rightGearUp
+      rightGearUp
+    end
+    if event.rightGearDown and not @current_event.rightGearDown
+      rightGearDown
+    end
+    if event.leftControlUp and not @current_event.leftControlUp
+      leftControlUp
+    end
+    if event.leftControlDown and not @current_event.leftControlDown
+      leftControlDown
+    end
+    if event.leftControlLeft and not @current_event.leftControlLeft
+      leftControlLeft
+    end
+    if event.leftControlRight and not @current_event.leftControlRight
+      leftControlRight
+    end
+    if event.rightControlLeft and not @current_event.rightControlLeft
+      rightControlLeft
+    end
+    if event.rightControlRight and not @current_event.rightControlRight
+      rightControlRight
+    end
+    if event.rightControlUp and not @current_event.rightControlUp
+      rightControlUp
+    end
+
+    # Process scaled events
+    if event.leftBrakePressure != @current_event.leftBrakePressure
+      leftBrakeChanged(event.leftBrakePressure)
+    end
+    if event.rightBrakePressure != @current_event.rightBrakePressure
+      rightBrakeChanged(event.rightBrakePressure)
+    end
+    if event.crankPosition != @current_event.crankPosition
+      crankPositionChanged(event)
+    end
+
+    # Seated or not
+    if event.seated != @current_event.seated
+      seatingChanged(event.seated)
+    end
+
+    @current_event = event
+  end
+
+  def seatingChanged(seated)
+    if seated
+      puts "Rider is seated"
+    else
+      puts "Rider is standing"
+    end
+  end
+
+  def crankPositionChanged(event)
+    if @last_crank_change == nil
+      @last_crank_change = event.time
+      return
+    end
+
+    forward = false
+    ticks = 0
+    past_position = @current_event.crankPosition.hex
+    current_position = event.crankPosition.hex
+    if past_position < current_position
+      ticks = current_position - past_position
+      if ticks < 30
+        forward = true
+      else
+        ticks = 60 - ticks
+      end
+    else
+      ticks = past_position - current_position
+      if ticks > 30
+        ticks = 60 - ticks
+        forward = true
+      end
+    end
+
+    change = CrankChange.new(ticks, event.time, event.time - @last_crank_change)
+    @crank_changes << change
+    to_delete = []
+    while (event.time - @crank_changes.first.time) > 0.4
+      @crank_changes.delete_at(0)
+    end
+
+    smoothed_ticks = 0
+    for change in @crank_changes
+      smoothed_ticks += change.ticks
+    end
+
+    first_change = @crank_changes.first
+
+    if event.time - first_change.time == 0
+      puts "Crank RPM: 0"
+    else
+      rpms = (smoothed_ticks/60.0) * (60.0/(event.time - first_change.time))
+      puts "Crank RPM: #{rpms}"
+    end  
+    
+    @last_crank_change = event.time
+  end
+
+  def leftBrakeChanged(strength)
+    puts "Left brake engaged at #{strength}"
+  end
+
+  def rightBrakeChanged(strength)
+    puts "Right brake engaged at #{strength}"
+  end
+
+  def leftGearUp
+    puts "Left gear up!"
+  end
+
+  def leftGearDown
+    puts "Left gear down!"
+  end
+
+  def rightGearUp
+    puts "Right gear up!"
+  end
+
+  def rightGearDown
+    puts "Right gear down!"
+  end
+
+  def leftControlLeft
+    puts "Left control left pushed"
+  end
+
+  def leftControlRight
+    puts "Left control right pushed"
+  end
+
+  def leftControlUp
+    puts "Left control up pushed"
+  end
+
+  def leftControlDown
+    puts "Left control down pushed"
+  end
+
+  def rightControlLeft
+    puts "Right control left pushed"
+  end
+
+  def rightControlRight
+    puts "Right control right pushed"
+  end
+
+  def rightControlUp
+    puts "Right control up pushed"
+  end
+end

data/lib/trixter/trixter.rb

@@ -0,0 +1,144 @@
+#!/usr/bin/env ruby
+
+require 'serialport'
+require 'trixter/handler'
+require 'trixter/difficulties'
+require 'trixter/event'
+
+class Trixter < EventHandler
+  def initialize(port)
+    super()
+
+    @port = SerialPort.new(port, 115200, 8, 1)
+    @status_thread = nil
+    @control_thread = nil
+    @current_difficulty = 0
+    @saved_difficulty = 0
+    @right_brake = false
+    @left_brake = false
+
+    @should_run = false
+  end
+
+  def run
+    return if @should_run or @status_thread != nil
+    return if @should_run or @control_thread != nil
+
+    @should_run = true
+    @status_thread = Thread.new do
+      status
+    end
+    @control_thread = Thread.new do
+      control
+    end
+  end
+
+  def stop
+    @should_run = false
+  end
+
+  def rightGearUp
+    @current_difficulty += 1
+    if @current_difficulty >= Difficulties.size
+      @current_difficulty = Difficulties.size - 1
+    end
+    puts "Increased difficulty to level #{@current_difficulty}"
+  end
+
+  def rightGearDown
+    @current_difficulty -= 1
+    if @current_difficulty < 0
+      @current_difficulty = 0
+    end
+    puts "Decreased difficulty to level #{@current_difficulty}"
+  end
+
+  def leftBrakeChanged(strength)
+    pressure = 240 - strength.hex
+    if pressure > 10
+      if !@left_brake and !@right_brake
+        puts "Left brake engaged"
+        @saved_difficulty = @current_difficulty
+      end
+
+      @left_brake = true
+      applyBrake(pressure)
+    else
+      if @left_brake
+        puts "Left brake disengaged"
+        # restore selected difficulty
+        @left_brake = false
+        if !@right_brake
+          @current_difficulty = @saved_difficulty
+        end
+      end
+    end
+  end
+
+  def rightBrakeChanged(strength)
+    pressure = 240 - strength.hex
+    if pressure > 10
+      if !@left_brake and !@right_brake
+        puts "Right brake engaged"
+        @saved_difficulty = @current_difficulty
+      end
+
+      @right_brake = true
+      applyBrake(pressure)
+    else
+      if @right_brake
+        puts "Right brake disengaged"
+        # restore selected difficulty
+        @right_brake = false
+        if !@left_brake
+          @current_difficulty = @saved_difficulty
+        end
+      end
+    end
+  end
+
+  def applyBrake(pressure)
+    span = Difficulties.size - @saved_difficulty - 1
+    units = 100.0/span
+    new_difficulty = (pressure/units).to_i + @saved_difficulty
+    @current_difficulty = [new_difficulty, Difficulties.size - 1].min
+  end
+
+  def crankPositionChanged(event)
+    # pass
+  end
+
+  def status
+    while @should_run
+      raw_event = ""
+      while true
+        raw_event << @port.read(1).strip
+        if raw_event.size == 1
+          raw_event = "" unless raw_event[0] == '6'
+        elsif raw_event.size == 2
+          raw_event == "" unless raw_event[1] == 'a'
+        elsif raw_event.size > 2
+          raw_event == "" unless raw_event[0..1] == '6a'
+        end
+        break if raw_event.size == 32
+      end
+      handleEvent(Event.new(raw_event))
+    end
+  end
+
+  def control
+    while @should_run
+      sleep 0.01
+      @port.write([Difficulties[@current_difficulty]].pack("H*"))
+    end
+  end
+
+  def join
+    if @status_thread != nil
+      @status_thread.join
+    end
+    if @control_thread != nil
+      @control_thread.join
+    end
+  end
+end

metadata

@@ -0,0 +1,52 @@
+--- !ruby/object:Gem::Specification
+name: trixter
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Tom Metge
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-13 00:00:00.000000000 Z
+dependencies: []
+description: Turns the Trixter Dream Bike into a real training platform
+email: tom@accident-prone.com
+executables:
+- trixter
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/trixter.rb
+- lib/trixter/difficulties.rb
+- lib/trixter/event.rb
+- lib/trixter/handler.rb
+- lib/trixter/trixter.rb
+- bin/trixter
+homepage: http://www.accident-prone.com
+licenses:
+- MIT
+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.23
+signing_key: 
+specification_version: 3
+summary: Trixter Dream Bike Controller
+test_files: []