package org.apache.samoa.instances;

/*
 * #%L
 * SAMOA
 * %%
 * Copyright (C) 2014 - 2015 Apache Software Foundation
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

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

public class InstanceInformation implements Serializable {

  /**
   * The dataset's name.
   */
  protected String relationName;

  protected AttributesInformation attributesInformation;

  /**
   * The class index.
   */
  protected int classIndex = Integer.MAX_VALUE; //By default is multilabel

  /**
   * Range for multi-label instances.
   */
  protected Range range;

  public Attribute inputAttribute(int w) {
    return this.attributesInformation.attribute(inputAttributeIndex(w));
  }

  public Attribute outputAttribute(int w) {
    return this.attributesInformation.attribute(outputAttributeIndex(w));
  }

  /**
   * Instantiates a new instance information.
   *
   * @param chunk
   *          the chunk
   */
  public InstanceInformation(InstanceInformation chunk) {
    this.relationName = chunk.relationName;
    this.attributesInformation = chunk.attributesInformation;
    this.classIndex = chunk.classIndex;
  }

  /**
   * Instantiates a new instance information.
   *
   * @param st
   *          the st
   * @param input
   *          the input
   */
  public InstanceInformation(String st, List<Attribute> input) {
    this.relationName = st;
    this.attributesInformation = new AttributesInformation(input, input.size());
  }

  /**
   * Instantiates a new instance information.
   */
  public InstanceInformation() {
    this.relationName = null;
    this.attributesInformation = null;
  }

  //Information Instances
  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#setRelationName(java.lang.String)
   */
  public void setRelationName(String string) {
    this.relationName = string;
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#getRelationName()
   */
  public String getRelationName() {
    return this.relationName;
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#classIndex()
   */
  public int classIndex() {
    return this.classIndex;
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#setClassIndex(int)
   */
  public void setClassIndex(int classIndex) {
    this.classIndex = classIndex;
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#classAttribute()
   */
  public Attribute classAttribute() {
    return this.attribute(this.classIndex());
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#numAttributes()
   */
  public int numAttributes() {
    return this.attributesInformation.numberAttributes;
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#attribute(int)
   */
  public Attribute attribute(int w) {
    return this.attributesInformation.attribute(w);
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#numClasses()
   */
  public int numClasses() {
    return this.attributesInformation.attribute(classIndex()).numValues();
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#deleteAttributeAt(java.lang.Integer)
   */
  public void deleteAttributeAt(Integer integer) {
    this.attributesInformation.deleteAttributeAt(integer);
  }

  /* (non-Javadoc)
   * @see com.yahoo.labs.samoa.instances.InstanceInformationInterface#insertAttributeAt(com.yahoo.labs.samoa.instances.Attribute, int)
   */
  public void insertAttributeAt(Attribute attribute, int i) {
    this.attributesInformation.insertAttributeAt(attribute, i);
  }

  public void setAttributes(List<Attribute> v) {
    if (this.attributesInformation == null)
      this.attributesInformation = new AttributesInformation();
    this.attributesInformation.setAttributes(v);
  }

  public int inputAttributeIndex(int index) {
    int ret = 0;
    if (classIndex == Integer.MAX_VALUE) {//Multi Label
      if (index < range.getStart())//JD
        ret = index;
      else
        ret = index + range.getSelectionLength();

    } else { //Single Label
      ret = classIndex() > index ? index : index + 1;
    }
    return ret;
  }

  public int outputAttributeIndex(int attributeIndex) {
    int ret = 0;
    if (classIndex == Integer.MAX_VALUE) {//Multi Label
      ret = attributeIndex + range.getStart(); //JD - Range should be a "block"
    } else { //Single Label
      ret = classIndex;
    }
    return ret;
  }

  public int numInputAttributes() {
    int ret = 0;
    if (classIndex == Integer.MAX_VALUE) {//Multi Label
      ret = this.numAttributes() - range.getSelectionLength(); //JD
    } else { //Single Label
      ret = this.numAttributes() - 1;
    }
    return ret;
  }

  public int numOutputAttributes() {
    int ret = 0;
    if (classIndex == Integer.MAX_VALUE) {//Multi Label
      ret = range.getSelectionLength(); //JD
    } else { //Single Label
      ret = 1;
    }
    return ret;
  }

  public void setRangeOutputIndices(Range range) {
    this.setClassIndex(Integer.MAX_VALUE);
    this.range = range;
  }

  public void setAttributes(List<Attribute> v, List<Integer> indexValues) {
    if (this.attributesInformation == null)
      this.attributesInformation = new AttributesInformation();
    this.attributesInformation.setAttributes(v, indexValues);

  }

}