package tim.prune;

import java.util.EmptyStackException;
import java.util.Stack;

import javax.swing.JFrame;
import javax.swing.JOptionPane;

import tim.prune.data.DataPoint;
import tim.prune.data.Field;
import tim.prune.data.Track;
import tim.prune.data.TrackInfo;
import tim.prune.gui.MenuManager;
import tim.prune.gui.UndoManager;
import tim.prune.load.FileLoader;
import tim.prune.save.FileSaver;
import tim.prune.save.KmlExporter;
import tim.prune.undo.UndoCompress;
import tim.prune.undo.UndoDeleteDuplicates;
import tim.prune.undo.UndoDeletePoint;
import tim.prune.undo.UndoDeleteRange;
import tim.prune.undo.UndoException;
import tim.prune.undo.UndoInsert;
import tim.prune.undo.UndoLoad;
import tim.prune.undo.UndoOperation;
import tim.prune.undo.UndoRearrangeWaypoints;
import tim.prune.undo.UndoReverseSection;


/**
 * Main controller for the application
 */
public class App
{
	// Instance variables
	private JFrame _frame = null;
	private Track _track = null;
	private TrackInfo _trackInfo = null;
	private int _lastSavePosition = 0;
	private MenuManager _menuManager = null;
	private FileLoader _fileLoader = null;
	private Stack _undoStack = null;
	private UpdateMessageBroker _broker = null;
	private boolean _reversePointsConfirmed = false;

	// Constants
	public static final int REARRANGE_TO_START   = 0;
	public static final int REARRANGE_TO_END     = 1;
	public static final int REARRANGE_TO_NEAREST = 2;


	// TODO: Make waypoint window to allow list of waypoints, edit names etc

	/**
	 * Constructor
	 * @param inFrame frame object for application
	 * @param inBroker message broker
	 */
	public App(JFrame inFrame, UpdateMessageBroker inBroker)
	{
		_frame = inFrame;
		_undoStack = new Stack();
		_broker = inBroker;
		_track = new Track(_broker);
		_trackInfo = new TrackInfo(_track, _broker);
	}


	/**
	 * @return the current TrackInfo
	 */
	public TrackInfo getTrackInfo()
	{
		return _trackInfo;
	}

	/**
	 * Check if the application has unsaved data
	 * @return true if data is unsaved, false otherwise
	 */
	public boolean hasDataUnsaved()
	{
		return _undoStack.size() > _lastSavePosition;
	}

	/**
	 * @return the undo stack
	 */
	public Stack getUndoStack()
	{
		return _undoStack;
	}

	/**
	 * Set the MenuManager object to be informed about changes
	 * @param inManager MenuManager object
	 */
	public void setMenuManager(MenuManager inManager)
	{
		_menuManager = inManager;
	}


	/**
	 * Open a file containing track or waypoint data
	 */
	public void openFile()
	{
		if (_fileLoader == null)
			_fileLoader = new FileLoader(this, _frame);
		_fileLoader.openFile();
	}


	/**
	 * Save the file in the selected format
	 */
	public void saveFile()
	{
		if (_track == null)
		{
			JOptionPane.showMessageDialog(_frame, I18nManager.getText("error.save.nodata"),
				I18nManager.getText("error.save.dialogtitle"), JOptionPane.ERROR_MESSAGE);
		}
		else
		{
			FileSaver saver = new FileSaver(this, _frame, _track);
			saver.showDialog(_fileLoader.getLastUsedDelimiter());
		}
	}


	/**
	 * Export track data as Kml
	 */
	public void exportKml()
	{
		if (_track == null)
		{
			JOptionPane.showMessageDialog(_frame, I18nManager.getText("error.save.nodata"),
				I18nManager.getText("error.save.dialogtitle"), JOptionPane.ERROR_MESSAGE);
		}
		else
		{
			KmlExporter exporter = new KmlExporter(this, _frame, _track);
			exporter.showDialog();
		}
	}


