Manuel Bustillo 85e1ec453e
Some checks failed
Run unit tests / copyright_notice (pull_request) Successful in 2m15s
Run unit tests / rubocop (pull_request) Successful in 2m31s
Run unit tests / check-licenses (pull_request) Successful in 2m46s
Run unit tests / unit_tests (pull_request) Successful in 4m11s
Run unit tests / build-static-assets (pull_request) Has been cancelled
Add copyright notice
2025-01-25 08:41:22 +00:00

64 lines
1.4 KiB
Ruby

# Copyright (C) 2024 - 2025 LibreWeddingPlanner contributors
# Copyright (C) 2024-2025 LibreWeddingPlanner contributors
# frozen_string_literal: true
module VNS
class Engine
class << self
def sequence(elements)
elements = elements.to_a
(elements + elements.reverse).chunk(&:itself).map(&:first)
end
end
def target_function(&function)
@target_function = function
end
def add_perturbation(klass)
@perturbations ||= Set.new
@perturbations << klass
end
attr_writer :initial_solution
def run
raise 'No target function defined' unless @target_function
raise 'No perturbations defined' unless @perturbations
raise 'No initial solution defined' unless @initial_solution
@best_solution = @initial_solution
@best_score = @target_function.call(@best_solution)
self.class.sequence(@perturbations).each do |perturbation|
optimize(perturbation)
end
@best_solution
end
private
def optimize(perturbation_klass)
loop do
optimized = false
perturbation_klass.new(@best_solution).each do |alternative_solution|
score = @target_function.call(alternative_solution)
next if score >= @best_score
@best_solution = alternative_solution.deep_dup
@best_score = score
optimized = true
break
end
return unless optimized
end
end
end
end