/*
 * Carnegie Mellon University
 * Copyright (c) 2004, 2010
 * 
 * This software is distributed under the terms of the GNU Lesser General
 * Public License.  See the included COPYING and COPYING.LESSER files.
 * 
 */

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.Properties;

import edu.cmu.meteor.scorer.MeteorConfiguration;
import edu.cmu.meteor.scorer.MeteorScorer;
import edu.cmu.meteor.scorer.MeteorStats;
import edu.cmu.meteor.util.Constants;
import edu.cmu.meteor.util.SGMData;

public class Meteor {

	public static void main(String[] args) {

		// Usage
		if (args.length < 2) {
			printUsage();
			System.exit(2);
		}

		// Files
		String testFile = args[0];
		String refFile = args[1];

		// Use command line options to create props, configuration
		Properties props = createPropertiesFromArgs(args, 2);
		MeteorConfiguration config = new MeteorConfiguration(props);

		// Print settings
		Boolean ssOut = Boolean.parseBoolean(props.getProperty("ssOut"));
		Boolean sgml = Boolean.parseBoolean(props.getProperty("sgml"));
		Boolean stdio = Boolean.parseBoolean(props.getProperty("stdio"));
		Boolean quiet = Boolean.parseBoolean(props.getProperty("quiet"));

		String format = sgml ? "SGML" : "plaintext";
		if (!ssOut && !stdio && !quiet) {
			System.out.println("Meteor version: " + Constants.VERSION);
			System.out.println();
			System.out.println("Eval ID:        " + config.getConfigID());
			System.out.println();
			System.out.println("Language:       "
					+ config.getLanguage().substring(0, 1).toUpperCase()
					+ config.getLanguage().substring(1));
			System.out.println("Format:         " + format);
			System.out.println("Task:           " + config.getTaskDesc());
			System.out.println("Modules:        " + config.getModulesString());
			System.out.println("Weights:        "
					+ config.getModuleWeightsString());
			System.out.println("Parameters:     "
					+ config.getParametersString());
			System.out.println();
		}

		// Module / Weight check
		if (config.getModuleWeights().size() < config.getModules().size()) {
			System.err.println("Warning: More modules than weights specified "
					+ "- modules with no weights will not be counted.");
		}

		// Stdio check
		if (stdio && sgml) {
			System.err
					.println("Warning: Stdio incompatible with other modes - using Stdio only");
		}

		MeteorScorer scorer = new MeteorScorer(config);

		if (stdio) {
			try {
				scoreStdio(scorer);
			} catch (IOException ex) {
				System.err.println("Error: Could not score Stdio inputs");
				ex.printStackTrace();
				System.exit(1);
			}
		} else if (sgml) {
                            try {
                                
				scoreSGML(scorer, props, config, testFile, refFile);
			} catch (IOException ex) {
				System.err.println("Error: Could not score SGML files:");
				ex.printStackTrace();
				System.exit(1);
			}
		} else
			try {
				scorePlaintext(scorer, props, config, testFile, refFile);
			} catch (IOException ex) {
				System.err.println("Error: Could not score text files:");
				ex.printStackTrace();
				System.exit(1);
			}
	}

	private static int getRefCount(Properties props) {
		String refCountString = props.getProperty("refCount");
		if (refCountString == null)
			return 1;
		else
			return Integer.parseInt(refCountString);
	}

	/**
	 * Input is in plaintext format, output is in simple format
	 */

