RandomDistributor.h

Go to the documentation of this file.

/****************************************************************************/
// Represents a generic random distribution
/****************************************************************************/
// SUMO, Simulation of Urban MObility; see http://sumo.sourceforge.net/
// Copyright 2001-2010 DLR (http://www.dlr.de/) and contributors
/****************************************************************************/
//
//   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.
//
/****************************************************************************/
#ifndef RandomDistributor_h
#define RandomDistributor_h


// ===========================================================================
// included modules
// ===========================================================================
#ifdef _MSC_VER
#include <windows_config.h>
#else
#include <config.h>
#endif

#include <cassert>
#include <utils/common/RandHelper.h>
#include <utils/common/UtilExceptions.h>


// ===========================================================================
// class definitions
// ===========================================================================
template<class T>
class RandomDistributor {
public:
    RandomDistributor() : myProb(0) { }

    ~RandomDistributor() { }

    void add(SUMOReal prob, T val, bool checkDuplicates=true) {
        assert(prob>=0);
        myProb += prob;
        if (checkDuplicates) {
            for (size_t i=0; i<myVals.size(); i++) {
                if (val==myVals[i]) {
                    myProbs[i] += prob;
                    return;
                }
            }
        }
        myVals.push_back(val);
        myProbs.push_back(prob);
    }

    T get() const {
        if (myProb==0) {
            throw OutOfBoundsException();
        }
        SUMOReal prob = RandHelper::rand(myProb);
        for (size_t i=0; i<myVals.size(); i++) {
            if (prob<myProbs[i]) {
                return myVals[i];
            }
            prob -= myProbs[i];
        }
        return myVals.back();
    }

    SUMOReal getOverallProb() const {
        return myProb;
    }

    void clear() {
        myProb = 0;
        myVals.clear();
        myProbs.clear();
    }

    const std::vector<T> &getVals() const {
        return myVals;
    }

    const std::vector<SUMOReal> &getProbs() const {
        return myProbs;
    }

private:
    SUMOReal myProb;
    std::vector<T> myVals;
    std::vector<SUMOReal> myProbs;

};


#endif

/****************************************************************************/