/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#pragma once

#include <FileIOHandler_wwise_Platform.h>

#include <AK/SoundEngine/Common/AkTypes.h>
#include <AK/SoundEngine/Common/AkStreamMgrModule.h>

namespace Audio
{
    //! Wwise file IO device that access the Lumberyard file system through standard blocking file IO calls. Wwise will still
    //! run these in separate threads so it won't be blocking the audio playback, but it will interfere with the internal
    //! file IO scheduling of Lumberyard. This class can also write, so it's intended use is for one-off file reads and
    //! for tools to be able to write files.
    class CBlockingDevice_wwise
        : public AK::StreamMgr::IAkIOHookBlocking
    {
    public:
        ~CBlockingDevice_wwise() override;

        bool Init(size_t poolSize);
        void Destroy();
        AkDeviceID GetDeviceID() const { return m_deviceID; }
        bool Open(const char* filename, AkOpenMode openMode, AkFileDesc& fileDesc);

        AKRESULT Read(AkFileDesc& fileDesc, const AkIoHeuristics& heuristics, void* buffer, AkIOTransferInfo& transferInfo) override;
        AKRESULT Write(AkFileDesc& fileDesc, const AkIoHeuristics& heuristics, void* data, AkIOTransferInfo& transferInfo) override;

        AKRESULT Close(AkFileDesc& fileDesc) override;
        AkUInt32 GetBlockSize(AkFileDesc& fileDesc) override;
        void GetDeviceDesc(AkDeviceDesc& deviceDesc) override;
        AkUInt32 GetDeviceData() override;

    protected:
        AkDeviceID m_deviceID = AK_INVALID_DEVICE_ID;
    };

    //! Wwise file IO device that uses AZ::IO::Streamer to asynchronously handle file requests. By using AZ::IO::Streamer file requests
    //! can be scheduled along side other file requests for optimal disk usage. This class can't write and is intended to be used 
    //! as part of a streaming system.
    class CStreamingDevice_wwise
        : public AK::StreamMgr::IAkIOHookDeferred
    {
    public:
        ~CStreamingDevice_wwise() override;

        bool Init(size_t poolSize);
        void Destroy();
        AkDeviceID GetDeviceID() const { return m_deviceID; }
        bool Open(const char* filename, AkOpenMode openMode, AkFileDesc& fileDesc);

        AKRESULT Read(AkFileDesc& fileDesc, const AkIoHeuristics& heuristics, AkAsyncIOTransferInfo& transferInfo) override;
        AKRESULT Write(AkFileDesc& fileDesc, const AkIoHeuristics& heuristics, AkAsyncIOTransferInfo& transferInfo) override;
        void Cancel(AkFileDesc& fileDesc, AkAsyncIOTransferInfo& transferInfo, bool& cancelAllTransfersForThisFile) override {}

        AKRESULT Close(AkFileDesc& fileDesc) override;
        AkUInt32 GetBlockSize(AkFileDesc& fileDesc) override;
        void GetDeviceDesc(AkDeviceDesc& deviceDesc) override;
        AkUInt32 GetDeviceData() override;

    protected:
        AkDeviceID m_deviceID = AK_INVALID_DEVICE_ID;
    };

    class CFileIOHandler_wwise
        : public AK::StreamMgr::IAkFileLocationResolver
    {
    public:

        CFileIOHandler_wwise();
        ~CFileIOHandler_wwise() override = default;

        CFileIOHandler_wwise(const CFileIOHandler_wwise&) = delete;
        CFileIOHandler_wwise& operator=(const CFileIOHandler_wwise&) = delete;

        AKRESULT Init(size_t poolSize);
        void ShutDown();

        // IAkFileLocationResolver overrides.
        AKRESULT Open(const AkOSChar* fileName, AkOpenMode openMode, AkFileSystemFlags* flags, bool& syncOpen, AkFileDesc& fileDesc) override;
        AKRESULT Open(AkFileID fileID, AkOpenMode openMode, AkFileSystemFlags* flags, bool& syncOpen, AkFileDesc& fileDesc) override;
        // ~IAkFileLocationResolver overrides.

        void SetBankPath(const char* const bankPath);
        void SetLanguageFolder(const char* const languageFolder);

    private:
        CStreamingDevice_wwise m_streamingDevice;
        CBlockingDevice_wwise m_blockingDevice;
        AkOSChar m_bankPath[AK_MAX_PATH];
        AkOSChar m_languageFolder[AK_MAX_PATH];
        bool m_useAsyncOpen;
    };
} // namespace Audio