/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package qa.qcri.aidr.predict.dbentities;

import java.util.ArrayList;

/**
 *
 * @author Imran
 */
public class NominalAttributeEC {
    
    private int nominalAttributeID;
    private String name;
    private String description;
    private String code;
    private int userID;
    ArrayList<NominalLabelEC> labels = new ArrayList<>();

    /**
     * @return the nominalAttributeID
     */
    public int getNominalAttributeID() {
        return nominalAttributeID;
    }

    /**
     * @param nominalAttributeID the nominalAttributeID to set
     */
    public void setNominalAttributeID(int nominalAttributeID) {
        this.nominalAttributeID = nominalAttributeID;
    }

    /**
     * @return the name
     */
    public String getName() {
        return name;
    }

    /**
     * @param name the name to set
     */
    public void setName(String name) {
        this.name = name;
    }

    /**
     * @return the description
     */
    public String getDescription() {
        return description;
    }

    /**
     * @param description the description to set
     */
    public void setDescription(String description) {
        this.description = description;
    }

    /**
     * @return the code
     */
    public String getCode() {
        return code;
    }

    /**
     * @param code the code to set
     */
    public void setCode(String code) {
        this.code = code;
    }

    /**
     * @return the userID
     */
    public int getUserID() {
        return userID;
    }

    /**
     * @param userID the userID to set
     */
    public void setUserID(int userID) {
        this.userID = userID;
    }
    
    public NominalLabelEC[] getNominalLabels() {
        return labels.toArray(new NominalLabelEC[labels.size()]);
    }
    
    public NominalLabelEC getNominalLabel(String nominalLabelCode) {
        for (NominalLabelEC label : labels) {
            if (label.getNominalLabelCode().equals(nominalLabelCode))
                return label;
        }
        return null;
    }

    public NominalLabelEC getNominalLabel(int nominalLabelID) {
        for (NominalLabelEC label : labels) {
            if (label.getNominalLabelID() == nominalLabelID)
                return label;
        }
        return null;
    }
    
    public void addNominalLabel(NominalLabelEC label) {
        if (!labels.contains(label))
            labels.add(label);
    }
    
    public void resetNominalLabels() {
       labels = new ArrayList<NominalLabelEC>();
    }
}