/*

 * Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). 

 * All rights reserved.

 * This component and the accompanying materials are made available

 * under the terms of the License "Eclipse Public License v1.0"

 * which accompanies this distribution, and is available

 * at the URL "http://www.eclipse.org/legal/epl-v10.html".

 *

 * Initial Contributors:

 * Nokia Corporation - initial contribution.

 *

 * Contributors:

 *

 * Description: 

 *

 */

package com.nokia.carbide.cpp.pi.address;

import java.io.File;

import java.io.IOException;

import java.io.Serializable;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.Hashtable;

import java.util.Vector;

import org.eclipse.jface.action.Action;

import org.eclipse.jface.action.MenuManager;

import org.eclipse.jface.action.Separator;

import org.eclipse.jface.resource.ImageDescriptor;

import org.eclipse.swt.SWT;

import org.eclipse.swt.events.SelectionAdapter;

import org.eclipse.swt.events.SelectionEvent;

import org.eclipse.swt.graphics.Color;

import org.eclipse.swt.graphics.RGB;

import org.eclipse.swt.layout.FillLayout;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Display;

import org.eclipse.swt.widgets.Event;

import org.eclipse.swt.widgets.FileDialog;

import org.eclipse.swt.widgets.Shell;

import org.eclipse.swt.widgets.Text;

import org.eclipse.ui.PlatformUI;

import org.osgi.framework.BundleContext;

import com.nokia.carbide.cpp.internal.pi.actions.SetThresholdsDialog;

import com.nokia.carbide.cpp.internal.pi.analyser.NpiInstanceRepository;

import com.nokia.carbide.cpp.internal.pi.analyser.ProfileVisualiser;

import com.nokia.carbide.cpp.internal.pi.model.FunctionResolver;

import com.nokia.carbide.cpp.internal.pi.model.GenericSample;

import com.nokia.carbide.cpp.internal.pi.model.GenericTrace;

import com.nokia.carbide.cpp.internal.pi.model.ParsedTraceData;

import com.nokia.carbide.cpp.internal.pi.model.ProfiledBinary;

import com.nokia.carbide.cpp.internal.pi.model.ProfiledFunction;

import com.nokia.carbide.cpp.internal.pi.model.ProfiledGeneric;

import com.nokia.carbide.cpp.internal.pi.model.ProfiledThread;

import com.nokia.carbide.cpp.internal.pi.model.TraceDataRepository;

import com.nokia.carbide.cpp.internal.pi.plugin.model.AbstractPiPlugin;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IClassReplacer;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IEventListener;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IRecordable;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IReportable;

import com.nokia.carbide.cpp.internal.pi.plugin.model.ITrace;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IViewMenu;

import com.nokia.carbide.cpp.internal.pi.plugin.model.IVisualizable;

import com.nokia.carbide.cpp.internal.pi.resolvers.SymbolFileFunctionResolver;

import com.nokia.carbide.cpp.internal.pi.test.AnalysisInfoHandler;

import com.nokia.carbide.cpp.internal.pi.utils.PIUtilities;

import com.nokia.carbide.cpp.internal.pi.visual.Defines;

import com.nokia.carbide.cpp.internal.pi.visual.GenericTraceGraph;

import com.nokia.carbide.cpp.internal.pi.visual.GraphDrawRequest;

import com.nokia.carbide.cpp.internal.pi.visual.PIEvent;

import com.nokia.carbide.cpp.pi.core.SessionPreferences;

import com.nokia.carbide.cpp.pi.editors.PIPageEditor;

import com.nokia.carbide.cpp.pi.importer.SampleImporter;

import com.nokia.carbide.cpp.pi.util.ColorPalette;

import com.nokia.carbide.cpp.pi.util.GeneralMessages;

/**

 * The main plugin class to be used in the desktop.

 */

public class AddressPlugin extends AbstractPiPlugin

		implements ITrace, IViewMenu, IClassReplacer, //IExportItem,

					IReportable, IRecordable, IVisualizable, IEventListener

