Run yapf on all python files in the repo
Signed-off-by: Ravago Jones <ravagojones@gmail.com>
Change-Id: I221e04c3f517fab8535b22551553799e0fee7a80
diff --git a/y2022/control_loops/python/catapult.py b/y2022/control_loops/python/catapult.py
index 9e9adf7..e984478 100755
--- a/y2022/control_loops/python/catapult.py
+++ b/y2022/control_loops/python/catapult.py
@@ -40,7 +40,6 @@
M_cup = (1750 * 0.0254 * 0.04 * 2 * math.pi * (ball_diameter / 2.)**2.0)
J_cup = M_cup * lever**2.0 + M_cup * (ball_diameter / 2.)**2.0
-
J = (0.0 * J_ball + J_bar + J_cup * 0.0)
JEmpty = (J_bar + J_cup * 0.0)
@@ -83,6 +82,7 @@
# We really want our cost function to be robust so that we can tolerate the
# battery not delivering like we want at the end.
+
def quadratic_cost(catapult, X_initial, X_final, horizon):
Q_final = numpy.matrix([[10000.0, 0.0], [0.0, 10000.0]])
@@ -98,28 +98,32 @@
P_final = 2.0 * Bf.transpose() * Q_final * Bf
q_final = (2.0 * (Af * X_initial - X_final).transpose() * Q_final *
- Bf).transpose()
+ Bf).transpose()
constant_final = (Af * X_initial - X_final).transpose() * Q_final * (
Af * X_initial - X_final)
- m = numpy.matrix([[catapult.A[1, 1] ** (n + 1)] for n in range(horizon)])
+ m = numpy.matrix([[catapult.A[1, 1]**(n + 1)] for n in range(horizon)])
M = Bs[1:horizon * 2:2, :]
- W = numpy.matrix(numpy.identity(horizon) - numpy.eye(horizon, horizon, -1)) / catapult.dt
+ W = numpy.matrix(
+ numpy.identity(horizon) -
+ numpy.eye(horizon, horizon, -1)) / catapult.dt
w = -numpy.matrix(numpy.eye(horizon, 1, 0)) / catapult.dt
-
Pi = numpy.diag([
- (0.01 ** 2.0) + (0.02 * max(0.0, 20 - (horizon - n)) / 20.0) ** 2.0 for n in range(horizon)
+ (0.01**2.0) + (0.02 * max(0.0, 20 - (horizon - n)) / 20.0)**2.0
+ for n in range(horizon)
])
P_accel = 2.0 * M.transpose() * W.transpose() * Pi * W * M
- q_accel = 2.0 * (((W * m + w) * X_initial[1, 0]).transpose() * Pi * W * M).transpose()
+ q_accel = 2.0 * ((
+ (W * m + w) * X_initial[1, 0]).transpose() * Pi * W * M).transpose()
constant_accel = ((W * m + w) * X_initial[1, 0]).transpose() * Pi * (
(W * m + w) * X_initial[1, 0])
- return ((P_accel + P_final), (q_accel + q_final), (constant_accel + constant_final))
+ return ((P_accel + P_final), (q_accel + q_final),
+ (constant_accel + constant_final))
def new_cost(catapult, X_initial, X_final, u):
@@ -138,25 +142,28 @@
P_final = 2.0 * Bf.transpose() * Q_final * Bf
q_final = (2.0 * (Af * X_initial - X_final).transpose() * Q_final *
- Bf).transpose()
+ Bf).transpose()
constant_final = (Af * X_initial - X_final).transpose() * Q_final * (
Af * X_initial - X_final)
- m = numpy.matrix([[catapult.A[1, 1] ** (n + 1)] for n in range(len(u))])
+ m = numpy.matrix([[catapult.A[1, 1]**(n + 1)] for n in range(len(u))])
M = Bs[1:len(u) * 2:2, :]
- W = numpy.matrix(numpy.identity(len(u)) - numpy.eye(len(u), len(u), -1)) / catapult.dt
+ W = numpy.matrix(numpy.identity(len(u)) -
+ numpy.eye(len(u), len(u), -1)) / catapult.dt
w = -numpy.matrix(numpy.eye(len(u), 1, 0)) * X_initial[1, 0] / catapult.dt
accel = W * (M * u_matrix + m * X_initial[1, 0]) + w
Pi = numpy.diag([
- (0.01 ** 2.0) + (0.02 * max(0.0, 20 - (len(u) - n)) / 20.0) ** 2.0 for n in range(len(u))
+ (0.01**2.0) + (0.02 * max(0.0, 20 - (len(u) - n)) / 20.0)**2.0
+ for n in range(len(u))
])
P_accel = 2.0 * M.transpose() * W.transpose() * Pi * W * M
- q_accel = 2.0 * ((W * m * X_initial[1, 0] + w).transpose() * Pi * W * M).transpose()
+ q_accel = 2.0 * (
+ (W * m * X_initial[1, 0] + w).transpose() * Pi * W * M).transpose()
constant_accel = (W * m * X_initial[1, 0] +
w).transpose() * Pi * (W * m * X_initial[1, 0] + w)
@@ -205,6 +212,7 @@
def SolveCatapult(catapult, X_initial, X_final, u):
""" Solves for the optimal action given a seed, state, and target """
+
def vbat_constraint(z, i):
return 12.0 - z[i]
@@ -213,13 +221,11 @@
P, q, c = quadratic_cost(catapult, X_initial, X_final, len(u))
-
def mpc_cost2(u_solver):
u_matrix = numpy.matrix(u_solver).transpose()
cost = mpc_cost(catapult, X_initial, X_final, u_solver)
return cost
-
def mpc_cost3(u_solver):
u_matrix = numpy.matrix(u_solver).transpose()
return (0.5 * u_matrix.transpose() * P * u_matrix +
@@ -268,7 +274,6 @@
X_initial = numpy.matrix([[0.0], [0.0]])
X_final = numpy.matrix([[2.0], [25.0]])
-
X_initial = numpy.matrix([[0.0], [0.0]])
X = X_initial.copy()
@@ -382,7 +387,8 @@
)
else:
namespaces = ['y2022', 'control_loops', 'superstructure', 'catapult']
- catapult_lib.WriteCatapult([kCatapultWithBall, kCatapultEmpty], argv[1:3], argv[3:5], namespaces)
+ catapult_lib.WriteCatapult([kCatapultWithBall, kCatapultEmpty],
+ argv[1:3], argv[3:5], namespaces)
return 0
diff --git a/y2022/control_loops/python/catapult_lib.py b/y2022/control_loops/python/catapult_lib.py
index 4a88b14..eceb2d2 100644
--- a/y2022/control_loops/python/catapult_lib.py
+++ b/y2022/control_loops/python/catapult_lib.py
@@ -13,6 +13,7 @@
# TODO(austin): This is mostly the same as angular_system. Can we either wrap an angular_system or assign it?
class Catapult(angular_system.AngularSystem):
+
def __init__(self, params, name="Catapult"):
super(Catapult, self).__init__(params, name)
# Signal that we have a 2 cycle output delay to compensate for in
@@ -23,6 +24,7 @@
class IntegralCatapult(angular_system.IntegralAngularSystem):
+
def __init__(self, params, name="IntegralCatapult"):
super(IntegralCatapult, self).__init__(params, name=name)
# Signal that we have a 2 cycle output delay to compensate for in
diff --git a/y2022/control_loops/python/climber.py b/y2022/control_loops/python/climber.py
index 5f5ae38..12d9cd0 100755
--- a/y2022/control_loops/python/climber.py
+++ b/y2022/control_loops/python/climber.py
@@ -27,12 +27,15 @@
kalman_q_voltage=35.0,
kalman_r_position=0.05)
+
def main(argv):
if FLAGS.plot:
R = numpy.matrix([[0.2], [0.0]])
linear_system.PlotKick(kClimber, R, plant_params=kClimber)
- linear_system.PlotMotion(
- kClimber, R, max_velocity=5.0, plant_params=kClimber)
+ linear_system.PlotMotion(kClimber,
+ R,
+ max_velocity=5.0,
+ plant_params=kClimber)
# Write the generated constants out to a file.
if len(argv) != 5:
@@ -41,8 +44,8 @@
)
else:
namespaces = ['y2022', 'control_loops', 'superstructure', 'climber']
- linear_system.WriteLinearSystem(kClimber,
- argv[1:3], argv[3:5], namespaces)
+ linear_system.WriteLinearSystem(kClimber, argv[1:3], argv[3:5],
+ namespaces)
if __name__ == '__main__':
diff --git a/y2022/control_loops/python/polydrivetrain.py b/y2022/control_loops/python/polydrivetrain.py
index d749580..cf54470 100644
--- a/y2022/control_loops/python/polydrivetrain.py
+++ b/y2022/control_loops/python/polydrivetrain.py
@@ -12,20 +12,22 @@
FLAGS = gflags.FLAGS
try:
- gflags.DEFINE_bool('plot', False, 'If true, plot the loop response.')
+ gflags.DEFINE_bool('plot', False, 'If true, plot the loop response.')
except gflags.DuplicateFlagError:
- pass
+ pass
+
def main(argv):
- if FLAGS.plot:
- polydrivetrain.PlotPolyDrivetrainMotions(drivetrain.kDrivetrain)
- elif len(argv) != 7:
- glog.fatal('Expected .h file name and .cc file name')
- else:
- polydrivetrain.WritePolyDrivetrain(argv[1:3], argv[3:5], argv[5:7], 'y2022',
- drivetrain.kDrivetrain)
+ if FLAGS.plot:
+ polydrivetrain.PlotPolyDrivetrainMotions(drivetrain.kDrivetrain)
+ elif len(argv) != 7:
+ glog.fatal('Expected .h file name and .cc file name')
+ else:
+ polydrivetrain.WritePolyDrivetrain(argv[1:3], argv[3:5], argv[5:7],
+ 'y2022', drivetrain.kDrivetrain)
+
if __name__ == '__main__':
- argv = FLAGS(sys.argv)
- glog.init()
- sys.exit(main(argv))
+ argv = FLAGS(sys.argv)
+ glog.init()
+ sys.exit(main(argv))