/**
 @header@
 */
package org.pcmm.gates.impl;

import org.pcmm.base.impl.PCMMBaseObject;
import org.pcmm.gates.IClassifier;

import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.UnknownHostException;

/**
 * TODO - Cleanup this interface
 */
public class Classifier extends PCMMBaseObject implements IClassifier {

    /**
     *
     */
    public Classifier() {
        this(LENGTH, STYPE, SNUM);
    }

    /**
     * @param data - the data to add
     */
    public Classifier(byte[] data) {
        super(data);
    }

    /**
     * @param len - the classifier's length
     * @param sType - the sType value
     * @param sNum - the sNum value
     */
    public Classifier(short len, byte sType, byte sNum) {
        super(len, sType, sNum);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getDestinationIPAddress()
     */
    @Override
    public InetAddress getDestinationIPAddress() {
        try {
            return Inet4Address.getByAddress(getBytes((short) 8, (short) 4));
        } catch (UnknownHostException e) {
            return null;
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see
     * org.pcmm.gates.IClassifier#setDestinationIPAddress(java.net.InetAddress)
     */
    @Override
    public void setDestinationIPAddress(InetAddress address) {
        setBytes(address.getAddress(), (short) 8);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getDestinationPort()
     */
    @Override
    public short getDestinationPort() {
        return getShort((short) 14);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#setDestinationPort(short)
     */
    @Override
    public void setDestinationPort(short p) {
        setShort(p, (short) 14);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getSourceIPAddress()
     */
    @Override
    public InetAddress getSourceIPAddress() {
        try {
            return Inet4Address.getByAddress(getBytes((short) 4, (short) 4));
        } catch (UnknownHostException e) {
            return null;
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#setSourceIPAddress(java.net.InetAddress)
     */
    @Override
    public void setSourceIPAddress(InetAddress a) {
        setBytes(a.getAddress(), (short) 4);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getSourcePort()
     */
    @Override
    public short getSourcePort() {
        return getShort((short) 12);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#setSourcePort(short)
     */
    @Override
    public void setSourcePort(short p) {
        setShort(p, (short) 12);

    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getProtocol()
     */
    @Override
    public short getProtocol() {
        return getShort((short) 0);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#setProtocol(short)
     */
    @Override
    public void setProtocol(short p) {
        setShort(p, (short) 0);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#getPriority()
     */
    @Override
    public byte getPriority() {
        return getBytes((short) 16, (short) 1)[0];
    }

    /*
     * (non-Javadoc)
     *
     * @see org.pcmm.gates.IClassifier#setPriority(byte)
     */
    @Override
    public void setPriority(byte p) {
        setBytes(new byte[] { p }, (short) 16);
    }

    @Override
    public byte getDSCPTOS() {
        return getBytes((short) 2, (short) 1)[0];
    }

    @Override
    public void setDSCPTOS(byte v) {
        setBytes(new byte[] { v }, (short) 2);

    }

    @Override
    public byte getDSCPTOSMask() {
        return getBytes((short) 3, (short) 1)[0];
    }

    @Override
    public void setDSCPTOSMask(byte v) {
        setBytes(new byte[] { v }, (short) 3);

    }

}