	private static void scorePlaintext(MeteorScorer scorer, Properties props,
			MeteorConfiguration config, String testFile, String refFile)
			throws IOException {

		ArrayList<String> lines1 = new ArrayList<String>();
		ArrayList<String> lines2 = new ArrayList<String>();
		ArrayList<ArrayList<String>> lines2mref = new ArrayList<ArrayList<String>>();

		BufferedReader in = new BufferedReader(new InputStreamReader(
				new FileInputStream(testFile), "UTF-8"));
		String line;
		while ((line = in.readLine()) != null)
			lines1.add(line);
		in.close();

		int refCount = getRefCount(props);

		in = new BufferedReader(new InputStreamReader(new FileInputStream(
				refFile), "UTF-8"));
		if (refCount == 1)
			while ((line = in.readLine()) != null)
				lines2.add(line);
		else {
			while ((line = in.readLine()) != null) {
				ArrayList<String> refs = new ArrayList<String>();
				refs.add(line);
				for (int refNum = 1; refNum < refCount; refNum++)
					refs.add(in.readLine());
				lines2mref.add(refs);
			}
		}
		in.close();
		if ((refCount == 1 && lines1.size() != lines2.size())
				|| (refCount > 1 && lines1.size() != lines2mref.size())) {
			System.err.println("Error: test and reference not same length");
			return;
		}

		MeteorStats aggStats = new MeteorStats();

		// Write alignments?
		Boolean writeAlignments = Boolean.parseBoolean(props
				.getProperty("writeAlignments"));
		PrintWriter out = null;
		if (writeAlignments) {
			String filePrefix = props.getProperty("filePrefix");
			if (filePrefix == null)
				filePrefix = "meteor";
			out = new PrintWriter(new OutputStreamWriter(new FileOutputStream(
					filePrefix + "-align.out"), "UTF-8"));
		}

		Boolean ssOut = Boolean.parseBoolean(props.getProperty("ssOut"));
		Boolean vOut = Boolean.parseBoolean(props.getProperty("vOut"));
		Boolean quiet = Boolean.parseBoolean(props.getProperty("quiet"));

		for (int i = 0; i < lines1.size(); i++) {
			MeteorStats stats;
			if (refCount == 1) {
				stats = scorer.getMeteorStats(lines1.get(i), lines2.get(i));
			} else
				stats = scorer.getMeteorStats(lines1.get(i), lines2mref.get(i));
			if (ssOut) {
				System.out.println(stats.toString());
			} else if (vOut) {
				System.out.println("Segment " + (i + 1) + " score:\t"
						+ stats.precision + "\t" + stats.recall + "\t"
						+ stats.fragPenalty + "\t" + stats.score);
			} else if (quiet) {
				System.err.println(stats.score);
			} else {
				System.out.println("Segment " + (i + 1) + " score:\t"
						+ stats.score);
			}
			if (writeAlignments) {
				out.println(stats.alignment.toString("Alignment\t" + (i + 1)
						+ "\t" + stats.precision + "\t" + stats.recall + "\t"
						+ stats.fragPenalty + "\t" + stats.score));
			}
			aggStats.addStats(stats);
		}

		if (writeAlignments) {
			out.close();
		}

		if (!ssOut) {
			scorer.computeMetrics(aggStats);
			if (quiet) {
				System.out.println(aggStats.score);
			} else {
				printVerboseStats(aggStats, config);
			}
		}
	}

	/**
	 * Input is in SGML format, output to be in seg, doc, sys score format
	 */

