ant_colony 0.0.5

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 4e82b81f4994e2ae805fcfb1d9b4b87cf3d2c883
+  data.tar.gz: 11e0093725cb8e441de603d1eb2b1d4a91f28998
+SHA512:
+  metadata.gz: 23329bc7493d7d379cf6bff80b9ef12aa3465c3c4390dc852b03a0715e377db11e7d56d4b450bc3839304621dd8f08b2ea925536047710d8eb9a2a56f893b937
+  data.tar.gz: 350da0960564030351d9cae8e0d88c7cd50f2587460c6ae3c2de5d2c3d8b59e7080623c2b0eae48dee0ac8c1378e07963af0252f1f7510ed9b2f60e46d3f907b

data/.gitignore

@@ -0,0 +1,5 @@
+/.bundle/
+/Gemfile.lock
+*.bundle
+*.gem
+/.idea/

data/.rubocop.yml

@@ -0,0 +1,15 @@
+AllCops:
+  DisplayStyleGuide: true
+  DisplayCopNames: true
+
+Style/Documentation:
+  Enabled: false
+
+Style/GuardClause:
+  Enabled: false
+
+Metrics/LineLength:
+  Max: 110
+
+Metrics/MethodLength:
+  Max: 20

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/README.md

@@ -0,0 +1,34 @@
+# AntColony
+
+Simple implementation ant colony optimization algorithm
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'ant_colony'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ant_colony
+
+## Usage
+```ruby
+graph  = YAML.load_file('test_data/test1.yml')
+colony = AntColony::Colony.new(graph, beta: 0.8, alpha: 0.7, pop: 200, ph: 0.3, q: 5)
+colony.solve
+colony.find_path 1 # path from point 1
+```
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/ant_colony/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/ant_colony.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ant_colony/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ant_colony'
+  spec.version       = AntColony::VERSION
+  spec.date          = '2016-03-05'
+  spec.authors       = ['Alexandr Turchyn', 'Pete Matsyburka']
+  spec.email         = %w(lexfox777@gmail.com pete.matsy@gmail.com)
+  spec.summary       = 'Ant colony optimization algorithms.'
+  spec.description   = 'Simple implementation ant colony optimization algorithm.'
+  spec.homepage      = 'https://github.com/AlexandrToorchyn/AntColony'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = %w(lib)
+
+  spec.add_development_dependency 'rspec', '~> 3.4.0'
+  spec.add_development_dependency 'rubocop'
+
+  spec.add_dependency 'bundler', '~> 1.7'
+  spec.add_dependency 'rake', '~> 10.0'
+  spec.add_dependency 'thor'
+end

data/bin/ant_colony

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require_relative '../lib/ant_colony'
+require_relative '../lib/ant_colony/cli.rb'
+
+AntColony::CLI.start(ARGV)

data/lib/ant_colony.rb

@@ -0,0 +1,73 @@
+require_relative 'ant_colony/version'
+
+module AntColony
+  class Colony
+    DEFAULT_PHERO = 0.5
+
+    attr_reader :alpha, :beta, :ph, :q, :graph, :pop
+
+    def initialize(graph, alpha: 0.8, beta: 0.4, ph: 0.3, q: 500, pop: 200)
+      method(__method__).parameters.each do |_, name|
+        instance_variable_set(:"@#{name}", binding.local_variable_get(name))
+      end
+      set_default_tau
+    end
+
+    def find_path(*roadmap)
+      loop do
+        city = graph[roadmap.last] # current point(city) ant located
+
+        sum_paths = city.reduce(0.0) do |sum, (_, path)|
+          sum + (path[:tau]**alpha).to_f / (path[:length]**beta)
+        end
+
+        proposed_paths = city.each_with_object({}) do |(direction, path), paths|
+          unless roadmap.include? direction
+            paths[direction] = (((path[:tau]**alpha).to_f / path[:length])**beta) / sum_paths
+          end
+        end
+
+        roadmap << best_path_of(proposed_paths)
+        break unless roadmap.length < graph.size
+      end
+
+      roadmap
+    end
+
+    def solve
+      graph.size.times do |i|
+        pop.times do
+          path = find_path(i + 1)
+          increase_tau_for path
+        end
+      end
+    end
+
+    def length_of(path)
+      [*path, path.first].each_cons(2).reduce(0.0) do |sum, (cur, nxt)|
+        sum + graph[cur][nxt][:length]
+      end
+    end
+
+    private
+
+    def set_default_tau
+      graph.each do |_, roads|
+        roads.each do |_, props|
+          props[:tau] = DEFAULT_PHERO
+        end
+      end
+    end
+
+    def best_path_of(paths)
+      paths.max_by { |_, v| v }.first
+    end
+
+    def increase_tau_for(path)
+      [*path, path.first].each_cons(2) do |cur, nxt|
+        tau_next = (1.0 - ph) * graph[cur][nxt][:tau] + q.to_f / length_of(path)
+        graph[cur][nxt][:tau] = graph[nxt][cur][:tau] = tau_next
+      end
+    end
+  end
+end

