or-tools 0.2.0 → 0.3.4

data/lib/or_tools/tsp.rb

@@ -0,0 +1,60 @@
+module ORTools
+  class TSP
+    attr_reader :route, :route_indexes, :distances, :total_distance
+
+    DISTANCE_SCALE = 1000
+    DEGREES_TO_RADIANS = Math::PI / 180
+
+    def initialize(locations)
+      raise ArgumentError, "Locations must have latitude and longitude" unless locations.all? { |l| l[:latitude] && l[:longitude] }
+      raise ArgumentError, "Latitude must be between -90 and 90" unless locations.all? { |l| l[:latitude] >= -90 && l[:latitude] <= 90 }
+      raise ArgumentError, "Longitude must be between -180 and 180" unless locations.all? { |l| l[:longitude] >= -180 && l[:longitude] <= 180 }
+      raise ArgumentError, "Must be at least two locations" unless locations.size >= 2
+
+      distance_matrix =
+        locations.map do |from|
+          locations.map do |to|
+            # must be integers
+            (distance(from, to) * DISTANCE_SCALE).to_i
+          end
+        end
+
+      manager = ORTools::RoutingIndexManager.new(locations.size, 1, 0)
+      routing = ORTools::RoutingModel.new(manager)
+
+      distance_callback = lambda do |from_index, to_index|
+        from_node = manager.index_to_node(from_index)
+        to_node = manager.index_to_node(to_index)
+        distance_matrix[from_node][to_node]
+      end
+
+      transit_callback_index = routing.register_transit_callback(distance_callback)
+      routing.set_arc_cost_evaluator_of_all_vehicles(transit_callback_index)
+      assignment = routing.solve(first_solution_strategy: :path_cheapest_arc)
+
+      @route_indexes = []
+      @distances = []
+
+      index = routing.start(0)
+      while !routing.end?(index)
+        @route_indexes << manager.index_to_node(index)
+        previous_index = index
+        index = assignment.value(routing.next_var(index))
+        @distances << routing.arc_cost_for_vehicle(previous_index, index, 0) / DISTANCE_SCALE.to_f
+      end
+      @route_indexes << manager.index_to_node(index)
+      @route = locations.values_at(*@route_indexes)
+      @total_distance = @distances.sum
+    end
+
+    private
+
+    def distance(from, to)
+      from_lat = from[:latitude] * DEGREES_TO_RADIANS
+      from_lng = from[:longitude] * DEGREES_TO_RADIANS
+      to_lat = to[:latitude] * DEGREES_TO_RADIANS
+      to_lng = to[:longitude] * DEGREES_TO_RADIANS
+      2 * 6371 * Math.asin(Math.sqrt(Math.sin((to_lat - from_lat) / 2.0)**2 + Math.cos(from_lat) * Math.cos(to_lat) * Math.sin((from_lng - to_lng) / 2.0)**2))
+    end
+  end
+end

data/lib/or_tools/version.rb

@@ -1,3 +1,3 @@
 module ORTools
-  VERSION = "0.2.0"
+  VERSION = "0.3.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: or-tools
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.4
 platform: ruby
 authors:
 - Andrew Kane
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-05-23 00:00:00.000000000 Z
+date: 2021-01-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -80,7 +80,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '5'
-description: 
+description:
 email: andrew@chartkick.com
 executables: []
 extensions:
@@ -95,6 +95,7 @@ files:
 - ext/or-tools/extconf.rb
 - ext/or-tools/vendor.rb
 - lib/or-tools.rb
+- lib/or_tools/basic_scheduler.rb
 - lib/or_tools/bool_var.rb
 - lib/or_tools/comparison.rb
 - lib/or_tools/comparison_operators.rb
@@ -104,16 +105,20 @@ files:
 - lib/or_tools/int_var.rb
 - lib/or_tools/knapsack_solver.rb
 - lib/or_tools/linear_expr.rb
+- lib/or_tools/routing_index_manager.rb
 - lib/or_tools/routing_model.rb
 - lib/or_tools/sat_int_var.rb
 - lib/or_tools/sat_linear_expr.rb
+- lib/or_tools/seating.rb
 - lib/or_tools/solver.rb
+- lib/or_tools/sudoku.rb
+- lib/or_tools/tsp.rb
 - lib/or_tools/version.rb
 homepage: https://github.com/ankane/or-tools
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -128,8 +133,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
-signing_key: 
+rubygems_version: 3.2.3
+signing_key:
 specification_version: 4
 summary: Operations research tools for Ruby
 test_files: []