/*******************************************************************************
 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
 * in Industry THTH ry.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     VTT Technical Research Centre of Finland - initial API and implementation
 *******************************************************************************/
package org.simantics.district.network.profile;

import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.exception.DatabaseException;
import org.simantics.diagram.profile.DiagramElementGroup;
import org.simantics.district.network.ontology.DistrictNetworkResource;

public class EdgeNodeGroup extends DiagramElementGroup {

    public EdgeNodeGroup(ReadGraph graph) {
        super(graph);
        System.out.println("EdgeNodeGroup init");
    }

    @Override
    protected boolean test(ReadGraph graph, Resource element) throws DatabaseException {
        DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
        if (graph.isInstanceOf(element, DN.Edge)) {
            return true;
        }
        return false;
    }

    @Override
    public String toString() {
        return "Edge Nodes";
    }

}