cowboy 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in cowboy.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Alejandro Ciniglio
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# Cowboy
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'cowboy'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install cowboy
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/cowboy.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/cowboy/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Alejandro Ciniglio"]
+  gem.email         = ["mail@alejandrociniglio.com"]
+  gem.description   = %q{gem description}
+  gem.summary       = %q{gem summary}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "cowboy"
+  gem.require_paths = ["lib"]
+  gem.version       = Cowboy::VERSION
+
+  gem.extensions    = ["ext/cowboy/extconf.rb"]
+end

data/ext/cowboy/cowboy.c

@@ -0,0 +1,35 @@
+#include "cowboy.h"
+
+VALUE mCowboy;
+
+VALUE fft_1d(VALUE m, VALUE nums){
+  fftw_complex *in, *out;
+  fftw_plan fp;
+  long n;
+
+  Check_Type(nums, T_ARRAY);
+
+  n = size_of_ary(nums);
+  if (n == 0){
+    rb_raise(rb_eException, "Can't use blank array");
+  }
+  in = allocate_fftw_complex(n);
+  out = allocate_fftw_complex(n);
+  fp = fftw_plan_dft_1d(n, in, out, FFTW_FORWARD, FFTW_ESTIMATE);
+
+  cast_nums_to_complex(in, nums);
+
+  fftw_execute(fp);
+  free(in);
+  fftw_destroy_plan(fp);
+
+  return complex_to_real_nums(out, n);
+}
+
+void Init_cowboy(){
+  mCowboy = rb_define_module("Cowboy");
+
+  Init_cowboy_complex();
+
+  rb_define_module_function(mCowboy, "fft_1d", fft_1d, 1);
+}

data/ext/cowboy/cowboy.h

@@ -0,0 +1,11 @@
+#ifndef RUBY_COWBOY
+#define RUBY_COWBOY
+
+#include <ruby.h>;
+#include <fftw3.h>;
+
+#include "cowboy_complex.h";
+
+extern VALUE mCowboy;
+
+#endif

data/ext/cowboy/cowboy_complex.c

@@ -0,0 +1,56 @@
+#include "cowboy.h"
+#include <ruby.h>
+
+long size_of_ary(VALUE nums){
+  Check_Type(nums, T_ARRAY);
+  return RARRAY_LEN(nums);
+}
+
+void cast_nums_to_complex(fftw_complex * fc, VALUE nums){
+  long len;
+  int i;
+  VALUE n;
+  Check_Type(nums, T_ARRAY);
+  i = 0;
+
+  while((n =rb_ary_shift(nums)) != Qnil){
+    fc[i][0] = NUM2DBL(n);
+    fc[i][1] = 0;
+    i++;
+  }
+}
+
+fftw_complex * allocate_fftw_complex(long n){
+  fftw_complex * fc;
+  fc = ALLOC_N(fftw_complex, n);
+  return fc;
+}
+
+VALUE c_to_rb_complex(double r, double i){
+  VALUE cplx = rb_path2class("Complex");
+  return rb_funcall(cplx, rb_intern("rect"), 2,
+                    DBL2NUM(r), DBL2NUM(i));
+}
+
+VALUE complex_to_real_nums(fftw_complex *fc, long N){
+  VALUE ar = rb_ary_new();
+  int i;
+  for(i = 0; i < N; i++){
+    rb_ary_push(ar, c_to_rb_complex(fc[i][0],
+                                    fc[i][1]));
+  }
+  return ar;
+}
+
+VALUE test_back_and_from_complex(VALUE m, VALUE nums){
+  Check_Type(nums, T_ARRAY);
+  fftw_complex * fc;
+  fc = allocate_fftw_complex(size_of_ary(nums));
+  cast_nums_to_complex(fc, nums);
+  return complex_to_real_nums(fc,
+                              size_of_ary(nums));
+}
+
+void Init_cowboy_complex(){
+  rb_define_module_function(mCowboy, "test_complex", test_back_and_from_complex, 1);
+}

data/ext/cowboy/cowboy_complex.h

