#

Junjie

2024-04-26 0617722410f21916ae96d40d30e0778e2239217d

 zy-asrs-flow/src/pages/map/index.jsx

@@ -1,5 +1,5 @@
import * as React from 'react'
import * as PIXI from 'pixi.js';
import * as TWEEDLE from 'tweedle.js';
import { FormattedMessage, useIntl, useModel } from '@umijs/max';
import { Layout, Button, Flex, Row, Col, FloatButton, Select, notification, Segmented } from 'antd';
const { Header, Content } = Layout;
@@ -8,17 +8,18 @@
    FileAddOutlined,
    CompressOutlined,
    SettingOutlined,
    CloseOutlined
} from '@ant-design/icons';
import './index.css'
import { createStyles } from 'antd-style';
import Edit from './components/device';
import Settings from './components/settings'
import * as Utils from './utils'
import WebSocketClient from './websocket'
import Player from './player';
import MapSearch from './header/search';
import MapFloor from './header/floor';
import MapDrawer from './drawer';
import BatchDrawer from './batch';

const useStyles = createStyles(({ token }) => {
    let dark = token.colorBgBase === '#000';
@@ -59,7 +60,7 @@
    SETTINGS_MODEL: "3",
})

let player;
let player, websocket;

const Map = () => {
    const intl = useIntl();
@@ -78,26 +79,36 @@
    });
    const [app, setApp] = React.useState(null);
    const [mapContainer, setMapContainer] = React.useState(null);
    const [didClickSprite, setDidClickSprite] = React.useState(false);
    const [spriteBySettings, setSpriteBySettings] = React.useState(null);
    const prevSpriteBySettingsRef = React.useRef();
    const [drawerVisible, setDrawerVisible] = React.useState(false);
    const [dataFetched, setDataFetched] = React.useState(false);
    const [curSprite, setCurSPrite] = React.useState(null);
    const prevCurSpriteRef = React.useRef();
    const hasFloor = true;
    // const [hasFloor, setHasFloor] = React.useState(true);
    const [curFloor, setCurFloor] = React.useState(1);
    const [floorList, setFloorList] = React.useState([]);
    const [curFloor, setCurFloor] = React.useState(() => {
        const storedValue = localStorage.getItem('curFloor');
        return storedValue !== null ? JSON.parse(storedValue) : null;
    });
    const curFloorRef = React.useRef(curFloor);
    const [batchSprites, setBatchSprites] = React.useState([]);
    const [batchDrawerVisible, setBatchDrawerVisible] = React.useState(false);

    // init func
    React.useEffect(() => {
        const initialize = async () => {
            player = new Player(mapRef.current, styles.dark, didClickSprite);
            player = new Player(mapRef.current, styles.dark);
            setApp(player.app);
            setMapContainer(player.mapContainer);
            Utils.syncApp(player.app);
            Utils.syncMapContainer(player.mapContainer);
            Utils.syncNotify(notify);

            websocket = new WebSocketClient('/ws/map/websocket');
            websocket.connect();
            websocket.onMessage = (data) => {
                Utils.updateMapStatusInRealTime(data, () => curFloorRef.current, setCurSPrite);
            }

            const handleResize = () => {
                setWindowSize({
@@ -107,7 +118,11 @@
            };
            window.addEventListener('resize', handleResize);

            await Utils.fetchMapData(curFloor);
            const mapFloorData = await Utils.fetchMapFloor();
            setFloorList(mapFloorData);
            let defaultFloor = curFloor || mapFloorData?.[0]?.value;
            setCurFloor(defaultFloor);
            await Utils.fetchMapData(defaultFloor);
            setDataFetched(true);
            setModel(MapModel.OBSERVER_MODEL)
            setTimeout(() => {
@@ -116,6 +131,13 @@
            }, 200)
        }
        initialize();

        return () => {
            websocket.onMessage = (data) => {}
            if (websocket) {
                websocket.close();
            }
        }
    }, []);

    // resize
@@ -131,61 +153,69 @@
        }
    }, [app, mapContainer, windowSize])

    // model
    React.useEffect(() => {
        if (!mapContainer && !dataFetched) {
            return;
        }
    // fn switch model
    const switchModel = (model) => {
        Utils.removeSelectedEffect();

        setCurSPrite(null);
        setDrawerVisible(false);
        setSpriteBySettings(null);
        setSettingsVisible(false);
        setBatchSprites([]);
        setBatchDrawerVisible(false);

        switch (model) {
            case MapModel.OBSERVER_MODEL:

                player.hideGridlines();
                player.hideStarryBackground();

                player.activateMapEvent(null);

                Utils.removeSelectedEffect();
                setCurSPrite(null);
                setDeviceVisible(false);
                setSettingsVisible(false);
                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, (selectedSprites, resetFn) => {
                    setBatchSprites(selectedSprites);
                });

                mapContainer.children.forEach(child => {
                    Utils.viewFeature(child, setCurSPrite);
                })
                break
            case MapModel.MOVABLE_MODEL:

                player.showGridlines();
                player.hideStarryBackground();

                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, model);

                Utils.removeSelectedEffect();
                setSpriteBySettings(null);
                setSettingsVisible(false);
                setDrawerVisible(false);
                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, (selectedSprites, resetFn) => {
                    Utils.spriteListBeMovable(selectedSprites, () => {
                        resetFn();
                    });
                });

                mapContainer.children.forEach(child => {
                    Utils.beMovable(child, setDidClickSprite);
                    Utils.beMovable(child);
                })
                break
            case MapModel.SETTINGS_MODEL:

                player.showGridlines();
                player.showStarryBackground();

                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, model);

                setDeviceVisible(false);
                setDrawerVisible(false);
                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, (selectedSprites, resetFn) => {
                    setBatchSprites(selectedSprites);
                });

                mapContainer.children.forEach(child => {
                    Utils.beSettings(child, setSpriteBySettings, setDidClickSprite);
                    Utils.beSettings(child, setSpriteBySettings);
                })
                break
            default:
                break
        }
    }

    // model
    React.useEffect(() => {
        if (!mapContainer && !dataFetched) {
            return;
        }
        switchModel(model);
    }, [model]);

    // Add New Device