	private static void scoreSGML(MeteorScorer scorer, Properties props,
			MeteorConfiguration config, String testFile, String refFile)
			throws IOException {

		// Gather SGML data
		SGMData data = new SGMData();
		SGMData.populate(data, testFile, false);
		SGMData.populate(data, refFile, true);

		// Print scores (or stats)
		Boolean ssOut = Boolean.parseBoolean(props.getProperty("ssOut"));
		Boolean vOut = Boolean.parseBoolean(props.getProperty("vOut"));

		/*
		 * Score all sets, documents, and segments for a single system against
		 * one or more references. Check for missing documents and segments.
		 * Save lines to be sorted and output. Assume data exists for specified
		 * system.
		 */

		// Output prefix
		String filePrefix = props.getProperty("filePrefix");
		if (filePrefix == null)
			filePrefix = "meteor";

		// Output lines
		ArrayList<String> sysLines = new ArrayList<String>();
		ArrayList<String> docLines = new ArrayList<String>();
		ArrayList<String> segLines = new ArrayList<String>();
		Hashtable<String, MeteorStats> segStats = new Hashtable<String, MeteorStats>();

		// Set keys
		HashSet<String> allSetIDs = new HashSet<String>();
		allSetIDs.addAll(data.testData.keySet());
		allSetIDs.addAll(data.refData.keySet());
		// For each set ...
		Iterator<String> setIDs = allSetIDs.iterator();
		while (setIDs.hasNext()) {
			// *[set]*[sys][doc][seg]
			String setID = setIDs.next();
			// All references
			Hashtable<String, Hashtable<String, Hashtable<String, String>>> refSyss = data.refData
					.get(setID);
			// All systems
			Hashtable<String, Hashtable<String, Hashtable<String, String>>> testSyss = data.testData
					.get(setID);
			// For each system ...
			Iterator<String> sysIDs = testSyss.keySet().iterator();
			while (sysIDs.hasNext()) {
				// [set]*[sys]*[doc][seg]
				String sysID = sysIDs.next();
				// One set, one sys, multiple doc, multiple segs
				MeteorStats sysStats = new MeteorStats();
				//
				// One system (must exist)
				Hashtable<String, Hashtable<String, String>> testDocs = testSyss
						.get(sysID);
				// All references (must exist)
				ArrayList<Hashtable<String, Hashtable<String, String>>> refDocs = new ArrayList<Hashtable<String, Hashtable<String, String>>>();
				ArrayList<String> refDocNames = new ArrayList<String>();
				Iterator<String> refIDs = refSyss.keySet().iterator();
				while (refIDs.hasNext()) {
					String r = refIDs.next();
					refDocs.add(refSyss.get(r));
					refDocNames.add(r);
				}
				// Doc keys
				HashSet<String> allDocIDs = new HashSet<String>();
				allDocIDs.addAll(testDocs.keySet());
				for (Hashtable<String, Hashtable<String, String>> r : refDocs)
					allDocIDs.addAll(r.keySet());
				// For each doc ...
				Iterator<String> docIDs = allDocIDs.iterator();
				while (docIDs.hasNext()) {
					// [set][sys]*[doc]*[seg]
					String docID = docIDs.next();
					// One set, one sys, one doc, multiple segs
					MeteorStats docStats = new MeteorStats();
					//
					// One test
					Hashtable<String, String> testSegs = testDocs.get(docID);
					if (testSegs == null) {
						System.err.println("Warning: no data for system ["
								+ sysID + "] on document [" + setID + "]["
								+ docID + "] - skipping this document");
						continue;
					}
					// All references (that exist)
					ArrayList<Hashtable<String, String>> refSegs = new ArrayList<Hashtable<String, String>>();
					ArrayList<String> refSegNames = new ArrayList<String>();
					for (int i = 0; i < refDocs.size(); i++) {
						String r = refDocNames.get(i);
						Hashtable<String, String> refSegsElement = refDocs.get(
								i).get(docID);
						if (refSegsElement == null) {
							System.err
									.println("Warning: no data for reference ["
											+ r + "] on document [" + setID
											+ "][" + docID + "]");
							continue;
						}
						refSegNames.add(r);
						refSegs.add(refSegsElement);
					}
					// Require at least one reference
					if (refSegs.size() == 0) {
						System.err
								.println("Warning: no usable references for document ["
										+ setID
										+ "]["
										+ docID
										+ "] - skipping this document");
						continue;
					}
					// Seg keys
					HashSet<String> allSegIDs = new HashSet<String>();
					allSegIDs.addAll(testSegs.keySet());
					for (Hashtable<String, String> r : refSegs)
						allSegIDs.addAll(r.keySet());
					// For each seg
					Iterator<String> segIDs = allSegIDs.iterator();
					while (segIDs.hasNext()) {
						// [set][sys][doc]*[seg]*
						String segID = segIDs.next();
						// One test
						String testSeg = testSegs.get(segID);
						if (testSeg == null) {
							System.err.println("Warning: no data for system ["
									+ sysID + "] on segment [" + setID + "]["
									+ docID + "][" + segID
									+ "] - scoring empty segment");
							testSeg = "";
						}
						// All references
						ArrayList<String> refSeg = new ArrayList<String>();
						for (int i = 0; i < refSegs.size(); i++) {
							String refSegName = refSegNames.get(i);
							String refSegText = refSegs.get(i).get(segID);
							if (refSegText == null) {
								System.err
										.println("Warning: no data for reference ["
												+ refSegName
												+ "] on segment ["
												+ setID
												+ "]["
												+ docID
												+ "]["
												+ segID + "]");
								continue;
							}
							refSeg.add(refSegText);
						}
						// Require at least one reference
						if (refSeg.size() == 0) {
							System.err
									.println("Warning: no usable references for segment ["
											+ setID
											+ "]["
											+ docID
											+ "]["
											+ segID
											+ "] - skipping this segment");
							continue;
						}
						// One system, one set, one doc, one seg
						// Score segment
						MeteorStats stats = scorer.getMeteorStats(testSeg,
								refSeg);
						//
						// Aggreagate statistics
						docStats.addStats(stats);
						sysStats.addStats(stats);
						// Eng segment: store score or SS
						String segName = setID + "\t" + sysID + "\t" + docID
								+ "\t" + segID + "\t";
						String segScore;
						if (ssOut) {
							segScore = stats.toString();
						} else if (vOut) {
							segScore = String.valueOf(stats.precision + "\t"
									+ stats.recall + "\t" + stats.fragPenalty
									+ "\t" + stats.score);
						} else {
							segScore = String.valueOf(stats.score);
						}
						String segLine = segName + segScore;
						segLines.add(segLine);
						// Store alignment and stats using seg key
						segStats.put(segLine, stats);
					}
					// End document: store score or SS
					scorer.computeMetrics(docStats);
					String docName = setID + "\t" + sysID + "\t" + docID + "\t";
					String docScore;
					if (ssOut) {
						docScore = docStats.toString();
					} else if (vOut) {
						docScore = String.valueOf(docStats.precision + "\t"
								+ docStats.recall + "\t" + docStats.fragPenalty
								+ "\t" + docStats.score);
					} else {
						docScore = String.valueOf(docStats.score);
					}
					docLines.add(docName + docScore);
				}
				// End system: store score or SS
				scorer.computeMetrics(sysStats);
				String setName = setID + "\t" + sysID + "\t";
				String setScore;
				if (ssOut) {
					setScore = sysStats.toString();
				} else if (vOut) {
					setScore = String.valueOf(sysStats.precision + "\t"
							+ sysStats.recall + "\t" + sysStats.fragPenalty
							+ "\t" + sysStats.score);
				} else {
					setScore = String.valueOf(sysStats.score);
				}
				sysLines.add(setName + setScore);

				/*
				 * Report system performance
				 */
				if (!ssOut) {
					printVerboseStats(sysStats, config, "[" + setID + "]["
							+ sysID + "] ");
					System.out.println();
				}
			}
		}

		/*
		 * Output score files: system-sys.score system-doc.score
		 * system-seg.score
		 */

		// System
		// Sort
		Collections.sort(sysLines, new Comparator<String>() {
			public int compare(String s1, String s2) {
				// s1 Set Sys
				int scoreDelim1 = s1.lastIndexOf("\t");
				String set1 = s1.substring(0, scoreDelim1);
				// s2 Set Sys
				int scoreDelim2 = s2.lastIndexOf("\t");
				String set2 = s2.substring(0, scoreDelim2);
				return set1.compareTo(set2);
			}
		});
		// Write
		PrintWriter out = new PrintWriter(new OutputStreamWriter(
				new FileOutputStream(filePrefix + "-sys.scr"), "UTF-8"));
		for (String line : sysLines)
			out.println(line);
		out.close();

		// Document
		// Sort
		Collections.sort(docLines, new Comparator<String>() {
			public int compare(String s1, String s2) {
				// s1 Set Sys Doc
				int scoreDelim1 = s1.lastIndexOf("\t");
				String doc1 = s1.substring(0, scoreDelim1);
				// s2 Set Sys Doc
				int scoreDelim2 = s2.lastIndexOf("\t");
				String doc2 = s2.substring(0, scoreDelim2);
				return doc1.compareTo(doc2);
			}
		});
		// Write
		out = new PrintWriter(new OutputStreamWriter(new FileOutputStream(
				filePrefix + "-doc.scr"), "UTF-8"));
		for (String line : docLines)
			out.println(line);
		out.close();

		// Segment
		// Sort
		Collections.sort(segLines, new Comparator<String>() {
			public int compare(String s1, String s2) {
				// s1 Set Sys Doc
				int scoreDelim1 = s1.lastIndexOf("\t");
				int segDelim1 = s1.lastIndexOf("\t", scoreDelim1 - 1);
				String doc1 = s1.substring(0, segDelim1);
				// s2 Set Sys Doc
				int scoreDelim2 = s2.lastIndexOf("\t");
				int segDelim2 = s2.lastIndexOf("\t", scoreDelim2 - 1);
				String doc2 = s2.substring(0, segDelim2);
				int diff = doc1.compareTo(doc2);
				// Same set/sys/doc ?
				if (diff != 0)
					return diff;
				// Compare Segs
				String seg1 = s1.substring(segDelim1 + 1, scoreDelim1);
				String seg2 = s2.substring(segDelim2 + 1, scoreDelim2);
				try {
					// Try to compare as ints
					int seg1int = Integer.parseInt(seg1);
					int seg2int = Integer.parseInt(seg2);
					return seg1int - seg2int;
				} catch (Exception ex) {
					// Otherwise compare as strings
					return seg1.compareTo(seg2);
				}
			}
		});
		// Write
		out = new PrintWriter(new OutputStreamWriter(new FileOutputStream(
				filePrefix + "-seg.scr"), "UTF-8"));
		for (String line : segLines)
			out.println(line);
		out.close();

		// Write alignments?
		Boolean writeAlignments = Boolean.parseBoolean(props
				.getProperty("writeAlignments"));
		if (writeAlignments) {
			out = new PrintWriter(new OutputStreamWriter(new FileOutputStream(
					filePrefix + "-align.out"), "UTF-8"));
			for (int i = 0; i < segLines.size(); i++) {
				String line = segLines.get(i);
				MeteorStats stats = segStats.get(line);
				int idx = -1;
				for (int j = 0; j < 4; j++) {
					idx = line.indexOf('\t', idx + 1);
				}
				String lineNoScore = line.substring(0, idx);
				out.println(stats.alignment.toString("Alignment\t"
						+ lineNoScore + '\t' + stats.precision + "\t"
						+ stats.recall + "\t" + stats.fragPenalty + "\t"
						+ stats.score));
			}
			out.close();
		}
	}

