genfun_constructor.h

   1 #include <gmp.h>
   2 #include <NTL/mat_ZZ.h>
   3 #include <barvinok/util.h>
   4 #include <barvinok/genfun.h>
   5 #include <barvinok/options.h>
   6 #include "reducer.h"
   7 #include "bfcounter.h"
   8
   9 #ifdef NTL_STD_CXX
  10 using namespace NTL;
  11 #endif
  12
  13 /* base for generating function counting */
  14 struct gf_base {
  15     np_base *base;
  16     gen_fun *gf;
  17
  18     gf_base(np_base *npb, Polyhedron *context) : base(npb) {
  19         gf = new gen_fun(context);
  20     }
  21     virtual ~gf_base() {
  22     };
  23     static gf_base *create(Polyhedron *context, unsigned dim, unsigned nparam,
  24                            barvinok_options *options);
  25
  26     void start_gf(Polyhedron *P, barvinok_options *options) {
  27         base->start(P, options);
  28     }
  29 };
  30
  31 struct partial_ireducer : public ireducer, public gf_base {
  32     partial_ireducer(Polyhedron *context, unsigned dim, unsigned nparam) :
  33             ireducer(dim), gf_base(this, context) {
  34         lower = nparam;
  35     }
  36     ~partial_ireducer() {
  37     }
  38     virtual void base(const QQ& c, const vec_ZZ& num, const mat_ZZ& den_f);
  39     virtual gen_fun *get_gf() {
  40         return gf;
  41     }
  42 };
  43
  44 struct partial_reducer : public reducer, public gf_base {
  45     vec_ZZ lambda;
  46     vec_ZZ tmp;
  47
  48     partial_reducer(Polyhedron *context, unsigned dim, unsigned nparam) :
  49             reducer(dim), gf_base(this, context) {
  50         lower = nparam;
  51
  52         tmp.SetLength(dim - nparam);
  53     }
  54     virtual void init(Polyhedron *P) {
  55         randomvector(P, lambda, dim - lower);
  56     }
  57     virtual void reset() {
  58         gf->clear_terms();
  59     }
  60     ~partial_reducer() {
  61     }
  62     virtual void base(const QQ& c, const vec_ZZ& num, const mat_ZZ& den_f);
  63     virtual void base(const vec_QQ& c, const mat_ZZ& num, const mat_ZZ& den_f);
  64
  65     virtual void split(const mat_ZZ& num, vec_ZZ& num_s, mat_ZZ& num_p,
  66                        const mat_ZZ& den_f, vec_ZZ& den_s, mat_ZZ& den_r);
  67     virtual gen_fun *get_gf() {
  68         return gf;
  69     }
  70 };
  71
  72 struct partial_bfcounter : public bfcounter_base, public gf_base {
  73     partial_bfcounter(Polyhedron *context, unsigned dim, unsigned nparam) :
  74             bfcounter_base(dim), gf_base(this, context) {
  75         lower = nparam;
  76     }
  77     ~partial_bfcounter() {
  78     }
  79     virtual void base(mat_ZZ& factors, bfc_vec& v);
  80 };