Current section

Files

Jump to
fixpoint lib solver space space.ex
Raw

lib/solver/space/space.ex

defmodule CPSolver.Space do
@moduledoc """
Computation space.
The concept is taken from Chapter 12, "Concepts, Techniques, and Models
of Computer Programming" by Peter Van Roy and Seif Haridi.
"""
alias CPSolver.Utils
alias CPSolver.ConstraintStore
alias CPSolver.Solution, as: Solution
alias CPSolver.DefaultDomain, as: Domain
alias CPSolver.Propagator.ConstraintGraph
alias CPSolver.Utils
alias CPSolver.Space.Propagation
alias CPSolver.Objective
alias CPSolver.Shared
require Logger
@behaviour GenServer
def default_space_opts() do
[
store_impl: CPSolver.ConstraintStore.default_store(),
solution_handler: Solution.default_handler(),
search: CPSolver.Search.Strategy.default_strategy(),
postpone: false
]
end
def create(variables, propagators, space_opts \\ default_space_opts()) do
propagators = maybe_add_objective_propagator(propagators, space_opts[:objective])
{:ok, _space} =
create(%{
variables: variables,
propagators: propagators,
constraint_graph: ConstraintGraph.create(propagators),
opts: space_opts
})
end
defp maybe_add_objective_propagator(propagators, nil) do
propagators
end
defp maybe_add_objective_propagator(propagators, objective) do
[objective.propagator | propagators]
end
def create(data) do
if CPSolver.complete?(data.opts[:solver_data]) do
{:error, :complete}
else
GenServer.start(__MODULE__, data)
end
end
def start_propagation(space_pid) when is_pid(space_pid) do
:done = GenServer.call(space_pid, :propagate, :infinity)
end
@impl true
def init(%{variables: variables, opts: space_opts, constraint_graph: graph} = data) do
{:ok, space_variables, store} =
ConstraintStore.create_store(variables,
store_impl: space_opts[:store_impl],
space: self()
)
space_data =
data
|> maybe_bind_objective_variable(store)
|> Map.put(:id, make_ref())
|> Map.put(:variables, space_variables)
|> Map.put(:store, store)
|> Map.put(:constraint_graph, ConstraintGraph.remove_fixed(graph, space_variables))
Shared.add_active_spaces(data.opts[:solver_data], [self()])
{:ok, space_data, {:continue, :propagate}}
end
defp maybe_bind_objective_variable(data, store) do
case data.opts[:objective] do
nil ->
data
objective ->
put_in(data, [:opts, :objective], Objective.bind_to_store(objective, store))
end
end
@impl true
def handle_continue(:propagate, data) do
(data.opts[:postpone] && {:noreply, data}) ||
data
|> propagate()
|> tap(fn _ ->
caller = Map.get(data, :caller)
caller && GenServer.reply(caller, :done)
end)
end
@impl true
def handle_call(:propagate, caller, data) do
{:noreply,
data
|> Map.put(:caller, caller)
|> Map.put(:opts, Keyword.put(data.opts, :postpone, false)), {:continue, :propagate}}
end
defp propagate(
%{
propagators: propagators,
variables: variables,
constraint_graph: constraint_graph,
store: store
} =
data
) do
case Propagation.run(propagators, constraint_graph, store) do
:fail ->
handle_failure(data)
:solved ->
handle_solved(data)
{:stable, reduced_constraint_graph, reduced_propagators} ->
%{
data
| constraint_graph: reduced_constraint_graph,
propagators: reduced_propagators,
variables: variables
}
|> handle_stable()
end
end
defp handle_failure(data) do
shutdown(data, :failure)
end
defp handle_solved(data) do
data
|> solution()
|> then(fn
:fail ->
shutdown(data, :fail)
solution ->
maybe_tighten_objective_bound(data.opts[:objective])
Solution.run_handler(solution, data.opts[:solution_handler])
shutdown(data, :solved)
end)
end
defp solution(%{variables: variables, store: store} = _data) do
Enum.reduce_while(variables, Map.new(), fn var, acc ->
case ConstraintStore.get(store, var, :min) do
:fail -> {:halt, :fail}
val -> {:cont, Map.put(acc, var.name, val)}
end
end)
end
defp maybe_tighten_objective_bound(nil) do
:ok
end
defp maybe_tighten_objective_bound(objective) do
Objective.tighten(objective)
end
defp handle_stable(%{variables: variables} = data) do
{localized_vars, _all_fixed?} = Utils.localize_variables(variables)
distribute(%{data | variables: localized_vars})
end
def distribute(
%{
opts: opts,
variables: localized_variables
} = data
) do
{:ok, {var_to_branch_on, domain_partitions}} = branching(localized_variables, opts[:search])
Enum.take_while(domain_partitions, fn partition ->
variable_copies =
Enum.map(localized_variables, fn %{id: clone_id} = clone ->
if clone_id == var_to_branch_on.id do
Map.put(clone, :domain, Domain.new(partition))
else
clone
end
end)
case create(
data
|> Map.put(:variables, variable_copies)
|> Map.put(:opts, Keyword.put(data.opts, :postpone, true))
) do
{:ok, space_pid} -> spawn(fn -> start_propagation(space_pid) end)
{:error, :complete} -> false
end
end)
shutdown(data, :distribute)
end
defp branching(variables, search_strategy) do
case search_strategy.select_variable(variables) do
{:ok, var_to_branch_on} ->
var_domain = var_to_branch_on.domain
case search_strategy.partition(var_domain) do
:fail -> :fail
{:ok, partitions} -> {:ok, {var_to_branch_on, partitions}}
end
error ->
error
end
end
defp shutdown(data, reason) do
Shared.remove_space(data.opts[:solver_data], self(), reason)
{:stop, :normal, data}
end
end