MovementIn.h

#pragma once

// Forward declare EnvironmentalInstrumentsAudioProcessor
class EnvironmentalInstrumentsAudioProcessor;

#include "Types.h"

#include "JoyConBridge/src/JoyConBridge/JoyCon.h"
#include "JoyConBridge/src/JoyConBridge/protocol.h"
using JoyCon = joy_con_bridge::JoyCon;
using SensorData = joy_con_bridge::protocol::SensorData;
using ButtonsState = joy_con_bridge::ButtonsState;

#include <memory>
#include <thread>
#include <atomic>
#include <utility>
#include <juce_core/juce_core.h>  // For juce::Thread and other utilities

class MovementIn {
public:
    MovementIn(EnvironmentalInstrumentsAudioProcessor&);
    ~MovementIn();

    void startPolling();
    void stopPolling();
    
    std::pair<SensorDataFloat, SensorDataFloat> getJoyConData();
    SensorDataFloat getLeftJoyConData();
    SensorDataFloat getRightJoyConData();
    ButtonsState getLeftJoyConButtons();
    ButtonsState getRightJoyConButtons();

    bool outputMuted();
    bool waitForUnmute();

private:
    void connectJoyCon();
    std::unique_ptr<JoyCon> leftJoyCon;
    std::mutex leftJoyConLock;
    std::unique_ptr<JoyCon> rightJoyCon;
    std::mutex rightJoyConLock;

    EnvironmentalInstrumentsAudioProcessor& processorRef;

    std::thread connectionThread;  // For JoyCon connection handling
    std::thread dataPollingThread; // For movement data polling

    void pollForJoyCons();  // Handle JoyCon connection
    void pollForMovementData();  // Handle real-time JoyCon data polling
    void processMuteButton(bool buttonDepressed);

    struct AxisCalibration {
        const float coeff;
        const int16_t offset;
    };
    struct Calib {
        const struct {
            AxisCalibration x;
            AxisCalibration y;
            AxisCalibration z;
        } accel;
        const struct {
            AxisCalibration x;
            AxisCalibration y;
            AxisCalibration z;
        } gyro;
    };

    const Calib calibL = {
        { {1.0f, 162}, {1.0f, -197}, {1.0f, 356} },  // accel
        { {0.8719269644603848f, -18}, {0.8719269644603848f, 12}, {0.8719269644603848f, 17} }  // gyro
    };

    const Calib calibR = {
        { {1.0f, -78}, {1.0f, -53}, {1.0f, 306} },  // accel
        { {0.8719269644603848f, 0}, {0.8719269644603848f, -11}, {0.8719269644603848f, 44} }  // gyro
    };

    const SensorDataFloat zeroL = {
        {167, 197, 3810},
        {20.05432018258885f, -10.463123573524618f, 46.21212911640039f}
    };

    const SensorDataFloat zeroR = {
        {260.5, 30, -4311},
        {0.0f, 9.591196609064232f, -38.36478643625693f}
    };

    SensorDataFloat calibrateSensorData(const SensorData& data, const Calib& calib);

    SensorData leftJoyConData = {};
    SensorData rightJoyConData = {};
    ButtonsState leftJoyConButtons = {};
    ButtonsState rightJoyConButtons = {};
    std::mutex dataMutex;

    bool muteOutput = true;
    std::mutex muteMutex;

    std::atomic<bool> shouldStop{ false };
    std::condition_variable signpost;
};