1/*
2 * Copyright (c) 2015, Intel Corporation
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *
7 * * Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 * * Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 * * Neither the name of Intel Corporation nor the names of its contributors
13 * may be used to endorse or promote products derived from this software
14 * without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** \file
30 * \brief State for corpus generator.
31 */
32
33#include "config.h"
34
35#include "ng_corpus_properties.h"
36#include "ue2common.h"
37
38#include <boost/random/uniform_int_distribution.hpp>
39
40// default constructor
41CorpusProperties::CorpusProperties()
42 : matchness(100), unmatchness(0), randomness(0), prefixRange(0, 0),
43 suffixRange(0, 0), cycleMin(1), cycleMax(1),
44 corpusLimit(DEFAULT_CORPUS_GENERATOR_LIMIT), editDistance(0),
45 alphabetSize(~0) {
46 // empty
47}
48
49bool CorpusProperties::setPercentages(unsigned int match, unsigned int unmatch,
50 unsigned int random) {
51 if (match + unmatch + random != 100) {
52 // Do not update probabilities
53 return false;
54 }
55 matchness = match;
56 unmatchness = unmatch;
57 randomness = random;
58 return true;
59}
60
61void CorpusProperties::seed(unsigned val) {
62 rngSeed = val;
63 randomGen.seed(val);
64}
65
66unsigned CorpusProperties::getSeed() const {
67 return rngSeed;
68}
69
70unsigned CorpusProperties::rand(unsigned n, unsigned m) {
71 boost::random::uniform_int_distribution<> dist(n, m);
72 return dist(randomGen);
73}
74
75// not const because it stores state for the random number generator
76CorpusProperties::RollResult CorpusProperties::throwDice() {
77 if (matchness == 100) {
78 return ROLLED_MATCH;
79 }
80 if (unmatchness == 100) {
81 return ROLLED_UNMATCH;
82 }
83 if (randomness == 100) {
84 return ROLLED_RANDOM;
85 }
86
87 // This assumes a uniform distribution. Perhaps factor some 'depth' param
88 // and whether this 'depth' should increase or decrease the likelihood of
89 // unmatch or random rolls.
90 unsigned int outcome = rand(0, 99);
91 if (outcome < matchness) {
92 return ROLLED_MATCH;
93 }
94 if (outcome < matchness + unmatchness) {
95 return ROLLED_UNMATCH;
96 }
97
98 return ROLLED_RANDOM;
99}
100