package de.ugoe.cs.eventbench.commands;

import java.security.InvalidParameterException;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import de.ugoe.cs.eventbench.data.GlobalDataContainer;
import de.ugoe.cs.eventbench.data.Event;
import de.ugoe.cs.eventbench.models.IStochasticProcess;
import de.ugoe.cs.util.console.Command;
import de.ugoe.cs.util.console.Console;

/**
 * <p>
 * Command to generate random sessions.
 * </p>
 * 
 * @author Steffen Herbold
 * @version 1.0
 */
public class CMDgenerateRandomSequences implements Command {

	/*
	 * (non-Javadoc)
	 * 
	 * @see de.ugoe.cs.util.console.Command#run(java.util.List)
	 */
	@Override
	public void run(List<Object> parameters) {
		String modelname;
		String sequencesName;
		int numSessions;
		long maxIter;
		int minLength = 0;
		int maxLength = Integer.MAX_VALUE;
		try {
			modelname = (String) parameters.get(0);
			sequencesName = (String) parameters.get(1);
			numSessions = Integer.parseInt((String) parameters.get(2));
			maxIter = numSessions * 10;
			if (parameters.size() >= 4) {
				maxIter = Long.parseLong((String) parameters.get(3));
			}
			if (parameters.size() >= 5) {
				minLength = Integer.parseInt((String) parameters.get(4));
			}
			if (parameters.size() >= 6) {
				maxLength = Integer.parseInt((String) parameters.get(5));
			}
		} catch (Exception e) {
			throw new InvalidParameterException();
		}

		IStochasticProcess model = null;
		Object dataObject = GlobalDataContainer.getInstance()
				.getData(modelname);
		if (dataObject == null) {
			Console.println("Model " + modelname + " not found in storage.");
			return;
		}
		if (!(dataObject instanceof IStochasticProcess)) {
			Console.println("Object " + modelname + " not of type MarkovModel!");
			return;
		}
		model = (IStochasticProcess) dataObject;
		Set<List<? extends Event<?>>> sequences = new HashSet<List<? extends Event<?>>>(
				numSessions);
		long numIterations = 0;
		while (sequences.size() < numSessions && numIterations < maxIter) {
			List<? extends Event<?>> generatedSequence = model.randomSequence();
			if (generatedSequence.size() >= minLength
					&& generatedSequence.size() <= maxLength) {
				sequences.add(generatedSequence);
			}
			numIterations++;
		}
		if (sequences.size() < numSessions) {
			Console.println("Only " + sequences.size()
					+ " unique sessions generated after " + maxIter
					+ " iterations");
		}
		if (GlobalDataContainer.getInstance().addData(sequencesName, sequences)) {
			Console.traceln("Old data \"" + sequencesName + "\" overwritten");
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see de.ugoe.cs.util.console.Command#help()
	 */
	@Override
	public void help() {
		Console.println("Usage: generateRandomSequenecs <modelName> <sequencesName> <numSessions> {<maxIter> <minLength> <maxLength>}");
	}

}