1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 package org.sat4j.opt;
29
30 import org.sat4j.specs.ISolver;
31 import org.sat4j.specs.TimeoutException;
32 import org.sat4j.tools.SolverDecorator;
33
34
35
36
37
38
39
40
41
42
43
44 public abstract class AbstractSelectorVariablesDecorator extends
45 SolverDecorator<ISolver> {
46
47
48
49
50 private static final long serialVersionUID = 1L;
51
52 protected int nborigvars;
53
54 private int nbexpectedclauses;
55
56 protected int nbnewvar;
57
58 protected int[] prevfullmodel;
59
60 private int [] prevmodel;
61 private boolean [] prevboolmodel;
62
63 public AbstractSelectorVariablesDecorator(ISolver solver) {
64 super(solver);
65 }
66
67 @Override
68 public int[] model() {
69 return prevmodel;
70 }
71
72 @Override
73 public boolean model(int var) {
74 return prevboolmodel[var-1];
75 }
76
77 @Override
78 public int newVar(int howmany) {
79 nborigvars = super.newVar(howmany);
80 return nborigvars;
81 }
82
83 @Override
84 public void setExpectedNumberOfClauses(int nb) {
85 nbexpectedclauses = nb;
86 super.setExpectedNumberOfClauses(nb);
87 super.newVar(nborigvars + nbexpectedclauses);
88 }
89
90 public int getExpectedNumberOfClauses() {
91 return nbexpectedclauses;
92 }
93
94 @Override
95 public void reset() {
96 super.reset();
97 nbnewvar = 0;
98 }
99
100 public boolean admitABetterSolution() throws TimeoutException {
101 boolean result = super.isSatisfiable(true);
102 if (result) {
103 prevfullmodel = super.model();
104 prevboolmodel = new boolean[nVars()];
105 for (int i=0;i<nVars();i++) {
106 prevboolmodel[i]=decorated().model(i+1);
107 }
108 int end = nborigvars - 1;
109 while (Math.abs(prevfullmodel[end]) > nborigvars)
110 end--;
111 prevmodel = new int[end + 1];
112 for (int i = 0; i <= end; i++) {
113 prevmodel[i] = prevfullmodel[i];
114 }
115 }
116 return result;
117 }
118
119 }