tobspr_shapez.io/src/js/game/buildings/transistor.js

import { generateMatrixRotations } from "../../core/utils";
import { enumDirection, Vector } from "../../core/vector";
import { enumLogicGateType, LogicGateComponent } from "../components/logic_gate";
import { enumPinSlotType, WiredPinsComponent } from "../components/wired_pins";
import { Entity } from "../entity";
import { defaultBuildingVariant, MetaBuilding } from "../meta_building";
import { GameRoot } from "../root";
import { enumHubGoalRewards } from "../tutorial_goals";

export class MetaTransistorBuilding extends MetaBuilding {
    constructor() {
        super("transistor");
    }

    /**
     * @param {string} variant
     */
    getSilhouetteColor(variant) {
        return MetaTransistorBuilding.silhouetteColors[variant]();
    }

    /**
     * @param {string} variant
     */
    getIsRemovable(variant) {
        return MetaTransistorBuilding.isRemovable[variant]();
    }

    /**
     * @param {string} variant
     */
    getIsRotateable(variant) {
        return MetaTransistorBuilding.isRotateable[variant]();
    }

    /**
     * @param {GameRoot} root
     */
    getAvailableVariants(root) {
        const variants = MetaTransistorBuilding.avaibleVariants;

        let available = [];
        for (const variant in variants) {
            if (variants[variant](root)) available.push(variant);
        }

        return available;
    }

    /**
     * Returns the edit layer of the building
     * @param {GameRoot} root
     * @param {string} variant
     * @returns {Layer}
     */
    getLayer(root, variant) {
        // @ts-ignore
        return MetaTransistorBuilding.layerByVariant[variant](root);
    }

    /**
     * @param {string} variant
     */
    getDimensions(variant) {
        return MetaTransistorBuilding.dimensions[variant]();
    }

    /**
     * @param {string} variant
     */
    getShowLayerPreview(variant) {
        return MetaTransistorBuilding.layerPreview[variant]();
    }

    /**
     * @param {number} rotation
     * @param {number} rotationVariant
     * @param {string} variant
     * @param {Entity} entity
     * @returns {Array<number>|null}
     */
    getSpecialOverlayRenderMatrix(rotation, rotationVariant, variant, entity) {
        let matrices = MetaTransistorBuilding.overlayMatrices[variant](entity, rotationVariant);
        return matrices ? matrices[rotation] : null;
    }

    /**
     * @param {string} variant
     */
    getRenderPins(variant) {
        return MetaTransistorBuilding.renderPins[variant]();
    }

    /**
     * Creates the entity at the given location
     * @param {Entity} entity
     */
    setupEntityComponents(entity) {
        MetaTransistorBuilding.setupEntityComponents.forEach(func => func(entity));
    }

    /**
     * @param {Entity} entity
     * @param {number} rotationVariant
     * @param {string} variant
     */
    updateVariants(entity, rotationVariant, variant) {
        MetaTransistorBuilding.componentVariations[variant](entity, rotationVariant);
    }
}

MetaTransistorBuilding.setupEntityComponents = [
    entity =>
        entity.addComponent(
            new WiredPinsComponent({
                slots: [
                    {
                        pos: new Vector(0, 0),
                        direction: enumDirection.top,
                        type: enumPinSlotType.logicalEjector,
                    },
                    {
                        pos: new Vector(0, 0),
                        direction: enumDirection.left,
                        type: enumPinSlotType.logicalAcceptor,
                    },
                    {
                        pos: new Vector(0, 0),
                        direction: enumDirection.bottom,
                        type: enumPinSlotType.logicalAcceptor,
                    },
                ],
            })
        ),
    entity =>
        entity.addComponent(
            new LogicGateComponent({
                type: enumLogicGateType.transistor,
            })
        ),
];

MetaTransistorBuilding.variants = {
    mirrored: "mirrored",
};

MetaTransistorBuilding.overlayMatrices = {
    [defaultBuildingVariant]: (entity, rotationVariant) =>
        generateMatrixRotations([0, 1, 0, 1, 1, 0, 0, 1, 0]),
    [MetaTransistorBuilding.variants.mirrored]: (entity, rotationVariant) =>
        generateMatrixRotations([0, 1, 0, 0, 1, 1, 0, 1, 0]),
};

MetaTransistorBuilding.dimensions = {
    [defaultBuildingVariant]: () => new Vector(1, 1),
    [MetaTransistorBuilding.variants.mirrored]: () => new Vector(1, 1),
};

MetaTransistorBuilding.silhouetteColors = {
    [defaultBuildingVariant]: () => "#823cab",
    [MetaTransistorBuilding.variants.mirrored]: () => "#823cab",
};

MetaTransistorBuilding.isRemovable = {
    [defaultBuildingVariant]: () => true,
    [MetaTransistorBuilding.variants.mirrored]: () => true,
};

MetaTransistorBuilding.isRotateable = {
    [defaultBuildingVariant]: () => true,
    [MetaTransistorBuilding.variants.mirrored]: () => true,
};

MetaTransistorBuilding.renderPins = {
    [defaultBuildingVariant]: () => false,
    [MetaTransistorBuilding.variants.mirrored]: () => false,
};

MetaTransistorBuilding.layerPreview = {
    [defaultBuildingVariant]: () => "wires",
    [MetaTransistorBuilding.variants.mirrored]: () => "wires",
};

MetaTransistorBuilding.avaibleVariants = {
    [defaultBuildingVariant]: root => root.hubGoals.isRewardUnlocked(enumHubGoalRewards.reward_logic_gates),
    [MetaTransistorBuilding.variants.mirrored]: root =>
        root.hubGoals.isRewardUnlocked(enumHubGoalRewards.reward_logic_gates),
};

MetaTransistorBuilding.layerByVariant = {
    [defaultBuildingVariant]: root => "wires",
    [MetaTransistorBuilding.variants.mirrored]: root => "wires",
};

MetaTransistorBuilding.componentVariations = {
    [defaultBuildingVariant]: (entity, rotationVariant) => {
        entity.components.WiredPins.slots[1].direction = enumDirection.left;
    },

    [MetaTransistorBuilding.variants.mirrored]: (entity, rotationVariant) => {
        entity.components.WiredPins.slots[1].direction = enumDirection.right;
    },
};