DataTablesModel.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.desktop.datalab;

import org.gephi.data.attributes.api.AttributeColumn;
import org.gephi.data.attributes.api.AttributeTable;

/**
 *
 * @author Mathieu Bastian
 */
public class DataTablesModel {

    private AvailableColumnsModel nodeAvailableColumnsModel;
    private AvailableColumnsModel edgeAvailableColumnsModel;

    public DataTablesModel(AttributeTable nodeTable, AttributeTable edgeTable) {
        nodeAvailableColumnsModel = new AvailableColumnsModel();
        edgeAvailableColumnsModel = new AvailableColumnsModel();

        //Try to make available all columns at start by default:
        for(AttributeColumn column:nodeTable.getColumns()){
            nodeAvailableColumnsModel.addAvailableColumn(column);
        }

        for(AttributeColumn column:edgeTable.getColumns()){
            edgeAvailableColumnsModel.addAvailableColumn(column);
        }
    }

    public AvailableColumnsModel getEdgeAvailableColumnsModel() {
        return edgeAvailableColumnsModel;
    }

    public AvailableColumnsModel getNodeAvailableColumnsModel() {
        return nodeAvailableColumnsModel;
    }
}