univer/packages/sheets-plugin-formula/src/Model/FormulaDataModel.ts

import { ArrayFormulaDataType, FormulaDataType } from '@univerjs/base-formula-engine';
import { ObjectMatrix } from '@univerjs/core';
import { IFormulaConfig, RecalculationModeType } from '../Basics/Interfaces/IFormula';

export class FormulaDataModel {
    private _formulaData: FormulaDataType = {};

    private _arrayFormulaData: ArrayFormulaDataType = {};

    private _calculationChain: string[] = [];

    private _recalculationMode = RecalculationModeType.AUTOMATIC;

    constructor(config?: IFormulaConfig) {
        if (config?.formulaData) {
            this._formulaData = config.formulaData;
        }

        if (config?.calculationChain) {
            this._calculationChain = config.calculationChain;
        }

        if (config?.recalculationMode) {
            this._recalculationMode = config.recalculationMode;
        }
    }

    getFormulaData() {
        return this._formulaData;
    }

    setFormulaData(value: FormulaDataType) {
        this._formulaData = value;
    }

    getArrayFormulaData(): ArrayFormulaDataType {
        return this._arrayFormulaData;
    }

    setArrayFormulaData(value: ArrayFormulaDataType) {
        Object.keys(value).forEach((sheetId) => {
            const arrayFormula = value[sheetId];
            if (!this._arrayFormulaData[sheetId]) {
                this._arrayFormulaData[sheetId] = new ObjectMatrix();
            }
            arrayFormula.forValue((r, c, v) => {
                this._arrayFormulaData[sheetId].setValue(r, c, v);
            });
        });
    }
}