/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.uima.ruta.textruler.ui;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import org.apache.commons.lang3.StringUtils;
import org.apache.uima.UIMAFramework;
import org.apache.uima.resource.metadata.TypeDescription;
import org.apache.uima.resource.metadata.TypeSystemDescription;
import org.apache.uima.ruta.textruler.TextRulerPlugin;
import org.apache.uima.ruta.textruler.core.TextRulerToolkit;
import org.apache.uima.util.InvalidXMLException;
import org.apache.uima.util.XMLInputSource;
import org.apache.uima.util.XMLizable;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.viewers.CheckboxTableViewer;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.KeyAdapter;
import org.eclipse.swt.events.KeyEvent;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.FormAttachment;
import org.eclipse.swt.layout.FormData;
import org.eclipse.swt.layout.FormLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.FileDialog;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.TableItem;
public class AddRemoveList extends Composite {
private boolean error = false;
private String name;
private CheckboxTableViewer listViewer;
private Button add;
private Button remove;
private Button up;
private Button down;
private Button ts;
private TextRulerViewComposite parent;
protected final HashMap<String, Image> images;
private ArrayList<String> types;
private boolean ordered;
public AddRemoveList(String name, HashMap<String, Image> images, TextRulerViewComposite parent,
boolean ordered) {
super(parent, SWT.NULL);
this.name = name;
this.images = images;
this.parent = parent;
this.ordered = ordered;
init();
}
private void setTypes() {
error = false;
types.clear();
String preFilePath = TextRulerViewComposite.getScriptPath();
File preFile = new File(preFilePath);
if (preFile.exists() == false || StringUtils.isEmpty(preFilePath)) {
printErrorDialog("The preprocessing file was not found!");
error = true;
return;
}
Path scriptPath = new Path(preFilePath);
String defaultTypeSystemDescriptorLocation = TextRulerToolkit
.getTypeSystemDescriptorFromTMSourceFile(scriptPath);
TypeSystemDescription defaultTypeSystemDescription = null;
try {
defaultTypeSystemDescription = UIMAFramework.getXMLParser().parseTypeSystemDescription(
new XMLInputSource(new File(defaultTypeSystemDescriptorLocation)));
defaultTypeSystemDescription.resolveImports();
TypeDescription[] systemTypes = defaultTypeSystemDescription.getTypes();
for (TypeDescription typeDescription : systemTypes) {
types.add(typeDescription.getName());
}
Collections.sort(types);
} catch (InvalidXMLException e) {
TextRulerPlugin.error(e);
} catch (IOException e) {
TextRulerPlugin.error(e);
}
}
public static void printErrorDialog(String error) {
ErrorDialog.openError(Display.getCurrent().getActiveShell(), "File not Found!", error,
new Status(IStatus.ERROR, "-1", "File not found!"));
}
private void init() {
this.setLayout(new FormLayout());
this.setSize(200, 300);
types = new ArrayList<String>();
listViewer = CheckboxTableViewer.newCheckList(this, SWT.BORDER | SWT.H_SCROLL | SWT.V_SCROLL
| SWT.MULTI);
FormData listViewerLData = new FormData();
listViewerLData.width = 160;
listViewerLData.height = 150;
listViewerLData.left = new FormAttachment(0, 1000, 0);
listViewerLData.right = new FormAttachment(1000, 1000, -40);
listViewerLData.top = new FormAttachment(0, 1000, 0);
listViewerLData.bottom = new FormAttachment(1000, 1000, -10);
listViewer.getControl().setLayoutData(listViewerLData);
listViewer.setContentProvider(parent.new ListContentProvider());
listViewer.setLabelProvider(parent.new DefaultLabelProvider(name));
// listViewer.setLabelProvider(parent.new ViewLabelProvider());
// listViewer.setSorter(parent.new NameSorter());
listViewer.getTable().setVisible(true);
if (name.equals("FilterTypes")) {
String input = TextRulerToolkit.getStandardFilterSetString();
addAll(input);
} else {
listViewer.setAllChecked(true);
}
listViewer.getControl().addKeyListener(new KeyAdapter() {
@Override
public void keyPressed(KeyEvent e) {
int keyCode = e.keyCode;
// backspace or delete
if (keyCode == SWT.BS || keyCode == SWT.DEL) {
String selected = listViewer.getSelection().toString();
selected = selected.substring(1, selected.length() - 1);
String[] types = selected.split(", ");
for (String string : types) {
listViewer.remove(string);
}
}
}
});
ts = new Button(this, SWT.PUSH | SWT.CENTER);
FormData tsLData = new FormData();
tsLData.width = 25;
tsLData.height = 25;
tsLData.top = new FormAttachment(0, 1000, 0);
tsLData.right = new FormAttachment(1000, 1000, -10);
ts.setLayoutData(tsLData);
Image tsIcon = images.get("prepFolder");
ts.setImage(tsIcon);
ts.addSelectionListener(new SelectionAdapter() {
@Override
public void widgetSelected(SelectionEvent event) {
FileDialog fd = new FileDialog(getShell(), SWT.OPEN);
fd.setText("Choose Type System Descriptor");
String[] filterExt = { "*.xml", "*.*" };
fd.setFilterExtensions(filterExt);
fd.setFileName(parent.getPreprocessorTMFile());
String file = fd.open();
if (file != null) {
XMLizable descriptor = null;
try {
descriptor = UIMAFramework.getXMLParser().parse(new XMLInputSource(file));
} catch (InvalidXMLException e) {
} catch (IOException e) {
}
if (descriptor instanceof TypeSystemDescription) {
TypeSystemDescription tsd = (TypeSystemDescription) descriptor;
try {
tsd.resolveImports();
} catch (InvalidXMLException e) {
}
List<String> tsTypes = new ArrayList<String>();
TypeDescription[] typeDescs = tsd.getTypes();
for (TypeDescription typeDescription : typeDescs) {
String n = typeDescription.getName();
tsTypes.add(n);
}
addAll(tsTypes);
}
}
}
});
add = new Button(this, SWT.PUSH | SWT.CENTER);
FormData addLData = new FormData();
addLData.width = 25;
addLData.height = 25;
addLData.top = new FormAttachment(0, 1000, 27);
addLData.right = new FormAttachment(1000, 1000, -10);
add.setLayoutData(addLData);
Image addIcon = images.get("add");
add.setImage(addIcon);
add.addSelectionListener(new SelectionAdapter() {
@Override
public void widgetSelected(SelectionEvent event) {
setTypes();
if (error)
return;
Display display = Display.getDefault();
Shell shell = new Shell(display, SWT.RESIZE | SWT.APPLICATION_MODAL | SWT.DIALOG_TRIM);
if (name.equals("FilterTypes"))
shell.setText("Filter Types");
else
shell.setText("Slot Types");
new SelectTypesDialog(shell, types, AddRemoveList.this);
}
});
remove = new Button(this, SWT.PUSH | SWT.CENTER);
FormData removeLData = new FormData();
removeLData.width = 25;
removeLData.height = 25;
removeLData.top = new FormAttachment(0, 1000, 54);
removeLData.right = new FormAttachment(1000, 1000, -10);
remove.setLayoutData(removeLData);
Image deleteIcon = images.get("delete");
remove.setImage(deleteIcon);
remove.addSelectionListener(new SelectionAdapter() {
@Override
public void widgetSelected(SelectionEvent event) {
String selected = listViewer.getSelection().toString();
selected = selected.substring(1, selected.length() - 1);
String[] types = selected.split(", ");
for (String string : types) {
listViewer.remove(string);
}
}
});
if (ordered) {
up = new Button(this, SWT.PUSH | SWT.CENTER);
FormData upLData = new FormData();
upLData.width = 25;
upLData.height = 25;
upLData.top = new FormAttachment(0, 1000, 81);
upLData.right = new FormAttachment(1000, 1000, -10);
up.setLayoutData(upLData);
Image upIcon = images.get("up");
up.setImage(upIcon);
up.addSelectionListener(new SelectionAdapter() {
@Override
public void widgetSelected(SelectionEvent event) {
ISelection selection = listViewer.getSelection();
List<String> dataList = new ArrayList<String>();
if (selection instanceof StructuredSelection) {
StructuredSelection ss = (StructuredSelection) selection;
Iterator<?> iterator = ss.iterator();
while (iterator.hasNext()) {
Object object = (Object) iterator.next();
if (object instanceof String) {
String obj = (String) object;
dataList = new ArrayList<String>();
TableItem[] items = listViewer.getTable().getItems();
for (TableItem tableItem : items) {
dataList.add((String) tableItem.getData());
}
int indexOf = dataList.indexOf(obj);
Collections.swap(dataList, indexOf, indexOf - 1);
}
}
}
listViewer.remove(dataList.toArray());
addAll(dataList);
}
});
down = new Button(this, SWT.PUSH | SWT.CENTER);
FormData downLData = new FormData();
downLData.width = 25;
downLData.height = 25;
downLData.top = new FormAttachment(0, 1000, 108);
downLData.right = new FormAttachment(1000, 1000, -10);
down.setLayoutData(downLData);
Image downIcon = images.get("down");
down.setImage(downIcon);
down.addSelectionListener(new SelectionAdapter() {
@Override
public void widgetSelected(SelectionEvent event) {
ISelection selection = listViewer.getSelection();
List<String> dataList = new ArrayList<String>();
if (selection instanceof StructuredSelection) {
StructuredSelection ss = (StructuredSelection) selection;
Iterator<?> iterator = ss.iterator();
while (iterator.hasNext()) {
Object object = (Object) iterator.next();
if (object instanceof String) {
String obj = (String) object;
dataList = new ArrayList<String>();
TableItem[] items = listViewer.getTable().getItems();
for (TableItem tableItem : items) {
dataList.add((String) tableItem.getData());
}
int indexOf = dataList.indexOf(obj);
Collections.swap(dataList, indexOf, indexOf + 1);
}
}
}
listViewer.remove(dataList.toArray());
addAll(dataList);
}
});
}
}
public String[] getAll() {
ArrayList<String> elements = new ArrayList<String>();
Object[] objectElements = listViewer.getCheckedElements();
for (Object object : objectElements) {
elements.add(object.toString());
}
return elements.toArray(new String[objectElements.length]);
}
public String getAllElementsAsString() {
String result = "";
listViewer.setAllChecked(true);
for (int i = 0; i < listViewer.getCheckedElements().length; i++) {
if (i == listViewer.getCheckedElements().length - 1) {
result += listViewer.getCheckedElements()[i];
} else {
result += listViewer.getCheckedElements()[i] + ", ";
}
}
return result;
}
public void addAll(String input) {
if (input == null)
return;
int index = 0;
while (index != -1) {
index = input.indexOf(",");
String filter;
if (index == -1)
filter = input;
else {
filter = input.substring(0, index);
input = input.substring(index + 2);
}
boolean isAlreadyAdded = false;
int i = 0;
while (listViewer.getElementAt(i) != null) {
if (filter.equals(listViewer.getElementAt(i))) {
isAlreadyAdded = true;
break;
}
i++;
}
if (!isAlreadyAdded && !StringUtils.isBlank(filter))
listViewer.add(filter);
}
listViewer.setAllChecked(true);
}
public void addAll(List<String> input) {
addAll(input, true);
}
public void addAll(List<String> input, boolean checkAll) {
for (String string : input) {
boolean isAlreadyAdded = false;
int i = 0;
while (listViewer.getElementAt(i) != null) {
if (string.equals(listViewer.getElementAt(i))) {
isAlreadyAdded = true;
break;
}
i++;
}
if (!isAlreadyAdded)
listViewer.add(string);
}
if (checkAll)
listViewer.setAllChecked(true);
}
}