import numpy as np


class Pid:
    def __init__(self, params):
        self.params = params

        # Integrator
        self.yi = 0

        # Differentiator
        self.err = 0

        # Anti windup
        self.y_min = -1.0
        self.y_max = 1.0
        self.awu = 0

        # Output
        self.y = 0

    @staticmethod
    def params(kp, ki, kd, kt):
        p = dict(kp=kp, ki=ki, kd=kd, kt=kt)
        return p

    def reset(self):
        self.yi = 0
        self.err = 0
        self.awu = 0

    def process(self, err):
        dt = self.params['dt']
        kp = self.params['kp']
        ki = self.params['ki']
        kd = self.params['kd']
        kt = self.params['kt']

        self.yi = self.yi + ki*dt * err + kt*dt * self.awu
        yd = kd/dt*(err - self.err)
        yp = kp * err

        self.err = err

        y = yp + self.yi + yd

        self.y = max(self.y_min, min(self.y_max, y))

        # Anti windup
        self.awu = self.y - y

    def get_y(self):
        return self.y