Audio2FaceRESTHandler.h

Go to the documentation of this file.

// Copyright(c) 2024 grrimgrriefer & DZnnah, see LICENSE for details.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Dom/JsonObject.h"
#include "HttpModule.h"
#include "Interfaces/IHttpRequest.h"
#include "Interfaces/IHttpResponse.h"

class VOXTAUTILITY_A2F_API Audio2FaceRESTHandler : public TSharedFromThis<Audio2FaceRESTHandler>
{
#pragma region public API
public:
    void TryInitialize();

    void GetBlendshapes(const FString& wavFileName, const FString& shapesFilePath, const FString& shapesFileName,
        TFunction<void(const FString&, bool /*success*/)> callback);

    bool IsInitializing() const;

    bool IsAvailable() const;
#pragma endregion
 
#pragma region private helper classes
private:
    enum class CurrentA2FState : uint8
    {
        NotConnected,
        Initializing,
        Idle,
        Busy
    };
#pragma endregion
 
#pragma region data
private:
    std::atomic<CurrentA2FState> m_currentState = CurrentA2FState::NotConnected;
#pragma endregion
 
#pragma region private API
    void GetStatus(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
    void LoadUsdFile(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
    void SetPlayerRootPath(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
    void SetPlayerTrack(const FString& fileName, TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
    void GenerateBlendShapes(const FString& filePath, const FString& fileName,
        TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
 
    // TODO: implement these, maybe? idk yet
    //void GetInstance(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback);
    //void GetSettings(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback);
    //void GetPlayerInstance(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback);
    //void GetPlayerTracks(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback);
    //void GetBlendshapeSolver(TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback);

    FString JsonToString(TSharedRef<FJsonObject> jsonObject) const;

    TSharedRef<IHttpRequest, ESPMode::ThreadSafe> GetBaseRequest(
        TFunction<void(FHttpRequestPtr, FHttpResponsePtr, bool)> callback) const;
#pragma endregion
};