{

	private static final String HELP_CONTEXT_ID = PIPageEditor.PI_ID + ".address";  //$NON-NLS-1$

//	private static HashMap<Integer,Long> uidToAddrThreadPeriod = new HashMap<Integer,Long>();

	// There will be three graphs - one each for editor pages 0, 1, 2

	// This code assumes that page 0 has the threads graph, 1 the binaries, and 2 the functions

	private final static int GRAPH_COUNT = 3;

	private boolean pagesCreated = false;

	// The shared instance.

	private static AddressPlugin plugin;

	private int functionsShown = 10;

	// string that should contain the maximum number of functions to display

	private String functionCountString;

	private static void setPlugin(AddressPlugin newPlugin)

	{

		plugin = newPlugin;

	}

	/**

	 * The constructor.

	 */

	public AddressPlugin() {

		super();

		setPlugin(this);

	}

	public Class getTraceClass()

	{

		return GppTrace.class;

	}

	public Class getReplacedClass(String className)

	{

		if (   className.indexOf("com.nokia.carbide.cpp.pi.address.GppTrace") != -1 //$NON-NLS-1$

			|| className.indexOf("fi.vtt.bappea.model.GppTrace") != -1 //$NON-NLS-1$

			|| className.indexOf("com.nokia.carbide.pi.address.GppTrace") != -1 //$NON-NLS-1$

			|| className.indexOf("fi.vtt.bappea.gppTracePlugin.GppTrace") != -1) //$NON-NLS-1$

		{

			return GppTrace.class;

		}

		else if (   className.indexOf("com.nokia.carbide.cpp.pi.address.GppSample") != -1 //$NON-NLS-1$

				 || className.indexOf("com.nokia.carbide.pi.address.GppSample") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.model.GppSample") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.gppTracePlugin.GppSample") != -1) //$NON-NLS-1$

		{

			return GppSample.class;

		}

		else if (   className.indexOf("com.nokia.carbide.cpp.pi.address.GppThread") != -1 //$NON-NLS-1$

				 || className.indexOf("com.nokia.carbide.pi.address.GppThread") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.model.GppThread") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.gppTracePlugin.GppThread") != -1) //$NON-NLS-1$

		{

			return GppThread.class;

		}

		else if (   className.indexOf("com.nokia.carbide.cpp.pi.address.GppProcess") != -1 //$NON-NLS-1$

				 || className.indexOf("com.nokia.carbide.pi.address.GppProcess") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.model.GppProcess") != -1 //$NON-NLS-1$

				 || className.indexOf("fi.vtt.bappea.gppTracePlugin.GppProcess") != -1) //$NON-NLS-1$

		{

			return GppProcess.class;

		}

		else

		{

			return null;

		}

	}

	public void initialiseTrace(GenericTrace genericTrace)

	{

		if (!(genericTrace instanceof GppTrace))

			return;

		GppTrace trace = (GppTrace)genericTrace;

		NpiInstanceRepository.getInstance().activeUidAddTrace("com.nokia.carbide.cpp.pi.address", trace); //$NON-NLS-1$

		// initialize the address/thread base sampling rate

		int samplingInterval = 1;

		if (trace.samples.size() > 2) {

			// because of a problem in older samplers, the first address/thread sample may have been thrown out

			samplingInterval = (int) (((GppSample) trace.samples.get(1)).sampleSynchTime - ((GppSample) trace.samples.get(0)).sampleSynchTime); 

		}

		NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.samplingInterval", new Integer(samplingInterval)); //$NON-NLS-1$

		// make sure that the sorted samples array exists

		if (trace.getSortedGppSamples() == null)

			trace.sortGppSamples();

		GppTraceGraph.refreshDataFromTrace(trace);

		PIPageEditor.setTime(0.0f, 0.0f);

		long lastSampleTime = trace.getSample(trace.samples.size() - 1).sampleSynchTime;

		PIPageEditor.currentPageEditor().setMaxEndTime(((double)lastSampleTime) / 1000.0f);

	}

	public GppTrace getTrace()

	{

		return (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

	}

	public GenericTraceGraph getTraceGraph(int graphIndex)

	{

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		if (trace != null) {

			int uid = NpiInstanceRepository.getInstance().activeUid();

			return trace.getTraceGraph(graphIndex, uid);

		} else

			return null;

	}

	public GenericTrace getTrace(int graphIndex)

	{	

		return (GenericTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

	}

/*

	public GenericTraceGraph getTraceGraph(int graphIndex, int uid)

	{	

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		// if we already had it, then we can pass in the formatted trace data

		return trace.getTraceGraph(graphIndex, uid);

	}

*/

	public Integer getLastSample(int graphIndex)

	{

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		if (trace == null)

			return null;

	  	//this sets GPP thread list visible by default

	  	((GppTraceGraph)trace.getTraceGraph(graphIndex)).piEventReceived(new PIEvent(null, PIEvent.MOUSE_PRESSED));

	  	return new Integer(trace.getLastSampleNumber());

	}

	public GraphDrawRequest getDrawRequest(int graphIndex) {

		return null;

	}

	public void receiveSelectionEvent(String eventString)

	{

		if (eventString == null)

			return;

		int currentPage = PIPageEditor.currentPageIndex();

		if (   (currentPage != PIPageEditor.THREADS_PAGE)

			&& (currentPage != PIPageEditor.BINARIES_PAGE)

			&& (currentPage != PIPageEditor.FUNCTIONS_PAGE))

			  return;

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		if (eventString.equals("fillSelected")) //$NON-NLS-1$

	    {

	    	PIEvent be = new PIEvent(null, PIEvent.SET_FILL_SELECTED_THREAD);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.THREADS_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.BINARIES_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.FUNCTIONS_PAGE)).piEventReceived(be);

	    	NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.fillAll", Boolean.FALSE); //$NON-NLS-1$

	    }

		else if (eventString.equals("fillAll")) //$NON-NLS-1$

		{

			PIEvent be = new PIEvent(null, PIEvent.SET_FILL_ALL_THREADS);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.THREADS_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.BINARIES_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.FUNCTIONS_PAGE)).piEventReceived(be);

	    	NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.fillAll", Boolean.TRUE); //$NON-NLS-1$

		}

		else if (eventString.equals("fillNone")) //$NON-NLS-1$

		{

			PIEvent be = new PIEvent(null, PIEvent.SET_FILL_OFF);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.THREADS_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.BINARIES_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.FUNCTIONS_PAGE)).piEventReceived(be);

	    	NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.fillAll", Boolean.FALSE); //$NON-NLS-1$

		}

		else if (eventString.equals("setBarOn")) //$NON-NLS-1$

		{

			PIEvent be = new PIEvent(null, PIEvent.GPP_SET_BAR_GRAPH_ON);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.THREADS_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.BINARIES_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.FUNCTIONS_PAGE)).piEventReceived(be);

	    	NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.bar", Boolean.TRUE); //$NON-NLS-1$

		}

		else if (eventString.equals("setBarOff")) //$NON-NLS-1$

		{

			PIEvent be = new PIEvent(null, PIEvent.GPP_SET_BAR_GRAPH_OFF);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.THREADS_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.BINARIES_PAGE)).piEventReceived(be);

	    	((GppTraceGraph)trace.getTraceGraph(PIPageEditor.FUNCTIONS_PAGE)).piEventReceived(be);

	    	NpiInstanceRepository.getInstance().activeUidSetPersistState("com.nokia.carbide.cpp.pi.address.bar", Boolean.FALSE); //$NON-NLS-1$

		}

	  	else if (eventString.equals("resetToCurrentMode")) //$NON-NLS-1$

	  	{

	  		((GppTraceGraph)trace.getTraceGraph(currentPage)).action(eventString);

	  	}

	}

	public String getTraceName() {

		return "Address/Thread";	//$NON-NLS-1$

	}

	public int getTraceId() {

		return 1;

	}

	public ParsedTraceData parseTraceFile(File file /*, ProgressBar progressBar*/) throws IOException

	{

//		progressBar.setString("Parsing address and thread trace");

        GppTraceParser gppParser = new GppTraceParser();

        ParsedTraceData parsed = gppParser.parse(file);

        SymbolFileFunctionResolver sffp = this.resolveSymbolFileParser(/*progressBar*/);

        parsed.functionResolvers = new FunctionResolver[]{sffp};

        return parsed;

	}

	private SymbolFileFunctionResolver resolveSymbolFileParser(/*ProgressBar progressBar*/)

    {

        //        	progressBar.setString("Parsing symbol file");

        //reads the symbol name path from configuration data

        File symbolFile = new File(

        		SampleImporter.getInstance().getRomSymbolFile());

        SymbolFileFunctionResolver sffp = new SymbolFileFunctionResolver();

        sffp.parseAndProcessSymbolFile(symbolFile);

        return sffp;

    }

	public Hashtable<Integer,Object> getSummaryTable(double startTime, double endTime)

	{

		if (plugin == null)

			return null;

		// This code relies on currentPage being the same as graphIndex

		int currentPage = PIPageEditor.currentPageIndex();

		if (   (currentPage != PIPageEditor.THREADS_PAGE)

			&& (currentPage != PIPageEditor.BINARIES_PAGE)

			&& (currentPage != PIPageEditor.FUNCTIONS_PAGE))

			  return null;

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		Hashtable<Integer,Object> summary = new Hashtable<Integer,Object>();

		Enumeration<ProfiledGeneric> e = trace.getSortedThreadsElements();

		while (e.hasMoreElements())

		{

			Vector<Object> data = new Vector<Object>();

			ProfiledThread pt = (ProfiledThread)e.nextElement();

			data.add(pt.getNameString());

			data.add(pt.getAverageLoadValueString(currentPage));

			summary.put(new Integer(pt.getThreadId()), data);

		}

		return summary;

	}

	public String getGeneralInfo()

	{

		return null;

	}

	public ArrayList<String> getColumnNames()

	{

		ArrayList<String> names = new ArrayList<String>();

		names.add(Messages.getString("AddressPlugin.0"));  //$NON-NLS-1$

		names.add(Messages.getString("AddressPlugin.1"));  //$NON-NLS-1$

		return names;

	}

	public ArrayList<Boolean> getColumnSortTypes()

	{

		ArrayList<Boolean> sortTypes = new ArrayList<Boolean>();

		sortTypes.add(SORT_BY_NAME);

		sortTypes.add(SORT_BY_NUMBER);

		return sortTypes;

	}

	public String getActiveInfo(Object key, double startTime, double endTime)

	{

		int threadId;

		if (key instanceof Integer)

			threadId = ((Integer)key).intValue();

		else

			return null;

		String threadName = ""; //$NON-NLS-1$

		Hashtable<Integer,Object> tmpTable = this.getSummaryTable(startTime, endTime);

		if (tmpTable != null)

		{

			threadName = (String)((Vector<Object>)tmpTable.get(key)).elementAt(0);

		}

		GppTrace trace = (GppTrace)NpiInstanceRepository.getInstance().activeUidGetTrace("com.nokia.carbide.cpp.pi.address"); //$NON-NLS-1$

		Vector<GenericSample> samples = trace.getSamplesInsideTimePeriod((long)startTime, (long)endTime);

		Hashtable<String,Integer> functionLoad = new Hashtable<String,Integer>();

		int threadSampleAmount = 0;

		for (Enumeration<GenericSample> e = samples.elements(); e.hasMoreElements();)

		{

			GppSample sample = (GppSample)e.nextElement();

			String functionName = Messages.getString("AddressPlugin.2");  //$NON-NLS-1$

			if (sample.currentFunctionSym != null)

				functionName = sample.currentFunctionSym.functionName;

			else if (sample.currentFunctionItt != null)

				functionName = sample.currentFunctionItt.functionName;

			if (sample.thread.threadId.intValue() == threadId)

			{

				Integer load = (Integer)functionLoad.get(functionName);

				if (load != null)

				{

					functionLoad.remove(functionName);

					functionLoad.put(functionName, new Integer(load.intValue()+1));

				}

				else

				{

					functionLoad.put(functionName, new Integer(1));

				}

				threadSampleAmount++;

			}

		}

		float totalLoad = 0;

		int functions = 0;

//		int functionsShown = 10;

		for (Enumeration<String> e = functionLoad.keys();e.hasMoreElements();)

		{

			String testFunction = e.nextElement();

			float load = ((Integer)functionLoad.get(testFunction)).floatValue()*100/threadSampleAmount;

			totalLoad += load;

			functions++;

		}

		ArrayList<Object> sortedFunctions = new ArrayList<Object>();

//		functionsShown = new Integer(functionsShownField.getText()).intValue();

		while (functionLoad.size() != 0 && sortedFunctions.size() <= (functionsShown * 2) - 1)

		{

			String maxFunction = null;

			int maxCount = 0;

			for (Enumeration<String> e = functionLoad.keys(); e.hasMoreElements();)

			{

				String testFunction = e.nextElement();

				int count = functionLoad.get(testFunction).intValue();

				if (count > maxCount)

				{

					maxFunction = testFunction;

					maxCount = count;

				}

			}

			sortedFunctions.add(maxFunction);

			sortedFunctions.add(new Integer(maxCount));

			functionLoad.remove(maxFunction);

		}

		if (threadSampleAmount > 0)

		{

			String finalString = "\r\n" + Messages.getString("AddressPlugin.4")+threadName+"\r\n";   //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$

			for (int i = 0; i < sortedFunctions.size(); )

			{

				String name = (String)sortedFunctions.get(i++);

				//String percent = ((Integer)e.nextElement()).floatValue()*100/threadSampleAmount+"%";

				float load = ((Integer)sortedFunctions.get(i++)).floatValue()*100/threadSampleAmount;

				String percent = new Float(load/totalLoad*100).toString();

				finalString += percent+"% "+name+"\r\n"; //$NON-NLS-1$ //$NON-NLS-2$ 

			}

			return finalString;

		}

		else return "\r\n"  + Messages.getString("AddressPlugin.8")+threadName+"\r\n";   //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$

	}

	public Action getIncludeAction()

	{

		return null;

	}

	public Action getExportAction() {

		Action action = new Action(Messages.getString("AddressPlugin.10"), Action.AS_PUSH_BUTTON) {  //$NON-NLS-1$

			public void run() {

				ParsedTraceData ptd = TraceDataRepository.getInstance().getTrace(NpiInstanceRepository.getInstance().activeUid(), getTraceClass());

				if (ptd != null) {

					final GppTrace gppTraceTmp = (GppTrace)ptd.traceData;

					final FileDialog dlg = new FileDialog(PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(), SWT.SAVE);

					new Thread()

					{

						public void run()

						{

							dlg.setFilterNames(new String[] {

									Messages.getString("AddressPlugin.11") + " (*.csv)"  //$NON-NLS-1$ //$NON-NLS-2$

							});

							dlg.setFilterExtensions(new String[] {

									"*.csv" //$NON-NLS-1$

							});

							dlg.setText(Messages.getString("AddressPlugin.13"));  //$NON-NLS-1$

							String fileName = dlg.open();

							if (fileName == null) {

								return;

							}

							File file = new File(fileName);

						    GppTraceCsvPrinter csvPrinter = new GppTraceCsvPrinter(gppTraceTmp);

						    String csvPrint = csvPrinter.getCsvPrint();

						    if (csvPrint != null)

						    {

						    	try {

							        PIUtilities.saveCsvPrint(file, csvPrint);

						    	}

								catch (Exception e) 

								{

								    e.printStackTrace();

								}

						    }

						}

					}.start();

				}

			}