Laboratoire ICAR » Plateforme TXM

import java.io.File;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.UnsupportedEncodingException;

import java.util.ArrayList;

import java.util.List;

import java.util.Properties;

import org.eclipse.jface.dialogs.IDialogConstants;

import org.eclipse.swt.custom.ScrolledComposite;

import org.eclipse.swt.events.ControlAdapter;

import org.eclipse.swt.events.ControlEvent;

import org.eclipse.swt.graphics.Rectangle;

import org.eclipse.swt.widgets.Button;

import org.eclipse.swt.widgets.Combo;

import org.eclipse.swt.widgets.Event;

import org.eclipse.swt.widgets.FileDialog;

import org.eclipse.swt.widgets.Listener;

import org.txm.rcp.swt.widget.QueryWidget;

import org.txm.searchengine.cqp.CQPSearchEngine;

import org.txm.searchengine.cqp.clientExceptions.CqiClientException;

import org.txm.searchengine.cqp.corpus.StructuralUnit;

import org.txm.searchengine.cqp.corpus.StructuralUnitProperty;

import org.txm.utils.io.IOUtils;

// TODO: Auto-generated Javadoc

	// For the "simple" tab

	/** The selected structural unit. */

	private StructuralUnit selectedStructuralUnit;

	/** The selected structural unit property. */

	private StructuralUnitProperty selectedStructuralUnitProperty;

	// for the "advanced" tab

	/** The queries. */

	private List<QueryWidget> queries = new ArrayList<>();

	/** The part names. */

	private List<Text> partNames = new ArrayList<>();

	/** The queries string. */

	List<String> queriesString = new ArrayList<>();

	/** The is simple selected. */

	private boolean isSimpleSelected = true;

	/** The is assisted selected. */

	private boolean isAssistedSelected = false;

	/** The is advanced selected. */

	private boolean isAdvancedSelected = false;

	/** The name. */

	private String name;

	/** The composite for assisted. */

	private PartitionComposer compositeForAssisted;

	/** The structural units combo. */

	private Combo structuralUnitsCombo;

	/** The structural units final. */

	private List<StructuralUnit> structuralUnitsFinal;

	/** The property combo. */

	private Combo propertyCombo;

	/** The composite for advanced. */

	private Composite compositeForAdvanced;

	/** The scroll composite. */

	private ScrolledComposite scrollComposite;

		nameLabel.setLayoutData(new GridData(GridData.FILL, GridData.CENTER,

				false, false));

		TabItem simpleTab = new TabItem(tabFolder, SWT.NONE);

		Composite composite = new Composite(tabFolder, SWT.NONE);

		simpleTab.setControl(composite);

		GridLayout layout = new GridLayout(2, false);

		composite.setLayout(layout);

		// StructuralUnit

		// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

		Label structureLabel = new Label(composite, SWT.NONE);

		structureLabel.setText(TXMUIMessages.ampStructureColon);

		structureLabel.setLayoutData(new GridData(GridData.END,

				GridData.CENTER, false, false));

		structuralUnitsCombo = new Combo(composite, SWT.READ_ONLY);

		structuralUnitsCombo.setLayoutData(new GridData(GridData.FILL, GridData.CENTER,

				true, false));

		List<StructuralUnit> structuralUnits = null;

		try {

			structuralUnits = corpus.getOrderedStructuralUnits();

		}

		catch (CqiClientException e) {

			Log.severe(TXMCoreMessages.bind(TXMCoreMessages.error_errorP0, Log.toString(e)));

			return null;

		}

		structuralUnitsFinal = new ArrayList<>();

		for (StructuralUnit unit : structuralUnits) {

			if (unit.getName().equals("txmcorpus")) continue; // ignore txmcorpus structure //$NON-NLS-1$

			if (unit.getOrderedProperties() != null

					&& unit.getOrderedProperties().size() != 0) {

				/*

				 * if ((unit.getName().equals("text") &&

				 * unit.getProperties().size() > 3)) //$NON-NLS-1$ {

				 * structuralUnitsCombo.add(unit.getName());

				 * structuralUnitsFinal.add(unit); } else

				 * if (!unit.getName().equals("text")) //$NON-NLS-1$ {

				 */

				structuralUnitsCombo.add(unit.getName());

				structuralUnitsFinal.add(unit);

				// }

			}

		}

		if (structuralUnitsCombo.getItemCount() == 0) {

			this.getButton(IDialogConstants.OK_ID).setEnabled(false);

		}

		else {

			structuralUnitsCombo.select(0);

			String[] items = structuralUnitsCombo.getItems();

			for (int i = 0; i < items.length; i++) {

				if (items[i].equals("text")) //$NON-NLS-1$

				{

					structuralUnitsCombo.select(i);

					break;

				}

			}

			selectedStructuralUnit = structuralUnitsFinal.get(0);

		}

		// combo.setSize (200, 200);

		// StructuralUnitProperty

		// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

		Label propertyLabel = new Label(composite, SWT.NONE);

		propertyLabel.setText(TXMUIMessages.ampPropertyColon);

		propertyLabel.setLayoutData(new GridData(GridData.END, GridData.CENTER,

				false, false));

		propertyCombo = new Combo(composite, SWT.READ_ONLY);

		propertyCombo.setLayoutData(new GridData(GridData.FILL, GridData.CENTER,

				true, false));

		// combo.setSize (200, 200);

		structuralUnitsCombo.addSelectionListener(new SelectionListener() {

			@Override

			public void widgetDefaultSelected(SelectionEvent e) {}

			@Override

			public void widgetSelected(SelectionEvent event) {

				reloadProperties();

			}

		});

		for (StructuralUnitProperty property : structuralUnitsFinal.get(0).getOrderedProperties()) {

			if (!(structuralUnitsFinal.get(0).getName().equals("text") &&  //$NON-NLS-1$

					(property.getName().equals("project") || property.getName().equals("base")))) //$NON-NLS-1$ //$NON-NLS-2$

				propertyCombo.add(property.getName());

		}

		propertyCombo.select(0);

		selectedStructuralUnitProperty = selectedStructuralUnit.getOrderedProperties()

				.get(0);

		propertyCombo.addSelectionListener(new SelectionListener() {

			@Override

			public void widgetDefaultSelected(SelectionEvent e) {}

			@Override

			public void widgetSelected(SelectionEvent event) {

				selectedStructuralUnitProperty = selectedStructuralUnit

						.getOrderedProperties().get(propertyCombo.getSelectionIndex());

			}

		});

		reloadProperties();

		TabItem assistedTab = new TabItem(tabFolder, SWT.NONE);

		// Third tab: avanced mode

		// --------------------------------------------------------

		TabItem advancedTab = new TabItem(tabFolder, SWT.NONE);

		Composite advanceContainer = new Composite(tabFolder, SWT.NONE);

		GridLayout advanceLayout = new GridLayout(2, false);

		advanceContainer.setLayout(advanceLayout);

		advancedTab.setControl(advanceContainer);

		scrollComposite = new ScrolledComposite(advanceContainer, SWT.V_SCROLL | SWT.BORDER);

		scrollComposite.setLayoutData(new GridData(GridData.FILL, GridData.FILL, true, true, 2, 1));

		Button plusButton = new Button(advanceContainer, SWT.NONE);

		plusButton.setLayoutData(new GridData(GridData.HORIZONTAL_ALIGN_CENTER, GridData.FILL, false, false));

		plusButton.setText("+"); //$NON-NLS-1$

		plusButton.addListener(SWT.Selection, new Listener() {

			@Override

			public void handleEvent(Event event) {

				addQueryField();

			}

		});

		// TODO enable when specified

		// Button importButton = new Button(advanceContainer, SWT.NONE);

		// importButton.setLayoutData(new GridData(GridData.HORIZONTAL_ALIGN_CENTER,

		// GridData.FILL, false, false));

		// importButton.setText("Import from properties file..."); //$NON-NLS-1$

		// importButton.addListener(SWT.Selection, new Listener() {

		// @Override

		// public void handleEvent(Event event) {

		// FileDialog d = new FileDialog(event.display.getActiveShell());

		// d.setFilterExtensions(new String[] {"*.properties"});

		// d.setText("Select a properties file containing CQL queries");

		// String path = d.open();

		// if (path != null) {

		// Properties props = new Properties();

		// try {

		// props.load(IOUtils.getReader(new File(path)));

		// for (Object key : props.keySet()) {

		// String value = props.getProperty(key.toString());

		// QueryWidget qw = addQueryField(key.toString().substring(0, Math.min(20, key.toString().length())), value);

		// }

		// } catch (Exception e) {

		// // TODO Auto-generated catch block

		// e.printStackTrace();

		// }

		// }

		//

		// }

		// });

		compositeForAdvanced = new Composite(scrollComposite, SWT.NONE);

		GridLayout layoutForAdvanced = new GridLayout(3, false);

		compositeForAdvanced.setLayout(layoutForAdvanced);

		scrollComposite.setContent(compositeForAdvanced);

		scrollComposite.setExpandVertical(true);

		scrollComposite.setExpandHorizontal(true);

		scrollComposite.addControlListener(new ControlAdapter() {

			@Override

			public void controlResized(ControlEvent e) {

				Rectangle r = scrollComposite.getClientArea();

				scrollComposite.setMinSize(compositeForAdvanced.computeSize(

						r.width, SWT.DEFAULT));

			}

		});

		createAdvancedQueryField();

				// record the tab currently in use

				if (tabFolder.getSelectionIndex() == 0) {

					isSimpleSelected = true;

					isAssistedSelected = false;

					isAdvancedSelected = false;

				}

				else if (tabFolder.getSelectionIndex() == 1) {

					isSimpleSelected = false;

					isAssistedSelected = true;

					isAdvancedSelected = false;

				}

				else {

					isSimpleSelected = false;

					isAssistedSelected = false;

					isAdvancedSelected = true;

				}

	/**

	 * Reload properties.

	 */

	private void reloadProperties() {

		List<StructuralUnitProperty> properties = null;

		selectedStructuralUnit = structuralUnitsFinal.get(structuralUnitsCombo

				.getSelectionIndex());

		properties = selectedStructuralUnit.getOrderedProperties();

		propertyCombo.removeAll();

		for (StructuralUnitProperty property : properties) {

			/*

			 * if (!(selectedStructuralUnit.getName().equals("text") &&

			 * //$NON-NLS-1$ (property.getName().equals("id") || //$NON-NLS-1$

			 * property.getName().equals("base") || //$NON-NLS-1$

			 * property.getName().equals("project")))) //$NON-NLS-1$

			 */

			if (selectedStructuralUnit.getName().equals("text") && //$NON-NLS-1$

					(property.getName().equals("project") || property.getName().equals("base"))) continue; // ignore technical properties //$NON-NLS-1$ //$NON-NLS-2$

			propertyCombo.add(property.getName());

		}

		if (propertyCombo.getItemCount() > 0) {

			propertyCombo.select(0);

			String[] items = propertyCombo.getItems();

			for (int i = 0; i < items.length; i++)

				if (items[i].equals("id")) //$NON-NLS-1$

				{

					propertyCombo.select(i);

					break;

				}

			selectedStructuralUnitProperty = selectedStructuralUnit

					.getOrderedProperties().get(0);

			// this.getButton(IDialogConstants.OK_ID).setEnabled(true);

		}

	}

	/**

	 * Return true if no query is currently given in the "advanced" tab (ie if

	 * all Text field opened are empty).

	 *

	 * @return true, if successful

	 */

	private boolean noquery() {

		for (QueryWidget query : queries) {

			if (query.getText().length() == 0)

				return true;

		}

		return false;

	}

	/**

	 * Creates the advanced query field.

	 *

	 */

	private void createAdvancedQueryField() {

		addQueryField();

		Rectangle r = scrollComposite.getClientArea();

		scrollComposite.setMinSize(compositeForAdvanced.computeSize(r.width,

				SWT.DEFAULT));

	}

	/**

	 * Adds the query field.

	 *

	 */

	private QueryWidget addQueryField() {

		return addQueryField(null, null);

	}

	/**

	 * Adds the query field.

	 *

	 * @param name the query name

	 * @param value the query

	 */

	private QueryWidget addQueryField(String name, String value) {

		final Text advancedPartLabel = new Text(compositeForAdvanced, SWT.BORDER);

		advancedPartLabel.setText(TXMUIMessages.part

				+ (partNames.size() + 1));

		advancedPartLabel.setLayoutData(new GridData(GridData.FILL,

				GridData.CENTER, false, false));

		if (name != null && name.length() > 0) {

			advancedPartLabel.setText(name);

		}

		partNames.add(advancedPartLabel);

		final QueryWidget queryText;

		queryText = new QueryWidget(compositeForAdvanced, SWT.DROP_DOWN, corpus.getProject(), CQPSearchEngine.getEngine());

		queryText.setLayoutData(new GridData(GridData.FILL, GridData.FILL, true, false));

		GridData gridData = new GridData(GridData.FILL, GridData.FILL, true, false);

		// gridData.widthHint = convertHeightInCharsToPixels(20);

		queryText.setLayoutData(gridData);

		queries.add(queryText);

		if (value != null) {

			queryText.setText(value);

		}

		final Button minusButton = new Button(compositeForAdvanced, SWT.NONE);

		minusButton.setText("-"); //$NON-NLS-1$

		// minusButton.setData(queryText);

		minusButton.addListener(SWT.Selection, new Listener() {

			@Override

			public void handleEvent(Event event) {

				queries.remove(queryText);

				partNames.remove(advancedPartLabel);

				advancedPartLabel.dispose();

				queryText.dispose();

				minusButton.dispose();

				compositeForAdvanced.layout(true);

			}

		});

		Rectangle r = scrollComposite.getClientArea();

		scrollComposite.setMinSize(compositeForAdvanced.computeSize(r.width,

				SWT.DEFAULT));

		scrollComposite.update();

		scrollComposite.layout();

		compositeForAdvanced.update();

		compositeForAdvanced.layout();

		return queryText;

	}

		if (isAdvancedSelected() && noquery()) {

		/*

		 * if (isAssistedSelected) {

		 * if (compositeForAssisted.hasUnassigned()) {

		 * System.out.println(Messages.CreatePartitionDialog_11);

		 * return;

		 * }

		 * }

		 */

		for (QueryWidget query : queries) {

			queriesString.add(query.getQueryString());

			query.memorize();

		}

	/**

	 * Tell if the user is using the "advanced" tab.

	 *

	 * @return true, if is advanced selected

	 */

	public boolean isAdvancedSelected() {

		return isAdvancedSelected;

	}

	 * Gets the structural unit.

	 *

	 * @return the structural unit

	 */

	public StructuralUnit getStructuralUnit() {

		if (isAdvancedSelected())

			throw new IllegalStateException(TXMUIMessages.cannotAskForTheSimpleTabWhenTheUserHasSelectedTheAdvancedTab);

		return selectedStructuralUnit;

	}

	/**

	 * Gets the structural unit property.

	 *

	 * @return the structural unit property

	 */

	public StructuralUnitProperty getStructuralUnitProperty() {

		if (isAdvancedSelected())

			throw new IllegalStateException(TXMUIMessages.cannotAskForTheSimpleTabWhenTheUserHasSelectedTheAdvancedTab);

		return selectedStructuralUnitProperty;

	}

	/**

	 * Gets the queries.

	 *

	 * @return the queries

	 */

	public List<String> getQueries() {

		if (!isAdvancedSelected())

			throw new IllegalStateException(TXMUIMessages.cannotAskForTheSimpleTabWhenTheUserHasSelectedTheAdvancedTab);

		return queriesString;

	}

	/**

		// Simple mode

		if (isSimpleSelected) {

			StructuralUnit structuralUnit = this.structuralUnitsFinal.get(this.structuralUnitsCombo.getSelectionIndex());

			StructuralUnitProperty property = structuralUnit.getProperty(this.propertyCombo.getText());

			try {

				// auto-naming if needed

				if (name.isEmpty()) {

					name = property.asFullNameString();

				}

				partition = new Partition(corpus);

				partition.setParameters(name, property, null);

			}

			catch (Exception e) {

				org.txm.utils.logger.Log.printStackTrace(e);

				return null;

			}

		// Assisted mode

		else if (isAssistedSelected) {

		// Advanced mode

		else if (isAdvancedSelected) {

			try {

				// auto-naming if needed

				if (name.isEmpty()) {

					name = "partition" + (this.corpus.getPartitions().size() + 1); //$NON-NLS-1$

					Log.info(TXMCoreMessages.bind(TXMUIMessages.noPartitionNameWasSpecifiedTheNameIsP0, name));

				}

				partition = new Partition(this.corpus);

				partition.setParameters(name, this.getQueries(), this.getAdvancePartNames());

			}

			catch (Exception e) {

				org.txm.utils.logger.Log.printStackTrace(e);

				return null;

			}

	 * Gets the advance part names.

	 *

	 * @return the advance part names

	 */

	private List<String> getAdvancePartNames() {

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

		int count = 0;

		for (Text t : partNames) {

			String name = t.getText().trim();

			if (name.length() == 0)

				name = TXMUIMessages.part + count++;

			values.add(name);

		}

		return values;

	}

	/**

import org.eclipse.swt.dnd.DropTargetListener;

import org.eclipse.swt.dnd.TransferData;

			titleLabel.setLayoutData(new GridData(GridData.FILL,

					GridData.CENTER, false, false, 1, 1));

			Font f = titleLabel.getFont();

			FontData defaultFont = f.getFontData()[0];

			defaultFont.setStyle(SWT.BOLD);

			Font newf = new Font(Display.getCurrent(), defaultFont);

			titleLabel.setFont(newf);

			title.setLayoutData(new GridData(GridData.FILL, GridData.FILL,

					true, false, 2, 1));

			delete.setLayoutData(new GridData(GridData.FILL, GridData.FILL,

					false, false, 1, 1));

					for (String s : p.getColumns())

			add.setLayoutData(new GridData(GridData.FILL, GridData.FILL, true,

					false, 2, 1));

					for (String s : composer.getSelection())

					if (composer.availableList.getItemCount() > 0)

			rmv.setLayoutData(new GridData(GridData.FILL, GridData.FILL, true,

					false, 2, 1));

			columns = new org.eclipse.swt.widgets.List(this, SWT.MULTI

					| SWT.READ_ONLY | SWT.BORDER | SWT.V_SCROLL);

			columns.setLayoutData(new GridData(GridData.FILL, GridData.FILL,

					true, true, 4, 1));