54 lines
1.0 KiB
Python
54 lines
1.0 KiB
Python
import numpy as np
|
|
|
|
|
|
class Pid:
|
|
def __init__(self):
|
|
# Integrator
|
|
self.yi = 0
|
|
|
|
# Differentiator
|
|
self.err = 0
|
|
|
|
# Auto windup
|
|
self.y_min = -1.0
|
|
self.y_max = 1.0
|
|
|
|
# Output
|
|
self.y = 0
|
|
|
|
@staticmethod
|
|
def params(kp, ki, kd, rho):
|
|
p = dict(kp=kp, ki=ki, kd=kd, rho=rho)
|
|
return p
|
|
|
|
def reset(self):
|
|
self.yi = 0
|
|
self.err = 0
|
|
|
|
def process(self, dt, params, err):
|
|
kp = params['kp']
|
|
ki = params['ki']
|
|
kd = params['kd']
|
|
rho = params['rho']
|
|
|
|
yi = rho*self.yi + ki*dt * err
|
|
yd = err - self.err
|
|
|
|
_yp = kp * err
|
|
_yi = yi
|
|
_yd = kd/dt * yd
|
|
|
|
# Reset yi on error sign changes
|
|
if np.sign(err) != np.sign(self.err):
|
|
_yi = 0
|
|
|
|
y = _yp + _yi + _yd
|
|
|
|
self.y = max(self.y_min, min(self.y_max, y))
|
|
print("y = {}".format(_yi))
|
|
self.yi = yi
|
|
self.err = err
|
|
|
|
def get_y(self):
|
|
return self.y
|