// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.

#pragma once

#include <functional>
#include <memory>

#include <hal/Interrupts.h>
#include <units/time.h>

#include "frc/AnalogTriggerType.h"

namespace frc {

class InterruptableSensorBase {
 public:
  enum WaitResult {
    kTimeout = 0x0,
    kRisingEdge = 0x1,
    kFallingEdge = 0x100,
    kBoth = 0x101,
  };

  /**
   * Handler for interrupts.
   *
   * First parameter is if rising, 2nd is if falling.
   */
  using InterruptEventHandler = std::function<void(WaitResult)>;

  InterruptableSensorBase() = default;

  /**
   * Free the resources for an interrupt event.
   */
  virtual ~InterruptableSensorBase();

  InterruptableSensorBase(InterruptableSensorBase&&) = default;
  InterruptableSensorBase& operator=(InterruptableSensorBase&&) = default;

  virtual HAL_Handle GetPortHandleForRouting() const = 0;
  virtual AnalogTriggerType GetAnalogTriggerTypeForRouting() const = 0;

  /**
   * Request one of the 8 interrupts asynchronously on this digital input.
   *
   * Request interrupts in asynchronous mode where the user's interrupt handler
   * will be called when the interrupt fires. Users that want control over the
   * thread priority should use the synchronous method with their own spawned
   * thread. The default is interrupt on rising edges only.
   */
  virtual void RequestInterrupts(HAL_InterruptHandlerFunction handler,
                                 void* param);

  /**
   * Request one of the 8 interrupts asynchronously on this digital input.
   *
   * Request interrupts in asynchronous mode where the user's interrupt handler
   * will be called when the interrupt fires. Users that want control over the
   * thread priority should use the synchronous method with their own spawned
   * thread. The default is interrupt on rising edges only.
   */
  virtual void RequestInterrupts(InterruptEventHandler handler);

  /**
   * Request one of the 8 interrupts synchronously on this digital input.
   *
   * Request interrupts in synchronous mode where the user program will have to
   * explicitly wait for the interrupt to occur using WaitForInterrupt.
   * The default is interrupt on rising edges only.
   */
  virtual void RequestInterrupts();

  /**
   * Cancel interrupts on this device.
   *
   * This deallocates all the chipobject structures and disables any interrupts.
   */
  virtual void CancelInterrupts();

  /**
   * In synchronous mode, wait for the defined interrupt to occur.
   *
   * You should <b>NOT</b> attempt to read the sensor from another thread while
   * waiting for an interrupt. This is not threadsafe, and can cause memory
   * corruption
   *
   * @param timeout        Timeout
   * @param ignorePrevious If true, ignore interrupts that happened before
   *                       WaitForInterrupt was called.
   * @return What interrupts fired
   */
  virtual WaitResult WaitForInterrupt(units::second_t timeout,
                                      bool ignorePrevious = true);

  /**
   * Enable interrupts to occur on this input.
   *
   * Interrupts are disabled when the RequestInterrupt call is made. This gives
   * time to do the setup of the other options before starting to field
   * interrupts.
   */
  virtual void EnableInterrupts();

  /**
   * Disable Interrupts without without deallocating structures.
   */
  virtual void DisableInterrupts();

  /**
   * Return the timestamp for the rising interrupt that occurred most recently.
   *
   * This is in the same time domain as GetClock(). The rising-edge interrupt
   * should be enabled with SetUpSourceEdge().
   *
   * @return Timestamp in seconds since boot.
   */
  virtual units::second_t ReadRisingTimestamp();

  /**
   * Return the timestamp for the falling interrupt that occurred most recently.
   *
   * This is in the same time domain as GetClock().
   * The falling-edge interrupt should be enabled with
   * {@link #DigitalInput.SetUpSourceEdge}
   *
   * @return Timestamp in seconds since boot.
   */
  virtual units::second_t ReadFallingTimestamp();

  /**
   * Set which edge to trigger interrupts on
   *
   * @param risingEdge  true to interrupt on rising edge
   * @param fallingEdge true to interrupt on falling edge
   */
  virtual void SetUpSourceEdge(bool risingEdge, bool fallingEdge);

 protected:
  hal::Handle<HAL_InterruptHandle> m_interrupt;
  std::unique_ptr<InterruptEventHandler> m_interruptHandler{nullptr};

  void AllocateInterrupts(bool watcher);
};

}  // namespace frc