/usr/include/shark/Rng/DiffGeometric.h is in libshark-dev 3.1.3+ds1-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | /*!
*
*
* \brief Diff geometric distribution.
*
*
*
* \author O. Krause
* \date 2010-01-01
*
*
* \par Copyright 1995-2015 Shark Development Team
*
* <BR><HR>
* This file is part of Shark.
* <http://image.diku.dk/shark/>
*
* Shark 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 3 of the License, or
* (at your option) any later version.
*
* Shark 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 Shark. If not, see <http://www.gnu.org/licenses/>.
*
*/
#ifndef SHARK_RNG_DIFFGEOMETRIC_H
#define SHARK_RNG_DIFFGEOMETRIC_H
#include <shark/Rng/Rng.h>
#include <boost/random.hpp>
#include <boost/random/geometric_distribution.hpp>
#ifndef BOOST_RANDOM_NO_STREAM_OPERATORS
#include <iostream>
#endif
#include <cmath>
namespace shark{
/**
* \brief Implements a diff geometric distribution.
*/
template<class IntType = int, class RealType = double>
class DiffGeometric_distribution
{
public:
typedef RealType input_type;
typedef IntType result_type;
explicit DiffGeometric_distribution(const RealType& p = RealType(0.5))
:geom(p) {}
RealType p() const
{
return geom.p();
}
void reset() { }
template<class Engine>
result_type operator()(Engine& eng)
{
return geom(eng)-geom(eng);
}
#ifndef BOOST_RANDOM_NO_STREAM_OPERATORS
template<class CharT, class Traits>
friend std::basic_ostream<CharT,Traits>&
operator<<(std::basic_ostream<CharT,Traits>& os, const DiffGeometric_distribution& gd)
{
os << gd.geom;
return os;
}
template<class CharT, class Traits>
friend std::basic_istream<CharT,Traits>&
operator>>(std::basic_istream<CharT,Traits>& is, DiffGeometric_distribution& gd)
{
is >> gd.geom;
return is;
}
#endif
private:
boost::geometric_distribution<IntType,RealType> geom;
};
/**
* \brief Random variable with diff geometric distribution.
*/
template<typename RngType = shark::DefaultRngType>
class DiffGeometric: public boost::variate_generator<RngType*,DiffGeometric_distribution<> >
{
private:
typedef boost::variate_generator<RngType*,DiffGeometric_distribution<> > Base;
public:
DiffGeometric( RngType & rng, double mean = 0.5 )
:Base(&rng,DiffGeometric_distribution<>(1.0-mean))
{}
using Base::operator();
double operator()(double mean)
{
DiffGeometric_distribution<> dist(mean);
return dist(Base::engine());
}
double mean()const
{
return 1-Base::distribution().p();
}
void mean(double newMean)
{
Base::distribution()=DiffGeometric_distribution<>(1-newMean);
}
double p(double x)const
{
return( mean() * std::pow( 1 - mean(), std::abs(x) ) / ( 2 - mean() ) );
}
};
}
#endif
|