// This file is part of Background Music.
//
// Background Music is free software: you can redistribute it and/or
// modify it under the terms of the GNU General Public License as
// published by the Free Software Foundation, either version 2 of the
// License, or (at your option) any later version.
//
// Background Music 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Background Music. If not, see <http://www.gnu.org/licenses/>.

//
//  VC_TaskQueue.h
//  VCDriver
//
//  Copyright © 2016 Kyle Neideck
//

#ifndef __VCDriver__VC_TaskQueue__
#define __VCDriver__VC_TaskQueue__

// PublicUtility Includes
#include "CAPThread.h"
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wsign-conversion"
#include "CAAtomicStack.h"
#pragma clang diagnostic pop

// STL Includes
#include <functional>

// System Includes
#include <mach/semaphore.h>
#include <CoreAudio/AudioHardware.h>


#pragma clang assume_nonnull begin

//==================================================================================================
//	VC_TaskQueue
//
//  This class has two worker threads, one with real-time priority and one with default priority,
//  that tasks can be dispatched to. The two main use cases are dispatching work from a real-time
//  thread to be done async, and dispatching work from a non-real-time thread that needs to run on
//  a real-time thread to avoid priority inversions.
//==================================================================================================

class VC_TaskQueue
{
    
private:
    enum VC_TaskID {
        kVCTaskUninitialized,
        kVCTaskStopWorkerThread,

        // Non-realtime thread only
        kVCTaskSendPropertyNotification
    };
    
    class VC_Task
    {
    public:
                                        VC_Task(VC_TaskID inTaskID = kVCTaskUninitialized, bool inIsSync = false, UInt64 inArg1 = 0, UInt64 inArg2 = 0) : mNext(NULL), mTaskID(inTaskID), mIsSync(inIsSync), mArg1(inArg1), mArg2(inArg2) { };
        
        VC_TaskID                      GetTaskID() { return mTaskID; }
        
        // True if the thread that queued this task is blocking until the task is completed
        bool                            IsSync() { return mIsSync; }
        
        UInt64                          GetArg1() { return mArg1; }
        UInt64                          GetArg2() { return mArg2; }
        
        UInt64                          GetReturnValue() { return mReturnValue; }
        void                            SetReturnValue(UInt64 inReturnValue) { mReturnValue = inReturnValue; }
        
        bool                            IsComplete() { return mIsComplete; }
        void                            MarkCompleted() { mIsComplete = true; }
        
        // Used by TAtomicStack
        VC_Task* __nullable &          next() { return mNext; }
        VC_Task* __nullable            mNext;
        
    private:
        VC_TaskID                      mTaskID;
        bool                            mIsSync;
        UInt64                          mArg1;
        UInt64                          mArg2;
        UInt64                          mReturnValue = INT64_MAX;
        bool                            mIsComplete = false;
    };
    
public:
                                        VC_TaskQueue();
                                        ~VC_TaskQueue();
                                        // Disallow copying
                                        VC_TaskQueue(const VC_TaskQueue&) = delete;
                                        VC_TaskQueue& operator=(const VC_TaskQueue&) = delete;
    
private:
    static UInt32                       NanosToAbsoluteTime(UInt32 inNanos);
    
public:
    // Sends a property changed notification to the VCDevice host. Assumes the scope and element are kAudioObjectPropertyScopeGlobal and
    // kAudioObjectPropertyElementMaster because currently those are the only ones we use.
    void                                QueueAsync_SendPropertyNotification(AudioObjectPropertySelector inProperty, AudioObjectID inDeviceID);

private:
    UInt64                              QueueSync(VC_TaskID inTaskID, bool inRunOnRealtimeThread, UInt64 inTaskArg1 = 0, UInt64 inTaskArg2 = 0);
    
    void                                QueueOnNonRealtimeThread(VC_Task inTask);
    
public:
    void                                AssertCurrentThreadIsRTWorkerThread(const char* inCallerMethodName);
    
private:
    static void* __nullable             RealTimeThreadProc(void* inRefCon);
    static void* __nullable             NonRealTimeThreadProc(void* inRefCon);
    
    void                                WorkerThreadProc(semaphore_t inWorkQueuedSemaphore, semaphore_t inSyncTaskCompletedSemaphore, TAtomicStack<VC_Task>* inTasks, TAtomicStack2<VC_Task>* __nullable inFreeList, std::function<bool(VC_Task*)> inProcessTask);
    
    // These return true when the thread should be stopped
    bool                                ProcessRealTimeThreadTask(VC_Task* inTask);
    bool                                ProcessNonRealTimeThreadTask(VC_Task* inTask);
    
private:
    // The worker threads that perform the queued tasks
    CAPThread                           mRealTimeThread;
    CAPThread                           mNonRealTimeThread;
    
    // The approximate amount of time we'll need whenever our real-time thread is scheduled. This is currently just
    // set to the minimum (see sched_prim.c) because our real-time tasks do very little work.
    //
    // TODO: Would it be better to specify these in absolute time, which would make them relative to the system's bus
    //       speed? Or even calculate them from the system's CPU/RAM speed? Note that none of our tasks actually have
    //       a deadline (though that might change). They just have to run with real-time priority to avoid causing
    //       priority inversions on the IO thread.
    static const UInt32                 kRealTimeThreadNominalComputationNs = 50 * NSEC_PER_USEC;
    // The maximum amount of time the real-time thread can take to finish its computation after being scheduled.
    static const UInt32                 kRealTimeThreadMaximumComputationNs = 60 * NSEC_PER_USEC;
    
    // We use Mach semaphores for communication with the worker threads because signalling them is real-time safe.
    
    // Signalled to tell the worker threads when there are tasks for them process.
    semaphore_t                         mRealTimeThreadWorkQueuedSemaphore;
    semaphore_t                         mNonRealTimeThreadWorkQueuedSemaphore;
    // Signalled when a worker thread completes a task, if the thread that queued that task is blocking on it.
    semaphore_t                         mRealTimeThreadSyncTaskCompletedSemaphore;
    semaphore_t                         mNonRealTimeThreadSyncTaskCompletedSemaphore;
    
    // When a task is queued we add it to one of these, depending on which worker thread it will run on. Using
    // TAtomicStack lets us safely add and remove tasks on real-time threads.
    //
    // We use TAtomicStack rather than TAtomicStack2 because we need pop_all_reversed() to make sure we process the
    // tasks in order. (It might have been better to use OSAtomicFifoEnqueue/OSAtomicFifoDequeue, but I only
    // recently found out about them.)
    TAtomicStack<VC_Task>              mRealTimeThreadTasks;
    TAtomicStack<VC_Task>              mNonRealTimeThreadTasks;
    
    // The number of tasks to pre-allocate and add to the non-realtime task free list. Should be large enough that
    // the free list is never emptied. (At least not while IO could be running.)
    static const UInt32                 kNonRealTimeThreadTaskBufferSize = 512;
    // Realtime threads can't safely allocate memory, so when they queue a task the memory for it comes from this
    // free list. We pre-allocate as many tasks as they should ever need in the constructor. (But if the free list
    // runs out of tasks somehow the realtime thread will allocate a new one.)
    //
    // There's a similar free list used in Apple's CAThreadSafeList.h.
    //
    // We can use TAtomicStack2 instead of TAtomicStack because we never call pop_all on the free list.
    TAtomicStack2<VC_Task>             mNonRealTimeThreadTasksFreeList;
    
};

#pragma clang assume_nonnull end

#endif /* __VCDriver__VC_TaskQueue__ */