/*
 * ARX: Powerful Data Anonymization
 * Copyright 2012 - 2017 Fabian Prasser, Florian Kohlmayer and contributors
 * 
 * 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.
 */

package org.deidentifier.arx.gui.model;

import org.deidentifier.arx.criteria.EqualDistanceTCloseness;
import org.deidentifier.arx.criteria.HierarchicalDistanceTCloseness;
import org.deidentifier.arx.criteria.OrderedDistanceTCloseness;
import org.deidentifier.arx.criteria.PrivacyCriterion;
import org.deidentifier.arx.gui.resources.Resources;
import org.deidentifier.arx.gui.view.SWTUtil;

/**
 * This class implements a model for the t-closeness criterion.
 *
 * @author Fabian Prasser
 */
public class ModelTClosenessCriterion extends ModelExplicitCriterion{

    /** SVUID. */
    private static final long serialVersionUID     = 4901053938589514626L;

    /** Variant. */
    public static final int   VARIANT_EQUAL        = 0;

    /** Variant. */
    public static final int   VARIANT_HIERARCHICAL = 1;

    /** Variant. */
    public static final int   VARIANT_ORDERED      = 2;

    /** The variant. */
    private int               variant              = 0;

    /** T. */
    private double            t                    = 0.001d;

    /**
     * Creates a new instance.
     *
     * @param attribute
     */
    public ModelTClosenessCriterion(String attribute) {
        super(attribute);
    }
    
    /**
     * Creates a new instance.
     *
     * @param attribute
     */
    public ModelTClosenessCriterion(String attribute,
                                    int variant,
                                    double t) {
        super(attribute);
        this.variant = variant;
        this.t = t;
    }
    
    @Override
    public ModelTClosenessCriterion clone() {
        ModelTClosenessCriterion result = new ModelTClosenessCriterion(this.getAttribute());
        result.t = this.t;
        result.variant = this.variant;
        result.setEnabled(this.isEnabled());
        return result;
    }
	
	@Override
	public PrivacyCriterion getCriterion(Model model) {
	    switch (variant) {
            case VARIANT_EQUAL: return new EqualDistanceTCloseness(getAttribute(), t);
            case VARIANT_ORDERED: return new OrderedDistanceTCloseness(getAttribute(), t);
            case VARIANT_HIERARCHICAL: return new HierarchicalDistanceTCloseness(getAttribute(), t,
                                                                                 model.getInputConfig().getHierarchy(getAttribute()));
            default: throw new RuntimeException(Resources.getMessage("Model.0c")); //$NON-NLS-1$
	    }
	}
	
	@Override
    public String getLabel() {
        return Resources.getMessage("Model.1b"); //$NON-NLS-1$
    }
	
	/**
     * Returns T.
     *
     * @return
     */
	public double getT() {
		return t;
	}

	/**
     * Returns the variant.
     *
     * @return
     */
	public int getVariant() {
		return variant;
	}

    @Override
    public void parse(ModelCriterion criterion, boolean _default) {
        if (!(criterion instanceof ModelTClosenessCriterion)) {
            return;
        }
        ModelTClosenessCriterion other = (ModelTClosenessCriterion)criterion;
        this.t = other.t;
        this.variant = other.variant;
        if (!_default) {
            this.setEnabled(other.isEnabled());
        }
    }
    
	@Override
    public void pull(ModelExplicitCriterion criterion) {
        if (!(criterion instanceof ModelTClosenessCriterion)) {
            throw new RuntimeException(Resources.getMessage("Model.2d")); //$NON-NLS-1$
        }
        ModelTClosenessCriterion other = (ModelTClosenessCriterion)criterion;
        this.variant = other.variant;
        this.t = other.t;
    }

    /**
     * Sets T.
     *
     * @param t
     */
	public void setT(double t) {
		this.t = t;
	}

    /**
     * Sets the variant.
     *
     * @param variant
     */
	public void setVariant(int variant) {
		this.variant = variant;
	}
    
    @Override
    public String toString() {
        switch (variant) {
            case VARIANT_EQUAL: return SWTUtil.getPrettyString(t)+Resources.getMessage("Model.3b"); //$NON-NLS-1$
            case VARIANT_ORDERED: return SWTUtil.getPrettyString(t)+Resources.getMessage("Model.34"); //$NON-NLS-1$
            case VARIANT_HIERARCHICAL: return SWTUtil.getPrettyString(t)+Resources.getMessage("Model.4b"); //$NON-NLS-1$
            default: throw new RuntimeException(Resources.getMessage("Model.5b")); //$NON-NLS-1$
        }
    }
}