Commit | Line | Data |
60cda014 |
1 | package DX::State; |
2 | |
5622b4df |
3 | use Return::MultiLevel qw(with_return); |
71217e42 |
4 | use DX::Op::Backtrack; |
b373788e |
5 | use Scalar::Util qw(blessed); |
60cda014 |
6 | use Moo; |
7 | |
8 | has next_op => (is => 'ro', required => 1); |
9 | |
10 | has return_stack => (is => 'ro', required => 1); |
11 | |
12 | has by_id => (is => 'ro', required => 1); |
13 | |
14 | has scope => (is => 'ro', required => 1); |
15 | |
16 | has last_choice => (is => 'ro', required => 1); |
17 | |
b373788e |
18 | has id_gen => (is => 'ro', default => sub { {} }); |
19 | |
b40e416a |
20 | has rule_set => (is => 'ro'); |
21 | |
22 | has facts => (is => 'ro'); |
23 | |
60cda014 |
24 | sub scope_var { |
25 | my ($self, $name) = @_; |
26 | $self->by_id->{$self->scope->{$name}}; |
27 | } |
28 | |
b373788e |
29 | sub allocate_var { |
606537d1 |
30 | my ($self, $name, $var, $id_gen) = @_; |
31 | my $id = join('_', $name, ++($id_gen->{$name}||='000')); |
b373788e |
32 | DX::Var->new(id => $id, %$var); |
33 | } |
34 | |
606537d1 |
35 | sub expand_vars { |
b373788e |
36 | my ($self, %vars) = @_; |
37 | my %by_id = %{$self->by_id}; |
606537d1 |
38 | my %id_gen = %{$self->id_gen}; |
b373788e |
39 | foreach my $name (keys %vars) { |
b373788e |
40 | unless (blessed($vars{$name})) { |
606537d1 |
41 | my $var = $vars{$name} = $self->allocate_var( |
42 | $name, $vars{$name}, \%id_gen |
43 | ); |
b373788e |
44 | $by_id{$var->id} = $var; |
45 | } |
46 | } |
606537d1 |
47 | $self->but( |
48 | by_id => \%by_id, id_gen => \%id_gen, |
49 | ), %vars; |
50 | } |
51 | |
52 | sub assign_vars { |
53 | my ($self, %vars) = @_; |
54 | my ($state, %expanded) = $self->expand_vars(%vars); |
b40e416a |
55 | $state->but( |
606537d1 |
56 | scope => { %{$self->scope}, map +($_ => $expanded{$_}->id), keys %expanded } |
b373788e |
57 | ); |
58 | } |
59 | |
0de441cd |
60 | sub bind_value { |
61 | my ($self, $var, $value) = @_; |
b40d5c51 |
62 | my $bound = $var->with_value($value); |
63 | $self->but( |
64 | by_id => { %{$self->by_id}, $var->id => $bound }, |
b40d5c51 |
65 | ); |
66 | } |
67 | |
60cda014 |
68 | sub bind_stream_then { |
69 | my ($self, $var, $stream, $then) = @_; |
60cda014 |
70 | my $bound = $var->with_stream($stream); |
03079510 |
71 | $self->but( |
60cda014 |
72 | by_id => { %{$self->by_id}, $var->id => $bound }, |
73 | next_op => $then |
74 | )->mark_choice($bound); |
75 | } |
76 | |
d95799c4 |
77 | sub bind_root_set_then { |
1b6999da |
78 | my ($self, $var_id, $set, $then) = @_; |
79 | my $bound = $self->by_id->{$var_id}->with_root_set($set); |
d95799c4 |
80 | $self->but( |
1b6999da |
81 | by_id => { %{$self->by_id}, $var_id => $bound }, |
d95799c4 |
82 | next_op => $then |
83 | )->mark_choice($bound); |
84 | } |
85 | |
60cda014 |
86 | sub mark_choice { |
87 | my ($self, $var) = @_; |
03079510 |
88 | $self->but(last_choice => [ $self, $var ]); |
60cda014 |
89 | } |
90 | |
91 | sub backtrack { |
92 | my ($self) = @_; |
94565614 |
93 | my ($state, $var) = ($self); |
94 | while (($state, $var) = @{$state->last_choice}) { |
60cda014 |
95 | $var->bound_value; $var->clear_bound_value; |
94565614 |
96 | return $state->mark_choice($var) unless $var->bound_stream->is_exhausted; |
60cda014 |
97 | } |
b40d5c51 |
98 | $self->return_from_run(undef); |
60cda014 |
99 | } |
100 | |
101 | sub then { |
102 | my ($self, $then) = @_; |
03079510 |
103 | $self->but(next_op => $then); |
60cda014 |
104 | } |
105 | |
b40d5c51 |
106 | sub return_from_run { |
107 | my (undef, $return) = @_; |
71217e42 |
108 | (our $Run_Return)->($return); |
109 | } |
110 | |
111 | sub return_from_op { |
112 | my (undef, $return) = @_; |
113 | if (our $Op_Return) { |
114 | $Op_Return->($return); |
115 | } |
b40d5c51 |
116 | } |
117 | |
94565614 |
118 | sub run { |
119 | my ($state) = @_; |
5622b4df |
120 | with_return { |
121 | my ($return) = @_; |
71217e42 |
122 | local our $Run_Return = $return; |
54817920 |
123 | while (my $op = $state->next_op) { |
71217e42 |
124 | my $backtrack = with_return { |
125 | my ($return) = @_; |
126 | local our $Op_Return = $return; |
127 | $state = $op->run($state); |
128 | return; |
129 | }; |
130 | if ($backtrack) { $state = $state->backtrack } |
5622b4df |
131 | } |
132 | return $state; |
94565614 |
133 | } |
5622b4df |
134 | } |
135 | |
136 | sub push_backtrack { |
71217e42 |
137 | $_[0]->then(DX::Op::Backtrack->new); |
94565614 |
138 | } |
139 | |
03079510 |
140 | sub but { |
141 | my ($self, @but) = @_; |
142 | $self->new(%$self, @but); |
143 | } |
144 | |
b40d5c51 |
145 | sub pop_return_stack { |
146 | my ($self) = @_; |
147 | my @stack = @{$self->return_stack}; |
148 | my $top = pop @stack; |
734376d9 |
149 | $self->but(return_stack => \@stack, next_op => $top->[0]); |
b40d5c51 |
150 | } |
151 | |
152 | sub push_return_then { |
153 | my ($self, $return, $then) = @_; |
154 | $self->but( |
734376d9 |
155 | return_stack => [ @{$self->return_stack}, [ $return, $self ] ], |
b40d5c51 |
156 | next_op => $then |
157 | ); |
158 | } |
159 | |
60cda014 |
160 | 1; |