BernoulliRng.h

Go to the documentation of this file.

/*
// $Id: //open/dev/fennel/common/BernoulliRng.h#5 $
// Fennel is a library of data storage and processing components.
// Copyright (C) 2007-2009 The Eigenbase Project
// Copyright (C) 2007-2009 SQLstream, Inc.
// Copyright (C) 2007-2009 LucidEra, Inc.
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 2 of the License, or (at your option)
// any later version approved by The Eigenbase Project.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#ifndef Fennel_BernoulliRng_Included
#define Fennel_BernoulliRng_Included

#include <boost/random.hpp>
#include <boost/cstdint.hpp>

FENNEL_BEGIN_NAMESPACE

class FENNEL_COMMON_EXPORT BernoulliRng
{
private:
    boost::mt19937 uniformRng;

    boost::bernoulli_distribution<float> bernoulliDist;

    boost::variate_generator<
        boost::mt19937 &,
        boost::bernoulli_distribution<float> > rng;

public:
    explicit BernoulliRng(float successProbability);

    void reseed(boost::uint32_t seed);

    bool nextValue();
};

FENNEL_END_NAMESPACE

#endif

// End BernoulliRng.h

---