/*
  Ticker.h - Ticker library that calls functions periodically

  Copyright (c) 2017 Bert Melis. All rights reserved.

  Based on the original work of:
  Copyright (c) 2014 Ivan Grokhotkov. All rights reserved.
  The original version is part of the esp8266 core for Arduino environment.

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library 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
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#pragma once

#include "drv_timer.h"
#include <functional>
#include <atomic>
#include <pthread.h>

namespace arduino {

class Ticker {
public:
    Ticker();
    ~Ticker();

    typedef void (*callback_with_arg_t)(void *);
    typedef std::function<void(void)> callback_function_t;

    void attach(float seconds, callback_function_t callback) {
        _callback_function = std::move(callback);
        _attach_us(1000000ULL * seconds, true, _static_callback, this);
    }

    void attach_ms(uint64_t milliseconds, callback_function_t callback) {
        _callback_function = std::move(callback);
        _attach_us(1000ULL * milliseconds, true, _static_callback, this);
    }

    template<typename TArg> void attach(float seconds, void (*callback)(TArg), TArg arg) {
        static_assert(sizeof(TArg) <= sizeof(void *), "attach() callback argument size must be <= sizeof(void*)");
        _attach_us(1000000ULL * seconds, true, reinterpret_cast<callback_with_arg_t>(callback), reinterpret_cast<void *>(arg));
    }

    template<typename TArg> void attach_ms(uint64_t milliseconds, void (*callback)(TArg), TArg arg) {
        static_assert(sizeof(TArg) <= sizeof(void *), "attach() callback argument size must be <= sizeof(void*)");
        _attach_us(1000ULL * milliseconds, true, reinterpret_cast<callback_with_arg_t>(callback), reinterpret_cast<void *>(arg));
    }

    void once(float seconds, callback_function_t callback) {
        _callback_function = std::move(callback);
        _attach_us(1000000ULL * seconds, false, _static_callback, this);
    }

    void once_ms(uint64_t milliseconds, callback_function_t callback) {
        _callback_function = std::move(callback);
        _attach_us(1000ULL * milliseconds, false, _static_callback, this);
    }

    template<typename TArg> void once(float seconds, void (*callback)(TArg), TArg arg) {
        static_assert(sizeof(TArg) <= sizeof(void *), "attach() callback argument size must be <= sizeof(void*)");
        _attach_us(1000000ULL * seconds, false, reinterpret_cast<callback_with_arg_t>(callback), reinterpret_cast<void *>(arg));
    }

    template<typename TArg> void once_ms(uint64_t milliseconds, void (*callback)(TArg), TArg arg) {
        static_assert(sizeof(TArg) <= sizeof(void *), "attach() callback argument size must be <= sizeof(void*)");
        _attach_us(1000ULL * milliseconds, false, reinterpret_cast<callback_with_arg_t>(callback), reinterpret_cast<void *>(arg));
    }

    void detach();
    bool active() const;

protected:
    static void _static_callback(void *arg);
    callback_function_t _callback_function = nullptr;

private:
    void _attach_us(uint64_t micros, bool repeat, callback_with_arg_t callback, void *arg);

    struct TickerNode {
        Ticker* ticker;
        callback_with_arg_t callback;
        void* arg;
        uint64_t period_us;          // 使用微秒存储，提高精度
        uint64_t start_tick;         // 起始tick
        uint64_t trigger_count;      // 已触发次数
        bool repeat;
        bool active;
        bool in_callback;
        bool pending_delete;
        TickerNode* next;
    };

    TickerNode* _node;

    static drv_soft_timer_inst_t* _shared_timer;
    static TickerNode* _ticker_list_head;
    static volatile uint64_t _tick_count;
    static volatile bool _timer_initialized;
    static std::atomic<int> _ticker_count;  // 活跃的 Ticker 实例计数
    static pthread_mutex_t _param_lock;
    static pthread_mutex_t _init_lock;

    static void _timer_callback(void* arg);
    static bool _init_shared_timer();
    static void _cleanup_shared_timer();
    static void _add_ticker(TickerNode* node);
    static void _remove_ticker(TickerNode* node);
};
}
