/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.ignite.examples.ml.genetic.knapsack;

import java.util.List;
import java.util.function.Consumer;
import org.apache.ignite.Ignite;
import org.apache.ignite.ml.genetic.Chromosome;
import org.apache.ignite.ml.genetic.Gene;
import org.apache.ignite.ml.genetic.parameter.ITerminateCriteria;
import org.apache.ignite.ml.genetic.utils.GAGridUtils;

/**
 * Represents the terminate condition for {@link KnapsackGAExample}.
 * <p>
 * Class terminates Genetic algorithm when once GA Grid has performed 30 generations.</p>
 */
public class KnapsackTerminateCriteria implements ITerminateCriteria {
    /**
     * Ignite instance.
     */
    private final Ignite ignite;

    /**
     *
     */
    private final Consumer<String> logConsumer;

    /**
     * Create class instance.
     *
     * @param ignite      Ignite instance.
     * @param logConsumer Logging consumer.
     */
    KnapsackTerminateCriteria(Ignite ignite, Consumer<String> logConsumer) {
        this.ignite = ignite;
        this.logConsumer = logConsumer;
    }

    /**
     * {@inheritDoc}
     */
    @Override public boolean isTerminationConditionMet(Chromosome fittestChromosome, double averageFitnessScore,
        int currGeneration) {
        boolean isTerminate = true;

        logConsumer.accept(
            "\n##########################################################################################"
                + "\n Generation: " + currGeneration
                + "\n Fittest is Chromosome Key: " + fittestChromosome
                + "\nTotal value is: " + fittestChromosome.getFitnessScore()
                + "\nTotal weight is: " + calculateTotalWeight(
                GAGridUtils.getGenesInOrderForChromosome(ignite, fittestChromosome))
                + "\nChromosome: " + fittestChromosome
                + "\n" + reportItems(GAGridUtils.getGenesInOrderForChromosome(ignite, fittestChromosome))
                + "\n##########################################################################################");

        if (!(currGeneration > 29))
            isTerminate = false;

        return isTerminate;
    }

    /**
     * Calculate total weight.
     *
     * @param genes List of Genes.
     * @return Calculated value.
     */
    private double calculateTotalWeight(List<Gene> genes) {
        double totalWeight = 0;
        for (Gene gene : genes)
            totalWeight += ((Item)gene.getVal()).getWeight();

        return totalWeight;
    }

    /**
     * Helper to print items in knapsack.
     *
     * @param genes List of Genes.
     * @return Items to print.
     */
    private String reportItems(List<Gene> genes) {
        StringBuilder sb = new StringBuilder();

        for (Gene gene : genes) {
            sb.append("\n------------------------------------------------------------------------------------------")
                .append("\nName: ").append(((Item)gene.getVal()).getName())
                .append("\nWeight: ").append(((Item)gene.getVal()).getWeight())
                .append("\nValue: ").append(((Item)gene.getVal()).getVal());
        }

        return sb.toString();
    }
}