/**
 * Copyright (c) 2020-present, Goldman Sachs
 *
 * 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.
 */
import { PRIMITIVE_TYPE, matchFunctionName, } from '@finos/legend-graph';
import { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
import { QueryBuilderSimpleProjectionColumnState, } from '../../projection/QueryBuilderProjectionColumnState.js';
import { buildAggregateExpression } from './QueryBuilderAggregateOperatorValueSpecificationBuilder.js';
import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graph/QueryBuilderMetaModelConst.js';
import { hashArray, guaranteeType } from '@finos/legend-shared';
import { QUERY_BUILDER_STATE_HASH_STRUCTURE } from '../../../../QueryBuilderStateHashUtils.js';
import { action, makeObservable, observable } from 'mobx';
export class QueryBuilderAggregateOperator_Wavg extends QueryBuilderAggregateOperator {
    weight;
    constructor() {
        super();
        makeObservable(this, {
            weight: observable,
            setWeight: action,
        });
    }
    setWeight(val) {
        this.weight = val;
    }
    getLabel(projectionColumnState) {
        return 'wavg';
    }
    getName(projectionColumnState) {
        return projectionColumnState.columnName;
    }
    isCompatibleWithColumn(projectionColumnState) {
        if (projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState) {
            const propertyType = projectionColumnState.propertyExpressionState.propertyExpression.func
                .value.genericType.value.rawType;
            return [
                PRIMITIVE_TYPE.NUMBER,
                PRIMITIVE_TYPE.INTEGER,
                PRIMITIVE_TYPE.DECIMAL,
                PRIMITIVE_TYPE.FLOAT,
            ].includes(propertyType.path);
        }
        return false;
    }
    buildAggregateExpression(propertyExpression, variableName, graph) {
        return buildAggregateExpression(QUERY_BUILDER_SUPPORTED_FUNCTIONS.WAVG, variableName, graph);
    }
    buildAggregateColumnState(expression, lambdaParam, projectionColumnState) {
        if (matchFunctionName(expression.functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.WAVG)) {
            const aggregateColumnState = new QueryBuilderAggregateColumnState(projectionColumnState.tdsState.aggregationState, projectionColumnState, new QueryBuilderAggregateOperator_Wavg());
            const currentOperator = guaranteeType(aggregateColumnState.operator, QueryBuilderAggregateOperator_Wavg);
            aggregateColumnState.setLambdaParameterName(lambdaParam.name);
            aggregateColumnState.setOperator(currentOperator);
            return aggregateColumnState;
        }
        return undefined;
    }
    get getOperator() {
        return new QueryBuilderAggregateOperator_Wavg();
    }
    get hashCode() {
        return hashArray([
            QUERY_BUILDER_STATE_HASH_STRUCTURE.AGGREGATE_OPERATOR_WAVG,
        ]);
    }
}
//# sourceMappingURL=QueryBuilderAggregateOperator_Wavg.js.map