cosmos 3.0.0

data/lib/cosmos/core_ext/array.rb

@@ -0,0 +1,354 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'cosmos/ext/array'
+
+# COSMOS specific additions to the Ruby Array class
+class Array
+  # Redefine inspect to only print for small numbers of
+  # items. Prevents exceptions taking forever to be raise with
+  # large objects. See http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/105145
+  alias old_inspect inspect
+
+  # @param max_elements [Integer] The maximum number of elements in the array to
+  #   print out before simply displaying the array class and object id
+  # @return [String] String representation of the array
+  def inspect(max_elements = 10)
+    if self.length <= max_elements
+      old_inspect()
+    else
+      '#<' + self.class.to_s + ':' + self.object_id.to_s + '>'
+    end
+  end
+
+  # @return [Array] Cloned array after all elements called to_f
+  def clone_to_f
+    new_array = self.class.new(0)
+    self.each do |value|
+      new_array << value.to_f
+    end
+    new_array
+  end
+
+  # Returns the array index nearest to the passed in value. This only makes sense
+  # for numerical arrays containing integers or floats. It has an optimized
+  # algorithm if the array is sorted but will fail if passed unsorted data with
+  # the sorted option.
+  #
+  # @param value [Numeric] A value to search for in the array
+  # @param ordered_data [Boolean] Whether or not the data is sorted
+  # @return [Integer] The index of the element closest to value
+  def nearest_index(value, ordered_data = true)
+    raise "Cannot search on empty array" if self.empty?
+
+    if ordered_data
+      last_index  = self.length - 1
+      first_value = self[0].to_f
+      last_value = self[-1].to_f
+      return 0 if first_value == last_value
+      slope = last_index.to_f / (last_value - first_value)
+      offset = -(slope * first_value)
+      guess_index = ((slope * value.to_f) + offset).to_i
+
+      # Return immediately for boundary conditions
+      return 0 if guess_index < 0
+      return last_index if guess_index > last_index
+
+      # Verify guess index
+      previous_guess_index = nil
+      previous_guess_value = nil
+
+      # While in the valid range of indexes
+      while (guess_index >= 0 and guess_index <= last_index)
+
+        # Retrieve the value at our current guess index
+        guess_value = self[guess_index]
+
+        # We're done if we found the exact value
+        return guess_index if guess_value == value
+
+        if previous_guess_value # Determine if we did better or worse
+          # Was previous guess better or worse?
+          if (guess_value - value).abs <= (previous_guess_value - value).abs
+            # Previous Guess Worse or the same
+            if guess_value > value # Moving with decreasing indexes
+              if previous_guess_value > value # Still moving in right direction
+                previous_guess_index = guess_index
+                guess_index -= 1
+              else # We passed the value
+                return guess_index
+              end
+            else # guess_value < value and moving with increasing indexes
+              if previous_guess_value < value # Still moving in right direction
+                previous_guess_index = guess_index
+                guess_index += 1
+              else # We passed the value
+                return guess_index
+              end
+            end
+          else
+            # Previous Guess Better
+            return previous_guess_index
+          end
+        else # Move to the next point
+          previous_guess_index = guess_index
+          if guess_value > value
+            guess_index -= 1
+          else #guess_value < value
+            guess_index += 1
+          end
+        end
+        previous_guess_value = guess_value
+      end
+
+      # Return our best guess
+      return 0 if guess_index < 0
+      return last_index
+    else # Brute force search
+      # Calculate the initial delta
+      min_delta     = (self[0] - value).abs
+      closest_index = 0
+      self.each_with_index do |self_value, index|
+        # Calculate the delta between the current value and value we are
+        # searching for
+        delta = (value - self_value).abs
+        # If the newly calculate delta is less than or equal to are previous
+        # minimum delta then we proceed
+        if delta <= min_delta
+          # There is a special case if the delta is equal to the previously
+          # calculated delta. We want to round up in this case so we check if
+          # the value we are searching for is greater than the current value.
+          # If so we skip this value since we don't want to round down.
+          next if (delta == min_delta) and (value > self_value)
+          min_delta = delta
+          closest_index = index
+        end
+      end
+      return closest_index
+    end
+  end
+
+  # Returns the index of the first element which is less than or equal to the
+  # passed in value.
+  #
+  # NOTE: This routine only works on sorted data!
+  #
+  # @param value [Numeric] The value to search for in the array
+  # @return [Fixnum] The index of the element which is less than or equal to
+  #   the value
+  def index_lt_eq(value)
+    index = nearest_index(value)
+
+    # Keep backing up if self[index - 1] == value to move past duplicates
+    while (index > 0 and self[index - 1] == value)
+      index -= 1
+    end
+
+    return index if self[index] <= value
+    index -= 1 if index > 0
+    return index
+  end
+
+  # Returns the index of the last element which is greater than or equal to the
+  # passed in value.
+  #
+  # NOTE: This routine only works on sorted data!
+  #
+  # @param value [Numeric] The value to search for in the array
+  # @return [Fixnum] The index of the element which is greater than or equal to
+  #   the value
+  def index_gt_eq(value)
+    index = nearest_index(value)
+    last_index = self.length - 1
+
+    # Keep moving forward if self[index - 1] == value to move past duplicates
+    while (index < last_index and self[index + 1] == value)
+      index += 1
+    end
+
+    return index if self[index] >= value
+    index += 1 if (self.length - 1) > index
+    return index
+  end
+
+  # Returns the range of array elements which contain both the start value and
+  # end value.
+  #
+  # NOTE: This routine only works on sorted data!
+  #
+  # @param start_value [Numeric] The start value to search for (must be less
+  #   than end_value)
+  # @param end_value [Numeric] The end value to search for
+  # @return [Range] The range of array elements which contain both the
+  #   start_value and end_value
+  def range_containing(start_value, end_value)
+    raise "end_value: #{end_value} must be greater than start_value: #{start_value}" if end_value < start_value
+    Range.new(index_lt_eq(start_value), index_gt_eq(end_value))
+  end
+
+  # Returns the range of array elements which within both the start value and
+  # end value.
+  #
+  # NOTE: This routine only works on sorted data!
+  #
+  # @param start_value [Numeric] The start value to search for (must be less
+  #   than end_value)
+  # @param end_value [Numeric] The end value to search for
+  # @return [Range] The range of array elements which contain both the
+  #   start_value and end_value
+  def range_within(start_value, end_value)
+    raise "end_value: #{end_value} must be greater than start_value: #{start_value}" if end_value < start_value
+    Range.new(index_gt_eq(start_value), index_lt_eq(end_value))
+  end
+
+  if !(self.method_defined?(:sum))
+    # @return [Numeric] The sum of all the elements in the array
+    def sum
+      self.inject(0, :+)
+    end
+  end
+
+  # return [Float] The mean of all the elements in the array
+  def mean
+    return 0.0 if self.empty?
+    return self.sum / self.length.to_f
+  end
+
+  # return [Array] A new array with each value of the original squared
+  def squared
+    self.map {|value| value * value}
+  end
+
+  # return [Numeric, Fixnum] The first maximum value and its index
+  # def max_with_index
+
+  # return [Numeric, Fixnum] The first minimum value and its index
+  # def min_with_index
+
+  # @param num_buckets [Integer] The number of buckets (groups of numbers) that
+  #   will be used when histogramming. nil indicates to use as many buckets as
+  #   necessary to cause each bucket to have a unique element.
+  # @param numeric [Boolean] Whether the array data is numeric
+  # @param block [Proc] If a block is given it will be called to sort buckets
+  #   with the same object. This might be necessary if your data is not numeric
+  #   and you want to override the way your objects compare.
+  # @return [Array<Array(first_value, last_value, num_values)>] Array of buckets
+  #   which are arrays containing the first value that is found in the bucket,
+  #   the last value found in the bucket, and the total number of values in the
+  #   bucket.
+  def histogram(num_buckets = nil, numeric = false, &block)
+    buckets = {}
+
+    # Count the occurance of each value
+    self.each do |value|
+      buckets[value] ||= 0
+      buckets[value] += 1
+    end
+
+    # Sort buckets by value, use block for sorting if given
+    if block_given?
+      sorted_buckets = buckets.sort {|x, y| yield(x,y)}
+    else
+      sorted_buckets = buckets.sort
+    end
+
+    reduced_buckets = []
+    if num_buckets
+      # Validate num_buckets
+      raise "Invalid num_buckets #{num_buckets}" if num_buckets.to_i <= 0
+
+      # Handle histogram types
+      if numeric
+        # Numeric histograms use the same sized range for each bucket
+        first_value   = sorted_buckets[0][0]
+        last_value    = sorted_buckets[-1][0]
+        delta         = last_value - first_value
+        bucket_size   = delta.to_f / num_buckets.to_f
+        integers      = false
+        integers      = true if first_value.kind_of?(Integer) and last_value.kind_of?(Integer)
+        if integers
+          bucket_size = bucket_size.ceil
+          last_value = first_value + bucket_size*num_buckets - 1
+          delta = last_value - first_value
+          (delta + 1).times do |index|
+            buckets[first_value + index] ||= 0
+          end
+          if block_given?
+            sorted_buckets = buckets.sort {|val1, val2| yield(val1, val2) }
+          else
+            sorted_buckets = buckets.sort
+          end
+        end
+        bucket_ranges = []
+        current_value = first_value
+        num_buckets.times do |bucket_index|
+          if bucket_index == (num_buckets - 1)
+            bucket_ranges[bucket_index] = (current_value)..(last_value)
+          else
+            if integers
+              bucket_ranges[bucket_index] = (current_value)..(current_value + bucket_size - 1)
+            else
+              bucket_ranges[bucket_index] = (current_value)..(current_value + bucket_size)
+            end
+          end
+          current_value += bucket_size
+        end
+
+        # Build the final buckets
+        first_index  = 0
+        sorted_index = 0
+        num_buckets.times do |bucket_index|
+          break if sorted_index > (sorted_buckets.length - 1)
+          sum = 0
+          bucket_range = bucket_ranges[bucket_index]
+          while (bucket_range.include?(sorted_buckets[sorted_index][0]))
+            sum += sorted_buckets[sorted_index][1]
+            sorted_index += 1
+            break if sorted_index > (sorted_buckets.length - 1)
+          end
+          reduced_buckets[bucket_index] = [bucket_range.first, bucket_range.last, sum]
+        end
+      else
+        # Non-numeric histograms use the same number of items per bucket
+        items_per_bucket = sorted_buckets.length / num_buckets.to_i
+        items_per_bucket = 1 if items_per_bucket < 1
+        bucket_sizes     = [items_per_bucket] * num_buckets
+        excess_items     = sorted_buckets.length - (items_per_bucket * num_buckets)
+        if excess_items > 0
+          bucket_sizes.length.times do |bucket_size_index|
+            break if excess_items <= 0
+            bucket_sizes[bucket_size_index] += 1
+            excess_items -= 1
+          end
+        end
+
+        # Build the final buckets
+        first_index = 0
+        num_buckets.times do |bucket_index|
+          break if first_index > (sorted_buckets.length - 1)
+          if bucket_index == (num_buckets - 1)
+            last_index = sorted_buckets.length - 1
+          else
+            last_index = first_index + bucket_sizes[bucket_index] - 1
+            last_index = sorted_buckets.length - 1 if last_index > (sorted_buckets.length - 1)
+          end
+          sum = 0
+          sorted_buckets[first_index..last_index].each {|key, value| sum += value}
+          reduced_buckets[bucket_index] = [sorted_buckets[first_index][0], sorted_buckets[last_index][0], sum]
+          first_index = first_index + bucket_sizes[bucket_index]
+        end
+      end
+    else
+      sorted_buckets.each {|key, value| reduced_buckets << [key, key, value]}
+    end
+    reduced_buckets
+  end
+
+end # class Array

