arbol 0.0.2

data/lib/functions/create_lookup.rb

@@ -0,0 +1,63 @@
+$tables = {}
+
+class CreateLookup < Base
+  Arbol.add_mapped_class(
+    'create_lookup', 
+    CreateLookup, 
+    nil
+  )
+  
+  attr_accessor :identifier
+  attr_accessor :value
+  
+  def initialize(params)
+    super(params)
+    @identifier = params[:identifier]
+    unless $tables.has_key?(@identifier)
+      $tables[@identifier] = params[:value]
+      @value = params[:value]
+    else
+      raise "duplicate table definition #{params[:name]}"
+    end
+  end
+
+  def param_keys
+    []
+  end
+  
+  def arduino_code
+    []
+  end
+
+  def top_level_scope_code
+    [
+      "long #{@identifier}[#{@value.length}][3] = #{table_to_cplusplus_array(@value)};"
+    ]
+  end
+  
+  def table_to_cplusplus_array(t)
+    "{#{t.map {|f| f.to_s.gsub(/\[/, '{').gsub(/\]/, '}') }.join(',')}}"
+  end
+end
+
+
+def resolve_lookup(val)
+  puts val
+  puts val.class
+  raise "table definition must be an array" unless val.class == Array
+  val.map do |v|
+    case
+      when [Integer, Float].include?(v.class)
+        then 3.times.map { scale_correctly(v) }
+      when v.class == Array then v.map { |i| scale_correctly(i) }
+    end
+  end
+end
+
+def create_table_ref(identifier, value)
+  h = ArbolHash.new
+  h[:type] = 'create_lookup'
+  h[:identifier] = identifier
+  h[:value] = resolve_lookup(value)
+  h
+end

data/lib/functions/create_ref.rb

@@ -0,0 +1,48 @@
+$refs = {}
+$refs_frame_optimized = {}
+
+class CreateRef < Base
+  Arbol.add_mapped_class('create_ref', CreateRef, nil)
+  attr_accessor :identifier
+  attr_accessor :value
+  
+  def initialize(params)
+    super(params)
+    @identifier = params[:identifier]
+    resolve_frame_optimized
+    unless $refs.has_key?(@identifier)
+      $refs[@identifier] = @name
+      $refs_frame_optimized[@identifier] = @frame_optimized
+    else
+      raise "duplicate ref definition #{params[:name]}"
+    end
+  end
+
+  def param_keys
+    [:value]
+  end
+  
+  def arduino_code
+    []
+  end
+
+  def top_level_scope_code
+    [
+      "long *#{@name} = #{@value.name};"
+    ]
+  end
+end
+
+def create_ref(identifier, value)
+  h = ArbolHash.new
+  if value.class == ArbolTable
+    h[:type] = 'create_lookup'
+    h[:identifier] = identifier
+    h[:value] = resolve_lookup(value.table)
+  else
+    h[:type] = 'create_ref'
+    h[:identifier] = identifier
+    h[:value] = resolve(value)
+  end
+  h
+end

data/lib/functions/crossfade.rb

@@ -0,0 +1,73 @@
+class Crossfade < Base
+  Arbol.add_mapped_class(
+    'crossfade', 
+    Crossfade,
+%{void crossfade(long fader[3], long channel1[3], long channel2[3], long out[3]) {
+  out[0] = long_mult(INTEGER_SCALE - 1 - fader[0], channel1[0]) + long_mult(fader[0], channel2[0]);
+  out[1] = long_mult(INTEGER_SCALE - 1 - fader[1], channel1[1]) + long_mult(fader[1], channel2[1]);
+  out[2] = long_mult(INTEGER_SCALE - 1 - fader[2], channel1[2]) + long_mult(fader[2], channel2[2]);
+}}
+  )
+  
+  attr_accessor :fader
+  attr_accessor :channel1
+  attr_accessor :channel2
+  
+  def param_keys
+    [:fader, :channel1, :channel2]
+  end
+
+  def arduino_code
+    unless @frame_optimized 
+      [
+        "crossfade(#{@fader.name}, #{@channel1.name}, #{@channel2.name}, #{@name});"
+      ]
+    else
+      []
+    end
+  end
+  
+  def cycle_level_arduino_code
+    if @frame_optimized
+      [
+        "crossfade(#{@fader.name}, #{@channel1.name}, #{@channel2.name}, #{@name});"
+      ]
+    else
+      []
+    end
+  end
+  
+  def top_level_scope_code
+    [
+      "long #{@name}[3];",
+    ]
+  end
+end
+
+module Arbol
+  class Documentation
+
+  def crossfade   
+%{--
+### crossfade(fader, channel1, channel2)
+
+* **fader** - fade amount between channels
+* **channel1** - channel1
+* **channel2** - channel2
+
+Returns a mix of channels 1 and 2 based on the fader amount between 0-~1.0.
+
+}
+  end
+
+  end
+end
+
+def crossfade(fader, channel1, channel2)
+  h = ArbolHash.new
+  h[:type] = 'crossfade'
+  h[:fader] = resolve(fader)
+  h[:channel1] = resolve(channel1)
+  h[:channel2] = resolve(channel2)
+  h
+end

