432 lines
8.7 KiB
C++
432 lines
8.7 KiB
C++
#ifndef YAP_PACKAGES_CLPBN_HORUS_UTIL_H_
|
|
#define YAP_PACKAGES_CLPBN_HORUS_UTIL_H_
|
|
|
|
#include <cmath>
|
|
#include <cassert>
|
|
|
|
#include <vector>
|
|
#include <queue>
|
|
#include <set>
|
|
#include <unordered_map>
|
|
#include <algorithm>
|
|
#include <limits>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <functional>
|
|
|
|
#include "Horus.h"
|
|
|
|
namespace Horus {
|
|
|
|
namespace {
|
|
|
|
const double NEG_INF = -std::numeric_limits<double>::infinity();
|
|
|
|
}
|
|
|
|
|
|
namespace Util {
|
|
|
|
template <typename T> void
|
|
addToVector (std::vector<T>&, const std::vector<T>&);
|
|
|
|
template <typename T> void
|
|
addToSet (std::set<T>&, const std::vector<T>&);
|
|
|
|
template <typename T> void
|
|
addToQueue (std::queue<T>&, const std::vector<T>&);
|
|
|
|
template <typename T> bool
|
|
contains (const std::vector<T>&, const T&);
|
|
|
|
template <typename T> bool contains
|
|
(const std::set<T>&, const T&);
|
|
|
|
template <typename K, typename V> bool
|
|
contains (const std::unordered_map<K, V>&, const K&);
|
|
|
|
template <typename T> size_t
|
|
indexOf (const std::vector<T>&, const T&);
|
|
|
|
template <class Operation> void
|
|
apply_n_times (Params& v1, const Params& v2, unsigned reps, Operation);
|
|
|
|
template <typename T> void
|
|
log (std::vector<T>&);
|
|
|
|
template <typename T> void
|
|
exp (std::vector<T>&);
|
|
|
|
template <typename T> std::string
|
|
elementsToString (const std::vector<T>& v, std::string sep = " ");
|
|
|
|
template <typename T> std::string
|
|
toString (const T&);
|
|
|
|
template <> std::string
|
|
toString (const bool&);
|
|
|
|
double logSum (double, double);
|
|
|
|
unsigned maxUnsigned();
|
|
|
|
unsigned stringToUnsigned (std::string);
|
|
|
|
double stringToDouble (std::string);
|
|
|
|
double factorial (unsigned);
|
|
|
|
double logFactorial (unsigned);
|
|
|
|
unsigned nrCombinations (unsigned, unsigned);
|
|
|
|
size_t sizeExpected (const Ranges&);
|
|
|
|
unsigned nrDigits (int);
|
|
|
|
bool isInteger (const std::string&);
|
|
|
|
std::string parametersToString (
|
|
const Params&, unsigned = Constants::precision);
|
|
|
|
std::vector<std::string> getStateLines (const Vars&);
|
|
|
|
bool setHorusFlag (std::string option, std::string value);
|
|
|
|
void printHeader (std::string, std::ostream& os = std::cout);
|
|
|
|
void printSubHeader (std::string, std::ostream& os = std::cout);
|
|
|
|
void printAsteriskLine (std::ostream& os = std::cout);
|
|
|
|
void printDashedLine (std::ostream& os = std::cout);
|
|
|
|
} // namespace Util
|
|
|
|
|
|
|
|
template <typename T> void
|
|
Util::addToVector (std::vector<T>& v, const std::vector<T>& elements)
|
|
{
|
|
v.insert (v.end(), elements.begin(), elements.end());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
Util::addToSet (std::set<T>& s, const std::vector<T>& elements)
|
|
{
|
|
s.insert (elements.begin(), elements.end());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
Util::addToQueue (std::queue<T>& q, const std::vector<T>& elements)
|
|
{
|
|
for (size_t i = 0; i < elements.size(); i++) {
|
|
q.push (elements[i]);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
template <typename T> bool
|
|
Util::contains (const std::vector<T>& v, const T& e)
|
|
{
|
|
return std::find (v.begin(), v.end(), e) != v.end();
|
|
}
|
|
|
|
|
|
|
|
template <typename T> bool
|
|
Util::contains (const std::set<T>& s, const T& e)
|
|
{
|
|
return s.find (e) != s.end();
|
|
}
|
|
|
|
|
|
|
|
template <typename K, typename V> bool
|
|
Util::contains (const std::unordered_map<K, V>& m, const K& k)
|
|
{
|
|
return m.find (k) != m.end();
|
|
}
|
|
|
|
|
|
|
|
template <typename T> size_t
|
|
Util::indexOf (const std::vector<T>& v, const T& e)
|
|
{
|
|
return std::distance (v.begin(),
|
|
std::find (v.begin(), v.end(), e));
|
|
}
|
|
|
|
|
|
|
|
template <class Operation> void
|
|
Util::apply_n_times (
|
|
Params& v1,
|
|
const Params& v2,
|
|
unsigned repetitions,
|
|
Operation unary_op)
|
|
{
|
|
Params::iterator first = v1.begin();
|
|
Params::const_iterator last = v1.end();
|
|
Params::const_iterator first2 = v2.begin();
|
|
Params::const_iterator last2 = v2.end();
|
|
while (first != last) {
|
|
for (first2 = v2.begin(); first2 != last2; ++first2) {
|
|
std::transform (first, first + repetitions, first,
|
|
std::bind1st (unary_op, *first2));
|
|
first += repetitions;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
Util::log (std::vector<T>& v)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(), ::log);
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
Util::exp (std::vector<T>& v)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(), ::exp);
|
|
}
|
|
|
|
|
|
|
|
template <typename T> std::string
|
|
Util::elementsToString (const std::vector<T>& v, std::string sep)
|
|
{
|
|
std::stringstream ss;
|
|
for (size_t i = 0; i < v.size(); i++) {
|
|
ss << ((i != 0) ? sep : "") << v[i];
|
|
}
|
|
return ss.str();
|
|
}
|
|
|
|
|
|
|
|
template <typename T> std::string
|
|
Util::toString (const T& t)
|
|
{
|
|
std::stringstream ss;
|
|
ss << t;
|
|
return ss.str();
|
|
}
|
|
|
|
|
|
|
|
inline double
|
|
Util::logSum (double x, double y)
|
|
{
|
|
// std::log (std::exp (x) + std::exp (y)) can overflow!
|
|
assert (std::isnan (x) == false);
|
|
assert (std::isnan (y) == false);
|
|
if (x == NEG_INF) {
|
|
return y;
|
|
}
|
|
if (y == NEG_INF) {
|
|
return x;
|
|
}
|
|
// if one value is much smaller than the other,
|
|
// keep the larger value
|
|
const double tol = 460.517; // log (1e200)
|
|
if (x < y - tol) {
|
|
return y;
|
|
}
|
|
if (y < x - tol) {
|
|
return x;
|
|
}
|
|
assert (std::isnan (x - y) == false);
|
|
const double exp_diff = std::exp (x - y);
|
|
if (std::isfinite (exp_diff) == false) {
|
|
// difference is too large
|
|
return x > y ? x : y;
|
|
}
|
|
// otherwise return the sum
|
|
return y + std::log (static_cast<double>(1.0) + exp_diff);
|
|
}
|
|
|
|
|
|
|
|
inline unsigned
|
|
Util::maxUnsigned()
|
|
{
|
|
return std::numeric_limits<unsigned>::max();
|
|
}
|
|
|
|
|
|
|
|
namespace LogAware {
|
|
|
|
inline double one() { return Globals::logDomain ? 0.0 : 1.0; }
|
|
inline double zero() { return Globals::logDomain ? NEG_INF : 0.0; }
|
|
inline double addIdenty() { return Globals::logDomain ? NEG_INF : 0.0; }
|
|
inline double multIdenty() { return Globals::logDomain ? 0.0 : 1.0; }
|
|
inline double withEvidence() { return Globals::logDomain ? 0.0 : 1.0; }
|
|
inline double noEvidence() { return Globals::logDomain ? NEG_INF : 0.0; }
|
|
inline double log (double v) { return Globals::logDomain ? ::log (v) : v; }
|
|
inline double exp (double v) { return Globals::logDomain ? ::exp (v) : v; }
|
|
|
|
void normalize (Params&);
|
|
|
|
double getL1Distance (const Params&, const Params&);
|
|
|
|
double getMaxNorm (const Params&, const Params&);
|
|
|
|
double pow (double, unsigned);
|
|
|
|
double pow (double, double);
|
|
|
|
void pow (Params&, unsigned);
|
|
|
|
void pow (Params&, double);
|
|
|
|
} // namespace LogAware
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator+=(std::vector<T>& v, double val)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::plus<double>(), val));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator-=(std::vector<T>& v, double val)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::minus<double>(), val));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator*=(std::vector<T>& v, double val)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::multiplies<double>(), val));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator/=(std::vector<T>& v, double val)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::divides<double>(), val));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator+=(std::vector<T>& a, const std::vector<T>& b)
|
|
{
|
|
assert (a.size() == b.size());
|
|
std::transform (a.begin(), a.end(), b.begin(), a.begin(),
|
|
std::plus<double>());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator-=(std::vector<T>& a, const std::vector<T>& b)
|
|
{
|
|
assert (a.size() == b.size());
|
|
std::transform (a.begin(), a.end(), b.begin(), a.begin(),
|
|
std::minus<double>());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator*=(std::vector<T>& a, const std::vector<T>& b)
|
|
{
|
|
assert (a.size() == b.size());
|
|
std::transform (a.begin(), a.end(), b.begin(), a.begin(),
|
|
std::multiplies<double>());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator/=(std::vector<T>& a, const std::vector<T>& b)
|
|
{
|
|
assert (a.size() == b.size());
|
|
std::transform (a.begin(), a.end(), b.begin(), a.begin(),
|
|
std::divides<double>());
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator^=(std::vector<T>& v, double exp)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::ptr_fun<double, double, double> (std::pow), exp));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> void
|
|
operator^=(std::vector<T>& v, int iexp)
|
|
{
|
|
std::transform (v.begin(), v.end(), v.begin(),
|
|
std::bind2nd (std::ptr_fun<double, int, double> (std::pow), iexp));
|
|
}
|
|
|
|
|
|
|
|
template <typename T> std::ostream&
|
|
operator<< (std::ostream& os, const std::vector<T>& v)
|
|
{
|
|
os << "[" ;
|
|
os << Util::elementsToString (v, ", ");
|
|
os << "]" ;
|
|
return os;
|
|
}
|
|
|
|
|
|
namespace FuncObj {
|
|
|
|
template<typename T>
|
|
struct max : public std::binary_function<T, T, T> {
|
|
T operator() (const T& x, const T& y) const {
|
|
return x < y ? y : x;
|
|
}};
|
|
|
|
|
|
|
|
template <typename T>
|
|
struct abs_diff : public std::binary_function<T, T, T> {
|
|
T operator() (const T& x, const T& y) const {
|
|
return std::abs (x - y);
|
|
}};
|
|
|
|
|
|
|
|
template <typename T>
|
|
struct abs_diff_exp : public std::binary_function<T, T, T> {
|
|
T operator() (const T& x, const T& y) const {
|
|
return std::abs (std::exp (x) - std::exp (y));
|
|
}};
|
|
|
|
} // namespace FuncObj
|
|
|
|
} // namespace Horus
|
|
|
|
#endif // YAP_PACKAGES_CLPBN_HORUS_UTIL_H_
|
|
|