/* XMRig
 * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
 * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
 * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
 * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
 * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
 * Copyright 2017-2018 XMR-Stak    <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
 * Copyright 2018-2019 SChernykh   <https://github.com/SChernykh>
 * Copyright 2016-2019 XMRig       <https://github.com/xmrig>, <support@xmrig.com>
 *
 *   This program 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 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 General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef XMRIG_WORKERS_H
#define XMRIG_WORKERS_H


#include <atomic>
#include <list>
#include <uv.h>
#include <vector>

#ifdef XMRIG_ALGO_RANDOMX
#   include <randomx.h>
#endif

#include "base/net/stratum/Job.h"
#include "net/JobResult.h"
#include "rapidjson/fwd.h"


class Hashrate;
class IWorker;
class ThreadHandle;


namespace xmrig {
    class Controller;
    class IJobResultListener;
}


class Workers
{
public:
    static xmrig::Job job();
    static size_t hugePages();
    static size_t threads();
    static void printHashrate(bool detail);
    static void setEnabled(bool enabled);
    static void setJob(const xmrig::Job &job, bool donate);
    static void start(xmrig::Controller *controller);
    static void stop();
    static void submit(const xmrig::JobResult &result);

    static inline bool isEnabled()                                      { return m_enabled; }
    static inline bool isOutdated(uint64_t sequence)                    { return m_sequence.load(std::memory_order_relaxed) != sequence; }
    static inline bool isPaused()                                       { return m_paused.load(std::memory_order_relaxed) == 1; }
    static inline Hashrate *hashrate()                                  { return m_hashrate; }
    static inline uint64_t sequence()                                   { return m_sequence.load(std::memory_order_relaxed); }
    static inline void pause()                                          { m_active = false; m_paused = 1; m_sequence++; }
    static inline void setListener(xmrig::IJobResultListener *listener) { m_listener = listener; }

#   ifdef XMRIG_FEATURE_API
    static void threadsSummary(rapidjson::Document &doc);
#   endif

#   ifdef XMRIG_ALGO_RANDOMX
    static void updateDataset(const uint8_t* seed_hash, uint32_t num_threads);
    static randomx_dataset* getDataset();
#   endif

private:
    static void onReady(void *arg);
    static void onResult(uv_async_t *handle);
    static void onTick(uv_timer_t *handle);
    static void start(IWorker *worker);

    class LaunchStatus
    {
    public:
        inline LaunchStatus() :
            hugePages(0),
            pages(0),
            started(0),
            threads(0),
            ways(0),
            algo(xmrig::CRYPTONIGHT),
            variant(xmrig::VARIANT_AUTO)
        {}

        size_t hugePages;
        size_t pages;
        size_t started;
        size_t threads;
        size_t ways;
        xmrig::Algo algo;
        xmrig::Variant variant;
    };

    static bool m_active;
    static bool m_enabled;
    static Hashrate *m_hashrate;
    static xmrig::IJobResultListener *m_listener;
    static xmrig::Job m_job;
    static LaunchStatus m_status;
    static std::atomic<int> m_paused;
    static std::atomic<uint64_t> m_sequence;
    static std::list<xmrig::JobResult> m_queue;
    static std::vector<ThreadHandle*> m_workers;
    static uint64_t m_ticks;
    static uv_async_t *m_async;
    static uv_mutex_t m_mutex;
    static uv_rwlock_t m_rwlock;
    static uv_timer_t *m_timer;
    static xmrig::Controller *m_controller;

#   ifdef XMRIG_ALGO_RANDOMX
    static uv_rwlock_t m_rx_dataset_lock;
    static randomx_cache *m_rx_cache;
    static randomx_dataset *m_rx_dataset;
    static uint8_t m_rx_seed_hash[32];
    static std::atomic<uint32_t> m_rx_dataset_init_thread_counter;
#   endif
};


#endif /* XMRIG_WORKERS_H */