data/lib/functions/divide.rb

@@ -0,0 +1,70 @@
+class Divide < Base
+  Arbol.add_mapped_class(
+    'divide', 
+    Divide,
+%{void divide(long numerator[3], long denominator[3], long out[3]) {
+  out[0] = long_div(numerator[0], denominator[0]);
+  out[1] = long_div(numerator[1], denominator[1]);
+  out[2] = long_div(numerator[2], denominator[2]);
+}}
+  )
+  
+  attr_accessor :numerator
+  attr_accessor :denominator
+  
+  def param_keys
+    [:numerator, :denominator]
+  end
+  
+  def arduino_code
+    unless @frame_optimized 
+      [
+        "divide(#{@numerator.name}, #{@denominator.name}, #{@name});"
+      ]
+    else
+      []
+    end
+  end
+  
+  def cycle_level_arduino_code
+    if @frame_optimized
+      [
+        "divide(#{@numerator.name}, #{@denominator.name}, #{@name});"
+      ]
+    else
+      []
+    end
+  end
+  
+  def top_level_scope_code
+    [
+      "long #{@name}[3];"
+    ]
+  end
+end
+
+module Arbol
+  class Documentation
+
+  def divide   
+%{--
+### divide(numerator, denominator)
+
+* **numerator**
+* **denominator**
+
+Division. Also accepts the form `numerator / denominator`.
+
+}
+  end
+
+  end
+end
+
+def divide(numerator, denominator)
+  h = ArbolHash.new
+  h[:type] = 'add'
+  h[:numerator] = resolve(numerator)
+  h[:denominator] = resolve(denominator)
+  h
+end

data/lib/functions/feedback.rb

@@ -0,0 +1,65 @@
+class Feedback < Base
+  Arbol.add_mapped_class(
+    'feedback', 
+    Feedback,
+%{void feedback(int pixel, long input[3], long feedback[3], long storage[][3], long out[3]) {
+  out[0] = max(input[0], storage[pixel][0]);
+  out[1] = max(input[1], storage[pixel][1]);
+  out[2] = max(input[2], storage[pixel][2]);
+  storage[pixel][0] = constrain(long_mult(out[0], feedback[0]), 0, INTEGER_SCALE);
+  storage[pixel][1] = constrain(long_mult(out[1], feedback[1]), 0, INTEGER_SCALE);
+  storage[pixel][2] = constrain(long_mult(out[2], feedback[2]), 0, INTEGER_SCALE);
+}}
+  )
+  
+  attr_accessor :input
+  attr_accessor :feedback
+
+  def initialize(params)
+    super(params)
+    @frame_optimized = false
+  end
+ 
+  def param_keys
+    [:input, :feedback]
+  end
+  
+  def arduino_code
+    [
+      "feedback(i, #{@input.name}, #{@feedback.name}, #{@name}_storage, #{@name});"
+    ]
+  end
+
+  def top_level_scope_code
+    [
+      "long #{@name}[3];",
+      "long #{@name}_storage[PIXEL_COUNT][3];"
+    ]
+  end
+end
+
+module Arbol
+  class Documentation
+
+  def feedback  
+%{--
+### feedback(input, feedback)
+
+* **input**
+* **feedback**
+
+Returns the greatest of the input or the feedback value.
+
+}
+  end
+
+  end
+end
+
+def feedback(input, feedback)
+  h = ArbolHash.new
+  h[:type] = 'feedback'
+  h[:input] = resolve(input)
+  h[:feedback] = resolve(feedback)
+  h
+end

data/lib/functions/feedback_offset.rb

