/* ========================================

  Copyright (C) 2024-2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.

======================================== */

#ifndef __KERNEL_KIT_TIMER_H__
#define __KERNEL_KIT_TIMER_H__

#include <ArchKit/ArchKit.h>
#include <KernelKit/KPC.h>

namespace Kernel {
class SoftwareTimer;
class ITimer;

inline constexpr Int16 kTimeUnit = 1000;

class ITimer {
 public:
  /// @brief Default constructor
  explicit ITimer() = default;
  virtual ~ITimer() = default;

 public:
  NE_COPY_DEFAULT(ITimer)

 public:
  virtual BOOL Wait();
};

class SoftwareTimer final : public ITimer {
 public:
  explicit SoftwareTimer(Int64 seconds);
  ~SoftwareTimer() override;

 public:
  NE_COPY_DEFAULT(SoftwareTimer)

 public:
  BOOL Wait() override;

 private:
  UIntPtr* fDigitalTimer{nullptr};
  Int64    fWaitFor{0};
};

class HardwareTimer final : public ITimer {
 public:
  explicit HardwareTimer(UInt64 seconds);
  ~HardwareTimer() override;

 public:
  NE_COPY_DEFAULT(HardwareTimer)

 public:
  BOOL Wait() override;

 private:
  volatile UInt8* fDigitalTimer{nullptr};
  Int64           fWaitFor{0};
};

inline UInt64 rtl_microseconds(UInt64 time) {
  if (time < 1) return 0;

  // TODO: nanoseconds maybe?
  return time / kTimeUnit;
}

inline UInt64 rtl_milliseconds(UInt64 time) {
  if (time < 1) return 0;

  return time;
}
}  // namespace Kernel

#endif  // !__KERNEL_KIT_TIMER_H__