data/lib/cosmos/core_ext/class.rb

@@ -0,0 +1,51 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+# COSMOS specific additions to the Ruby Class class
+class Class
+  # Creates instance variables in the class which have corresponding class
+  # method accessors. NOTE: You must define self.instance for this to work.
+  #
+  # For example:
+  #   class MyClass
+  #     instance_attr_reader :test
+  #     @@instance = nil
+  #     def self.instance
+  #       @@instance ||= self.new
+  #       return @@instance
+  #     end
+  #     def initialize
+  #       @test = "Test"
+  #       @@instance = self
+  #     end
+  #
+  # Will allow the following:
+  #   my = MyClass.new
+  #   my.test # returns "Test"
+  #   MyClass.test # returns "Test"
+  #
+  # @param args [Array<Symbol>] Array of symbols which should be turned into
+  #   instance variables with class method accessors
+  def instance_attr_reader(*args)
+    args.each do |arg|
+      self.class_eval("def #{arg};@#{arg};end")
+      self.instance_eval("def #{arg};self.instance.#{arg};end")
+    end
+  end
+
+  def instance_attr_accessor(*args)
+    args.each do |arg|
+      self.class_eval("def #{arg};@#{arg};end")
+      self.instance_eval("def #{arg};self.instance.#{arg};end")
+      self.class_eval("def #{arg}=(arg);@#{arg} = arg;end")
+      self.instance_eval("def #{arg}=(arg);self.instance.#{arg} = arg;end")
+    end
+  end
+end

