enumerate_ext.h

Go to the documentation of this file.

/*
   (C) 2016 Marc Stevens.
 
   This file is part of fplll. fplll is free software: you
   can redistribute it and/or modify it under the terms of the GNU Lesser
   General Public License as published by the Free Software Foundation,
   either version 2.1 of the License, or (at your option) any later version.
 
   fplll is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
   GNU Lesser General Public License for more details.
 
   You should have received a copy of the GNU Lesser General Public License
   along with fplll. If not, see <http://www.gnu.org/licenses/>. */
 
#ifndef FPLLL_ENUMERATE_EXT_H
#define FPLLL_ENUMERATE_EXT_H
 
#include "enumerate_ext_api.h"
 
#include <fplll/enum/enumerate_base.h>
#include <fplll/enum/evaluator.h>
#include <fplll/gso_interface.h>
 
FPLLL_BEGIN_NAMESPACE
 
/* function callback API for external enumeration library (extenum) */
 
// typedef void(extenum_cb_set_config)(enumf *mu, size_t mudim, bool mutranspose, enumf *rdiag,
//                                    enumf *pruning);
using ::extenum_cb_set_config;
 
// typedef enumf(extenum_cb_process_sol)(enumf dist, enumf *sol);
using ::extenum_cb_process_sol;
 
// typedef void(extenum_cb_process_subsol)(enumf dist, enumf *subsol, int offset);
using ::extenum_cb_process_subsol;
 
// typedef array<uint64_t, FPLLL_EXTENUM_MAX_EXTENUM_DIM>(extenum_fc_enumerate)(
//    const int dim, enumf maxdist, std::function<extenum_cb_set_config> cbfunc,
//    std::function<extenum_cb_process_sol> cbsol, std::function<extenum_cb_process_subsol>
//    cbsubsol, bool dual /*=false*/, bool findsubsols /*=false*/
//);
using ::extenum_fc_enumerate;
 
/* set & get external enumerator. If extenum = nullptr then this interface is disabled,
                                  and fplll will use the internal enumerator.
                                  Otherwise, fplll will use the enumeration function pointed to
                                  by extenum.
*/
void set_external_enumerator(std::function<extenum_fc_enumerate> extenum = nullptr);
std::function<extenum_fc_enumerate> get_external_enumerator();
 
template <typename ZT, typename FT> class ExternalEnumeration
{
public:
  ExternalEnumeration(MatGSOInterface<ZT, FT> &gso, Evaluator<FT> &evaluator)
      : _gso(gso), _evaluator(evaluator)
  {
  }
 
  bool enumerate(int first, int last, FT &fmaxdist, long fmaxdistexpo,
                 const vector<enumf> &pruning = vector<enumf>(), bool dual = false);
 
  // get_nodes. This returns the number of nodes visited by the external enumeration process.
  // If this returns 0, then fplll will fall back to the internal enumerator.
  inline uint64_t get_nodes(const int level = -1) const
  {
    if (level == -1)
    {
      return std::accumulate(_nodes.begin(), _nodes.end(), 0);
    }
    return _nodes[level];
  }
 
  inline std::array<uint64_t, FPLLL_EXTENUM_MAX_EXTENUM_DIM> get_nodes_array() const
  {
    return _nodes;
  }
 
private:
  void callback_set_config(enumf *mu, size_t mudim, bool mutranspose, enumf *rdiag, enumf *pruning);
 
  enumf callback_process_sol(enumf dist, enumf *sol);
 
  void callback_process_subsol(enumf dist, enumf *subsol, int offset);
 
  MatGSOInterface<ZT, FT> &_gso;
  Evaluator<FT> &_evaluator;
  vector<enumf> _pruning;
  long _normexp;
 
  array<uint64_t, FPLLL_EXTENUM_MAX_EXTENUM_DIM> _nodes;
  bool _dual;
  int _d, _first;
  enumf _maxdist;
  vector<FT> _fx;
};
 
FPLLL_END_NAMESPACE
 
#endif