/*
 * Copyright (C) 2023 The Android Open Source Project
 *
 * 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 com.android.server.healthconnect.fitness.aggregation;

import static com.android.server.healthconnect.fitness.aggregation.AggregateParams.PriorityAggregationExtraParams.VALUE_TYPE_DOUBLE;
import static com.android.server.healthconnect.fitness.aggregation.AggregateParams.PriorityAggregationExtraParams.VALUE_TYPE_LONG;

import android.database.Cursor;

import com.android.internal.annotations.VisibleForTesting;
import com.android.server.healthconnect.fitness.aggregation.AggregateParams.PriorityAggregationExtraParams.ValueColumnType;
import com.android.server.healthconnect.storage.utils.StorageUtils;

import java.util.Map;

/**
 * Represents priority aggregation data for one column.
 *
 * @hide
 */
class ValueColumnAggregationData extends AggregationRecordData {
    private final double mValue;

    ValueColumnAggregationData(
            Cursor cursor,
            boolean useLocalTime,
            Map<Long, Integer> appIdToPriority,
            String valueColumnName,
            @ValueColumnType int type) {
        super(cursor, useLocalTime, appIdToPriority);
        if (type == VALUE_TYPE_DOUBLE) {
            mValue = StorageUtils.getCursorDouble(cursor, valueColumnName);
        } else if (type == VALUE_TYPE_LONG) {
            mValue = StorageUtils.getCursorLong(cursor, valueColumnName);
        } else {
            throw new IllegalArgumentException("Unknown aggregation column type.");
        }
    }

    @VisibleForTesting
    ValueColumnAggregationData(
            long recordStartTime,
            long recordEndTime,
            int priority,
            long lastModifiedTime,
            double value) {
        super(
                recordStartTime,
                recordEndTime,
                priority,
                lastModifiedTime,
                /* startTimeZoneOffset= */ null);
        this.mValue = value;
    }

    @Override
    double getResultOnInterval(AggregationTimestamp startPoint, AggregationTimestamp endPoint) {
        double intervalDuration = getEndTime() - getStartTime();
        double overlapDuration =
                Math.min(getEndTime(), endPoint.getTime())
                        - Math.max(getStartTime(), startPoint.getTime());

        // Case when this record start time equals to end time.
        // We check types of timestamps as if
        // multiple instant (start time = end time) records have the same time, we want to account
        // only one value. In this case sorted timestamps will look like
        // [start1, start2, end1, end2] and we output non-zero value only after calling
        // getResultOnInterval(start2, end1).
        if (intervalDuration == 0
                && startPoint.getType() == AggregationTimestamp.INTERVAL_START
                && endPoint.getType() == AggregationTimestamp.INTERVAL_END) {
            return mValue;
        }

        if (intervalDuration < 0 || overlapDuration <= 0) {
            return 0;
        }

        return mValue * overlapDuration / intervalDuration;
    }
}