	/**
	 * Input is in Stdio format
	 */

	private static void scoreStdio(MeteorScorer scorer) throws IOException {

		BufferedReader in = new BufferedReader(new InputStreamReader(System.in,
				"UTF-8"));
		String line;

		while ((line = in.readLine()) != null) {

			if (line.startsWith("SCORE") || line.startsWith("score")) {
				String[] part = line.split("\\|\\|\\|");
				if (part.length < 3) {
					System.out
							.println("Error: specify hypothesis and at least one reference");
					continue;
				}
				ArrayList<String> refs = new ArrayList<String>();
				for (int i = 1; i < part.length - 1; i++)
					refs.add(part[i].trim());
				String test = part[part.length - 1].trim();
				MeteorStats stats = scorer.getMeteorStats(test, refs);
				System.out.println(stats.toString());

			} else if (line.startsWith("EVAL") || line.startsWith("eval")) {
				String[] part = line.split("\\|\\|\\|");
				if (part.length < 2) {
					System.out.println("Error: specify Meteor stats");
					continue;
				}
				MeteorStats stats = new MeteorStats(part[1].trim());
				scorer.computeMetrics(stats);
				System.out.println(stats.score);

			} else {
				System.out.println("Error: specify SCORE or EVAL");
			}
		}
	}

