org.coode.matrix.model.impl.PropertyAssertionsMatrixModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.coode.matrix Show documentation
A Swing Component that combines a JTree and a JTable.
The newest version!
package org.coode.matrix.model.impl;

import java.util.Collections;
import java.util.List;
import java.util.Set;
/*
* Copyright (C) 2007, University of Manchester
*
* Modifications to the initial code base are copyright of their
* respective authors, or their employers as appropriate.  Authorship
* of the modifications may be determined from the ChangeLog placed at
* the end of this file.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.

* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.

* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

import org.coode.matrix.model.api.AbstractMatrixModel;
import org.coode.matrix.model.helper.IndividualsHelper;
import org.protege.editor.owl.model.OWLModelManager;
import org.protege.editor.owl.ui.tree.OWLObjectTree;
import org.semanticweb.owlapi.model.OWLAnnotationProperty;
import org.semanticweb.owlapi.model.OWLDataProperty;
import org.semanticweb.owlapi.model.OWLIndividual;
import org.semanticweb.owlapi.model.OWLLiteral;
import org.semanticweb.owlapi.model.OWLNamedIndividual;
import org.semanticweb.owlapi.model.OWLObject;
import org.semanticweb.owlapi.model.OWLObjectProperty;
import org.semanticweb.owlapi.model.OWLOntologyChange;

/**
 * Author: Nick Drummond

 * http://www.cs.man.ac.uk/~drummond/


 * 
 * The University Of Manchester

 * Bio Health Informatics Group

 * Date: Jul 4, 2007


 */
public class PropertyAssertionsMatrixModel extends AbstractMatrixModel {

    private IndividualsHelper helper;

    public PropertyAssertionsMatrixModel(OWLObjectTree tree, OWLModelManager mngr) {
        super(tree, mngr);
        helper = new IndividualsHelper(mngr.getOWLOntologyManager(), mngr.getActiveOntologies());
    }


    public String getTreeColumnLabel() {
        return "Individual";
    }


    @Override
    public Object getMatrixValue(OWLObject entity, Object prop) {
        if (entity instanceof OWLIndividual){
            if (prop instanceof OWLObjectProperty) {
                return helper.getRelationships((OWLIndividual) entity, (OWLObjectProperty)prop);
            }
            else if (prop instanceof OWLDataProperty) {
                return helper.getRelationships((OWLIndividual) entity, (OWLDataProperty)prop);
            }
            else {
                return super.getMatrixValue(entity, prop);
            }
        }
        return null;
    }


    @Override
    public List setMatrixValue(OWLObject ind, Object prop, Object value) {
        if (prop instanceof OWLObjectProperty){
            return helper.setRelationships((OWLNamedIndividual) ind,
                                           (OWLObjectProperty) prop,
                                           (Set) value,
                                           mngr.getActiveOntology());
        }
        else if (prop instanceof OWLDataProperty){
            return helper.setRelationships((OWLNamedIndividual) ind,
                                           (OWLDataProperty) prop,
                                           (Set) value,
                                           mngr.getActiveOntology());
        }
        else{
            return super.setMatrixValue(ind, prop, value);
        }
    }


    @Override
    public List addMatrixValue(OWLObject rowObj, Object columnObj, Object value) {
        if (columnObj instanceof OWLObjectProperty){
            Set values = null;
            if (value instanceof OWLNamedIndividual){
                values = Collections.singleton((OWLNamedIndividual)value);
            }
            else if (value instanceof Set){
                // @@TODO check the contents of the set
                values = (Set)value;
            }
            if (values != null){
                return helper.addRelationships((OWLNamedIndividual) rowObj,
                                               (OWLObjectProperty) columnObj,
                                               values,
                                               mngr.getActiveOntology());
            }
        }
        else if (columnObj instanceof OWLDataProperty){
            Set values = null;
            if (value instanceof OWLLiteral){
                values = Collections.singleton((OWLLiteral)value);
            }
            else if (value instanceof Set){
                // @@TODO check the contents of the set
                values = (Set)value;
            }
            if (values != null){
                return helper.addRelationships((OWLNamedIndividual) rowObj,
                                               (OWLDataProperty) columnObj,
                                               values,
                                               mngr.getActiveOntology());
            }
        }
        return super.addMatrixValue(rowObj, columnObj, value);
    }


    public boolean isSuitableCellValue(Object value, int row, int col) {
        if (isCellEditable(row, col)){
            Object colObj = getColumnObjectAtModelIndex(col);
            if (colObj instanceof OWLObjectProperty){
                return value instanceof OWLIndividual;
            }
            else if (colObj instanceof OWLDataProperty){
                return value instanceof OWLLiteral;
            }
        }
        return false;
    }


    public Object getSuitableColumnObject(Object columnObject) {
        if (columnObject instanceof OWLObjectProperty ||
            columnObject instanceof OWLDataProperty ||
            columnObject instanceof OWLAnnotationProperty){
            return columnObject;
        }
        return null;
    }


    public boolean isValueRestricted(OWLObject rowObject, Object columnObject) {
        return false;
    }


    public Set getSuggestedFillers(OWLObject rowObject, Object columnObject, int threshold) {
        return Collections.EMPTY_SET;
    }


    @Override
    public boolean isCellEditable(OWLObject rowObject, int column) {
        return rowObject instanceof OWLIndividual;
    }
}