// 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.

package edu.wpi.first.wpilibj;

import edu.wpi.first.hal.ControlWord;
import edu.wpi.first.hal.DriverStationJNI;
import edu.wpi.first.util.WPIUtilJNI;
import java.util.LinkedHashSet;
import java.util.Set;

/**
 * The Motor Safety feature acts as a watchdog timer for an individual motor. It operates by
 * maintaining a timer that tracks how long it has been since the feed() method has been called for
 * that actuator. Code in the Driver Station class initiates a comparison of these timers to the
 * timeout values for any actuator with safety enabled every 5 received packets (100ms nominal).
 *
 * <p>The subclass should call feed() whenever the motor value is updated.
 */
public abstract class MotorSafety {
  private static final double kDefaultSafetyExpiration = 0.1;

  private double m_expiration = kDefaultSafetyExpiration;
  private boolean m_enabled;
  private double m_stopTime = Timer.getFPGATimestamp();
  private final Object m_thisMutex = new Object();
  private static final Set<MotorSafety> m_instanceList = new LinkedHashSet<>();
  private static final Object m_listMutex = new Object();
  private static Thread m_safetyThread;

  @SuppressWarnings("PMD.AssignmentInOperand")
  private static void threadMain() {
    int event = WPIUtilJNI.createEvent(false, false);
    DriverStationJNI.provideNewDataEventHandle(event);
    ControlWord controlWord = new ControlWord();

    int safetyCounter = 0;
    while (true) {
      boolean timedOut;
      try {
        timedOut = WPIUtilJNI.waitForObjectTimeout(event, 0.1);
      } catch (InterruptedException e) {
        DriverStationJNI.removeNewDataEventHandle(event);
        WPIUtilJNI.destroyEvent(event);
        Thread.currentThread().interrupt();
        return;
      }
      if (!timedOut) {
        DriverStationJNI.getControlWord(controlWord);
        if (!(controlWord.getEnabled() && controlWord.getDSAttached())) {
          safetyCounter = 0;
        }
        if (++safetyCounter >= 4) {
          checkMotors();
          safetyCounter = 0;
        }
      } else {
        safetyCounter = 0;
      }
    }
  }

  /** MotorSafety constructor. */
  @SuppressWarnings("this-escape")
  public MotorSafety() {
    synchronized (m_listMutex) {
      m_instanceList.add(this);
      if (m_safetyThread == null) {
        m_safetyThread = new Thread(MotorSafety::threadMain, "MotorSafety Thread");
        m_safetyThread.setDaemon(true);
        m_safetyThread.start();
      }
    }
  }

  /**
   * Feed the motor safety object.
   *
   * <p>Resets the timer on this object that is used to do the timeouts.
   */
  public void feed() {
    synchronized (m_thisMutex) {
      m_stopTime = Timer.getFPGATimestamp() + m_expiration;
    }
  }

  /**
   * Set the expiration time for the corresponding motor safety object.
   *
   * @param expirationTime The timeout value in seconds.
   */
  public void setExpiration(double expirationTime) {
    synchronized (m_thisMutex) {
      m_expiration = expirationTime;
    }
  }

  /**
   * Retrieve the timeout value for the corresponding motor safety object.
   *
   * @return the timeout value in seconds.
   */
  public double getExpiration() {
    synchronized (m_thisMutex) {
      return m_expiration;
    }
  }

  /**
   * Determine of the motor is still operating or has timed out.
   *
   * @return a true value if the motor is still operating normally and hasn't timed out.
   */
  public boolean isAlive() {
    synchronized (m_thisMutex) {
      return !m_enabled || m_stopTime > Timer.getFPGATimestamp();
    }
  }

  /**
   * Check if this motor has exceeded its timeout. This method is called periodically to determine
   * if this motor has exceeded its timeout value. If it has, the stop method is called, and the
   * motor is shut down until its value is updated again.
   */
  public void check() {
    boolean enabled;
    double stopTime;

    synchronized (m_thisMutex) {
      enabled = m_enabled;
      stopTime = m_stopTime;
    }

    if (!enabled || RobotState.isDisabled() || RobotState.isTest()) {
      return;
    }

    if (stopTime < Timer.getFPGATimestamp()) {
      DriverStation.reportError(
          getDescription()
              + "... Output not updated often enough. See https://docs.wpilib.org/motorsafety for more information.",
          false);

      stopMotor();
    }
  }

  /**
   * Enable/disable motor safety for this device.
   *
   * <p>Turn on and off the motor safety option for this PWM object.
   *
   * @param enabled True if motor safety is enforced for this object
   */
  public void setSafetyEnabled(boolean enabled) {
    synchronized (m_thisMutex) {
      m_enabled = enabled;
    }
  }

  /**
   * Return the state of the motor safety enabled flag.
   *
   * <p>Return if the motor safety is currently enabled for this device.
   *
   * @return True if motor safety is enforced for this device
   */
  public boolean isSafetyEnabled() {
    synchronized (m_thisMutex) {
      return m_enabled;
    }
  }

  /**
   * Check the motors to see if any have timed out.
   *
   * <p>This static method is called periodically to poll all the motors and stop any that have
   * timed out.
   */
  public static void checkMotors() {
    synchronized (m_listMutex) {
      for (MotorSafety elem : m_instanceList) {
        elem.check();
      }
    }
  }

  /** Called to stop the motor when the timeout expires. */
  public abstract void stopMotor();

  /**
   * Returns a description to print when an error occurs.
   *
   * @return Description to print when an error occurs.
   */
  public abstract String getDescription();
}