/** Generate a sentence that includes (if possible) the specified word. */
public String getSentence(String word) {
LinkedList parts = new LinkedList();
Quad[] quads;
if (words.containsKey(word)) {
quads = (Quad[]) ((HashSet) words.get(word)).toArray(new Quad[0]);
} else {
quads = (Quad[]) this.quads.keySet().toArray(new Quad[0]);
}
if (quads.length == 0) {
return "";
}
Quad middleQuad = quads[rand.nextInt(quads.length)];
Quad quad = middleQuad;
for (int i = 0; i < 4; i++) {
parts.add(quad.getToken(i));
}
while (quad.canEnd() == false) {
String[] nextTokens = (String[]) ((HashSet) next.get(quad)).toArray(new String[0]);
String nextToken = nextTokens[rand.nextInt(nextTokens.length)];
quad =
(Quad)
this.quads.get(
new Quad(quad.getToken(1), quad.getToken(2), quad.getToken(3), nextToken));
parts.add(nextToken);
}
quad = middleQuad;
while (quad.canStart() == false) {
String[] previousTokens = (String[]) ((HashSet) previous.get(quad)).toArray(new String[0]);
String previousToken = previousTokens[rand.nextInt(previousTokens.length)];
quad =
(Quad)
this.quads.get(
new Quad(previousToken, quad.getToken(0), quad.getToken(1), quad.getToken(2)));
parts.addFirst(previousToken);
}
StringBuffer sentence = new StringBuffer();
Iterator it = parts.iterator();
while (it.hasNext()) {
String token = (String) it.next();
sentence.append(token);
}
return sentence.toString();
}
