/**
 * Copyright (c) 2024 Opal Kelly Incorporated
 *
 * This source code is licensed under the FrontPanel license.
 * See the LICENSE file found in the root directory of this project.
 */

import React from "react";

import type { Decorator } from "@storybook/react";

import { FrontPanel } from "../../components";

import { MockFrontPanel, ByteCount } from "@opalkelly/frontpanel-platform-api";

const mockDevice: MockFrontPanel = new MockFrontPanel(32, 32);

const InitializeMockDevice = () => {
    // Initialize the WireOuts with unique values.
    const wireOutBlock = mockDevice.WireOutBlock;

    let byteId: ByteCount = 0;

    for (let elementIndex = 0; elementIndex < wireOutBlock.Count; elementIndex++) {
        let elementValue: number = byteId++;
        for (let byteIndex = 1; byteIndex < 4; byteIndex++) {
            elementValue |= byteId++ << (8 * byteIndex);
        }

        wireOutBlock.SetValue(
            wireOutBlock.BaseAddress + elementIndex,
            elementValue,
            wireOutBlock.Mask
        );
    }
};

const withFrontPanel: Decorator = (Story) => {
    InitializeMockDevice();
    return (
        <FrontPanel device={mockDevice}>
            <Story />
        </FrontPanel>
    );
};

export default withFrontPanel;