	/**
	 * Exit the application if confirmed
	 */
	public void exit()
	{
		// check if ok to exit
		Object[] buttonTexts = {I18nManager.getText("button.exit"), I18nManager.getText("button.cancel")};
		if (!hasDataUnsaved()
			|| JOptionPane.showOptionDialog(_frame, I18nManager.getText("dialog.exit.confirm.text"),
				I18nManager.getText("dialog.exit.confirm.title"), JOptionPane.YES_NO_OPTION,
				JOptionPane.WARNING_MESSAGE, null, buttonTexts, buttonTexts[1])
			== JOptionPane.YES_OPTION)
		{
			System.exit(0);
		}
	}


	/**
	 * Delete the currently selected point
	 */
	public void deleteCurrentPoint()
	{
		if (_track != null)
		{
			DataPoint currentPoint = _trackInfo.getCurrentPoint();
			if (currentPoint != null)
			{
				// add information to undo stack
				int pointIndex = _trackInfo.getSelection().getCurrentPointIndex();
				UndoOperation undo = new UndoDeletePoint(pointIndex, currentPoint);
				// call track to delete point
				if (_trackInfo.deletePoint())
				{
					_undoStack.push(undo);
				}
			}
		}
	}


	/**
	 * Delete the currently selected range
	 */
	public void deleteSelectedRange()
	{
		if (_track != null)
		{
			// add information to undo stack
			UndoOperation undo = new UndoDeleteRange(_trackInfo);
			// call track to delete point
			if (_trackInfo.deleteRange())
			{
				_undoStack.push(undo);
			}
		}
	}


	/**
	 * Delete all the duplicate points in the track
	 */
	public void deleteDuplicates()
	{
		if (_track != null)
		{
			// Save undo information
			UndoOperation undo = new UndoDeleteDuplicates(_track);
			// tell track to do it
			int numDeleted = _trackInfo.deleteDuplicates();
			if (numDeleted > 0)
			{
				_undoStack.add(undo);
				String message = null;
				if (numDeleted == 1)
				{
					message = "1 " + I18nManager.getText("dialog.deleteduplicates.single.text");
				}
				else
				{
					message = "" + numDeleted + " " + I18nManager.getText("dialog.deleteduplicates.multi.text");
				}
				JOptionPane.showMessageDialog(_frame, message,
					I18nManager.getText("dialog.deleteduplicates.title"), JOptionPane.INFORMATION_MESSAGE);
			}
			else
			{
				JOptionPane.showMessageDialog(_frame,
					I18nManager.getText("dialog.deleteduplicates.nonefound"),
					I18nManager.getText("dialog.deleteduplicates.title"), JOptionPane.INFORMATION_MESSAGE);
			}
		}
	}


	/**
	 * Compress the track
	 */
	public void compressTrack()
	{
		UndoCompress undo = new UndoCompress(_track);
		// Get compression parameter
		Object compParam = JOptionPane.showInputDialog(_frame,
			I18nManager.getText("dialog.compresstrack.parameter.text"),
			I18nManager.getText("dialog.compresstrack.title"),
			JOptionPane.QUESTION_MESSAGE, null, null, "100");
		int compNumber = parseNumber(compParam);
		if (compNumber <= 0) return;
		// call track to do compress
		int numPointsDeleted = _trackInfo.compress(compNumber);
		// add to undo stack if successful
		if (numPointsDeleted > 0)
		{
			undo.setNumPointsDeleted(numPointsDeleted);
			_undoStack.add(undo);
			JOptionPane.showMessageDialog(_frame,
				I18nManager.getText("dialog.compresstrack.text") + " - "
				 + numPointsDeleted + " "
				 + (numPointsDeleted==1?I18nManager.getText("dialog.compresstrack.single.text"):I18nManager.getText("dialog.compresstrack.multi.text")),
				I18nManager.getText("dialog.compresstrack.title"), JOptionPane.INFORMATION_MESSAGE);
		}
		else
		{
			JOptionPane.showMessageDialog(_frame, I18nManager.getText("dialog.compresstrack.nonefound"),
				I18nManager.getText("dialog.compresstrack.title"), JOptionPane.WARNING_MESSAGE);
		}
	}


