radio 0.0.1 → 0.0.2

data/lib/radio/http/session.rb

@@ -0,0 +1,43 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+# This is unfinished, unused, and needs expiration management.
+# We'll definitely want sessions working for authentication,
+
+class Radio
+  class HTTP
+    class Session
+  
+      CODES = ('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a
+      COOKIE_KEY = 'ham21-radio-session'
+  
+      def self.prepare request, response
+        @sessions ||= {}
+        session_id = request.cookies[COOKIE_KEY]
+        session = @sessions[session_id]
+        unless session
+          session_id = (0...24).collect{CODES.sample}.join
+          session = @sessions[session_id] = new
+          Rack::Utils.set_cookie_header!(response.headers, COOKIE_KEY, session_id)
+        end
+        session
+      end
+  
+      def initialize
+      end
+  
+    end
+  end
+end

data/lib/radio/input.rb

@@ -0,0 +1,62 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+class Radio
+  module Input
+    
+    # Keep this namespace clean because we search inputs by
+    # finding the classes from Radio::Inputs.constants.
+    
+    # Drivers are supposed to fail silently if they can't find
+    # dependencies. This allows us to present a basic debug screen.
+    def self.status
+      s = {}
+      Radio::Input.constants.collect do |input_name|
+        s[input_name] = eval(input_name.to_s).status
+      end
+      s
+    end
+    
+    # Consolidate all sources from all inputs and add the class name to the keys.
+    def self.sources
+      s = {}
+      Radio::Input.constants.each do |type|
+        eval(type.to_s).sources.each do |id, source|
+          s[[type, id]] = source
+        end
+      end
+      s
+    end
+    
+    # You can't new a module so this switches into the specific class.
+    def self.new type, id, rate, channel_i, channel_q=nil
+      # defend the eval
+      unless Radio::Input.constants.include? type.to_sym
+        raise NameError, "uninitialized constant Radio::Input::#{type}"
+      end
+      input = eval(type.to_s).new id, rate, channel_i, channel_q
+      # Ask for and discard the first sample to report errors here
+      begin
+        input.call 1
+      rescue Exception => e
+        input.stop
+        raise e
+      end
+      input
+    end
+    
+  end
+end
+

data/lib/radio/inputs/alsa.rb

@@ -0,0 +1,135 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+begin
+  require 'alsa'
+rescue LoadError => e
+end
+
+
+class Radio
+  module Input
+
+    class ALSA
+      
+      def self.status
+        if defined? ::ALSA::PCM
+          return "Loaded: %d input devices" % sources.count
+        end
+        unless defined? @is_linux
+          @is_linux = (`uname`.strip == 'Linux') rescue false
+        end
+        return "Unsupported: requires Linux" unless @is_linux
+        if defined? ::ALSA
+          'Unavailable: install ALSA to your OS'
+        else
+          'Unavailable: gem install ruby-alsa'
+        end
+      end
+      
+      CP_REGEX = /:\s*(capture|playback)\s*(\d+)\s*$/
+      CD_REGEX = /^(\d+)-(\d+):\s*/
+      
+      def self.sources
+        return {} unless defined? ::ALSA::PCM
+        # Funky read to get around old Linux bug
+        # http://kerneltrap.org/mailarchive/git-commits-head/2009/4/17/5510664
+        pcm = ::File.open('/proc/asound/pcm') do |io|
+          io.read_nonblock 32768
+        end
+        result = {}
+        pcm.each_line do |line|
+          capture = 0
+          2.times do
+            line.gsub! CP_REGEX, ''
+            capture = $2.to_i if $1 == 'capture'
+          end
+          line.gsub! CD_REGEX, ''
+          device = "hw:#{$1.to_i},#{$2.to_i}"
+          ::ALSA::PCM::Capture.open(device) do |stream|
+            params = stream.hardware_parameters
+            result[device] = {
+              name: line,
+              rates: [params.sample_rate],
+              channels: params.channels
+            }
+          end rescue nil
+        end
+        result
+      end
+      
+      def initialize id, rate, channel_i, channel_q
+        @stream = ::ALSA::PCM::Capture.new
+        @stream.open id
+        @channel_i = channel_i
+        @channel_q = channel_q
+      end
+      
+      def rate
+        @stream.hardware_parameters.sample_rate
+      end
+      
+      def channels
+        return 2 if @channel_q and @stream.hardware_parameters.channels > 1
+        1
+      end
+      
+      def call samples
+        
+        out=nil
+        buf_size = @stream.hw_params.buffer_size_for(samples)
+        FFI::MemoryPointer.new(:char, buf_size) do |buffer|
+          @stream.read_buffer buffer, samples
+          out = buffer.read_string(buf_size)
+          NArray.to_na(out, NArray::SINT).to_f.div! 32767
+        end
+        stream_channels = @stream.hardware_parameters.channels
+        out = case buf_size / samples / stream_channels
+        when 1 then NArray.to_na(out,NArray::BYTE).to_f.collect!{|v|(v-128)/127}
+        when 2 then NArray.to_na(out,NArray::SINT).to_f.div! 32767
+        # when 3 then NArray.to_na(d,NArray::???).to_f.collect!{|v|(v-8388608)/8388607}
+        else
+          raise "Unsupported sample size: #{@bit_sample}" 
+        end
+        return out if channels == 1 and stream_channels == 1
+        out.reshape! stream_channels, out.size/stream_channels
+        if channels == 1
+          out[@channel_i,true]
+        else
+          c_out = NArray.scomplex out[0,true].size
+          c_out[0..-1] = out[@channel_i,true]
+          c_out.imag = out[@channel_q,true]
+          c_out
+        end
+      end
+      
+      def stop
+        @stream.close
+      end
+      
+    end
+    
+  end
+end
+
+
+if $0 == __FILE__
+  
+  require 'narray'
+  a = Radio::Input::ALSA
+  i = a.new 'default', 44100, 0, nil
+  p i.call 8
+  
+end

data/lib/radio/inputs/coreaudio.rb

@@ -0,0 +1,102 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+begin
+  require 'coreaudio'
+rescue LoadError => e
+end
+
+
+class Radio
+  module Input
+
+    class CoreAudio
+      
+      def self.status
+        if defined? ::CoreAudio
+          return "Loaded: %d input devices" % sources.count
+        end
+        unless defined? @is_darwin
+          @is_darwin = (`uname`.strip == 'Darwin') rescue false
+        end
+        return "Unsupported: requires Apple OS" unless @is_darwin
+        return 'Unavailable: gem install coreaudio'
+      end
+  
+      # I don't see a way to automatically set the CoreAudio CODEC rate.
+      # We'll present the nominal_rate as the only option.
+      def self.sources
+        return {} unless defined? ::CoreAudio
+        result = {}
+        ::CoreAudio.devices.each do |dev|
+          channels = dev.input_stream.channels
+          if channels > 0
+            result[dev.devid] = {
+              name: dev.name,
+              rates: [dev.nominal_rate.to_i],
+              channels: channels
+            }
+          end
+        end
+        result
+      end
+      
+      attr_reader :rate, :channels
+
+      # id is the key from the sources hash.
+      # rate is the desired hardware rate.  do not decimate/interpolate here.
+      def initialize id, rate, channel_i, channel_q
+        @device = ::CoreAudio::AudioDevice.new id.to_i
+        @rate = @device.nominal_rate
+        raise 'I channel fail' unless channel_i < @device.input_stream.channels
+        @channel_i = channel_i
+        @channels = 1
+        if channel_q
+          raise 'Q channel fail' unless channel_q < @device.input_stream.channels
+          @channel_q = channel_q
+          extend IQ
+          @channels = 2
+        end
+        # Half second of buffer
+        coreaudio_input_buffer_size = @channels * rate / 2
+        @buf = @device.input_buffer coreaudio_input_buffer_size
+        @buf.start
+      end
+
+      # This is called on its own thread in Rig and is expected to block.
+      def call samples
+        # CoreAudio range of -32767..32767 makes easy conversion to -1.0..1.0
+        @buf.read(samples)[@channel_i,true].to_f.div!(32767)
+      end
+  
+      # Once stopped, rig won't attempt starting again on this object.
+      def stop
+        @buf.stop
+      end
+
+      module IQ
+        def call samples
+          b = @buf.read samples
+          c_out = NArray.scomplex samples
+          c_out[0..-1] = b[@channel_i,true].to_f.div!(32767)
+          c_out.imag = b[@channel_q,true].to_f.div!(32767)
+          c_out
+        end
+      end
+  
+    end
+
+  end
+end

data/lib/radio/inputs/file.rb

@@ -0,0 +1,60 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+
+class Radio
+  module Input
+    class File
+      
+      def self.status
+        "Loaded: %d files found" % sources.size
+      end
+      
+      def self.sources
+        result = {}
+        files = Dir.glob ::File.expand_path '../../../../test/wav/**/*.wav', __FILE__
+        files.each do |file|
+          begin
+            f = new file, 0, 0, 1
+          rescue Exception => e
+            next
+          end
+          result[file] = {
+            name: file,
+            rates: [f.rate],
+            channels: f.channels
+          }
+        end
+        result
+      end
+      
+      # You can load any file, not just the ones in sources.
+      def initialize id, rate, channel_i, channel_q
+        self.class.constants.each do |x|
+          klass = eval x.to_s
+          @reader = klass.new id, rate, channel_i, channel_q rescue nil
+          break if @reader
+        end
+        raise 'Unknown format' unless @reader
+      end
+      
+      def method_missing meth, *args, &block
+        @reader.send meth, *args, &block
+      end
+      
+    end
+  end
+end
+

data/lib/radio/inputs/wav.rb

@@ -0,0 +1,124 @@
+# Copyright 2012 The ham21/radio Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+
+class Radio
+  module Input
+    class File
+      class WAV
+        
+        attr_reader :rate
+        
+        def initialize id, rate, channel_i, channel_q
+          @file = ::File.new id
+          #TODO validate header instead?
+          @file.read 12 # discard header
+          @rate = rate
+          @channel_i = channel_i
+          @channel_q = channel_q
+          @data = [next_data]
+          @time = Time.now
+        end
+        
+        def call samples
+          sample_size = @channels * (@bit_sample/8)
+          @time += 1.0/(rate/samples)
+          sleep [0,@time-Time.now].max
+          while @data.reduce(0){|a,b|a+b.size} < samples * sample_size
+            @data.push next_data
+          end
+          if channels > 1
+            out = NArray.scomplex samples
+          else
+            out = NArray.sfloat samples
+          end
+          i = 0
+          while i < samples
+            if @data.first.size/sample_size > samples-i
+              out[i..-1] = convert @data.first[0...(samples-i)*sample_size]
+              @data[0] = @data.first[(samples-i)*sample_size..-1]
+              i = samples
+            else
+              converted_data = convert @data.shift
+              out[i...(i+converted_data.size)] = converted_data
+              i += converted_data.size
+            end
+          end
+          out
+        end
+        
+        def channels
+          return 2 if @channel_q and @channels > 1
+          1
+        end
+        
+        def stop
+          @file.close
+        end
+        
+        private
+        
+        def convert d
+          out = case @bit_sample
+          when 8 then NArray.to_na(d,NArray::BYTE).to_f.collect!{|v|(v-128)/127}
+          when 16 then NArray.to_na(d,NArray::SINT).to_f.div! 32767
+          # when 24 then NArray.to_na(d,NArray::???).to_f.collect!{|v|(v-8388608)/8388607}
+          else
+            raise "Unsupported sample size: #{@bit_sample}" 
+          end
+          return out if channels == 1 and @channels == 1
+          out.reshape! @channels, out.size/@channels
+          if channels == 1
+            out[@channel_i,true]
+          else
+            c_out = NArray.scomplex out[0,true].size
+            c_out[0..-1] = out[@channel_i,true]
+            c_out.imag = out[@channel_q,true]
+            c_out
+          end
+        end
+        
+        #TODO read data in chunks smaller than size (which is often the whole file)
+        def next_data
+          loop do
+            until @file.eof?
+              type = @file.read(4)
+              size = @file.read(4).unpack("V")[0].to_i
+              case type
+              when 'fmt '
+                fmt = @file.read(size)
+                @id = fmt.slice(0,2).unpack('c')[0]
+                @channels = fmt.slice(2,2).unpack('c')[0]
+                @rate = fmt.slice(4,4).unpack('V').join.to_i
+                @byte_sec = fmt.slice(8,4).unpack('V').join.to_i
+                @block_size = fmt.slice(12,2).unpack('c')[0]
+                @bit_sample = fmt.slice(14,2).unpack('c')[0]
+                next
+              when 'data'
+                return @file.read size
+              else
+                raise "Unknown GIF type: #{type}"
+              end
+            end
+            @file.rewind
+            @file.read 12 # discard header
+          end
+        end
+        
+      end
+    end
+  end
+end
+