Add 2019 field target locations
Change-Id: Ie3ac15f4de3f4f2f97b88a7160c3a02d65d41740
diff --git a/y2019/BUILD b/y2019/BUILD
index ab4a842..9896b78 100644
--- a/y2019/BUILD
+++ b/y2019/BUILD
@@ -24,7 +24,9 @@
"//aos/mutex",
"//aos/network:team_number",
"//frc971:constants",
+ "//frc971/control_loops:pose",
"//frc971/control_loops:static_zeroing_single_dof_profiled_subsystem",
+ "//y2019/control_loops/drivetrain:camera",
"//y2019/control_loops/drivetrain:polydrivetrain_plants",
"//y2019/control_loops/superstructure/elevator:elevator_plants",
"//y2019/control_loops/superstructure/intake:intake_plants",
diff --git a/y2019/constants.cc b/y2019/constants.cc
index a30d13f..791a6c4 100644
--- a/y2019/constants.cc
+++ b/y2019/constants.cc
@@ -30,6 +30,16 @@
const uint16_t kPracticeTeamNumber = 9971;
const uint16_t kCodingRobotTeamNumber = 7971;
+constexpr double FeetToMeters(double ft) {
+ return 0.3048 * ft;
+}
+constexpr double InchToMeters(double in) {
+ return 0.0254 * in;
+}
+constexpr double DegToRad(double deg) {
+ return deg * M_PI / 180.0;
+}
+
const Values *DoGetValuesForTeam(uint16_t team) {
Values *const r = new Values();
Values::PotAndAbsConstants *const elevator = &r->elevator;
@@ -210,5 +220,154 @@
return *values[team_number];
}
+constexpr size_t Field::kNumTargets;
+constexpr size_t Field::kNumObstacles;
+
+Field::Field() {
+ // TODO(james): These values need to re-verified. I got them by skimming the
+ // manual and they all seem to be pretty much correct.
+ //
+ // Note: Per //frc971/control_loops/pose.h, coordinate system is:
+ // -In meters
+ // -Origin at center of our driver's station wall
+ // -Positive X-axis pointing straight out from driver's station
+ // -Positive Y-axis pointing straight left from the driver's perspective
+ // -Positive Z-axis is straight up.
+ // -The angle of the target is such that the angle is the angle you would
+ // need to be facing to see it straight-on. I.e., if the target angle is
+ // pi / 2.0, then you would be able to see it face on by facing straight
+ // left from the driver's point of view (if you were standing in the right
+ // spot).
+ constexpr double kCenterFieldX = FeetToMeters(27.0) + InchToMeters(1.125);
+
+ constexpr double kFarSideCargoBayX =
+ kCenterFieldX - InchToMeters(20.875);
+ constexpr double kMidSideCargoBayX = kFarSideCargoBayX - InchToMeters(21.75);
+ constexpr double kNearSideCargoBayX = kMidSideCargoBayX - InchToMeters(21.75);
+ constexpr double kSideCargoBayY = InchToMeters(24 + 3 + 0.875);
+ constexpr double kSideCargoBayTheta = -M_PI_2;
+
+ constexpr double kFaceCargoBayX =
+ kCenterFieldX - InchToMeters(7 * 12 + 11.75 + 9);
+ constexpr double kFaceCargoBayY = InchToMeters(10.875);
+ constexpr double kFaceCargoBayTheta = 0.0;
+
+ constexpr double kRocketX = kCenterFieldX - FeetToMeters(8);
+ constexpr double kRocketY = InchToMeters((26 * 12 + 10.5) / 2.0);
+
+ constexpr double kRocketPortX = kRocketX;
+ constexpr double kRocketPortY = kRocketY - 0.70;
+ constexpr double kRocketPortTheta = M_PI_2;
+
+ // Half of portal + guess at width * cos(61.5 deg)
+ const double kRocketHatchXOffset = InchToMeters(14.634);
+ const double kRocketHatchY = kRocketPortY + InchToMeters(9.326);
+ const double kRocketNearX = kRocketX - kRocketHatchXOffset;
+ const double kRocketFarX = kRocketX + kRocketHatchXOffset;
+ constexpr double kRocketNearTheta = DegToRad(28.5);
+ constexpr double kRocketFarTheta = M_PI - kRocketNearTheta;
+
+ constexpr double kHpSlotY = InchToMeters((26 * 12 + 10.5) / 2.0 - 25.9);
+ constexpr double kHpSlotTheta = M_PI;
+
+ constexpr double kNormalZ = 0.80;
+ constexpr double kPortZ = 0.99;
+
+ const Pose far_side_cargo_bay({kFarSideCargoBayX, kSideCargoBayY, kNormalZ},
+ kSideCargoBayTheta);
+ const Pose mid_side_cargo_bay({kMidSideCargoBayX, kSideCargoBayY, kNormalZ},
+ kSideCargoBayTheta);
+ const Pose near_side_cargo_bay({kNearSideCargoBayX, kSideCargoBayY, kNormalZ},
+ kSideCargoBayTheta);
+
+ const Pose face_cargo_bay({kFaceCargoBayX, kFaceCargoBayY, kNormalZ},
+ kFaceCargoBayTheta);
+
+ const Pose rocket_port({kRocketPortX, kRocketPortY, kPortZ},
+ kRocketPortTheta);
+
+ const Pose rocket_near({kRocketNearX, kRocketHatchY, kNormalZ},
+ kRocketNearTheta);
+ const Pose rocket_far({kRocketFarX, kRocketHatchY, kNormalZ},
+ kRocketFarTheta);
+
+ const Pose hp_slot({0.0, kHpSlotY, kNormalZ}, kHpSlotTheta);
+
+ const ::std::array<Pose, 8> quarter_field_targets{
+ {far_side_cargo_bay, mid_side_cargo_bay, near_side_cargo_bay,
+ face_cargo_bay, rocket_port, rocket_near, rocket_far, hp_slot}};
+
+ // Mirror across center field mid-line (short field axis):
+ ::std::array<Pose, 16> half_field_targets;
+ ::std::copy(quarter_field_targets.begin(), quarter_field_targets.end(),
+ half_field_targets.begin());
+ for (int ii = 0; ii < 8; ++ii) {
+ const int jj = ii + 8;
+ half_field_targets[jj] = quarter_field_targets[ii];
+ half_field_targets[jj].mutable_pos()->x() =
+ 2.0 * kCenterFieldX - half_field_targets[jj].rel_pos().x();
+ half_field_targets[jj].set_theta(
+ aos::math::NormalizeAngle(M_PI - half_field_targets[jj].rel_theta()));
+ }
+
+ ::std::array<Pose, 32> target_poses_;
+
+ // Mirror across x-axis (long field axis):
+ ::std::copy(half_field_targets.begin(), half_field_targets.end(),
+ target_poses_.begin());
+ for (int ii = 0; ii < 16; ++ii) {
+ const int jj = ii + 16;
+ target_poses_[jj] = half_field_targets[ii];
+ target_poses_[jj].mutable_pos()->y() *= -1;
+ target_poses_[jj].set_theta(-target_poses_[jj].rel_theta());
+ }
+ for (int ii = 0; ii < 32; ++ii) {
+ targets_[ii] = {target_poses_[ii]};
+ }
+
+ // Define rocket obstacles as just being a single line that should block any
+ // cameras trying to see through the rocket up and down the field.
+ // This line is parallel to the driver's station wall and extends behind
+ // the portal.
+ Obstacle rocket_obstacle({{kRocketPortX, kRocketY, 0.0}, 0.0},
+ {{kRocketPortX, kRocketPortY + 0.01, 0.0}, 0.0});
+ // First, we mirror rocket obstacles across x-axis:
+ Obstacle rocket_obstacle2({{kRocketPortX, -kRocketY, 0.0}, 0.0},
+ {{kRocketPortX, -kRocketPortY - 0.01, 0.0}, 0.0});
+
+ // Define an obstacle for the Hab that extends striaght out a few feet from
+ // the driver's station wall.
+ // TODO(james): Does this actually block our view?
+ const double kHabL3X = FeetToMeters(4.0);
+ Obstacle hab_obstacle({}, {{kHabL3X, 0.0, 0.0}, 0.0});
+ ::std::array<Obstacle, 3> half_obstacles{
+ {rocket_obstacle, rocket_obstacle2, hab_obstacle}};
+ ::std::copy(half_obstacles.begin(), half_obstacles.end(), obstacles_.begin());
+
+ // Next, we mirror across the mid-line (short axis) to duplicate the
+ // rockets and hab to opposite side of the field.
+ for (int ii = 0; ii < 3; ++ii) {
+ const int jj = ii + 3;
+ obstacles_[jj] = half_obstacles[ii];
+ obstacles_[jj].mutable_pose1()->mutable_pos()->x() =
+ 2.0 * kCenterFieldX - obstacles_[jj].mutable_pose1()->rel_pos().x();
+ obstacles_[jj].mutable_pose2()->mutable_pos()->x() =
+ 2.0 * kCenterFieldX - obstacles_[jj].mutable_pose2()->rel_pos().x();
+ }
+
+ // Finally, define a rectangular cargo ship.
+ const double kCargoCornerX = kFaceCargoBayX + 0.1;
+ const double kCargoCornerY = kSideCargoBayY - 0.1;
+ ::std::array<Pose, 4> cargo_corners{
+ {{{kCargoCornerX, kCargoCornerY, 0.0}, 0.0},
+ {{kCargoCornerX, -kCargoCornerY, 0.0}, 0.0},
+ {{2.0 * kCenterFieldX - kCargoCornerX, -kCargoCornerY, 0.0}, 0.0},
+ {{2.0 * kCenterFieldX - kCargoCornerX, kCargoCornerY, 0.0}, 0.0}}};
+ for (int ii = 6; ii < 10; ++ii) {
+ obstacles_[ii] = Obstacle(cargo_corners[ii % cargo_corners.size()],
+ cargo_corners[(ii + 1) % cargo_corners.size()]);
+ }
+}
+
} // namespace constants
} // namespace y2019
diff --git a/y2019/constants.h b/y2019/constants.h
index 46d28f1..24d8b92 100644
--- a/y2019/constants.h
+++ b/y2019/constants.h
@@ -1,6 +1,7 @@
#ifndef Y2019_CONSTANTS_H_
#define Y2019_CONSTANTS_H_
+#include <array>
#include <math.h>
#include <stdint.h>
@@ -11,6 +12,8 @@
#include "y2019/control_loops/superstructure/intake/intake_plant.h"
#include "y2019/control_loops/superstructure/stilts/stilts_plant.h"
#include "y2019/control_loops/superstructure/wrist/wrist_plant.h"
+#include "y2019/control_loops/drivetrain/camera.h"
+#include "frc971/control_loops/pose.h"
namespace y2019 {
namespace constants {
@@ -25,6 +28,33 @@
//
// All ratios are from the encoder shaft to the output units.
+
+class Field {
+ public:
+ typedef ::frc971::control_loops::TypedPose<double> Pose;
+ typedef ::y2019::control_loops::TypedTarget<double> Target;
+ typedef ::frc971::control_loops::TypedLineSegment<double> Obstacle;
+
+ static constexpr size_t kNumTargets = 32;
+ static constexpr size_t kNumObstacles = 10;
+
+ Field();
+
+ ::std::array<Target, kNumTargets> targets() const { return targets_; }
+ ::std::array<Obstacle, kNumObstacles> obstacles() const { return obstacles_; }
+
+ private:
+ // All target locations are defined as being at the center of the target,
+ // except for the height, for which we use the top of the target.
+ ::std::array<Target, kNumTargets> targets_;
+ // Obstacle locations are approximate, as we are just trying to roughly
+ // approximate what will block our view when on the field.
+ // If anything, we should err on the side of making obstacles too small so
+ // that if there is any error in our position, we don't assume that it must
+ // be hidden behind a target when it really is not.
+ ::std::array<Obstacle, kNumObstacles> obstacles_;
+};
+
struct Values {
static const int kZeroingSampleSize = 200;
diff --git a/y2019/control_loops/drivetrain/BUILD b/y2019/control_loops/drivetrain/BUILD
index 959e839..38c73f0 100644
--- a/y2019/control_loops/drivetrain/BUILD
+++ b/y2019/control_loops/drivetrain/BUILD
@@ -84,6 +84,7 @@
cc_library(
name = "camera",
srcs = ["camera.h"],
+ visibility = ["//y2019:__pkg__"],
deps = [
"//aos/containers:sized_array",
"//frc971/control_loops:pose",