com.kdcloud.lib.domain.DataSpecification.java Source code

Introduction

Here is the source code for com.kdcloud.lib.domain.DataSpecification.java

Source

/**
 * Copyright (C) 2012 Vincenzo Pirrone
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option) any later
 * version.
 *
 * This program 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 General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */
package com.kdcloud.lib.domain;

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

import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlType;

import weka.core.Attribute;
import weka.core.Instances;

@XmlType(name = "")
@XmlAccessorType(XmlAccessType.NONE)
@XmlRootElement
public class DataSpecification implements Serializable {

    /**
     * 
     */
    private static final long serialVersionUID = 1L;

    @XmlAccessorType(XmlAccessType.FIELD)
    static class Column implements Serializable {
        private static final long serialVersionUID = 1L;

        String name;
        DataType type;
        InputSource source;
    }

    public enum DataType {
        DOUBLE, TIMESTAMP, INTEGER
    }

    public enum InputSource {
        HEARTBEAT, CLOCK
    }

    @XmlElement(name = "column")
    List<Column> columns;

    @XmlElement
    String view;

    public Instances newInstances(String relationalName) {
        return new Instances(relationalName, getAttrInfo(), 1000);
    }

    public boolean matchingSpecification(Instances instances) {
        return new Instances("", getAttrInfo(), 0).equalHeaders(instances);
    }

    public ArrayList<Attribute> getAttrInfo() {
        ArrayList<Attribute> info = new ArrayList<Attribute>(columns.size());
        for (Column c : columns) {
            info.add(new Attribute(c.name));
        }
        return info;
    }

    public String getView() {
        return view;
    }

    public static Instances newInstances(String name, int columns) {
        ArrayList<Attribute> info = new ArrayList<Attribute>(columns);
        for (int i = 0; i < columns; i++) {
            info.add(new Attribute("attr" + i));
        }
        return new Instances(name, info, 0);
    }

    public List<InputSource> sources() {
        LinkedList<InputSource> list = new LinkedList<DataSpecification.InputSource>();
        for (Column c : columns)
            list.add(c.source);
        return list;
    }

}