@@ -196,7 +226,7 @@

        Utils.initSprite(sprite, type);
        mapContainer.addChild(sprite);
        Utils.beMovable(sprite, setDidClickSprite);
        Utils.beMovable(sprite);
    };

    // watch curSprite
@@ -223,20 +253,7 @@
    const switchFloor = async (floor) => {
        await Utils.fetchMapData(floor);

        player.hideGridlines();
        player.hideStarryBackground();

        player.activateMapEvent(null);

        Utils.removeSelectedEffect();
        setCurSPrite(null);
        setDeviceVisible(false);
        setSettingsVisible(false);
        setDrawerVisible(false);

        mapContainer.children.forEach(child => {
            Utils.viewFeature(child, setCurSPrite);
        })
        switchModel(model);

        setTimeout(() => {
            player.adaptScreen();
@@ -245,16 +262,13 @@

    // watch curFloor
    React.useEffect(() => {
        curFloorRef.current = curFloor;
        if (!mapContainer && !dataFetched) {
            return;
        }
        switchFloor(curFloor);
        localStorage.setItem('curFloor', JSON.stringify(curFloor));
    }, [curFloor]);

    // didClickSprite, stop triggers both sprite click and play's selection boxs
    React.useEffect(() => {
        player.updateDidClickSprite(didClickSprite);
    }, [didClickSprite])

    // watch spriteBySettings
    React.useEffect(() => {
@@ -274,6 +288,19 @@
    }, [spriteBySettings])
    const prevSpriteBySettings = prevSpriteBySettingsRef.current;

    // watch batchSprites
    React.useEffect(() => {
        if (!mapContainer) {
            return;
        }
        if (batchSprites?.length > 0) {
            setBatchDrawerVisible(true)
        } else {
            player.clearSelectedSprites();
            setBatchDrawerVisible(false)
        }
    }, [batchSprites])

    return (
        <>
            {contextHolder}
@@ -286,6 +313,7 @@
                                    model={model}
                                    setModel={setModel}
                                    ModelEnum={MapModel}
                                    curFloor={curFloor}
                                    curSprite={curSprite}
                                    setCurSPrite={setCurSPrite}
                                    setSpriteBySettings={setSpriteBySettings}
@@ -358,8 +386,9 @@
                <Content ref={contentRef} className={styles.content}>
                    <div ref={mapRef} />

                    {hasFloor && (
                    {floorList.length > 0 && (
                        <MapFloor
                            floorList={floorList}
                            curFloor={curFloor}
                            setCurFloor={setCurFloor}
                        />
@@ -397,8 +426,8 @@
                                if (deviceVisible) {
                                    setDeviceVisible(false);
                                } else {
                                    setDeviceVisible(true);
                                    setModel(MapModel.MOVABLE_MODEL);
                                    setDeviceVisible(true);
                                }
                            }}
                        />
@@ -418,10 +447,23 @@
            <MapDrawer
                open={drawerVisible}
                curSprite={curSprite}
                curFloor={curFloor}
                refCurr={mapRef.current}
                onCancel={() => {
                    setCurSPrite(null);
                    setDrawerVisible(false);
                }}
            />

            <BatchDrawer
                open={batchDrawerVisible}
                batchSprites={batchSprites}
                refCurr={mapRef.current}
                model={model}
                ModelEnum={MapModel}
                onCancel={() => {
                    setBatchSprites([]);
                    setBatchDrawerVisible(false);
                }}
            />

@@ -442,7 +484,6 @@
                    setSpriteBySettings(null);
                }}
                setSpriteBySettings={setSpriteBySettings}
                setDidClickSprite={setDidClickSprite}
                refCurr={mapRef.current}
            />
        </>