| 125 | } |
| 126 | |
| 127 | MarkovSource PatternedNameGenerator::makeMarkovSource(size_t prefixSize, size_t endSize, StringList sourceNames) { |
| 128 | MarkovSource newSource; |
| 129 | newSource.prefixSize = prefixSize; |
| 130 | newSource.endSize = endSize; |
| 131 | for (auto sourceName : sourceNames) { |
| 132 | if (sourceName.length() < prefixSize || sourceName.length() < endSize) |
| 133 | continue; |
| 134 | |
| 135 | sourceName = sourceName.toLower(); |
| 136 | newSource.ends.add(sourceName.slice(sourceName.length() - endSize, sourceName.length())); |
| 137 | for (int i = 0; i + prefixSize <= sourceName.length(); ++i) { |
| 138 | auto prefix = sourceName.slice(i, i + prefixSize); |
| 139 | if (i == 0) |
| 140 | newSource.starts.append(prefix); |
| 141 | |
| 142 | if (i + prefixSize < sourceName.length()) { |
| 143 | if (!newSource.chains.contains(prefix)) |
| 144 | newSource.chains.insert(prefix, StringList()); |
| 145 | |
| 146 | newSource.chains.get(prefix).append(sourceName.slice(i + prefixSize, i + prefixSize + 1)); |
| 147 | } |
| 148 | } |
| 149 | } |
| 150 | return newSource; |
| 151 | } |
| 152 | |
| 153 | } |