godotVmcSharp/Marionette.cs

/*
    godotVmcSharp
    Copyright (C) 2023  Cassandra de la Cruz-Munoz

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published
    by the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
    */

using System;
using System.Net;
using System.Collections.Generic;
using Godot;

namespace godotVmcSharp
{
    public class Marionette
    {
        private godotOscSharp.OscReceiver receiver;
        private godotOscSharp.OscSender sender;
        public Marionette(int port)
        {
            receiver = new godotOscSharp.OscReceiver(port);
            receiver.MessageReceived += (sender, e) =>
            {
                if (sender == null) {
                    sender = new godotOscSharp.OscSender(IPAddress.Parse(e.IPAddress), port);
                }
                GD.Print($"Received a message from {e.IPAddress}:{e.Port}");
                ProcessMessage(e.Message);
            };
            receiver.ErrorReceived += (sender, e) =>
            {
                GD.Print($"Error: {e.ErrorMessage}");
            };
        }
        private void ProcessMessage(godotOscSharp.OscMessage m)
        {
            switch (m.Address.ToString())
            {
                case "/VMC/Ext/OK":
                    new VmcExtOk(m);
                    break;
                case "/VMC/Ext/T":
                    new VmcExtT(m);
                    break;
                case "/VMC/Ext/Root/Pos":
                    new VmcExtRootPos(m);
                    break;
                case "/VMC/Ext/Bone/Pos":
                    new VmcExtBonePos(m);
                    break;
                case "/VMC/Ext/Blend/Val":
                    new VmcExtBlendVal(m);
                    break;
                case "/VMC/Ext/Blend/Apply":
                    new VmcMessage(m.Address);
                    break;
                case "/VMC/Ext/Cam":
                    new VmcExtCam(m);
                    break;
                case "/VMC/Ext/Con":
                    new VmcExtCon(m);
                    break;
                case "/VMC/Ext/Key":
                    new VmcExtKey(m);
                    break;
                case "/VMC/Ext/Midi/Note":
                    new VmcExtMidiNote(m);
                    break;
                case "/VMC/Ext/Midi/CC/Val":
                    MidiValue(m.Data);
                    break;
                case "/VMC/Ext/Midi/CC/Bit":
                    MidiButton(m.Data);
                    break;
                case "/VMC/Ext/Hmd/Pos":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Con/Pos":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Tra/Pos":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Hmd/Pos/Local":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Con/Pos/Local":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Tra/Pos/Local":
                    new VmcExtDevicePos(m);
                    break;
                case "/VMC/Ext/Rcv":
                    ValidateReceiveEnable(m.Data);
                    break;
            }
        }
        private void MidiValue(List<godotOscSharp.OscArgument> data)
        {
            var addr = "/VMC/Ext/Midi/CC/Val";
            if (data[0].Type != 'i')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "knob", 'i', data[0].Type));
                return;
            }
            if (data[1].Type != 'f')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "value", 'f', data[1].Type));
                return;
            }
            GD.Print($"CC value input on knob {(int)data[0].Value} with value {(float)data[1].Value}.");
        }
        private void MidiButton(List<godotOscSharp.OscArgument> data)
        {
            var addr = "/VMC/Ext/Midi/CC/Bit";
            if (data.Count != 2)
            {
                GD.Print($"Invalid argument count for \"{addr}\". Expected 2, received {data.Count}");
                return;
            }
            if (data[0].Type != 'i')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "knob", 'i', data[0].Type));
                return;
            }
            if (data[1].Type != 'i')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "active", 'i', data[1].Type));
                return;
            }
            switch ((int)data[1].Value)
            {
                case 0:
                    GD.Print("Button released on knob {(int)data[0].Value}");
                    break;
                case 1:
                    GD.Print("Button pressed on knob {(int)data[0].Value}");
                    break;
                default:
                    GD.Print($"Invalid value for \"active\" argument of {addr}. Expected 0 or 1, received {(int)data[1].Value}.");
                    return;
            }
        }
        private bool ValidateReceiveEnable(List<godotOscSharp.OscArgument> data)
        {
            var addr = "/VMC/Ext/Rcv";
            switch (data.Count)
            {
                case 2:
                    break;
                case 3:
                    if (data[2].Type != 's')
                    {
                        GD.Print(InvalidArgumentType.GetErrorString(addr, "IpAddress", 's', data[2].Type));
                        return false;
                    }
                    break;
                default:
                    return false;
            }
            if (data[0].Type != 'i')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "enable", 'i', data[0].Type));
                return false;
            }
            if (data[1].Type != 'i')
            {
                GD.Print(InvalidArgumentType.GetErrorString(addr, "port", 'i', data[0].Type));
                return false;
            }
            if ((int)data[1].Value < 0 || (int)data[1].Value > 65535)
            {
                GD.Print($"Invalid value for \"port\" argument of {addr}. Expected 0-65535, received {(int)data[1].Value}.");
                return false;
            }
            return true;
        }
    }
}