Commit a5574307 authored by Arne Köhn's avatar Arne Köhn

initial spinoff

parents
/build/
*~
/.idea/
/.gradle/
*.conll
*.svg
This diff is collapsed.
Deptreeviz -- produce good looking dependency trees
===================================================
This library allows you to easily generate dependency tree SVGs. You
can even use a swing widget with interactivity to manipulate dependency trees.
The edges don't need to form a tree and you can visualize several
annotations at once, e.g. syntactic and anaphora edges at the same time.
If you build this library with "gradle shadowJar", the resulting jar
converts a conll file to SVG by default.
java -jar build/libs/deptreeviz-VERSION-all.jar sentence.conll > sentence.svg
Patches welcome!
plugins {
id 'com.github.johnrengelman.shadow' version '1.2.4'
id "com.palantir.git-version" version "0.5.3"
}
apply plugin: 'java'
apply plugin: 'maven-publish'
apply plugin: 'application'
mainClassName = "io.gitlab.nats.deptreeviz.ConllToSVG"
group="io.gitlab.nats"
version gitVersion()
repositories {
mavenCentral()
}
dependencies {
compile group:'org.apache.xmlgraphics', name:'batik-awt-util', version:'1.7'
compile group:'org.apache.xmlgraphics', name:'batik-svggen', version:'1.7'
compile group:'org.apache.xmlgraphics', name:'batik-ext', version:'1.7'
compile group:'org.apache.xmlgraphics', name:'batik-svg-dom', version:'1.7'
compile group:'org.apache.xmlgraphics', name:'batik-swing', version:'1.7'
}
publishing {
publications {
mavenJava(MavenPublication) {
from components.java
}
}
}
#Tue Feb 14 14:36:58 CET 2017
distributionBase=GRADLE_USER_HOME
distributionPath=wrapper/dists
zipStoreBase=GRADLE_USER_HOME
zipStorePath=wrapper/dists
distributionUrl=https\://services.gradle.org/distributions/gradle-3.3-bin.zip
#!/usr/bin/env sh
##############################################################################
##
## Gradle start up script for UN*X
##
##############################################################################
# Attempt to set APP_HOME
# Resolve links: $0 may be a link
PRG="$0"
# Need this for relative symlinks.
while [ -h "$PRG" ] ; do
ls=`ls -ld "$PRG"`
link=`expr "$ls" : '.*-> \(.*\)$'`
if expr "$link" : '/.*' > /dev/null; then
PRG="$link"
else
PRG=`dirname "$PRG"`"/$link"
fi
done
SAVED="`pwd`"
cd "`dirname \"$PRG\"`/" >/dev/null
APP_HOME="`pwd -P`"
cd "$SAVED" >/dev/null
APP_NAME="Gradle"
APP_BASE_NAME=`basename "$0"`
# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
DEFAULT_JVM_OPTS=""
# Use the maximum available, or set MAX_FD != -1 to use that value.
MAX_FD="maximum"
warn ( ) {
echo "$*"
}
die ( ) {
echo
echo "$*"
echo
exit 1
}
# OS specific support (must be 'true' or 'false').
cygwin=false
msys=false
darwin=false
nonstop=false
case "`uname`" in
CYGWIN* )
cygwin=true
;;
Darwin* )
darwin=true
;;
MINGW* )
msys=true
;;
NONSTOP* )
nonstop=true
;;
esac
CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
# Determine the Java command to use to start the JVM.
if [ -n "$JAVA_HOME" ] ; then
if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
# IBM's JDK on AIX uses strange locations for the executables
JAVACMD="$JAVA_HOME/jre/sh/java"
else
JAVACMD="$JAVA_HOME/bin/java"
fi
if [ ! -x "$JAVACMD" ] ; then
die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
else
JAVACMD="java"
which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
# Increase the maximum file descriptors if we can.
if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
MAX_FD_LIMIT=`ulimit -H -n`
if [ $? -eq 0 ] ; then
if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
MAX_FD="$MAX_FD_LIMIT"
fi
ulimit -n $MAX_FD
if [ $? -ne 0 ] ; then
warn "Could not set maximum file descriptor limit: $MAX_FD"
fi
else
warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
fi
fi
# For Darwin, add options to specify how the application appears in the dock
if $darwin; then
GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
fi
# For Cygwin, switch paths to Windows format before running java
if $cygwin ; then
APP_HOME=`cygpath --path --mixed "$APP_HOME"`
CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
JAVACMD=`cygpath --unix "$JAVACMD"`
# We build the pattern for arguments to be converted via cygpath
ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
SEP=""
for dir in $ROOTDIRSRAW ; do
ROOTDIRS="$ROOTDIRS$SEP$dir"
SEP="|"
done
OURCYGPATTERN="(^($ROOTDIRS))"
# Add a user-defined pattern to the cygpath arguments
if [ "$GRADLE_CYGPATTERN" != "" ] ; then
OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
fi
# Now convert the arguments - kludge to limit ourselves to /bin/sh
i=0
for arg in "$@" ; do
CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
else
eval `echo args$i`="\"$arg\""
fi
i=$((i+1))
done
case $i in
(0) set -- ;;
(1) set -- "$args0" ;;
(2) set -- "$args0" "$args1" ;;
(3) set -- "$args0" "$args1" "$args2" ;;
(4) set -- "$args0" "$args1" "$args2" "$args3" ;;
(5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
(6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
(7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
(8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
(9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
esac
fi
# Escape application args
save ( ) {
for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
echo " "
}
APP_ARGS=$(save "$@")
# Collect all arguments for the java command, following the shell quoting and substitution rules
eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong
if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then
cd "$(dirname "$0")"
fi
exec "$JAVACMD" "$@"
@if "%DEBUG%" == "" @echo off
@rem ##########################################################################
@rem
@rem Gradle startup script for Windows
@rem
@rem ##########################################################################
@rem Set local scope for the variables with windows NT shell
if "%OS%"=="Windows_NT" setlocal
set DIRNAME=%~dp0
if "%DIRNAME%" == "" set DIRNAME=.
set APP_BASE_NAME=%~n0
set APP_HOME=%DIRNAME%
@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
set DEFAULT_JVM_OPTS=
@rem Find java.exe
if defined JAVA_HOME goto findJavaFromJavaHome
set JAVA_EXE=java.exe
%JAVA_EXE% -version >NUL 2>&1
if "%ERRORLEVEL%" == "0" goto init
echo.
echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
echo.
echo Please set the JAVA_HOME variable in your environment to match the
echo location of your Java installation.
goto fail
:findJavaFromJavaHome
set JAVA_HOME=%JAVA_HOME:"=%
set JAVA_EXE=%JAVA_HOME%/bin/java.exe
if exist "%JAVA_EXE%" goto init
echo.
echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
echo.
echo Please set the JAVA_HOME variable in your environment to match the
echo location of your Java installation.
goto fail
:init
@rem Get command-line arguments, handling Windows variants
if not "%OS%" == "Windows_NT" goto win9xME_args
:win9xME_args
@rem Slurp the command line arguments.
set CMD_LINE_ARGS=
set _SKIP=2
:win9xME_args_slurp
if "x%~1" == "x" goto execute
set CMD_LINE_ARGS=%*
:execute
@rem Setup the command line
set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
@rem Execute Gradle
"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
:end
@rem End local scope for the variables with windows NT shell
if "%ERRORLEVEL%"=="0" goto mainEnd
:fail
rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
rem the _cmd.exe /c_ return code!
if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
exit /b 1
:mainEnd
if "%OS%"=="Windows_NT" endlocal
:omega
/*
* Copyright 2017 Natural Language Systems Group, Uni Hamburg
*
* Licensed 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 io.gitlab.nats.deptreeviz;
import java.io.IOException;
import java.io.StringWriter;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
/**
* Example class that converts Conll to SVG
* @author Arne Köhn
*/
class ConllToSVG {
public static void main(String[] args) throws IOException {
List<String> conll = Files.readAllLines(Paths.get(args[0]));
SimpleParse p = SimpleParse.fromConll(conll);
DepTree<SimpleParse, SimpleWord> dt = new DepTree<>(p);
// write tree
StringWriter w = new StringWriter();
dt.writeTree(w);
System.out.println(w.toString());
}
}
This diff is collapsed.
/*
* Copyright 2017 Natural Language Systems Group, Uni Hamburg
*
* Licensed 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 io.gitlab.nats.deptreeviz;
import java.awt.geom.Rectangle2D;
import java.util.ArrayList;
import java.util.HashMap;
import org.w3c.dom.Element;
/**
* Contains graphical and textual data of a word for convenient use in a GUI
*
* @author Sven Zimmer
*
*/
public class DepTreeNode {
// private Elem[] _visibleElems = {Elem.LINE, Elem.TEXT, Elem.DOT,
// Elem.ARROWLINE1, Elem.ARROWLINE2};
private String _word;
private int _index;
// for every level, this node links to one other or to a root:
private HashMap<String, Integer> _links = new HashMap<>();
private HashMap<String, String> _labels = new HashMap<>();
private HashMap<ElementID, Element> _elements =
new HashMap<>();
// each level gets a group of elements that changes color when it is
// higlighted, for example
private HashMap<String, ArrayList<Element>> _coloringGroup =
new HashMap<>();
private boolean _top;
private boolean _isInCircle;
private ArrayList<String> _markedLevels;
private double _x;
private double _y;
private ArrayList<Integer> whoLinks;
private HashMap <String, Integer> _nonCurrentYLevel;
private HashMap <String, Rectangle2D> _nonCurrentLabelBorders;
public DepTreeNode(String w) {
_word = w;
}
public void init(ArrayList<String> levels) {
// setCatText("");
for (String level : levels) {
setLink(level, - 1);
setLabel(level, "");
}
setIsTop(false);
setIsInCircle(false);
setMarkedLevels(null);
setX(0);
setY(0);
_nonCurrentYLevel = new HashMap<>();
_nonCurrentLabelBorders = new HashMap<>();
setWhoLinks(new ArrayList<>());
_coloringGroup = new HashMap<>();
}
public boolean linkIsRoot(String level) {
return ((_links.get(level) == - 1) || (_links.get(level) == - 2));
}
public void setLink(String level, int synLink) {
this._links.put(level, synLink);
}
public int getLink(String level) {
return _links.get(level);
}
public void setElement(Element element, DepTree.Elem type, String level) {
ElementID id = new ElementID(type, level);
_elements.put(id, element);
}
public Element getElement(DepTree.Elem type, String level) {
ElementID id = new ElementID(type, level);
return _elements.get(id);
}
private class ElementID {
DepTree.Elem _type;
String _level;
public ElementID(DepTree.Elem type, String level) {
_type = type;
_level = level;
}
public boolean equals(Object o) {
if (o.getClass() == ElementID.class) {
ElementID id = (ElementID) o;
return equals(id);
}
else
return false;
}
public boolean equals(ElementID id) {
return (_type.equals(id._type) && _level.equals(id._level));
}
public int hashCode() {
return _type.hashCode() + _level.hashCode();
}
}
// each level gets a group of elements that changes color when it is
// highlighted, for example
public void addToColoringGroup(Element element, String level) {
ArrayList<Element> elements = _coloringGroup.get(level);
if (elements == null) {
elements = new ArrayList<>();
_coloringGroup.put(level, elements);
}
elements.add(element);
}
// TODO decide if to use this
// public ArrayList<Element> getVisibleElements (){
// ArrayList<Element> result = new ArrayList<Element>();
// for (ElementID id : _elementIDs){
// for (Enum<?> type : _visibleElems){
// if (id._type.equals(type)){
// result.add(_elements.get(id));
// }
// }
// }
// return result;
// }
public void setLabel(String level, String label) {
_labels.put(level, label);
}
public String getLabel(String level) {
return _labels.get(level);
}
public ArrayList<Element> getColoringGroup(String level) {
return _coloringGroup.get(level);
}
public void setWord(String _word) {
this._word = _word;
}
public String getWord() {
return _word;
}
public void setIndex(int _index) {
this._index = _index;
}
public int getIndex() {
return _index;
}
public void setIsTop(boolean top) {
this._top = top;
}
public boolean isTop() {
return _top;
}
public void setIsInCircle(boolean isInCircle) {
this._isInCircle = isInCircle;
}
public boolean isInCircle() {
return _isInCircle;
}
public void setMarkedLevels(ArrayList<String> markedLevels) {
this._markedLevels = markedLevels;
}
public ArrayList<String> getMarkedLevels() {
return _markedLevels;
}
public void setX(double x) {
this._x = x;
}
public double getX() {
return _x;
}
public void setY(double y) {
_y = y;
}
public double getY() {
return _y;
}
public void setWhoLinks(ArrayList<Integer> whoLinks) {
this.whoLinks = whoLinks;
}
public ArrayList<Integer> getWhoLinks() {
return whoLinks;
}
public int getNonCurrentYLevel(String level) {
Integer result = _nonCurrentYLevel.get(level);
if (result == null){
result = 0;
}
return result;
}
public void setNonCurrentYLevel(int nonCurrentYLevel, String level) {
_nonCurrentYLevel.put(level, nonCurrentYLevel);
}
public Rectangle2D getNonCurrentLabelBorder(String level) {
return _nonCurrentLabelBorders.get(level);
}
public void setNonCurrentLabelBorder(Rectangle2D nonCurrentYLabelBorder, String level) {
_nonCurrentLabelBorders.put(level, nonCurrentYLabelBorder);
}
}
/*
* Copyright 2017 Natural Language Systems Group, Uni Hamburg
*
* Licensed 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 io.gitlab.nats.deptreeviz;
import java.util.List;
import java.util.Map;
/**
* A general interface for everything that represents a possibly multi-leveled
* dependency tree / graph
* Created by Arne Köhn on 08.02.17.
*/
public interface ParseInterface<T extends WordInterface> {
String print();
List<T> getWords();
List<String> getLevels();
Map<String, List<Integer>> getVerticesStructure();
Map<String, List<String>> getVerticesLabels();
void redirectEdge(int pos, String level, int to);
}
/*
* Copyright 2017 Natural Language Systems Group, Uni Hamburg
*
* Licensed 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 io.gitlab.nats.deptreeviz;
import io.gitlab.nats.deptreeviz.DepTreeNode;
import org.apache.batik.dom.events.DOMMouseEvent;
import java.util.EventObject;
/**
* Used for signaling that a menu is to pop up after clicking on a graphical
* element of the dependency tree.
*
* @author 3zimmer
*
*/
public class PopupEvent extends EventObject {
private static final long serialVersionUID = 1L;
private DepTreeNode _node;
private String _level;
private DepTreeBaseInteractor.Com _type;
private DOMMouseEvent _devt;
public PopupEvent(
Object source,
DOMMouseEvent devt,
DepTreeBaseInteractor.Com type,
DepTreeNode node,
String level) {
super(source);
_node = node;
_level = level;
_type = type;
_devt = devt;
}
public DepTreeNode getNode() {
r