data/lib/cosmos/core_ext/cosmos_io.rb

@@ -0,0 +1,29 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'cosmos/ext/cosmos_io'
+
+# COSMOS specific additions to the Ruby IO and StringIO classes
+module CosmosIO
+  # Reads a length field and then return the String resulting from reading the
+  # number of bytes the length field indicates
+  #
+  # For example:
+  #   io = StringIO.new
+  #   # where io is "\x02\x01\x02\x03\x04...."
+  #   result = io.read_length_bytes(1)
+  #   # result will be "\x01x02" because the length field was given
+  #   # to be 1 byte. We read 1 byte which is a 2. So we then read two
+  #   # bytes and return.
+  #
+  # @param length_num_bytes [Integer] Number of bytes in the length field
+  # @return [String] A String of "length field" number of bytes
+  # def read_length_bytes(length_num_bytes)
+end

data/lib/cosmos/core_ext/exception.rb

@@ -0,0 +1,52 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+# COSMOS specific additions to the Ruby Exception class
+class Exception
+  # @param hide_runtime_error_class [Boolean] Whether to hide the Exception
+  #   error class if the class is RuntimeError. Other classes will continue to
+  #   be printed.
+  # @param include_backtrace [Boolean] Whether to include the full exception
+  #   backtrace
+  # @return [String] The formatted Exception
+  def formatted(hide_runtime_error_class = false, include_backtrace = true)
+    if include_backtrace and self.backtrace
+      if hide_runtime_error_class and self.class == RuntimeError
+        "#{self.message}\n#{self.backtrace.join("\n")}"
+      else
+        "#{self.class.to_s.split('::')[-1]} : #{self.message}\n#{self.backtrace.join("\n")}"
+      end
+    else
+      if hide_runtime_error_class and self.class == RuntimeError
+        "#{self.message}"
+      else
+        "#{self.class.to_s.split('::')[-1]} : #{self.message}"
+      end
+    end
+  end
+
+  # @return [Array(String, Fixnum)] The filename and line number where the Exception
+  #   occurred
+  def source
+    trace = self.backtrace[0]
+    split_trace = trace.split(':')
+    filename = ''
+    line_number = ''
+    if trace[1..1] == ':' # Windows Path
+      filename = split_trace[0] + ':' + split_trace[1]
+      line_number = split_trace[2].to_i
+    else
+      filename = split_trace[0]
+      line_number = split_trace[1].to_i
+    end
+
+    [filename, line_number]
+  end
+end

