com.codingame.gameengine.module.toggle.ToggleModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of module-toggle Show documentation
This module allows you to display or hide elements of the GraphicEntityModule using the viewer's options menu.
The newest version!
package com.codingame.gameengine.module.toggle;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import com.codingame.gameengine.core.AbstractPlayer;
import com.codingame.gameengine.core.GameManager;
import com.codingame.gameengine.core.Module;
import com.codingame.gameengine.module.entities.Entity;
import com.codingame.gameengine.module.entities.GraphicEntityModule;
import com.google.inject.Inject;
import com.google.inject.Singleton;

/**
 * @author Jean Porée
 *
 *         This module allows you to display or hide elements of the GraphicEntityModule using the viewer's options menu.
 *
 */
@Singleton
public class ToggleModule implements Module {

    GameManager gameManager;
    @Inject GraphicEntityModule entityModule;
    Map registered, newRegistration;

    class Toggle {
        public String name;
        public boolean state = true;

        public Toggle(String name, boolean state) {
            this.name = name;
            this.state = state;
        }

        public boolean equals(Toggle other) {
            return other != null && this.state == other.state && stringEquals(this.name, other.name);
        }

        private boolean stringEquals(String a, String b) {
            if (a == b) {
                return true;
            } else if (a != null && a.equals(b)) {
                return true;
            } else {
                return false;
            }
        }
    }

    @Inject
    ToggleModule(GameManager gameManager) {
        this.gameManager = gameManager;
        gameManager.registerModule(this);
        registered = new HashMap<>();
        newRegistration = new HashMap<>();
    }

    @Override
    public void onGameInit() {
        sendFrameData();
    }

    @Override
    public void onAfterGameTurn() {
        sendFrameData();
    }

    @Override
    public void onAfterOnEnd() {
    }

    private void sendFrameData() {
        if (newRegistration.size() > 0) {
            Map data = new HashMap<>();
            newRegistration.forEach((entityId, toggle) -> {
                data.put(toggle.name, data.getOrDefault(toggle.name, "") + entityId + (toggle.state ? "+" : "-"));
            });
            gameManager.setViewData("toggles", data);
            newRegistration.clear();
        }
    }

    /**
     * Will display the entity only when the toggle state matches the state you set
     *
     * @param entity
     *            which will be displayed
     * @param toggle
     *            the name of the toggle you want to use
     * @param state
     *            the state of the toggle where the entity will be displayed at
     */
    public void displayOnToggleState(Entity entity, String toggle, boolean state) {
        int id = entity.getId();
        Toggle associatedToggle = new Toggle(toggle, state);
        if (!associatedToggle.equals(registered.get(id))) {
            newRegistration.put(id, associatedToggle);
            registered.put(id, associatedToggle);
        }
    }
}