1 package net.bmahe.genetics4j.gp.mutation;
2
3 import java.util.ArrayList;
4 import java.util.List;
5 import java.util.Objects;
6 import java.util.Optional;
7
8 import org.apache.commons.lang3.Validate;
9 import org.apache.logging.log4j.LogManager;
10 import org.apache.logging.log4j.Logger;
11
12 import net.bmahe.genetics4j.core.Genotype;
13 import net.bmahe.genetics4j.core.chromosomes.Chromosome;
14 import net.bmahe.genetics4j.core.chromosomes.TreeChromosome;
15 import net.bmahe.genetics4j.core.chromosomes.TreeNode;
16 import net.bmahe.genetics4j.core.mutation.Mutator;
17 import net.bmahe.genetics4j.core.spec.AbstractEAConfiguration;
18 import net.bmahe.genetics4j.core.spec.chromosome.ChromosomeSpec;
19 import net.bmahe.genetics4j.gp.Operation;
20 import net.bmahe.genetics4j.gp.program.Program;
21 import net.bmahe.genetics4j.gp.spec.chromosome.ProgramTreeChromosomeSpec;
22 import net.bmahe.genetics4j.gp.spec.mutation.Rule;
23 import net.bmahe.genetics4j.gp.utils.TreeNodeUtils;
24
25 public class ProgramRulesApplicatorMutator implements Mutator {
26 final static public Logger logger = LogManager.getLogger(ProgramRulesApplicatorMutator.class);
27
28 private final List<Rule> rules;
29 private final AbstractEAConfiguration eaConfiguration;
30
31 public ProgramRulesApplicatorMutator(final List<Rule> _rules, final AbstractEAConfiguration _eaConfiguration) {
32 Objects.requireNonNull(_rules);
33 Validate.isTrue(_rules.isEmpty() == false);
34 Objects.requireNonNull(_eaConfiguration);
35
36 this.rules = _rules;
37 this.eaConfiguration = _eaConfiguration;
38 }
39
40 protected TreeNode<Operation<?>> duplicateAndApplyRule(final Program program, final TreeNode<Operation<?>> root) {
41 Objects.requireNonNull(root);
42
43 logger.trace("A - {}", TreeNodeUtils.toStringTreeNode(root));
44
45 final Operation<?> rootOperation = root.getData();
46 final List<TreeNode<Operation<?>>> children = root.getChildren();
47
48 TreeNode<Operation<?>> currentRoot = new TreeNode<Operation<?>>(rootOperation);
49
50 if (children.size() > 0) {
51 final List<TreeNode<Operation<?>>> newChildren = new ArrayList<>();
52 for (int i = 0; i < children.size(); i++) {
53 final TreeNode<Operation<?>> childNode = children.get(i);
54
55 final TreeNode<Operation<?>> childCopy = duplicateAndApplyRule(program, childNode);
56 newChildren.add(childCopy);
57
58 }
59 currentRoot.addChildren(newChildren);
60 }
61
62 boolean done = false;
63 while (done == false) {
64 final TreeNode<Operation<?>> localRoot = currentRoot;
65
66 logger.trace("B - {}", TreeNodeUtils.toStringTreeNode(localRoot));
67
68 final Optional<Rule> applicableRule = rules.stream().filter((rule) -> rule.test(localRoot)).findFirst();
69 final Optional<TreeNode<Operation<?>>> newRootOpt = applicableRule.map(x -> x.apply(program, localRoot));
70
71 done = newRootOpt.map(newRoot -> TreeNodeUtils.areSame(newRoot, localRoot)).orElse(true);
72
73 if (newRootOpt.isPresent()) {
74 currentRoot = newRootOpt.get();
75 }
76 }
77
78 return currentRoot;
79 }
80
81 @Override
82 public Genotype mutate(final long generation, final Genotype original) {
83 Objects.requireNonNull(original);
84
85 logger.trace("Mutating genotype: {}", original);
86
87 final Chromosome[] newChromosomes = new Chromosome[original.getSize()];
88 final Chromosome[] chromosomes = original.getChromosomes();
89 for (int chromosomeIndex = 0; chromosomeIndex < chromosomes.length; chromosomeIndex++) {
90 final ChromosomeSpec chromosomeSpec = eaConfiguration.getChromosomeSpec(chromosomeIndex);
91 final Chromosome chromosome = chromosomes[chromosomeIndex];
92
93 if (chromosomeSpec instanceof ProgramTreeChromosomeSpec == false) {
94 throw new IllegalArgumentException("This mutator does not support chromosome specs " + chromosomeSpec);
95 }
96
97 if (chromosome instanceof TreeChromosome<?> == false) {
98 throw new IllegalArgumentException(
99 "This mutator does not support chromosome of type " + chromosome.getClass().getSimpleName());
100 }
101
102 final ProgramTreeChromosomeSpec programTreeChromosomeSpec = (ProgramTreeChromosomeSpec) chromosomeSpec;
103
104 final TreeChromosome<Operation<?>> treeChromosome = (TreeChromosome<Operation<?>>) chromosome;
105 final TreeNode<Operation<?>> root = treeChromosome.getRoot();
106
107 if (logger.isTraceEnabled()) {
108 logger.trace("Original chromosome {} - {}", chromosomeIndex, TreeNodeUtils.toStringTreeNode(root));
109 }
110
111 final TreeNode<Operation<?>> newRoot = duplicateAndApplyRule(programTreeChromosomeSpec.program(), root);
112 final TreeChromosome<Operation<?>> newTreeChromosome = new TreeChromosome<>(newRoot);
113 newChromosomes[chromosomeIndex] = newTreeChromosome;
114
115 }
116
117 return new Genotype(newChromosomes);
118 }
119 }