/*
 * SPDX-License-Identifier: Apache-2.0
 *
 * The OpenSearch Contributors require contributions made to
 * this file be licensed under the Apache-2.0 license or a
 * compatible open source license.
 *
 * Modifications Copyright OpenSearch Contributors. See
 * GitHub history for details.
 */

package org.opensearch.timeseries.ml;

import java.time.Instant;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

import org.opensearch.timeseries.model.Config;
import org.opensearch.timeseries.model.Entity;
import org.opensearch.timeseries.model.FeatureData;
import org.opensearch.timeseries.model.IndexableResult;

public abstract class IntermediateResult<IndexableResultType extends IndexableResult> {
    protected final long totalUpdates;
    protected final double rcfScore;

    public IntermediateResult(long totalUpdates, double rcfScore) {
        this.totalUpdates = totalUpdates;
        this.rcfScore = rcfScore;
    }

    public long getTotalUpdates() {
        return totalUpdates;
    }

    public double getRcfScore() {
        return rcfScore;
    }

    @Override
    public int hashCode() {
        return Objects.hash(totalUpdates);
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        IntermediateResult<IndexableResultType> other = (IntermediateResult<IndexableResultType>) obj;
        return totalUpdates == other.totalUpdates && Double.doubleToLongBits(rcfScore) == Double.doubleToLongBits(other.rcfScore);
    }

    /**
     * convert intermediateResult into 1+ indexable results.
     * @param config Config accessor
     * @param dataStartInstant data start time
     * @param dataEndInstant data end time
     * @param executionStartInstant execution start time
     * @param executionEndInstant execution end time
     * @param featureData feature data
     * @param entity entity info
     * @param schemaVersion schema version
     * @param modelId Model id
     * @param taskId Task id
     * @param error Error
     * @return 1+ indexable results
     */
    public abstract List<IndexableResultType> toIndexableResults(
        Config config,
        Instant dataStartInstant,
        Instant dataEndInstant,
        Instant executionStartInstant,
        Instant executionEndInstant,
        List<FeatureData> featureData,
        Optional<Entity> entity,
        Integer schemaVersion,
        String modelId,
        String taskId,
        String error
    );
}