data/lib/cosmos/core_ext/file.rb

@@ -0,0 +1,75 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'find'
+
+# COSMOS specific additions to the Ruby File class
+class File
+  NON_ASCII_PRINTABLE = /[^\x21-\x7e\s]/
+
+  # @return [Boolean] Whether the file only contains ASCII characters
+  def self.is_ascii?(filename)
+    return_value = true
+    File.open(filename) do |file|
+      while buf = file.read(1024)
+        if buf =~ NON_ASCII_PRINTABLE
+          return_value = false
+          break
+        end
+      end
+    end
+    return return_value
+  end
+
+  # Builds a String for use in creating a file. The time is formatted as
+  # YYYY_MM_DD_HH_MM_SS. The tags and joined with an underscore and appended to
+  # the date before appending the extension.
+  #
+  # For example:
+  #   File.build_timestamped_filename(['test','only'], '.bin', Time.now)
+  #   # result is YYYY_MM_DD_HH_MM_SS_test_only.bin
+  #
+  # @param tags [Array<String>] An array of strings to be joined by underscores
+  #   after the date. Pass nil or an empty array to use no tags.
+  # @param extension [String] The filename extension
+  # @param time [Time] The time to format into the filename
+  # @return [String] The filename string containing the timestamp, tags, and
+  #   extension
+  def self.build_timestamped_filename(tags = nil, extension = '.txt', time = Time.now)
+    timestamp = sprintf("%04u_%02u_%02u_%02u_%02u_%02u", time.year, time.month, time.mday, time.hour, time.min, time.sec)
+    tags ||= []
+    tags.compact!
+    combined_tags = tags.join("_")
+    if combined_tags.length > 0
+      filename = timestamp + "_" + combined_tags + extension
+    else
+      filename = timestamp + extension
+    end
+    return filename
+  end
+
+  # @param filename [String] The file to search for
+  # @return [String] The full path to the filename if it was found in the Ruby
+  #   search path. nil if the fild was not found.
+  def self.find_in_search_path(filename)
+    $:.each do |load_path|
+      begin
+        Find.find(load_path) do |path|
+          Find.prune if path =~ /\.svn/
+          return path if File.basename(path) == filename
+        end
+      rescue Errno::ENOENT
+        # Ignore non-existent folders
+        next
+      end
+    end
+    return nil
+  end
+end

