Add 2019 field target locations

Change-Id: Ie3ac15f4de3f4f2f97b88a7160c3a02d65d41740
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