@@ -0,0 +1,69 @@
+class FeedbackOffset < Base
+  Arbol.add_mapped_class(
+    'feedback_offset',
+    FeedbackOffset,
+%{void feedback_offset(int pixel, int pixel_count, long input[3], long feedback[3], long offset[3], long storage[][3], long out[3]) {
+  out[0] = max(input[0], storage[(pixel + long_mult(offset[0], pixel_count)) % pixel_count][0]);
+  out[1] = max(input[1], storage[(pixel + long_mult(offset[1], pixel_count)) % pixel_count][1]);
+  out[2] = max(input[2], storage[(pixel + long_mult(offset[2], pixel_count)) % pixel_count][2]);
+  storage[pixel][0] = constrain(long_mult(out[0], feedback[0]), 0, INTEGER_SCALE);
+  storage[pixel][1] = constrain(long_mult(out[1], feedback[1]), 0, INTEGER_SCALE);
+  storage[pixel][2] = constrain(long_mult(out[2], feedback[2]), 0, INTEGER_SCALE);
+}}
+  )
+
+  attr_accessor :input
+  attr_accessor :feedback
+  attr_accessor :offset
+
+  def initialize(params)
+    super(params)
+    @frame_optimized = false
+  end
+
+  def param_keys
+    [:input, :feedback, :offset]
+  end
+
+  def arduino_code
+    [
+      "feedback_offset(i, PIXEL_COUNT, #{@input.name}, #{@feedback.name}, #{@offset.name}, #{@name}_storage, #{@name});"
+    ]
+  end
+
+  def top_level_scope_code
+    [
+      "long #{@name}[3];",
+      "long #{@name}_storage[PIXEL_COUNT][3];"
+    ]
+  end
+end
+
+module Arbol
+  class Documentation
+
+  def feedback_offset   
+%{--
+### feedback\_offset(input, feedback, offset)
+
+* **input**
+* **feedback**
+* **offset**
+
+Returns the greatest of the input or the feedback value at the offset (0-~1.0) 
+from the current pixel.
+
+}
+  end
+
+  end
+end
+
+def feedback_offset(input, feedback, offset)
+  h = ArbolHash.new
+  h[:type] = 'feedback_offset'
+  h[:input] = resolve(input)
+  h[:feedback] = resolve(feedback)
+  h[:offset] = resolve(offset)
+  h
+end

data/lib/functions/gamma.rb

@@ -0,0 +1,61 @@
+class Gamma < Base
+  Arbol.add_mapped_class(
+    'gamma', 
+    Gamma,
+%{long neopix_gamma[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 2, 2, 3, 4, 4, 5, 6, 7, 8, 9, 10, 12, 13, 15, 16, 18, 20, 22, 24, 26, 28, 31, 33, 36, 39, 42, 45, 48, 51, 55, 59, 62, 66, 71, 75, 79, 84, 89, 94, 99, 104, 110, 116, 122, 128, 134, 140, 147, 154, 161, 168, 176, 184, 192, 200, 208, 217, 225, 234, 244, 253, 263, 273, 283, 293, 304, 315, 326, 338, 349, 361, 373, 386, 398, 411, 425, 438, 452, 466, 480, 495, 510, 525, 541, 556, 572, 589, 605, 622, 640, 657, 675, 693, 712, 730, 750, 769, 789, 809, 829, 850, 871, 892, 914, 936, 959, 981, 1004, 1028, 1052, 1076, 1100, 1125, 1150, 1176, 1202, 1228, 1255, 1282, 1309, 1337, 1365, 1393, 1422, 1452, 1481, 1511, 1542, 1572, 1604, 1635, 1667, 1700, 1733, 1766, 1799, 1833, 1868, 1903, 1938, 1974, 2010, 2046, 2083, 2121, 2158, 2197, 2235, 2274, 2314, 2354, 2394, 2435, 2476, 2518, 2560, 2603, 2646, 2690, 2734, 2778, 2823, 2869, 2915, 2961, 3008, 3055, 3103, 3151, 3200, 3249, 3299, 3349, 3399, 3451, 3502, 3554, 3607, 3660, 3714, 3768, 3823, 3878, 3933, 3990, 4046, 4103, 4161, 4219, 4278, 4337, 4397, 4458, 4518, 4580, 4642, 4704, 4767, 4831, 4895, 4959, 5025, 5090, 5157, 5223, 5291, 5359, 5427, 5496, 5566, 5636, 5707, 5778, 5850, 5922, 5995, 6069, 6143, 6218, 6293, 6369, 6446, 6523, 6601, 6679, 6758, 6837, 6917, 6998, 7079, 7161, 7244, 7327, 7410, 7495, 7580, 7665, 7751, 7838, 7926, 8014, 8102};
+  
+long INTEGER_SCALE_MINUS_ONE = INTEGER_SCALE - 1;
+void gamma(long input[3], long out[3]) {
+  out[0] = neopix_gamma[map(input[0], 0, INTEGER_SCALE_MINUS_ONE, 0, 255) ];
+  out[1] = neopix_gamma[map(input[1], 0, INTEGER_SCALE_MINUS_ONE, 0, 255) ];
+  out[2] = neopix_gamma[map(input[2], 0, INTEGER_SCALE_MINUS_ONE, 0, 255) ];
+}}
+  )
+  
+  attr_accessor :input
+
+  def initialize(params)
+    super(params)
+    @frame_optimized = false
+  end
+
+  def param_keys
+    [:input]
+  end
+  
+  def arduino_code
+    [
+      "gamma(#{@input.name}, #{@name});"
+    ]
+  end
+
+  def top_level_scope_code
+    [
+      "long #{@name}[3];"
+    ]
+  end
+end
+
+module Arbol
+  class Documentation
+
+  def gamma  
+%{--
+### gamma(input)
+
+* **input**
+
+Returns gamma corrected input. This makes the colors look much better.
+
+}
+  end
+
+  end
+end
+
+def gamma(input)
+  h = ArbolHash.new
+  h[:type] = 'gamma'
+  h[:input] = resolve(input)
+  h
+end