data/lib/cosmos/core_ext/hash.rb

@@ -0,0 +1,28 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+# COSMOS specific additions to the Ruby Hash class
+class Hash
+  # Redefine inspect to only print for small numbers of
+  # items.  Prevents exceptions taking forever to be raised with
+  # large objects.  See http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/105145
+  alias old_inspect inspect
+
+  # @param max_elements [Integer] The maximum number of elements in the Hash to
+  #   print out before simply displaying the Hash class and object id
+  # @return [String] String representation of the hash
+  def inspect(max_elements = 10)
+    if self.length <= max_elements
+      old_inspect()
+    else
+      '#<' + self.class.to_s + ':' + self.object_id.to_s + '>'
+    end
+  end
+end

data/lib/cosmos/core_ext/io.rb

@@ -0,0 +1,75 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'cosmos/core_ext/cosmos_io'
+
+class IO
+  include CosmosIO
+
+  SELECT_BASE_TIMEOUT = 0.0004
+  SELECT_MAX_TIMEOUT = 0.016
+
+  def self.fast_select(read_sockets = nil, write_sockets = nil, error_array = nil, timeout = nil)
+    # Always try a zero timeout first
+    current_timeout = SELECT_BASE_TIMEOUT
+    total_timeout = 0.0
+
+    while true
+      result = IO.select(read_sockets, write_sockets, error_array, current_timeout)
+      return result if result or current_timeout.nil?
+      return nil if timeout and total_timeout >= timeout
+
+      if current_timeout <= 0.0001
+        # Always try the base timeout next
+        current_timeout = SELECT_BASE_TIMEOUT
+        total_timeout = SELECT_BASE_TIMEOUT
+      else
+        # Then start doubling the timeout
+        current_timeout = current_timeout * 2
+
+        # Until it is bigger than our max timeout
+        if current_timeout >= SELECT_MAX_TIMEOUT
+          if timeout
+            # Block for the remaining requested timeout
+            current_timeout = timeout - total_timeout
+            total_timeout = timeout
+          else
+            # Or block forever
+            current_timeout = nil
+          end
+        else
+          # Or it is bigger than the given timeout
+          if timeout and current_timeout >= timeout
+            # Block for the remaining requested timeout
+            current_timeout = timeout - total_timeout
+            total_timeout = timeout
+          else
+            # Up our total time in select
+            total_timeout += current_timeout
+          end
+          if timeout and total_timeout > timeout
+            # Block for the remaining requested timeout
+            current_timeout = timeout - total_timeout
+            total_timeout = timeout
+          end
+        end
+        return nil if current_timeout and current_timeout < 0
+      end
+    end # while true
+  end # fast_select
+
+  def self.fast_read_select(read_sockets, timeout)
+    return fast_select(read_sockets, nil, nil, timeout)
+  end
+
+  def self.fast_write_select(write_sockets, timeout)
+    return fast_select(nil, write_sockets, nil, timeout)
+  end
+end