4 #include <barvinok/util.h>
5 #include <barvinok/barvinok.h>
10 #include "omega/convert.h"
13 #include "verif_ehrhart.h"
14 #include "evalue_convert.h"
16 /* The input of this example program is a polytope in combined
17 * data and parameter space followed by two lines indicating
18 * the number of existential variables and parameters respectively.
19 * The first lines starts with "E ", followed by a number.
20 * The second lines starts with "P ", followed by a number.
21 * These two lines are (optionally) followed by the names of the parameters.
22 * The polytope is in PolyLib notation.
25 struct argp_option argp_options
[] = {
26 { "omega", 'o', 0, 0 },
28 { "series", 's', 0, 0 },
29 { "scarf", 'S', 0, 0 },
35 struct verify_options verify
;
36 struct convert_options convert
;
44 error_t
parse_opt(int key
, char *arg
, struct argp_state
*state
)
46 struct arguments
*arguments
= (struct arguments
*)(state
->input
);
50 state
->child_inputs
[0] = arguments
->verify
.barvinok
;
51 state
->child_inputs
[1] = &arguments
->verify
;
52 state
->child_inputs
[2] = &arguments
->convert
;
55 arguments
->series
= 1;
64 error(0, 0, "--omega option not supported");
71 error(0, 0, "--pip option not supported");
75 arguments
->verbose
= 1;
78 return ARGP_ERR_UNKNOWN
;
85 Polyhedron
*Omega_simplify(Polyhedron
*P
,
86 unsigned exist
, unsigned nparam
, char **parms
)
90 Relation r
= Polyhedron2relation(P
, exist
, nparam
, parms
);
92 return relation2Domain(r
, varv
, paramv
);
95 Polyhedron
*Omega_simplify(Polyhedron
*P
,
96 unsigned exist
, unsigned nparam
, char **parms
)
102 static void verify_results(Polyhedron
*P
, evalue
*EP
, int exist
, int nparam
,
103 verify_options
*options
);
105 int main(int argc
, char **argv
)
110 int exist
, nparam
, nvar
;
113 int print_solution
= 1;
114 struct arguments arguments
;
115 static struct argp_child argp_children
[] = {
116 { &barvinok_argp
, 0, 0, 0 },
117 { &verify_argp
, 0, "verification", 1 },
118 { &convert_argp
, 0, "output conversion", 2 },
121 static struct argp argp
= { argp_options
, parse_opt
, 0, 0, argp_children
};
122 struct barvinok_options
*options
= barvinok_options_new_with_defaults();
124 arguments
.verify
.barvinok
= options
;
128 arguments
.series
= 0;
129 arguments
.verbose
= 0;
131 argp_parse(&argp
, argc
, argv
, 0, 0, &arguments
);
133 if (arguments
.series
&& !arguments
.scarf
) {
135 "--series currently only available if --scarf is specified\n");
140 A
= Constraints2Polyhedron(MA
, options
->MaxRays
);
143 fgets(s
, 128, stdin
);
144 while ((*s
=='#') || (sscanf(s
, "E %d", &exist
)<1))
145 fgets(s
, 128, stdin
);
147 fgets(s
, 128, stdin
);
148 while ((*s
=='#') || (sscanf(s
, "P %d", &nparam
)<1))
149 fgets(s
, 128, stdin
);
151 /******* Read the options: initialize Min and Max ********/
153 if (arguments
.verify
.verify
) {
154 verify_options_set_range(&arguments
.verify
, A
->Dimension
);
155 if (!arguments
.verbose
)
159 if (print_solution
&& arguments
.verbose
) {
160 Polyhedron_Print(stdout
, P_VALUE_FMT
, A
);
161 printf("exist: %d, nparam: %d\n", exist
, nparam
);
163 param_name
= Read_ParamNames(stdin
, nparam
);
164 nvar
= A
->Dimension
- exist
- nparam
;
165 if (arguments
.omega
) {
166 A
= Omega_simplify(A
, exist
, nparam
, param_name
);
168 exist
= A
->Dimension
- nvar
- nparam
;
170 if (arguments
.series
) {
172 assert(arguments
.scarf
);
173 gf
= barvinok_enumerate_scarf_series(A
, exist
, nparam
, options
);
174 if (print_solution
) {
175 gf
->print(std::cout
, nparam
, param_name
);
181 EP
= barvinok_enumerate_scarf(A
, exist
, nparam
, options
);
182 else if (arguments
.pip
&& exist
> 0)
183 EP
= barvinok_enumerate_pip_with_options(A
, exist
, nparam
, options
);
185 EP
= barvinok_enumerate_e_with_options(A
, exist
, nparam
, options
);
187 if (evalue_convert(EP
, &arguments
.convert
, arguments
.verbose
, nparam
,
191 print_evalue(stdout
, EP
, param_name
);
192 if (arguments
.verify
.verify
) {
193 arguments
.verify
.params
= param_name
;
194 verify_results(A
, EP
, exist
, nparam
, &arguments
.verify
);
196 free_evalue_refs(EP
);
199 Free_ParamNames(param_name
, nparam
);
201 barvinok_options_free(options
);
205 void verify_results(Polyhedron
*P
, evalue
*EP
, int exist
, int nparam
,
206 verify_options
*options
)
212 unsigned MaxRays
= options
->barvinok
->MaxRays
;
213 Polyhedron
*C
= NULL
;
216 p
= (Value
*)malloc(sizeof(Value
) * (P
->Dimension
+2));
217 for(i
=0;i
<=P
->Dimension
;i
++) {
219 value_set_si(p
[i
],0);
222 value_set_si(p
[i
],1);
224 CS
= check_poly_context_scan(P
, &C
, nparam
, options
);
226 /* S = scanning list of polyhedra */
227 S
= Polyhedron_Scan(P
, C
, MaxRays
& POL_NO_DUAL
? 0 : MaxRays
);
229 check_poly_init(C
, options
);
231 /******* CHECK NOW *********/
233 if (S
&& !check_poly_EP(S
, CS
, EP
, exist
, nparam
, 0, p
, options
)) {
234 fprintf(stderr
,"Check failed !\n");
238 if (!options
->print_all
)
241 for(i
=0;i
<=(P
->Dimension
+1);i
++)