summaryrefslogtreecommitdiff
path: root/base/src/main/java/bjc/utils/gen
diff options
context:
space:
mode:
Diffstat (limited to 'base/src/main/java/bjc/utils/gen')
-rw-r--r--base/src/main/java/bjc/utils/gen/RandomGrammar.java20
-rw-r--r--base/src/main/java/bjc/utils/gen/WeightedGrammar.java223
-rw-r--r--base/src/main/java/bjc/utils/gen/WeightedRandom.java27
3 files changed, 126 insertions, 144 deletions
diff --git a/base/src/main/java/bjc/utils/gen/RandomGrammar.java b/base/src/main/java/bjc/utils/gen/RandomGrammar.java
index 978accb..b418c79 100644
--- a/base/src/main/java/bjc/utils/gen/RandomGrammar.java
+++ b/base/src/main/java/bjc/utils/gen/RandomGrammar.java
@@ -9,7 +9,7 @@ import bjc.utils.funcdata.IList;
* @author ben
*
* @param <E>
- * The type of grammar elements to use.
+ * The type of grammar elements to use.
*/
public class RandomGrammar<E> extends WeightedGrammar<E> {
/** Create a new random grammar. */
@@ -21,14 +21,14 @@ public class RandomGrammar<E> extends WeightedGrammar<E> {
* Add cases to a specified rule.
*
* @param rule
- * The name of the rule to add cases to.
+ * The name of the rule to add cases to.
*
* @param cases
- * The cases to add for this rule.
+ * The cases to add for this rule.
*/
@SafeVarargs
public final void addCases(final E rule, final IList<E>... cases) {
- for (final IList<E> currentCase : cases) {
+ for(final IList<E> currentCase : cases) {
super.addCase(rule, 1, currentCase);
}
}
@@ -37,16 +37,16 @@ public class RandomGrammar<E> extends WeightedGrammar<E> {
* Create a rule with the specified name and cases.
*
* @param rule
- * The name of the rule to add.
+ * The name of the rule to add.
*
* @param cases
- * The cases to add for this rule.
+ * The cases to add for this rule.
*/
@SafeVarargs
public final void makeRule(final E rule, final IList<E>... cases) {
super.addRule(rule);
- for (final IList<E> currentCase : cases) {
+ for(final IList<E> currentCase : cases) {
super.addCase(rule, 1, currentCase);
}
}
@@ -55,13 +55,13 @@ public class RandomGrammar<E> extends WeightedGrammar<E> {
* Create a rule with the specified name and cases.
*
* @param rule
- * The name of the rule to add.
+ * The name of the rule to add.
*
* @param cases
- * The cases to add for this rule.
+ * The cases to add for this rule.
*/
public void makeRule(final E rule, final IList<IList<E>> cases) {
- if (cases == null) throw new NullPointerException("Cases must not be null");
+ if(cases == null) throw new NullPointerException("Cases must not be null");
super.addRule(rule);
diff --git a/base/src/main/java/bjc/utils/gen/WeightedGrammar.java b/base/src/main/java/bjc/utils/gen/WeightedGrammar.java
index 1fbba02..472c3a3 100644
--- a/base/src/main/java/bjc/utils/gen/WeightedGrammar.java
+++ b/base/src/main/java/bjc/utils/gen/WeightedGrammar.java
@@ -19,7 +19,7 @@ import bjc.utils.funcdata.IMap;
* @author ben
*
* @param <E>
- * The values that make up sentences of this grammar.
+ * The values that make up sentences of this grammar.
*/
public class WeightedGrammar<E> {
/** The initial rule of the grammar */
@@ -55,12 +55,12 @@ public class WeightedGrammar<E> {
* randomness.
*
* @param source
- * The source of randomness to use
+ * The source of randomness to use
*/
public WeightedGrammar(final Random source) {
this();
- if (source == null) throw new NullPointerException("Source of randomness must be non-null");
+ if(source == null) throw new NullPointerException("Source of randomness must be non-null");
rng = source;
}
@@ -69,10 +69,10 @@ public class WeightedGrammar<E> {
* Configure the action to perform on special tokens.
*
* @param marker
- * The marker to find special tokens.
+ * The marker to find special tokens.
*
* @param action
- * The action to take on those tokens.
+ * The action to take on those tokens.
*/
public void configureSpecial(final Predicate<E> marker,
final BiFunction<E, WeightedGrammar<E>, IList<E>> action) {
@@ -84,15 +84,15 @@ public class WeightedGrammar<E> {
* Adds a special rule to the grammar.
*
* @param ruleName
- * The name of the special rule.
+ * The name of the special rule.
*
* @param cse
- * The case for the rule.
+ * The case for the rule.
*/
public void addSpecialRule(final E ruleName, final Supplier<IList<E>> cse) {
- if (ruleName == null) {
+ if(ruleName == null) {
throw new NullPointerException("Rule name must not be null");
- } else if (cse == null) {
+ } else if(cse == null) {
throw new NullPointerException("Case must not be null");
}
@@ -103,18 +103,18 @@ public class WeightedGrammar<E> {
* Add a case to an already existing rule.
*
* @param ruleName
- * The rule to add a case to.
+ * The rule to add a case to.
*
* @param probability
- * The probability for this rule to be chosen.
+ * The probability for this rule to be chosen.
*
* @param cse
- * The case being added.
+ * The case being added.
*/
public void addCase(final E ruleName, final int probability, final IList<E> cse) {
- if (ruleName == null) {
+ if(ruleName == null) {
throw new NullPointerException("Rule name must be not null");
- } else if (cse == null) {
+ } else if(cse == null) {
throw new NullPointerException("Case body must not be null");
}
@@ -125,24 +125,23 @@ public class WeightedGrammar<E> {
* Add a alias for an existing subgrammar.
*
* @param name
- * The name of the subgrammar to alias.
+ * The name of the subgrammar to alias.
*
* @param alias
- * The alias of the subgrammar.
+ * The alias of the subgrammar.
*
- * @return
- * Whether the alias was succesfully created.
+ * @return Whether the alias was succesfully created.
*/
public boolean addGrammarAlias(final E name, final E alias) {
- if (name == null) {
+ if(name == null) {
throw new NullPointerException("Subgrammar name must not be null");
- } else if (alias == null) {
+ } else if(alias == null) {
throw new NullPointerException("Subgrammar alias must not be null");
}
- if (subgrammars.containsKey(alias)) return false;
+ if(subgrammars.containsKey(alias)) return false;
- if (subgrammars.containsKey(name)) {
+ if(subgrammars.containsKey(name)) {
subgrammars.put(alias, subgrammars.get(name));
return true;
}
@@ -154,17 +153,16 @@ public class WeightedGrammar<E> {
* Add a new rule with no cases.
*
* @param name
- * The name of the rule to add.
+ * The name of the rule to add.
*
- * @return
- * Whether or not the rule was successfully added.
+ * @return Whether or not the rule was successfully added.
*/
public boolean addRule(final E name) {
- if (rng == null) {
+ if(rng == null) {
rng = new Random();
}
- if (name == null) throw new NullPointerException("Rule name must not be null");
+ if(name == null) throw new NullPointerException("Rule name must not be null");
return addRule(name, new WeightedRandom<>(rng));
}
@@ -173,22 +171,21 @@ public class WeightedGrammar<E> {
* Add a new rule with a set of cases.
*
* @param name
- * The name of the rule to add.
+ * The name of the rule to add.
*
* @param cases
- * The set of cases for the rule.
+ * The set of cases for the rule.
*
- * @return
- * Whether or not the rule was succesfully added.
+ * @return Whether or not the rule was succesfully added.
*/
public boolean addRule(final E name, final WeightedRandom<IList<E>> cases) {
- if (name == null) {
+ if(name == null) {
throw new NullPointerException("Name must not be null");
- } else if (cases == null) {
+ } else if(cases == null) {
throw new NullPointerException("Cases must not be null");
}
- if (rules.containsKey(name)) return false;
+ if(rules.containsKey(name)) return false;
rules.put(name, cases);
return true;
@@ -198,22 +195,21 @@ public class WeightedGrammar<E> {
* Add a subgrammar.
*
* @param name
- * The name of the subgrammar.
+ * The name of the subgrammar.
*
* @param subgrammar
- * The subgrammar to add.
+ * The subgrammar to add.
*
- * @return
- * Whether or not the subgrammar was succesfully added.
+ * @return Whether or not the subgrammar was succesfully added.
*/
public boolean addSubgrammar(final E name, final WeightedGrammar<E> subgrammar) {
- if (name == null) {
+ if(name == null) {
throw new NullPointerException("Subgrammar name must not be null");
- } else if (subgrammar == null) {
+ } else if(subgrammar == null) {
throw new NullPointerException("Subgrammar must not be null");
}
- if (subgrammars.containsKey(name)) return false;
+ if(subgrammars.containsKey(name)) return false;
subgrammars.put(name, subgrammar);
return true;
@@ -223,10 +219,10 @@ public class WeightedGrammar<E> {
* Remove a rule with the specified name.
*
* @param name
- * The name of the rule to remove.
+ * The name of the rule to remove.
*/
public void deleteRule(final E name) {
- if (name == null) throw new NullPointerException("Rule name must not be null");
+ if(name == null) throw new NullPointerException("Rule name must not be null");
rules.remove(name);
}
@@ -235,10 +231,10 @@ public class WeightedGrammar<E> {
* Remove a subgrammar with the specified name.
*
* @param name
- * The name of the subgrammar to remove.
+ * The name of the subgrammar to remove.
*/
public void deleteSubgrammar(final E name) {
- if (name == null) throw new NullPointerException("Rule name must not be null");
+ if(name == null) throw new NullPointerException("Rule name must not be null");
subgrammars.remove(name);
}
@@ -249,19 +245,18 @@ public class WeightedGrammar<E> {
* Only generates sentences one layer deep.
*
* @param ruleName
- * The rule to test.
+ * The rule to test.
*
- * @return
- * A set of sentences generated by the specified rule.
+ * @return A set of sentences generated by the specified rule.
*/
public IList<IList<E>> generateDebugValues(final E ruleName) {
- if (ruleName == null) throw new NullPointerException("Rule name must not be null");
+ if(ruleName == null) throw new NullPointerException("Rule name must not be null");
final IList<IList<E>> returnedList = new FunctionalList<>();
final WeightedRandom<IList<E>> ruleGenerator = rules.get(ruleName);
- for (int i = 0; i < 10; i++) {
+ for(int i = 0; i < 10; i++) {
returnedList.add(ruleGenerator.generateValue());
}
@@ -272,27 +267,27 @@ public class WeightedGrammar<E> {
* Generate a generic sentence from a initial rule.
*
* @param <T>
- * The type of the transformed output
+ * The type of the transformed output
*
* @param initRules
- * The initial rule to start with.
+ * The initial rule to start with.
*
* @param tokenTransformer
- * The function to transform grammar output into something.
+ * The function to transform grammar output into something.
*
* @param spacer
- * The spacer element to add in between output tokens.
+ * The spacer element to add in between output tokens.
*
- * @return
- * A randomly generated sentence from the specified initial rule.
+ * @return A randomly generated sentence from the specified initial
+ * rule.
*/
public <T> IList<T> generateGenericValues(final E initRules, final Function<E, T> tokenTransformer,
final T spacer) {
- if (initRules == null) {
+ if(initRules == null) {
throw new NullPointerException("Initial rule must not be null");
- } else if (tokenTransformer == null) {
+ } else if(tokenTransformer == null) {
throw new NullPointerException("Transformer must not be null");
- } else if (spacer == null) {
+ } else if(spacer == null) {
throw new NullPointerException("Spacer must not be null");
}
@@ -300,43 +295,42 @@ public class WeightedGrammar<E> {
IList<E> genRules = new FunctionalList<>(initRules);
- if (specialMarker != null) {
- if (specialMarker.test(initRules)) {
+ if(specialMarker != null) {
+ if(specialMarker.test(initRules)) {
genRules = specialAction.apply(initRules, this);
}
}
/*
- * @NOTE
- * Can this loop be simplified in some way?
+ * @NOTE Can this loop be simplified in some way?
*/
- for (final E initRule : genRules.toIterable()) {
- if (specialRules.containsKey(initRule)) {
- for (final E rulePart : specialRules.get(initRule).get().toIterable()) {
+ for(final E initRule : genRules.toIterable()) {
+ if(specialRules.containsKey(initRule)) {
+ for(final E rulePart : specialRules.get(initRule).get().toIterable()) {
final Iterable<T> generatedRuleParts = generateGenericValues(rulePart,
tokenTransformer, spacer).toIterable();
- for (final T generatedRulePart : generatedRuleParts) {
+ for(final T generatedRulePart : generatedRuleParts) {
returnedList.add(generatedRulePart);
returnedList.add(spacer);
}
}
- } else if (subgrammars.containsKey(initRule)) {
+ } else if(subgrammars.containsKey(initRule)) {
final Iterable<T> ruleParts = subgrammars.get(initRule)
.generateGenericValues(initRule, tokenTransformer, spacer).toIterable();
- for (final T rulePart : ruleParts) {
+ for(final T rulePart : ruleParts) {
returnedList.add(rulePart);
returnedList.add(spacer);
}
- } else if (rules.containsKey(initRule)) {
+ } else if(rules.containsKey(initRule)) {
final Iterable<E> ruleParts = rules.get(initRule).generateValue().toIterable();
- for (final E rulePart : ruleParts) {
+ for(final E rulePart : ruleParts) {
final Iterable<T> generatedRuleParts = generateGenericValues(rulePart,
tokenTransformer, spacer).toIterable();
- for (final T generatedRulePart : generatedRuleParts) {
+ for(final T generatedRulePart : generatedRuleParts) {
returnedList.add(generatedRulePart);
returnedList.add(spacer);
}
@@ -344,7 +338,7 @@ public class WeightedGrammar<E> {
} else {
final T transformedToken = tokenTransformer.apply(initRule);
- if (transformedToken == null)
+ if(transformedToken == null)
throw new NullPointerException("Transformer created null token");
returnedList.add(transformedToken);
@@ -359,14 +353,13 @@ public class WeightedGrammar<E> {
* Generate a random list of grammar elements from a given initial rule.
*
* @param initRule
- * The initial rule to start with.
+ * The initial rule to start with.
*
* @param spacer
- * The item to use to space the list.
+ * The item to use to space the list.
*
- * @return
- * A list of random grammar elements generated by the specified
- * rule.
+ * @return A list of random grammar elements generated by the specified
+ * rule.
*/
public IList<E> generateListValues(final E initRule, final E spacer) {
final IList<E> retList = generateGenericValues(initRule, strang -> strang, spacer);
@@ -377,8 +370,7 @@ public class WeightedGrammar<E> {
/**
* Get the initial rule of this grammar.
*
- * @return
- * The initial rule of this grammar.
+ * @return The initial rule of this grammar.
*/
public String getInitialRule() {
return initialRule;
@@ -387,8 +379,7 @@ public class WeightedGrammar<E> {
/**
* Returns the number of rules in this grammar.
*
- * @return
- * The number of rules in this grammar.
+ * @return The number of rules in this grammar.
*/
public int getRuleCount() {
return rules.size();
@@ -397,8 +388,7 @@ public class WeightedGrammar<E> {
/**
* Returns a set containing all of the rules in this grammar.
*
- * @return
- * The set of all rule names in this grammar.
+ * @return The set of all rule names in this grammar.
*/
public IList<E> getRuleNames() {
final IList<E> ruleNames = new FunctionalList<>();
@@ -413,13 +403,12 @@ public class WeightedGrammar<E> {
* Get the subgrammar with the specified name.
*
* @param name
- * The name of the subgrammar to get.
+ * The name of the subgrammar to get.
*
- * @return
- * The subgrammar with the specified name.
+ * @return The subgrammar with the specified name.
*/
public WeightedGrammar<E> getSubgrammar(final E name) {
- if (name == null) throw new NullPointerException("Subgrammar name must not be null");
+ if(name == null) throw new NullPointerException("Subgrammar name must not be null");
return subgrammars.get(name);
}
@@ -427,8 +416,7 @@ public class WeightedGrammar<E> {
/**
* Check if this grammar has an initial rule
*
- * @return
- * Whether or not this grammar has an initial rule
+ * @return Whether or not this grammar has an initial rule
*/
public boolean hasInitialRule() {
return initialRule != null && !initialRule.equalsIgnoreCase("");
@@ -438,10 +426,9 @@ public class WeightedGrammar<E> {
* Check if this grammar has a given rule.
*
* @param ruleName
- * The rule to check for.
+ * The rule to check for.
*
- * @return
- * Whether or not the grammar has a rule by that name.
+ * @return Whether or not the grammar has a rule by that name.
*/
public boolean hasRule(final E ruleName) {
return rules.containsKey(ruleName) || specialRules.containsKey(ruleName);
@@ -451,24 +438,24 @@ public class WeightedGrammar<E> {
* Prefix a given rule with a token multiple times.
*
* @param ruleName
- * The name of the rule to prefix.
+ * The name of the rule to prefix.
*
* @param prefixToken
- * The token to prefix to the rules.
+ * The token to prefix to the rules.
*
* @param additionalProbability
- * The additional probability of the tokens.
+ * The additional probability of the tokens.
*
* @param numberOfTimes
- * The number of times to prefix the token.
+ * The number of times to prefix the token.
*/
public void multiPrefixRule(final E ruleName, final E prefixToken, final int additionalProbability,
final int numberOfTimes) {
- if (ruleName == null)
+ if(ruleName == null)
throw new NullPointerException("Rule name must not be null");
- else if (prefixToken == null)
+ else if(prefixToken == null)
throw new NullPointerException("Prefix token must not be null");
- else if (numberOfTimes < 1)
+ else if(numberOfTimes < 1)
throw new IllegalArgumentException("Number of times to prefix must be positive.");
final WeightedRandom<IList<E>> rule = rules.get(ruleName);
@@ -476,24 +463,23 @@ public class WeightedGrammar<E> {
final IList<IPair<Integer, IList<E>>> newResults = new FunctionalList<>();
/*
- * @NOTE
- * Can this be simplified?
+ * @NOTE Can this be simplified?
*/
rule.getValues().forEach((pair) -> {
final IList<IList<E>> newRule = new FunctionalList<>();
- for (int i = 1; i <= numberOfTimes; i++) {
+ for(int i = 1; i <= numberOfTimes; i++) {
final IList<E> newCase = pair.merge((left, right) -> {
final IList<E> returnVal = new FunctionalList<>();
- for (final E val : right.toIterable()) {
+ for(final E val : right.toIterable()) {
returnVal.add(val);
}
return returnVal;
});
- for (int j = 1; j <= i; j++) {
+ for(int j = 1; j <= i; j++) {
newCase.prepend(prefixToken);
}
@@ -519,19 +505,18 @@ public class WeightedGrammar<E> {
* given token.
*
* @param additionalProbability
- * The amount to adjust the probability by.
+ * The amount to adjust the probability by.
*
* @param ruleName
- * The name of the rule to prefix.
+ * The name of the rule to prefix.
*
* @param prefixToken
- * The token to prefix to the rule.
+ * The token to prefix to the rule.
*/
public void prefixRule(final E ruleName, final E prefixToken, final int additionalProbability) {
- if (ruleName == null) {
+ if(ruleName == null) {
throw new NullPointerException("Rule name must not be null");
- }
- else if (prefixToken == null) {
+ } else if(prefixToken == null) {
throw new NullPointerException("Prefix token must not be null");
}
@@ -543,7 +528,7 @@ public class WeightedGrammar<E> {
final IList<E> newCase = pair.merge((left, right) -> {
final IList<E> returnVal = new FunctionalList<>();
- for (final E val : right.toIterable()) {
+ for(final E val : right.toIterable()) {
returnVal.add(val);
}
@@ -562,7 +547,7 @@ public class WeightedGrammar<E> {
* Set the initial rule of the grammar.
*
* @param initRule
- * The initial rule of this grammar.
+ * The initial rule of this grammar.
*/
public void setInitialRule(final String initRule) {
this.initialRule = initRule;
@@ -572,18 +557,18 @@ public class WeightedGrammar<E> {
* Suffix a token to a rule.
*
* @param ruleName
- * The rule to suffix.
+ * The rule to suffix.
*
* @param suffixToken
- * The token to prefix to the rule.
+ * The token to prefix to the rule.
*
* @param additionalProbability
- * Additional probability of the prefixed rule.
+ * Additional probability of the prefixed rule.
*/
public void suffixRule(final E ruleName, final E suffixToken, final int additionalProbability) {
- if (ruleName == null) {
+ if(ruleName == null) {
throw new NullPointerException("Rule name must not be null");
- } else if (suffixToken == null) {
+ } else if(suffixToken == null) {
throw new NullPointerException("Prefix token must not be null");
}
@@ -595,7 +580,7 @@ public class WeightedGrammar<E> {
final IList<E> newCase = par.merge((left, right) -> {
final IList<E> returnVal = new FunctionalList<>();
- for (final E val : right.toIterable()) {
+ for(final E val : right.toIterable()) {
returnVal.add(val);
}
diff --git a/base/src/main/java/bjc/utils/gen/WeightedRandom.java b/base/src/main/java/bjc/utils/gen/WeightedRandom.java
index abba36a..be8b8c2 100644
--- a/base/src/main/java/bjc/utils/gen/WeightedRandom.java
+++ b/base/src/main/java/bjc/utils/gen/WeightedRandom.java
@@ -15,7 +15,7 @@ import bjc.utils.funcdata.IList;
* @author ben
*
* @param <E>
- * The type of values that are randomly selected.
+ * The type of values that are randomly selected.
*/
public class WeightedRandom<E> {
/* The list of probabilities for each result */
@@ -33,13 +33,13 @@ public class WeightedRandom<E> {
* randomness.
*
* @param src
- * The source of randomness to use.
+ * The source of randomness to use.
*/
public WeightedRandom(final Random src) {
probabilities = new FunctionalList<>();
results = new FunctionalList<>();
- if (src == null) throw new NullPointerException("Source of randomness must not be null");
+ if(src == null) throw new NullPointerException("Source of randomness must not be null");
source = src;
}
@@ -48,10 +48,10 @@ public class WeightedRandom<E> {
* Add a probability for a specific result to be given.
*
* @param chance
- * The chance to get this result.
+ * The chance to get this result.
*
* @param result
- * The result to get when the chance comes up.
+ * The result to get when the chance comes up.
*/
public void addProbability(final int chance, final E result) {
probabilities.add(chance);
@@ -63,17 +63,16 @@ public class WeightedRandom<E> {
/**
* Generate a weighted random value.
*
- * @return
- * A random value selected in a weighted fashion.
+ * @return A random value selected in a weighted fashion.
*/
public E generateValue() {
- final IHolder<Integer> value = new Identity<>(source.nextInt(totalChance));
- final IHolder<E> current = new Identity<>();
+ final IHolder<Integer> value = new Identity<>(source.nextInt(totalChance));
+ final IHolder<E> current = new Identity<>();
final IHolder<Boolean> picked = new Identity<>(true);
probabilities.forEachIndexed((index, probability) -> {
- if (picked.unwrap(bool -> bool)) {
- if (value.unwrap((number) -> number < probability)) {
+ if(picked.unwrap(bool -> bool)) {
+ if(value.unwrap((number) -> number < probability)) {
current.transform((result) -> results.getByIndex(index));
picked.transform((bool) -> false);
@@ -89,8 +88,7 @@ public class WeightedRandom<E> {
/**
* Return a list of values that can be generated by this generator
*
- * @return
- * A list of all the values that can be generated
+ * @return A list of all the values that can be generated
*/
public IList<E> getResults() {
return results;
@@ -100,8 +98,7 @@ public class WeightedRandom<E> {
* Return a list containing values that can be generated paired with the
* probability of those values being generated
*
- * @return
- * A list of pairs of values and value probabilities
+ * @return A list of pairs of values and value probabilities
*/
public IList<IPair<Integer, E>> getValues() {
return probabilities.pairWith(results);
generated by cgit v1.2.3 (git 2.25.1) at 2026-05-10 08:54:01 +0000