data/lib/ant_colony/cli.rb

@@ -0,0 +1,22 @@
+require 'thor'
+require 'yaml'
+
+module AntColony
+  class CLI < Thor
+    desc 'solve [FILE_NAME, START_POINT]', 'find the shortest path for graph from start point'
+
+    option :alpha, type: :numeric, desc: 'alpha param'
+    option :beta,  type: :numeric, desc: 'beta param'
+    option :ph,    type: :numeric, desc: 'pheromone constant'
+    option :q,     type: :numeric, desc: 'Q'
+    option :pop,   type: :numeric, desc: 'number of ants'
+
+    def solve(file, point)
+      params = options.map { |k, v| [k.to_sym, v] }.to_h
+      graph  = YAML.load_file(file)
+      colony = AntColony::Colony.new(graph, **params)
+      colony.solve
+      $stdout.puts colony.find_path(point.to_i)
+    end
+  end
+end

data/lib/ant_colony/version.rb

@@ -0,0 +1,3 @@
+module AntColony
+  VERSION = '0.0.5'.freeze
+end

data/spec/ant_colony_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper.rb'
+require 'yaml'
+
+describe AntColony::Colony do
+  let(:graph)  { YAML.load_file('test_data/test1.yml') }
+  let(:colony) { AntColony::Colony.new(graph, pop: 100) }
+
+  before { colony.solve }
+
+  it 'should find the shortest way' do
+    paths = graph.keys.permutation.map do |path|
+      [path.to_s, colony.length_of(path)]
+    end
+    shortest_length = paths.min_by { |_, v| v }.last
+    shortest_paths = paths.select { |_, v| v == shortest_length }
+
+    expect(shortest_paths.map(&:first)).to include colony.find_path(1).to_s
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+if ENV['SIMPLECOV']
+  require 'simplecov'
+  SimpleCov.start
+end
+
+require 'yaml'
+
+require_relative '../lib/ant_colony.rb'

data/test_data/test1.yml

@@ -0,0 +1,67 @@
+---
+1:
+  2:
+    :length: 26
+  3:
+    :length: 42
+  4:
+    :length: 15
+  5:
+    :length: 29
+  6:
+    :length: 25
+2:
+  1:
+    :length: 26
+  3:
+    :length: 16
+  4:
+    :length: 1
+  5:
+    :length: 30
+  6:
+    :length: 8
+3:
+  1:
+    :length: 42
+  2:
+    :length: 16
+  4:
+    :length: 35
+  5:
+    :length: 5
+  6:
+    :length: 1
+4:
+  1:
+    :length: 15
+  2:
+    :length: 1
+  3:
+    :length: 35
+  5:
+    :length: 18
+  6:
+    :length: 18
+5:
+  1:
+    :length: 29
+  2:
+    :length: 30
+  3:
+    :length: 5
+  4:
+    :length: 18
+  6:
+    :length: 5
+6:
+  1:
+    :length: 25
+  2:
+    :length: 8
+  3:
+    :length: 1
+  4:
+    :length: 18
+  5:
+    :length: 5

metadata

@@ -0,0 +1,133 @@
+--- !ruby/object:Gem::Specification
+name: ant_colony
+version: !ruby/object:Gem::Version
+  version: 0.0.5
+platform: ruby
+authors:
+- Alexandr Turchyn
+- Pete Matsyburka
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-03-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.4.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.4.0
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Simple implementation ant colony optimization algorithm.
+email:
+- lexfox777@gmail.com
+- pete.matsy@gmail.com
+executables:
+- ant_colony
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rubocop.yml"
+- Gemfile
+- README.md
+- Rakefile
+- ant_colony.gemspec
+- bin/ant_colony
+- lib/ant_colony.rb
+- lib/ant_colony/cli.rb
+- lib/ant_colony/version.rb
+- spec/ant_colony_spec.rb
+- spec/spec_helper.rb
+- test_data/test1.yml
+homepage: https://github.com/AlexandrToorchyn/AntColony
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Ant colony optimization algorithms.
+test_files:
+- spec/ant_colony_spec.rb
+- spec/spec_helper.rb
+has_rdoc: