HIDtoVPADNetworkClient/src/net/ash/HIDToVPADNetworkClient/manager/ActiveControllerManager.java

/*******************************************************************************
 * Copyright (c) 2017 Ash (QuarkTheAwesome) & Maschell
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *******************************************************************************/
package net.ash.HIDToVPADNetworkClient.manager;

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

import lombok.extern.java.Log;
import net.ash.HIDToVPADNetworkClient.controller.Controller;
import net.ash.HIDToVPADNetworkClient.network.NetworkHIDDevice;
import net.ash.HIDToVPADNetworkClient.network.NetworkManager;
import net.ash.HIDToVPADNetworkClient.util.Settings;
import net.ash.HIDToVPADNetworkClient.util.Utilities;

@Log
public final class ActiveControllerManager implements Runnable {
    private static ActiveControllerManager instance = new ActiveControllerManager();
    private final Map<Controller, NetworkHIDDevice> activeControllers = new HashMap<Controller, NetworkHIDDevice>();

    private ActiveControllerManager() {
    }

    public static ActiveControllerManager getInstance() {
        return instance;
    }

    @Override
    public void run() { // TODO: Add mechanism to stop these threads? And add a mechanism start only once?
        new Thread(new Runnable() {
            @Override
            public void run() {
                while (true) {
                    updateControllerStates();
                    ControllerManager.detectControllers();
                    Utilities.sleep(Settings.DETECT_CONTROLLER_INTERVAL);
                }
            }
        }).start();

        new Thread(new Runnable() {
            @Override
            public void run() {
                while (true) {
                    handleControllerInputs();
                    Utilities.sleep(Settings.HANDLE_INPUTS_INTERVAL);
                }
            }
        }).start();
    }

    public void updateControllerStates() {
        List<Controller> currentControllers = ControllerManager.getActiveControllers();

        List<Controller> toAdd = new ArrayList<Controller>();
        List<Controller> toRemove = new ArrayList<Controller>();
        synchronized (activeControllers) {
            // Adding all missing.

            for (Controller c : currentControllers) {
                if (!activeControllers.containsKey(c)) {
                    log.info("Added " + c);
                    toAdd.add(c);
                }
            }

            // removing all old
            for (Controller c : activeControllers.keySet()) {
                if (!currentControllers.contains(c)) {
                    log.info("Removed " + c);
                    toRemove.add(c);
                }
            }
        }

        addController(toAdd);
        removeController(toRemove);
    }

    private void removeController(List<Controller> toRemove) {
        synchronized (activeControllers) {
            for (Controller c : toRemove) {
                NetworkManager.getInstance().removeHIDDevice(activeControllers.get(c));
                // c.destroyDriver(); Removing it from the list doesn't require to close the connection.
                activeControllers.remove(c);
            }
        }
    }

    private void addController(List<Controller> toAdd) {
        synchronized (activeControllers) {
            for (Controller c : toAdd) {
                NetworkHIDDevice hiddevice = new NetworkHIDDevice(c.getVID(), c.getPID());
                if (NetworkManager.getInstance().isConnected()) hiddevice.sendAttach(); // Only send the attach when we're connected. Because on connecting the
                                                                                        // TCPClient will automatically attach all active controller
                NetworkManager.getInstance().addHIDDevice(hiddevice);
                activeControllers.put(c, hiddevice);
            }
        }
    }

    private void handleControllerInputs() {
        synchronized (activeControllers) {
            for (Entry<Controller, NetworkHIDDevice> entry : activeControllers.entrySet()) {
                byte[] data = entry.getKey().getLatestData();
                if (data != null && data.length > 0) {
                    NetworkHIDDevice device = entry.getValue();
                    device.sendRead(data);
                }
            }
        }
    }

    public void attachAllActiveControllers() {
        synchronized (activeControllers) {
            for (Entry<Controller, NetworkHIDDevice> entry : activeControllers.entrySet()) {
                NetworkHIDDevice device = entry.getValue();
                device.sendAttach();
            }
        }
    }

    public void detachAllActiveControllers() {
        synchronized (activeControllers) {
            for (Entry<Controller, NetworkHIDDevice> entry : activeControllers.entrySet()) {
                NetworkHIDDevice device = entry.getValue();
                device.sendDetach();
            }
        }
    }

    /**
     * 
     * @param HIDhandle
     * @return returns the controller for the given handle. returns null if the controller with the given handle is not found.
     */
    public Controller getControllerByHIDHandle(int HIDhandle) {
        for (Entry<Controller, NetworkHIDDevice> entry : activeControllers.entrySet()) {
            if (entry.getValue().getHidHandle() == HIDhandle) {
                return entry.getKey();
            }
        }
        return null;
    }

}