PropertiesAssociations.java

/*
Copyright 2008-2010 Gephi
Authors : Mathieu Bastian <mathieu.bastian@gephi.org>
Website : http://www.gephi.org

This file is part of Gephi.

Gephi is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.

Gephi 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 Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with Gephi.  If not, see <http://www.gnu.org/licenses/>.
*/
package org.gephi.io.importer.api;

import java.io.Serializable;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

/**
 *
 * @author Mathieu Bastian
 */
public final class PropertiesAssociations implements Serializable {

    public enum NodeProperties {

        X, Y, Z, R, G, B, SIZE, ID, LABEL, FIXED;
    }

    public enum EdgeProperties {

        R, G, B, WEIGHT, ID, LABEL, ALPHA, SOURCE, TARGET;
    }
    //PropertiesAssociations association
    private List<PropertyAssociation<NodeProperties>> nodePropertyAssociations = new LinkedList<PropertyAssociation<NodeProperties>>();
    private List<PropertyAssociation<EdgeProperties>> edgePropertyAssociations = new LinkedList<PropertyAssociation<EdgeProperties>>();

    public void addEdgePropertyAssociation(EdgeProperties property, String title) {
        PropertyAssociation<EdgeProperties> association = new PropertyAssociation<EdgeProperties>(property, title);
        /*if (edgePropertyAssociations.contains(association)) {
            return;
        }
        //Avoid any double
        for (Iterator<PropertyAssociation<EdgeProperties>> itr = edgePropertyAssociations.iterator(); itr.hasNext();) {
            PropertyAssociation<EdgeProperties> p = itr.next();
            if (p.getTitle().equalsIgnoreCase(association.getTitle())) {
                itr.remove();
            } else if (p.getProperty().equals(association.getProperty())) {
                itr.remove();
            }
        }*/
        edgePropertyAssociations.add(association);
    }

    public void addNodePropertyAssociation(NodeProperties property, String title) {
        PropertyAssociation<NodeProperties> association = new PropertyAssociation<NodeProperties>(property, title);
        /*if (nodePropertyAssociations.contains(association)) {
            return;
        }
        //Avoid any double
        for (Iterator<PropertyAssociation<NodeProperties>> itr = nodePropertyAssociations.iterator(); itr.hasNext();) {
            PropertyAssociation<NodeProperties> p = itr.next();
            if (p.getTitle().equalsIgnoreCase(association.getTitle())) {
                itr.remove();
            } else if (p.getProperty().equals(association.getProperty())) {
                itr.remove();
            }
        }*/
        nodePropertyAssociations.add(association);
    }

    PropertyAssociation<EdgeProperties>[] getEdgePropertiesAssociation() {
        return edgePropertyAssociations.toArray(new PropertyAssociation[0]);
    }

    PropertyAssociation<NodeProperties>[] getNodePropertiesAssociation() {
        return nodePropertyAssociations.toArray(new PropertyAssociation[0]);
    }

    public NodeProperties getNodeProperty(String title) {
        for (PropertyAssociation<NodeProperties> p : nodePropertyAssociations) {
            if (p.getTitle().equalsIgnoreCase(title)) {
                return p.getProperty();
            }
        }
        return null;
    }

    public EdgeProperties getEdgeProperty(String title) {
        for (PropertyAssociation<EdgeProperties> p : edgePropertyAssociations) {
            if (p.getTitle().equalsIgnoreCase(title)) {
                return p.getProperty();
            }
        }
        return null;
    }

    public String getNodePropertyTitle(NodeProperties property) {
        for (PropertyAssociation<NodeProperties> p : nodePropertyAssociations) {
            if (p.getProperty().equals(property)) {
                return p.getTitle();
            }
        }
        return null;
    }

    public String getEdgePropertyTitle(EdgeProperties property) {
        for (PropertyAssociation<EdgeProperties> p : edgePropertyAssociations) {
            if (p.getProperty().equals(property)) {
                return p.getTitle();
            }
        }
        return null;
    }

    public String getInfos() {
        String res = "***Node Properties Associations***\n";
        for (PropertyAssociation<NodeProperties> p : nodePropertyAssociations) {
            res += "Property " + p.getProperty().toString() + " = " + p.getTitle() + " Column\n";
        }
        res += "*********************************\n";
        res = "***Edge Properties Associations***\n";
        for (PropertyAssociation<EdgeProperties> p : edgePropertyAssociations) {
            res += "Property " + p.getProperty().toString() + " = " + p.getTitle() + " Column\n";
        }
        res += "*********************************\n";
        return res;
    }
}