	/**
	 * Reverse a section of the track
	 */
	public void reverseRange()
	{
		// check whether Timestamp field exists, and if so confirm reversal
		int selStart = _trackInfo.getSelection().getStart();
		int selEnd = _trackInfo.getSelection().getEnd();
		if (!_track.hasData(Field.TIMESTAMP, selStart, selEnd)
			|| _reversePointsConfirmed
			|| (JOptionPane.showConfirmDialog(_frame,
				 I18nManager.getText("dialog.confirmreversetrack.text"),
				 I18nManager.getText("dialog.confirmreversetrack.title"),
				 JOptionPane.YES_NO_OPTION) == JOptionPane.YES_OPTION && (_reversePointsConfirmed = true)))
		{
			UndoReverseSection undo = new UndoReverseSection(selStart, selEnd);
			// call track to reverse range
			if (_track.reverseRange(selStart, selEnd))
			{
				_undoStack.add(undo);
			}
		}
	}


	/**
	 * Interpolate the two selected points
	 */
	public void interpolateSelection()
	{
		// Get number of points to add
		Object numPointsStr = JOptionPane.showInputDialog(_frame,
			I18nManager.getText("dialog.interpolate.parameter.text"),
			I18nManager.getText("dialog.interpolate.title"),
			JOptionPane.QUESTION_MESSAGE, null, null, "");
		int numPoints = parseNumber(numPointsStr);
		if (numPoints <= 0) return;

		UndoInsert undo = new UndoInsert(_trackInfo.getSelection().getStart() + 1,
			numPoints);
		// call track to interpolate
		if (_trackInfo.interpolate(numPoints))
		{
			_undoStack.add(undo);
		}
	}


	/**
	 * Rearrange the waypoints into track order
	 */
	public void rearrangeWaypoints(int inFunction)
	{
		UndoRearrangeWaypoints undo = new UndoRearrangeWaypoints(_track);
		boolean success = false;
		if (inFunction == REARRANGE_TO_START || inFunction == REARRANGE_TO_END)
		{
			// Collect the waypoints to the start or end of the track
			success = _track.collectWaypoints(inFunction == REARRANGE_TO_START);
		}
		else
		{
			// Interleave the waypoints into track order
			success = _track.interleaveWaypoints();
		}
		if (success)
		{
			_undoStack.add(undo);
		}
		else
		{
			JOptionPane.showMessageDialog(_frame, I18nManager.getText("error.rearrange.noop"),
				I18nManager.getText("error.function.noop.title"), JOptionPane.WARNING_MESSAGE);
		}
	}


	/**
	 * Open a new window with the 3d view
	 */
	public void show3dWindow()
	{
		// TODO: open 3d view window
		JOptionPane.showMessageDialog(_frame, I18nManager.getText("error.function.notimplemented"),
			I18nManager.getText("error.function.notimplemented.title"), JOptionPane.WARNING_MESSAGE);
	}


	/**
	 * Select all points
	 */
	public void selectAll()
	{
		_trackInfo.getSelection().select(0, 0, _track.getNumPoints()-1);
	}

	/**
	 * Select nothing
	 */
	public void selectNone()
	{
		_trackInfo.getSelection().clearAll();
	}

