package de.ugoe.cs.eventbench.swt;

import org.eclipse.swt.widgets.Dialog;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.MessageBox;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.List;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Group;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.FillLayout;

import de.ugoe.cs.eventbench.models.FirstOrderMarkovModel;
import de.ugoe.cs.eventbench.models.IStochasticProcess;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;

public class ModelPropertiesDialog extends Dialog {

	private IStochasticProcess process;
	
	protected Object result;
	protected Shell shlModelProperties;

	/**
	 * Create the dialog.
	 * @param parent
	 * @param style
	 */
	public ModelPropertiesDialog(Shell parent, int style) {
		super(parent, style);
		setText("SWT Dialog");
	}

	/**
	 * Open the dialog.
	 * @return the result
	 */
	public Object open() {
		createContents();
		shlModelProperties.open();
		shlModelProperties.layout();
		Display display = getParent().getDisplay();
		while (!shlModelProperties.isDisposed()) {
			if (!display.readAndDispatch()) {
				display.sleep();
			}
		}
		return result;
	}

	/**
	 * Create contents of the dialog.
	 */
	private void createContents() {
		shlModelProperties = new Shell(getParent(), SWT.DIALOG_TRIM | SWT.RESIZE);
		shlModelProperties.setSize(230, 318);
		shlModelProperties.setText("Model Properties");
		shlModelProperties.setLayout(new GridLayout(2, false));
		
		Group grpEvents = new Group(shlModelProperties, SWT.NONE);
		FillLayout fl_grpEvents = new FillLayout(SWT.HORIZONTAL);
		fl_grpEvents.marginHeight = 5;
		fl_grpEvents.marginWidth = 5;
		grpEvents.setLayout(fl_grpEvents);
		grpEvents.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 2, 1));
		grpEvents.setText("Events");
		
		List list = new List(grpEvents, SWT.BORDER | SWT.V_SCROLL);
		for( String symbol : process.getSymbolStrings() ) {
			list.add(symbol);
		}
		
		Group grpStatistics = new Group(shlModelProperties, SWT.NONE);
		grpStatistics.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false, 2, 1));
		grpStatistics.setText("Statistics");
		grpStatistics.setLayout(new GridLayout(2, false));
		
		Label lblNumEvents = new Label(grpStatistics, SWT.NONE);
		lblNumEvents.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false, 1, 1));
		lblNumEvents.setText("Num. Events");
		
		Label label = new Label(grpStatistics, SWT.RIGHT);
		label.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false, 1, 1));
		label.setText(""+process.getNumSymbols());
		
		Label lblNumTrieLeafs = new Label(grpStatistics, SWT.NONE);
		lblNumTrieLeafs.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false, 1, 1));
		lblNumTrieLeafs.setText("Size (Flattend FOM)");
		
		Label label_1 = new Label(grpStatistics, SWT.RIGHT);
		label_1.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false, 1, 1));
		label_1.setText(""+process.getNumFOMStates());
		
		Label lblEntropy = new Label(grpStatistics, SWT.NONE);
		lblEntropy.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false, 1, 1));
		lblEntropy.setText("Entropy");
		
		final Label label_2 = new Label(grpStatistics, SWT.RIGHT);
		label_2.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false, 1, 1));
		label_2.setText("####");
		
		Button btnCalculateEntropy = new Button(shlModelProperties, SWT.NONE);
		btnCalculateEntropy.addSelectionListener(new SelectionAdapter() {
			@Override
			public void widgetSelected(SelectionEvent e) {
				if( process instanceof FirstOrderMarkovModel) {
					label_2.setText(""+((FirstOrderMarkovModel) process).calcEntropy());
				} else {
					MessageBox messageBox = new MessageBox(shlModelProperties, SWT.NONE);
					messageBox.setText("Feature Not Available");
					messageBox.setMessage("The feature is currently only available for first-order Markov models.");
					messageBox.open();
				}
			}
		});
		btnCalculateEntropy.setText("Calculate Entropy");
		
		Button btnClose = new Button(shlModelProperties, SWT.NONE);
		btnClose.addSelectionListener(new SelectionAdapter() {
			@Override
			public void widgetSelected(SelectionEvent e) {
				shlModelProperties.dispose();
			}
		});
		btnClose.setText("Close");

	}
	
	public void setStochasticProcess(IStochasticProcess process) {
		this.process = process;
	}
}