James Kuszmaul | 92ba0e5 | 2019-03-29 17:19:30 -0700 | [diff] [blame] | 1 | #include <fstream> |
| 2 | #include <iostream> |
| 3 | |
Philipp Schrader | 790cb54 | 2023-07-05 21:06:52 -0700 | [diff] [blame] | 4 | #include "y2019/constants.h" |
| 5 | #include "y2019/vision/constants.h" |
| 6 | |
James Kuszmaul | 92ba0e5 | 2019-03-29 17:19:30 -0700 | [diff] [blame] | 7 | namespace y2019 { |
| 8 | namespace vision { |
| 9 | void DumpPose(std::basic_ostream<char> *o, const vision::CameraGeometry &pose) { |
| 10 | *o << "{x: " << pose.location[0] << ", y: " << pose.location[1] |
Philipp Schrader | 790cb54 | 2023-07-05 21:06:52 -0700 | [diff] [blame] | 11 | << ", theta: " << pose.heading << "}"; |
James Kuszmaul | 92ba0e5 | 2019-03-29 17:19:30 -0700 | [diff] [blame] | 12 | } |
| 13 | void DumpTypescriptConstants(const char *fname) { |
| 14 | ::std::ofstream out_file(fname); |
| 15 | out_file << "export const CAMERA_POSES = [\n"; |
| 16 | for (size_t ii = 0; ii < constants::Values::kNumCameras; ++ii) { |
| 17 | out_file << " "; |
James Kuszmaul | d6d37d1 | 2019-03-30 13:04:54 -0700 | [diff] [blame] | 18 | // TODO(james): Decide how to manage visualization for practice and code |
| 19 | // bots. |
James Kuszmaul | 92ba0e5 | 2019-03-29 17:19:30 -0700 | [diff] [blame] | 20 | DumpPose(&out_file, |
| 21 | GetCamera(CameraSerialNumbers(CompBotTeensyId())[ii])->geometry); |
| 22 | out_file << ",\n"; |
| 23 | } |
| 24 | out_file << "];\n"; |
| 25 | } |
Philipp Schrader | 790cb54 | 2023-07-05 21:06:52 -0700 | [diff] [blame] | 26 | } // namespace vision |
James Kuszmaul | 92ba0e5 | 2019-03-29 17:19:30 -0700 | [diff] [blame] | 27 | } // namespace y2019 |
| 28 | |
| 29 | int main(int argc, char *argv[]) { |
| 30 | if (argc != 2) { |
| 31 | ::std::cout << "Must provide a filename for output as an argument\n"; |
| 32 | return 1; |
| 33 | } |
| 34 | ::y2019::vision::DumpTypescriptConstants(argv[1]); |
| 35 | } |