	private static void printVerboseStats(MeteorStats stats,
			MeteorConfiguration config) {
		printVerboseStats(stats, config, "\nSystem level statistics:\n");
	}

	private static void printVerboseStats(MeteorStats stats,
			MeteorConfiguration config, String header) {
		System.out.println(header);
		System.out.println();
		System.out
				.println("           Test Matches                  Reference Matches");
		System.out
				.println("Stage      Content  Function    Total    Content  Function    Total");
		ArrayList<Double> weights = config.getModuleWeights();
		double testContentMatches = 0.0;
		double testFunctionMatches = 0.0;
		double referenceContentMatches = 0.0;
		double referenceFunctionMatches = 0.0;
		for (int i = 0; i < weights.size(); i++) {
			System.out
					.println(String.format(
							"%1d          %7s   %7s  %7s    %7s   %7s  %7s",
							i + 1,
							Constants.minFormat
									.format(stats.testStageMatchesContent
											.get(i)),
							Constants.minFormat
									.format(stats.testStageMatchesFunction
											.get(i)),
							Constants.minFormat
									.format(stats.testStageMatchesContent
											.get(i)
											+ stats.testStageMatchesFunction
													.get(i)),
							Constants.minFormat
									.format(stats.referenceStageMatchesContent
											.get(i)),
							Constants.minFormat
									.format(stats.referenceStageMatchesFunction
											.get(i)),
							Constants.minFormat.format(stats.referenceStageMatchesContent
									.get(i)
									+ stats.referenceStageMatchesFunction
											.get(i))));
			testContentMatches += stats.testStageMatchesContent.get(i);
			testFunctionMatches += stats.testStageMatchesFunction.get(i);
			referenceContentMatches += stats.referenceStageMatchesContent
					.get(i);
			referenceFunctionMatches += stats.referenceStageMatchesFunction
					.get(i);
		}
		System.out.println(String.format(
				"Total      %7s   %7s  %7s    %7s   %7s  %7s",
				Constants.minFormat.format(testContentMatches),
				Constants.minFormat.format(testFunctionMatches),
				Constants.minFormat.format(testContentMatches
						+ testFunctionMatches),
				Constants.minFormat.format(referenceContentMatches),
				Constants.minFormat.format(referenceFunctionMatches),
				Constants.minFormat.format(referenceContentMatches
						+ referenceFunctionMatches)));
		System.out.println();
		System.out.println("Test " + (config.getCharBased() ? "char" : "word")
				+ "s:             "
				+ Constants.minFormat.format(stats.testLength));
		System.out.println("Reference "
				+ (config.getCharBased() ? "char" : "word") + "s:        "
				+ Constants.minFormat.format(stats.referenceLength));
		System.out.println("Chunks:                 "
				+ Constants.minFormat.format(stats.chunks));
		System.out.println("Precision:              " + stats.precision);
		System.out.println("Recall:                 " + stats.recall);
		System.out.println("f1:                     " + stats.f1);
		System.out.println("fMean:                  " + stats.fMean);
		System.out.println("Fragmentation penalty:  " + stats.fragPenalty);
		System.out.println();
		System.out.println("Final score:            " + stats.score);
	}