@@ -0,0 +1,14 @@
+#ifndef RUBY_COWBOY_COMPLEX
+#define RUBY_COWBOY_COMPLEX
+
+#include <cowboy.h>
+
+void Init_cowboy_complex();
+
+void cast_nums_to_complex(fftw_complex * fc, VALUE nums);
+fftw_complex * allocate_fftw_complex(long n);
+VALUE complex_to_real_nums(fftw_complex * fc, long N);
+long size_of_ary(VALUE nums);
+
+
+#endif

data/ext/cowboy/extconf.rb

@@ -0,0 +1,29 @@
+require 'mkmf'
+LIBDIR = RbConfig::CONFIG['libdir']
+INCLUDEDIR = RbConfig::CONFIG['includedir']
+
+HEADER_DIRS = [
+              '/opt/local/include',
+              '/usr/local/include',
+              INCLUDEDIR,
+              '/usr/include',
+]
+
+LIB_DIRS = [
+            '/opt/local/lib',
+            '/usr/local/lib',
+            LIBDIR,
+            '/usr/lib',
+]
+
+dir_config('fftw', HEADER_DIRS, LIB_DIRS)
+
+unless find_header('fftw3.h')
+  abort "FFTw is missing, please install FFTw"
+end
+
+unless find_library('fftw3', "fftw_malloc")
+  abort "FFTw is missing, please install FFTw"
+end
+
+create_makefile('cowboy/cowboy')

data/lib/cowboy.rb

@@ -0,0 +1,9 @@
+require "cowboy/version"
+
+require 'cowboy/cowboy'
+module Cowboy
+  # Your code goes here...
+end
+
+require "cowboy/hamming"
+require "cowboy/fft"

data/lib/cowboy/fft.rb

@@ -0,0 +1,39 @@
+module Cowboy
+  class Frequencies
+    attr_reader :dc, :unshifted, :freq
+
+    def initialize(unshifted_array)
+      @unshifted = unshifted_array
+      @dc = unshifted_array[0].real
+      n = unshifted_array.size
+      @freq = []
+      for i in (n/2...n)
+        @freq << unshifted_array[i].abs
+      end
+      for i in (0...n/2)
+        @freq << unshifted_array[i].abs
+      end
+    end
+
+    def buckets(sample_rate=nil)
+      sample_rate ||= @freq.size
+      nyquist = sample_rate/2.0
+      a = []
+      n = @freq.size
+      for i in (0...n)
+        a << -nyquist + (2 * nyquist/n) * i
+      end
+      return a
+    end
+  end
+
+  def Cowboy::fft(arr, w=Hamming, n=29)
+    if !w.nil?
+      window = w.new(n)
+      input = window.convolve(arr)
+      fft_1d(input)
+    else
+      fft_1d(arr)
+    end
+  end
+end

data/lib/cowboy/hamming.rb

@@ -0,0 +1,30 @@
+class Cowboy::Hamming
+  def initialize(n)
+    @n = n
+    create_kernel
+  end
+
+  def convolve(arr)
+    raise "Array smaller than window size" if arr.size < @n
+    a = []
+    for i in (0...(arr.size - (@n - 1)))
+      n = 0.0
+      for j in (0...@n)
+        c = @kernel[j] * arr[i+j]
+        n += c
+      end
+      a << n
+    end
+    return a
+  end
+
+  private
+  def create_kernel
+    a = []
+    for j in (0...@n)
+      c = 0.54 - 0.46 * Math.cos((2*Math::PI * j)/(@n - 1))
+      a << c
+    end
+    @kernel = a
+  end
+end

data/lib/cowboy/version.rb

@@ -0,0 +1,3 @@
+module Cowboy
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: cowboy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Alejandro Ciniglio
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-01 00:00:00.000000000 Z
+dependencies: []
+description: gem description
+email:
+- mail@alejandrociniglio.com
+executables: []
+extensions:
+- ext/cowboy/extconf.rb
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- cowboy.gemspec
+- ext/cowboy/cowboy.c
+- ext/cowboy/cowboy.h
+- ext/cowboy/cowboy_complex.c
+- ext/cowboy/cowboy_complex.h
+- ext/cowboy/extconf.rb
+- lib/cowboy.rb
+- lib/cowboy/fft.rb
+- lib/cowboy/hamming.rb
+- lib/cowboy/version.rb
+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.24
+signing_key: 
+specification_version: 3
+summary: gem summary
+test_files: []