/*
 * Copyright (c) 2015-2018, IGN France.
 * Copyright (c) 2018-2026, Giro3D team.
 * SPDX-License-Identifier: MIT
 */

import { DataTexture, FloatType, LinearFilter, RGFormat } from 'three';

import type { DecodeBilTerrainResult, MessageMap, MessageType } from './bilWorker';
import type { DecodeOptions, DecodeResult } from './ImageFormat';

import WorkerPool from '../utils/WorkerPool';
import { decodeRaster } from './bilWorker';
import ImageFormat from './ImageFormat';

let workerPool: WorkerPool<MessageType, MessageMap> | null = null;

function createWorker(): Worker {
    return new Worker(new URL('./bilWorker.js', import.meta.url), {
        type: 'module',
        name: 'bil',
    });
}

/**
 * Decoder for [BIL](https://desktop.arcgis.com/en/arcmap/10.3/manage-data/raster-and-images/bil-bip-and-bsq-raster-files.htm) images.
 *
 * At the moment, only single band BIL files are supported and it is tested only on IGN elevation
 * WMS and WMTS layers.
 *
 * ```js
 *  // Create an elevation source
 * const source = new WmsSource({
 *     url: 'https://data.geopf.fr/wms-r',
 *     projection: 'EPSG:2154',
 *     layer: 'ELEVATION.ELEVATIONGRIDCOVERAGE.HIGHRES',
 *     imageFormat: 'image/x-bil;bits=32',
 *     format: new BilFormat(),
 * });
 *
 * const elevationLayer = new ElevationLayer({ source });
 *
 * map.addLayer(elevationLayer);
 *
 * ```
 * [See it in action](/examples/ign-data.html).
 *
 */
class BilFormat extends ImageFormat {
    public readonly isBilFormat: boolean = true as const;
    public override readonly type = 'BilFormat' as const;

    private _enableWorkers: boolean = true;
    private _workerConcurrency: number | undefined;

    /**
     * @param options - Decoder options.
     */
    public constructor(options?: {
        /**
         * Enables processing raster data in web workers.
         * @defaultValue true
         */
        enableWorkers?: boolean;
        /**
         * The maximum number of workers created by the worker pool.
         * If `undefined`, the maximum number of workers will be allowed.
         * @defaultValue undefined
         */
        workerConcurrency?: number;
    }) {
        super(true, FloatType);

        this._enableWorkers = options?.enableWorkers ?? true;
        this._workerConcurrency = options?.workerConcurrency ?? undefined;
    }

    /**
     * Decode a Bil blob into a
     * [DataTexture](https://threejs.org/docs/?q=texture#api/en/textures/DataTexture) containing
     * the elevation data. At the moment only one band BIL is supported.
     *
     * @param blob - the data to decode
     * @param options - the decoding options
     */

    public async decode(blob: Blob, options: DecodeOptions): Promise<DecodeResult> {
        const buffer = await blob.arrayBuffer();
        const floatArray = new Float32Array(buffer);

        const noData = options?.noDataValue;
        let result: DecodeBilTerrainResult;

        if (this._enableWorkers) {
            if (workerPool == null) {
                workerPool = new WorkerPool({ createWorker, concurrency: this._workerConcurrency });
            }

            result = await workerPool.queue('DecodeBilTerrainMessage', { buffer, noData });
        } else {
            result = decodeRaster(floatArray, noData);
        }

        const texture = new DataTexture(
            new Float32Array(result.data),
            options.width,
            options.height,
            RGFormat,
            FloatType,
        );

        texture.needsUpdate = true;
        texture.generateMipmaps = false;
        texture.magFilter = LinearFilter;
        texture.minFilter = LinearFilter;

        return { texture, min: result.min, max: result.max };
    }
}

export default BilFormat;