	public static Properties createPropertiesFromArgs(String[] args,
			int startIndex) {
		Properties props = new Properties();
		int curArg = startIndex;
		while (curArg < args.length) {
			if (args[curArg].equals("-l")) {
				props.setProperty("language", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-t")) {
				props.setProperty("task", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-p")) {
				props.setProperty("parameters", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-m")) {
				props.setProperty("modules", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-w")) {
				props.setProperty("moduleWeights", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-r")) {
				props.setProperty("refCount", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-x")) {
				props.setProperty("beamSize", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-s")) {
				props.setProperty("wordFile", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-emb")) {
				props.setProperty("embFile", args[curArg + 1]);
				props.setProperty("embType", "0");
				curArg += 2;
			} else if (args[curArg].equals("-embLemma")) {
				props.setProperty("embFile", args[curArg + 1]);
				props.setProperty("embType", "1");
				curArg += 2;
			} else if (args[curArg].equals("-embWord_POS")) {
				props.setProperty("embFile", args[curArg + 1]);
				props.setProperty("embType", "2");
				curArg += 2;
			} else if (args[curArg].equals("-embLemma_POS")) {
				props.setProperty("embFile", args[curArg + 1]);
				props.setProperty("embType", "3");
				curArg += 2;
			} else if (args[curArg].equals("-embWord_Lemma")) {
				props.setProperty("embFile", args[curArg + 1]);
				props.setProperty("embType", "4");
				curArg += 2;
			} else if (args[curArg].equals("-embTh")) {
				props.setProperty("embThreshold", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-d")) {
				props.setProperty("synDir", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-a")) {
				props.setProperty("paraFile", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-f")) {
				props.setProperty("filePrefix", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-new")) {
				props.setProperty("newLang", "true");
				props.setProperty("filesDir", args[curArg + 1]);
				curArg += 2;
			} else if (args[curArg].equals("-ch")) {
				props.setProperty("charBased", "true");
				curArg += 1;
			} else if (args[curArg].equals("-q")) {
				props.setProperty("quiet", "true");
				curArg += 1;
			} else if (args[curArg].equals("-writeAlignments")) {
				props.setProperty("writeAlignments", "true");
				curArg += 1;
			} else if (args[curArg].equals("-norm")) {
				props.setProperty("norm", "true");
				curArg += 1;
			} else if (args[curArg].equals("-lower")) {
				props.setProperty("lower", "true");
				curArg += 1;
			} else if (args[curArg].equals("-sgml")) {
				props.setProperty("sgml", "true");
				curArg += 1;
				// Include -mira for backward compatibility
			} else if (args[curArg].equals("-stdio")
					|| args[curArg].equals("-mira")) {
				props.setProperty("stdio", "true");
				curArg += 1;
			} else if (args[curArg].equals("-noPunct")) {
				props.setProperty("noPunct", "true");
				curArg += 1;
			} else if (args[curArg].equals("-ssOut")) {
				props.setProperty("ssOut", "true");
				curArg += 1;
			} else if (args[curArg].equals("-vOut")) {
				props.setProperty("vOut", "true");
				curArg += 1;
			} else {
				System.err.println("Unknown option \"" + args[curArg] + "\"");
				System.exit(1);
			}
			String params = props.getProperty("parameters");
			if (params != null)
				props.setProperty("task", "custom (" + params + ")");
		}
		return props;
	}

	private static void printUsage() {
		System.err.println("Meteor version " + Constants.VERSION);
		System.err.println();
		System.err
				.println("Usage: java -Xmx2G -jar meteor-*.jar <test> <reference> [options]");
		System.err.println();
		System.err.println("Options:");
		System.err
				.println("-l language                     Fully supported: en cz de es fr");
		System.err
				.println("                                Supported with language-independent parameters:");
		System.err
				.println("                                  da fi hu it nl no pt ro ru se tr");
		System.err.println("                                Experimental:");
		System.err.println("                                  ar-bw-red");

		System.err
				.println("-t task                         One of: rank util adq hter li tune");
		System.err
				.println("                                  util implies -ch");
		System.err
				.println("-p 'alpha beta gamma delta'     Custom parameters (overrides default)");
		System.err
				.println("-m 'module1 module2 ...'        Specify modules (overrides default)");
		System.err
				.println("                                  Any of: exact stem synonym paraphrase");
		System.err
				.println("-w 'weight1 weight2 ...'        Specify module weights (overrides default)");
		System.err
				.println("-r refCount                     Number of references (plaintext only)");
		System.err.println("-x beamSize                     (default 40)");
		System.err
				.println("-s wordListFile                 (if not default for language)");
		System.err
				.println("-d synonymDirectory             (if not default for language)");
        System.err
        .println("-emb embeddingsModelFile        specify the embeddings model in text format (if not default for language)");
        System.err
        .println("-embLemma embeddingsModelFile        specify the embeddings model in text format based on lemmas (if not default for language)");
        System.err
        .println("-embWord_POS embeddingsModelFile        specify the embeddings model in text format based on bot word and POS like \"is_VB\" (if not default for language)");
        System.err
        .println("-embLemma_POS embeddingsModelFile        specify the embeddings model in text format based on bot lemma and POS like \"be_VB\" (if not default for language)");
                System.err
                                .println("-embTh float                    specify the threshold alignment decision with embeddings (default is 0.8)");
		System.err
				.println("-a paraphraseFile               (if not default for language)");
		System.err
				.println("-f filePrefix                   Prefix for output files (default 'meteor')");
		System.err
				.println("-q                              Quiet: Segment scores to stderr, final to stdout,");
		System.err
				.println("                                  no additional output (plaintext only)");
		System.err
				.println("-new files-dir                  New language! (files-dir contains function.words and paraphrase.gz)");
		System.err.println("                                  implies -lower");
		System.err
				.println("-ch                             Character-based precision and recall");
		System.err
				.println("-norm                           Tokenize / normalize punctuation and lowercase");
		System.err
				.println("                                  (Recommended unless scoring raw output with");
		System.err
				.println("                                   pretokenized references)");
		System.err
				.println("-lower                          Lowercase only (not required if -norm specified)");
		System.err
				.println("-noPunct                        Do not consider punctuation when scoring");
		System.err
				.println("                                  (Not recommended unless special case)");
		System.err
				.println("-sgml                           Input is in SGML format");
		System.err
				.println("-stdio                           Input is from stdin, see README for format");
		System.err
				.println("                                  (Use '-' for test and reference files)");
		System.err
				.println("-vOut                           Output verbose scores (P / R / frag / score)");
		System.err
				.println("-ssOut                          Output sufficient statistics instead of scores");
		System.err
				.println("-writeAlignments                Output alignments annotated with Meteor scores");
		System.err
				.println("                                  (written to <prefix>-align.out)");
		System.err.println();
		System.err.println("Sample options for plaintext: -l <lang> -norm");
		System.err.println("Sample options for SGML: -l <lang> -norm -sgml");
		System.err
				.println("Sample options for raw output / pretokenized references: -l <lang> -lower");
		System.err
				.println("Sample options for new language (plaintext): -new meteor-files");
		System.err.println();

		System.err.println("See README file for additional information");
	}
}