	/**
	 * Receive loaded data and optionally merge with current Track
	 * @param inFieldArray array of fields
	 * @param inDataArray array of data
	 */
	public void informDataLoaded(Field[] inFieldArray, Object[][] inDataArray, int inAltFormat, String inFilename)
	{
		// Decide whether to load or append
		if (_track != null && _track.getNumPoints() > 0)
		{
			// ask whether to replace or append
			int answer = JOptionPane.showConfirmDialog(_frame,
				I18nManager.getText("dialog.openappend.text"),
				I18nManager.getText("dialog.openappend.title"),
				JOptionPane.YES_NO_CANCEL_OPTION);
			if (answer == JOptionPane.YES_OPTION)
			{
				// append data to current Track
				Track loadedTrack = new Track(_broker);
				loadedTrack.load(inFieldArray, inDataArray, inAltFormat);
				_undoStack.add(new UndoLoad(_track.getNumPoints(), loadedTrack.getNumPoints()));
				_track.combine(loadedTrack);
				_trackInfo.getFileInfo().addFile();
			}
			else if (answer == JOptionPane.NO_OPTION)
			{
				// Don't append, replace data
				_undoStack.add(new UndoLoad(_trackInfo, inDataArray.length));
				_lastSavePosition = _undoStack.size();
				_trackInfo.loadTrack(inFieldArray, inDataArray, inAltFormat);
				_trackInfo.getFileInfo().setFile(inFilename);
			}
		}
		else
		{
			// currently no data held, so use received data
			_undoStack.add(new UndoLoad(_trackInfo, inDataArray.length));
			_lastSavePosition = _undoStack.size();
			_trackInfo.loadTrack(inFieldArray, inDataArray, inAltFormat);
			_trackInfo.getFileInfo().setFile(inFilename);
		}
		_broker.informSubscribers();
		// update menu
		_menuManager.informFileLoaded();
	}


	/**
	 * Inform the app that the data has been saved
	 */
	public void informDataSaved()
	{
		_lastSavePosition = _undoStack.size();
	}


	/**
	 * Begin undo process
	 */
	public void beginUndo()
	{
		if (_undoStack.isEmpty())
		{
			JOptionPane.showMessageDialog(_frame, I18nManager.getText("dialog.undo.none.text"),
				I18nManager.getText("dialog.undo.none.title"), JOptionPane.INFORMATION_MESSAGE);
		}
		else
		{
			new UndoManager(this, _frame);
		}
	}


	/**
	 * Clear the undo stack (losing all undo information
	 */
	public void clearUndo()
	{
		// Exit if nothing to undo
		if (_undoStack == null || _undoStack.isEmpty())
			return;
		// Has track got unsaved data?
		boolean unsaved = hasDataUnsaved();
		// Confirm operation with dialog
		int answer = JOptionPane.showConfirmDialog(_frame,
			I18nManager.getText("dialog.clearundo.text"),
			I18nManager.getText("dialog.clearundo.title"),
			JOptionPane.YES_NO_OPTION);
		if (answer == JOptionPane.YES_OPTION)
		{
			_undoStack.clear();
			_lastSavePosition = 0;
			if (unsaved) _lastSavePosition = -1;
			_broker.informSubscribers();
		}
	}


	/**
	 * Undo the specified number of actions
	 * @param inNumUndos number of actions to undo
	 */
	public void undoActions(int inNumUndos)
	{
		try
		{
			for (int i=0; i<inNumUndos; i++)
			{
				((UndoOperation) _undoStack.pop()).performUndo(_trackInfo);
			}
			JOptionPane.showMessageDialog(_frame, "" + inNumUndos + " "
				 + (inNumUndos==1?I18nManager.getText("dialog.confirmundo.single.text"):I18nManager.getText("dialog.confirmundo.multiple.text")),
				I18nManager.getText("dialog.confirmundo.title"),
				JOptionPane.INFORMATION_MESSAGE);
		}
		catch (UndoException ue)
		{
			JOptionPane.showMessageDialog(_frame,
				I18nManager.getText("error.undofailed.text") + " : " + ue.getMessage(),
				I18nManager.getText("error.undofailed.title"),
				JOptionPane.ERROR_MESSAGE);
			_undoStack.clear();
			_broker.informSubscribers();
		}
		catch (EmptyStackException empty) {}
	}


	/**
	 * Helper method to parse an Object into an integer
	 * @param inObject object, eg from dialog
	 * @return int value given
	 */
	private static int parseNumber(Object inObject)
	{
		int num = 0;
		if (inObject != null)
		{
			try
			{
				num = Integer.parseInt(inObject.toString());
			}
			catch (NumberFormatException nfe)
			{}
		}
		return num;
	}
}