You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
2.3 KiB
63 lines
2.3 KiB
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include "src/builder/jit/DistributionEntry.h"
|
|
|
|
namespace storm {
|
|
namespace builder {
|
|
namespace jit {
|
|
|
|
template <typename IndexType, typename ValueType>
|
|
class Distribution {
|
|
public:
|
|
typedef std::vector<DistributionEntry<IndexType, ValueType>> ContainerType;
|
|
|
|
Distribution();
|
|
|
|
/*!
|
|
* Adds the given entry to the distribution.
|
|
*/
|
|
void add(DistributionEntry<IndexType, ValueType> const& entry);
|
|
|
|
/*!
|
|
* Adds the given entry to the distribution.
|
|
*/
|
|
void add(IndexType const& index, ValueType const& value);
|
|
|
|
/*!
|
|
* Adds the given other distribution to the distribution.
|
|
*/
|
|
void add(Distribution&& distribution);
|
|
|
|
/*!
|
|
* Compresses the internal storage by summing the values of entries which agree on the index. As a side
|
|
* effect, this sorts the entries in the distribution by their index.
|
|
*/
|
|
void compress();
|
|
|
|
/*!
|
|
* Divides all values in the distribution by the provided value.
|
|
*/
|
|
void divide(ValueType const& value);
|
|
|
|
/*!
|
|
* Access to iterators over the entries of the distribution. Note that there may be multiple entries for
|
|
* the same index. Also, no order is guaranteed. After a call to compress, the order is guaranteed to be
|
|
* ascending wrt. index and there are no elements with the same index.
|
|
*/
|
|
typename ContainerType::iterator begin();
|
|
typename ContainerType::const_iterator begin() const;
|
|
typename ContainerType::iterator end();
|
|
typename ContainerType::const_iterator end() const;
|
|
|
|
private:
|
|
// The underlying storage of the distribution.
|
|
ContainerType storage;
|
|
|
|
bool compressed;
|
|
};
|
|
|
|
}
|
|
}
|
|
}
|