summaryrefslogtreecommitdiff
path: root/RGens/src/main/java/bjc/rgens/parser/GrammarReaderApp.java
blob: b3beb813a57a1df4796a77bad3df0c183edfcb81 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package bjc.rgens.parser;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.PrintStream;

import javax.swing.UIManager;
import javax.swing.UnsupportedLookAndFeelException;

import bjc.utils.funcutils.ListUtils;
import bjc.utils.gen.WeightedGrammar;
import bjc.utils.gui.SimpleDialogs;
import bjc.utils.gui.awt.SimpleFileDialog;

/**
 * App that reads a grammar from a file and generates results
 * 
 * @author ben
 *
 */
public class GrammarReaderApp {
	/**
	 * Main method of class
	 * 
	 * @param args
	 *            CLI args
	 */
	public static void main(String[] args) {
		try {
			UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
		} catch (ClassNotFoundException
				| InstantiationException
				| IllegalAccessException
				| UnsupportedLookAndFeelException e) {
			e.printStackTrace();
			System.exit(1);
		}

		doSingleFile();
	}

	private static void doSingleFile() {
		File gramFile = SimpleFileDialog.getOpenFile(null, "Choose Grammar File", ".gram");

		WeightedGrammar<String> grammar = null;

		try {
			grammar = RBGrammarReader.fromPath(gramFile.toPath());
		} catch (IOException ioex) {
			ioex.printStackTrace();

			System.exit(1);
		}

		String initRule = "";

		if (!grammar.hasInitialRule()) {
			grammar.getRuleNames().sort((leftString, rightString) -> {
				return leftString.compareTo(rightString);
			});

			initRule = SimpleDialogs.getChoice(null,
					"Pick a initial rule",
					"Pick a initial rule to generate choices from",
					grammar.getRuleNames().toArray(new String[0]));
		} else {
			initRule = grammar.getInitialRule();
		}

		int count = SimpleDialogs.getWhole(null,
				"Enter number of repetitions",
				"Enter the number of items to generate from the rule");

		File outputFile = SimpleFileDialog.getSaveFile(null, "Choose Grammar File");

		PrintStream outputStream = null;

		try {
			outputStream = new PrintStream(outputFile);
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		}

		for (int i = 0; i < count; i++) {
			String ruleResult = ListUtils.collapseTokens(grammar.generateListValues(initRule, " "));

			outputStream.println(ruleResult.replaceAll("\\s+", " "));
		}

		outputStream.close();
	}
}