InputColumnFeature.java

package org.maltparser.core.syntaxgraph.feature;

import org.maltparser.core.exception.MaltChainedException;
import org.maltparser.core.feature.function.AddressFunction;
import org.maltparser.core.feature.value.AddressValue;
import org.maltparser.core.io.dataformat.ColumnDescription;
import org.maltparser.core.io.dataformat.DataFormatInstance;
import org.maltparser.core.symbol.nullvalue.NullValues.NullValueId;
import org.maltparser.core.syntaxgraph.SyntaxGraphException;
import org.maltparser.core.syntaxgraph.node.DependencyNode;

/**
 *
 *
 * @author Johan Hall
 */
public final class InputColumnFeature extends ColumnFeature {

    private final DataFormatInstance dataFormatInstance;
    private AddressFunction addressFunction;

    public InputColumnFeature(DataFormatInstance dataFormatInstance) throws MaltChainedException {
        super();
        this.dataFormatInstance = dataFormatInstance;
    }

    public void initialize(Object[] arguments) throws MaltChainedException {
        if (arguments.length != 2) {
            throw new SyntaxGraphException("Could not initialize InputColumnFeature: number of arguments are not correct. ");
        }
        if (!(arguments[0] instanceof String)) {
            throw new SyntaxGraphException("Could not initialize InputColumnFeature: the first argument is not a string. ");
        }
        if (!(arguments[1] instanceof AddressFunction)) {
            throw new SyntaxGraphException("Could not initialize InputColumnFeature: the second argument is not an address function. ");
        }
        ColumnDescription column = dataFormatInstance.getColumnDescriptionByName((String) arguments[0]);
        if (column == null) {
            throw new SyntaxGraphException("Could not initialize InputColumnFeature: the input column type '" + (String) arguments[0] + "' could not be found in the data format specification. ' ");
        }
        setColumn(column);
        setAddressFunction((AddressFunction) arguments[1]);
    }

    public Class<?>[] getParameterTypes() {
        Class<?>[] paramTypes = {java.lang.String.class, org.maltparser.core.feature.function.AddressFunction.class};
        return paramTypes;
    }

    public void update() throws MaltChainedException {
        final AddressValue a = addressFunction.getAddressValue();

        if (a.getAddress() == null) {
            featureValue.update(column.getSymbolTable().getNullValueCode(NullValueId.NO_NODE),
                    column.getSymbolTable().getNullValueSymbol(NullValueId.NO_NODE), true, 1);
        } else {
            final DependencyNode node = (DependencyNode) a.getAddress();

            if (!node.isRoot()) {
                int indexCode = node.getLabelCode(column.getSymbolTable());
                String symbol = column.getSymbolTable().getSymbolCodeToString(indexCode);
                if (column.getType() == ColumnDescription.STRING) {
                    featureValue.update(indexCode, symbol, false, 1);
                } else {
                    castFeatureValue(symbol);
                }
            } else {
                featureValue.update(column.getSymbolTable().getNullValueCode(NullValueId.ROOT_NODE),
                        column.getSymbolTable().getNullValueSymbol(NullValueId.ROOT_NODE), true, 1);
            }
        }

    }

    public AddressFunction getAddressFunction() {
        return addressFunction;
    }

    public void setAddressFunction(AddressFunction addressFunction) {
        this.addressFunction = addressFunction;
    }

    public DataFormatInstance getDataFormatInstance() {
        return dataFormatInstance;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (getClass() != obj.getClass()) {
            return false;
        }
        return obj.toString().equals(toString());
    }

    @Override
    public int hashCode() {
        return 217 + (null == toString() ? 0 : toString().hashCode());
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder();
        sb.append("InputColumn(");
        sb.append(super.toString());
        sb.append(", ");
        sb.append(addressFunction.toString());
        sb.append(")");
        return sb.toString();
    }
}