edomain.h

   1 #include <iostream>
   2 #include <vector>
   3 #include <gmp.h>
   4 extern "C" {
   5 #include <polylib/polylibgmp.h>
   6 }
   7 #include <barvinok/evalue.h>
   8 #include <barvinok/options.h>
   9
  10 struct EDomain_floor {
  11     int      refcount;
  12     evalue  *e;
  13     Vector  *v;
  14
  15     EDomain_floor(const evalue *f, int dim);
  16     ~EDomain_floor() {
  17         free_evalue_refs(e);
  18         delete e;
  19         Vector_Free(v);
  20     }
  21     EDomain_floor *ref() {
  22         ++refcount;
  23         return this;
  24     }
  25     static void unref(EDomain_floor* floor) {
  26         if (!--floor->refcount)
  27             delete floor;
  28     }
  29     void print(std::ostream& os, char **p) const;
  30 };
  31
  32 struct EDomain {
  33     Polyhedron          *D;
  34     Vector              *sample;
  35     std::vector<EDomain_floor *>        floors;
  36
  37     EDomain(Polyhedron *D) {
  38         this->D = Polyhedron_Copy(D);
  39         sample = NULL;
  40     }
  41     EDomain(Polyhedron *D, std::vector<EDomain_floor *>floors) {
  42         this->D = Polyhedron_Copy(D);
  43         add_floors(floors);
  44         sample = NULL;
  45     }
  46     EDomain(EDomain *ED) {
  47         this->D = Polyhedron_Copy(ED->D);
  48         add_floors(ED->floors);
  49         sample = NULL;
  50     }
  51     EDomain(Polyhedron *D, EDomain *ED, std::vector<EDomain_floor *>floors) {
  52         this->D = Polyhedron_Copy(D);
  53         add_floors(ED->floors);
  54         add_floors(floors);
  55         sample = NULL;
  56     }
  57     void add_floors(std::vector<EDomain_floor *>floors) {
  58         for (int i = 0; i < floors.size(); ++i)
  59             this->floors.push_back(floors[i]->ref());
  60     }
  61     int find_floor(evalue *needle) const {
  62         for (int i = 0; i < floors.size(); ++i)
  63             if (eequal(needle, floors[i]->e))
  64                 return i;
  65         return -1;
  66     }
  67     void print(FILE *out, char **p);
  68     void print_constraints(std::ostream& os, char **p,
  69                            barvinok_options *options) const;
  70     ~EDomain() {
  71         for (int i = 0; i < floors.size(); ++i)
  72             EDomain_floor::unref(floors[i]);
  73         Polyhedron_Free(D);
  74         if (sample)
  75             Vector_Free(sample);
  76     }
  77
  78     unsigned dimension() const {
  79         return D->Dimension - floors.size();
  80     }
  81
  82     Matrix *add_ge_constraint(evalue *constraint,
  83                                  std::vector<EDomain_floor *>& new_floors) const;
  84 };
  85
  86 int evalue2constraint(EDomain *D, evalue *E, Value *cons, int len);