/*
 * ARX Data Anonymization Tool
 * Copyright 2012 - 2024 Fabian Prasser 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 java.io.Serializable;

import org.deidentifier.arx.ARXClassificationConfiguration;
import org.deidentifier.arx.ARXFeatureScaling;
import org.deidentifier.arx.aggregates.ClassificationConfigurationLogisticRegression;
import org.deidentifier.arx.aggregates.ClassificationConfigurationNaiveBayes;
import org.deidentifier.arx.aggregates.ClassificationConfigurationRandomForest;

/**
 * This class represents a model
 *
 * @author Fabian Prasser
 */
public class ModelClassification implements Serializable {

    /** SVUID */
    private static final long                             serialVersionUID   = 5361564507029617616L;

    /** Modified */
    private boolean                                       modified           = false;
    /** Current configuration */
    private ARXClassificationConfiguration<?>             configCurrent      = null;
    /** Configuration logistic regression */
    private ClassificationConfigurationLogisticRegression config             = null;
    /** Configuration naive bayes */
    private ClassificationConfigurationNaiveBayes         configNaiveBayes   = null;
    /** Configuration random forest */
    private ClassificationConfigurationRandomForest       configRandomForest = null;
    /** Feature scaling */
    private ARXFeatureScaling                             featureScaling;

    /**
     * Returns the current classification configuration
     * @return
     */
    public ARXClassificationConfiguration<?> getCurrentConfiguration(){
        if (this.configCurrent == null) {
            this.configCurrent = getLogisticRegressionConfiguration();
        }
        return configCurrent;
    }

    /**
     * Returns the feature scaling
     * @return
     */
    public ARXFeatureScaling getFeatureScaling() {
        if (this.featureScaling == null) {
            this.featureScaling = ARXFeatureScaling.create();
        }
        return this.featureScaling;
    }
    
    /**
     * Returns a logistic regression configuration for ARX
     * @return
     */
    public ClassificationConfigurationLogisticRegression getLogisticRegressionConfiguration() {
        if (this.config == null) {
            this.config = ClassificationConfigurationLogisticRegression.create();
        }
        return this.config;
    }
    
    /**
     * Returns a naive bayes configuration for ARX
     * @return
     */
    public ClassificationConfigurationNaiveBayes getNaiveBayesConfiguration() {
        if (this.configNaiveBayes == null) {
            this.configNaiveBayes = ClassificationConfigurationNaiveBayes.create();
        }
        return this.configNaiveBayes;
    }

    /**
     * Returns a random forest configuration for ARX
     * @return
     */
    public ClassificationConfigurationRandomForest getRandomForestConfiguration() {
        if (this.configRandomForest == null) {
            this.configRandomForest = ClassificationConfigurationRandomForest.create();
        }
        return this.configRandomForest;
    }
    
    /**
     * Is this model or one of the configurations modified
     * 
     * @return
     */
    public boolean isModified() {
        return this.modified || getLogisticRegressionConfiguration().isModified() || getNaiveBayesConfiguration().isModified() ||
                                getRandomForestConfiguration().isModified();
    }
    
    /**
     * Sets the current classification configuration.
     * 
     * @param configCurrent
     */
    public void setCurrentConfiguration(ARXClassificationConfiguration<?> configCurrent){
        this.configCurrent = configCurrent;
    }
    
    /**
     * TODO: Ugly hack to set base-parameters for all methods
     * @param t
     */
    public void setDeterministic(Boolean t) {
        this.config.setDeterministic(t);
        this.configNaiveBayes.setDeterministic(t);
        this.configRandomForest.setDeterministic(t);
        this.setModified();
    }
    
    /**
     * TODO: Ugly hack to set base-parameters for all methods
     * @param t
     */
    public void setMaxRecords(Integer t) {
        this.config.setMaxRecords(t);
        this.configNaiveBayes.setMaxRecords(t);
        this.configRandomForest.setMaxRecords(t);
        this.setModified();
    }
    
    /**
     * TODO: Ugly hack to set base-parameters for all methods
     * @param t
     */
    public void setNumFolds(Integer t) {
        this.config.setNumFolds(t);
        this.configNaiveBayes.setNumFolds(t);
        this.configRandomForest.setNumFolds(t);
        this.setModified();
    }

    /**
     * Sets a feature scaling function
     * @param attribute
     * @param function
     */
    public void setScalingFunction(String attribute, String function) {
        this.setModified();
        this.getFeatureScaling().setScalingFunction(attribute, function);
    }

    /**
     * Set this model and all configurations unmodified
     */
    public void setUnmodified() {
        this.modified = false;
        getLogisticRegressionConfiguration().setUnmodified();
        getNaiveBayesConfiguration().setUnmodified();
        getRandomForestConfiguration().setUnmodified();
    }

    /**
     * TODO: Ugly hack to set base-parameters for all methods
     * @param t
     */
    public void setVectorLength(Integer t) {
        this.config.setVectorLength(t);
        this.configNaiveBayes.setVectorLength(t);
        this.configRandomForest.setVectorLength(t);
        this.setModified();
    }

    /**
     * Sets modified
     */
    private void setModified() {
        this.modified = true;
    }
}
