Open Source Projects Europe

/**
 * Copyright (c) 2013/2014 Verein zur Foerderung der IT-Sicherheit in Oesterreich (SBA).
 * The work has been developed in the TIMBUS Project and the above-mentioned are Members of the TIMBUS Consortium.
 * TIMBUS is supported by the European Union under the 7th Framework Programme for research and technological
 * development and demonstration activities (FP7/2007-2013) under grant agreement no. 269940.
 *
 * 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, including without
 * limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTIBITLY, or FITNESS FOR A PARTICULAR
 * PURPOSE. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise,
 * unless required by applicable law or agreed to in writing, shall any Contributor be liable for damages, including
 * any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this
 * License or out of the use or inability to use the Work.
 * See the License for the specific language governing permissions and limitation under the License.
 */

package net.timbusproject.context.ui;

import java.net.URL;
import java.util.HashMap;

import javax.swing.Icon;
import javax.swing.ImageIcon;

import net.timbusproject.context.OWLObjectRenderer;

import org.semanticweb.owlapi.model.OWLClassExpression;
import org.semanticweb.owlapi.model.OWLNamedIndividual;
import org.semanticweb.owlapi.model.OWLObjectProperty;

/**
 * @author Rudolf Mayer (rmayer@sba-research.at) 
 * @version $Id: StandaloneOWLObjectRenderer.java 694 2013-12-03 14:55:00Z RudolfMayer $
 */
public class StandaloneOWLObjectRenderer implements OWLObjectRenderer {
    private Icon classIcon;

    private Icon individualIcon;

    private static HashMap<String, URL> loadedIconURLs = new HashMap<String, URL>();

    private static HashMap<String, Icon> loadedIcons = new HashMap<String, Icon>();

    public StandaloneOWLObjectRenderer() {
        ClassLoader loader = getClass().getClassLoader();
        classIcon = getIcon(loader, "class.gif");
        individualIcon = getIcon(loader, "instance.gif");
    }

    private static String beautify(String string) {
        String res = string.substring(string.indexOf("#") + 1);
        return res.replace(">", "");
    }

    public static ImageIcon getIcon(ClassLoader loader, String string) {
        URL resource = loader.getResource(string);
        if (resource != null) {
            return new ImageIcon(resource);
        } else {
            return null;
        }
    }

    public static Icon getIconIndividual() {
        return getIcon("/rsc/icons/instance.gif");
    }

    public static Icon getIconClass() {
        return getIcon("/rsc/icons/class.gif");
    }

    public static Icon getSmallIconClass() {
        return getIcon("/rsc/icons/class-small.gif");
    }

    private static Icon getIcon(final String fileName) {
        if (!loadedIcons.containsKey(fileName)) {
            ClassLoader loader = StandaloneOWLObjectRenderer.class.getClassLoader();
            URL url = loader.getResource(fileName);
            if (url == null) {
                url = StandaloneOWLObjectRenderer.class.getResource(fileName);
            }
            Icon icon = new ImageIcon(url);
            loadedIcons.put(fileName, icon);
        }
        return loadedIcons.get(fileName);
    }

    public static URL getIconIndividualURL() {
        return getIconURL("/rsc/icons/instance.gif");
    }

    public static URL getIconClassURL() {
        return getIconURL("/rsc/icons/class.gif");
    }

    public static URL getSmallIconClassURL() {
        return getIconURL("/rsc/icons/class-small.gif");
    }

    private static URL getIconURL(final String fileName) {
        if (!loadedIconURLs.containsKey(fileName)) {
            ClassLoader loader = StandaloneOWLObjectRenderer.class.getClassLoader();
            URL url = loader.getResource(fileName);
            if (url == null) {
                url = StandaloneOWLObjectRenderer.class.getResource(fileName);
            }
            loadedIconURLs.put(fileName, url);
        }
        return loadedIconURLs.get(fileName);
    }

    @Override
    public Icon getIcon(OWLClassExpression owlClass) {
        return classIcon;
    }

    @Override
    public Icon getIcon(OWLNamedIndividual owlObject) {
        return individualIcon;
    }

    @Override
    public String getRendering(OWLClassExpression owlClass) {
        return beautify(owlClass.toString());
    }

    @Override
    public String getRendering(OWLNamedIndividual owlObject) {
        return beautify(owlObject.toString());
    }

    @Override
    public String getRendering(OWLObjectProperty property) {
        return beautify(property.toString());
    }

}