1 #include "mainwindow.h"
2 #include "post_to_main_thread.h"
3 #include "ui_mainwindow.h"
6 #include <netinet/in.h>
7 #include <netinet/tcp.h>
10 #include <sys/socket.h>
15 string escape_html(const string &str)
21 } else if (ch == '>') {
23 } else if (ch == '&') {
32 string escape_unicode(const string &str)
35 for (size_t pos = 0; pos < str.size(); ) {
37 int len = mbtowc(&wc, str.data() + pos, str.size() - pos);
46 } else if (isprint(wc)) {
50 snprintf(buf, sizeof(buf), "\\u%04x", wc);
57 string escape_quotes(const string &str)
69 string serialize_as_json(const map<string, string> ¶m)
74 for (const auto &key_value : param) {
75 if (!first) s += ", ";
79 s += escape_quotes(escape_unicode(key_value.first));
81 s += escape_quotes(escape_unicode(key_value.second));
88 MainWindow::MainWindow(QWidget *parent) :
90 ui(new Ui::MainWindow)
93 for (QComboBox *team_combo : { ui->initials_1_edit, ui->initials_2_edit, ui->match_2_initials_1_edit, ui->match_2_initials_2_edit, ui->match_3_initials_1_edit, ui->match_3_initials_2_edit }) {
94 // team_combo->addItem("BFK", "navy");
95 team_combo->addItem("BRI", "navy");
96 team_combo->addItem("BSI1", "white");
97 team_combo->addItem("BSI2", "black");
98 team_combo->addItem("ESK", "white");
99 team_combo->addItem("FRA", "navy");
100 team_combo->addItem("NHHI", "navy");
101 team_combo->addItem("NMBUI", "cyan");
102 team_combo->addItem("OSI", "black");
103 team_combo->addItem("PCL", "red");
104 team_combo->addItem("STO", "");
105 team_combo->addItem("TFK1", "yellow");
106 team_combo->addItem("TFK2", "yellow");
107 team_combo->addItem("TFK3", "yellow");
108 team_combo->addItem("TSI", "orange");
111 ws = new WSServer("127.0.0.1", 5250);
112 ws->set_connection_callback([this](bool connected) {
113 string msg = connected ? "Connected" : "Not connected";
114 post_to_main_thread([this, msg]() {
115 ui->ws_connected_label->setText(QString::fromStdString(msg));
119 udp_thread = std::thread(&MainWindow::udp_thread_func, this, 6000);
120 udp_thread2 = std::thread(&MainWindow::udp_thread_func, this, 6001);
121 udp_thread3 = std::thread(&MainWindow::udp_thread_func, this, 6002);
123 connect(ui->ws_disconnect_btn, &QPushButton::clicked, this, &MainWindow::ws_disconnect_clicked);
124 connect(ui->set_initials_btn, &QPushButton::clicked, this, &MainWindow::set_initials_clicked);
125 connect(ui->set_match_2_initials_btn, &QPushButton::clicked, this, &MainWindow::set_match_2_initials_clicked);
126 connect(ui->set_match_3_initials_btn, &QPushButton::clicked, this, &MainWindow::set_match_3_initials_clicked);
127 connect(ui->set_color_btn, &QPushButton::clicked, this, &MainWindow::set_color_clicked);
128 connect(ui->set_score_btn, &QPushButton::clicked, this, &MainWindow::set_score_clicked);
129 connect(ui->set_all_scorebug_btn, &QPushButton::clicked, this, &MainWindow::set_all_scorebug_clicked);
130 connect(ui->goal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, 1); });
131 connect(ui->ungoal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, -1); });
132 connect(ui->goal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, 1); });
133 connect(ui->ungoal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, -1); });
135 connect(ui->set_clock_btn, &QPushButton::clicked, this, &MainWindow::set_clock_clicked);
136 connect(ui->set_clock_limit_btn, &QPushButton::clicked, this, &MainWindow::set_clock_limit_clicked);
137 connect(ui->start_and_show_clock_btn, &QPushButton::clicked, this, &MainWindow::start_and_show_clock_clicked);
138 connect(ui->stop_clock_btn, &QPushButton::clicked, this, &MainWindow::stop_clock_clicked);
139 connect(ui->show_clock_btn, &QPushButton::clicked, this, &MainWindow::show_clock_clicked);
140 connect(ui->hide_clock_btn, &QPushButton::clicked, this, &MainWindow::hide_clock_clicked);
141 connect(ui->show_match_2_btn, &QPushButton::clicked, this, &MainWindow::show_match_2_clicked);
142 connect(ui->hide_match_2_btn, &QPushButton::clicked, this, &MainWindow::hide_match_2_clicked);
143 connect(ui->show_match_3_btn, &QPushButton::clicked, this, &MainWindow::show_match_3_clicked);
144 connect(ui->hide_match_3_btn, &QPushButton::clicked, this, &MainWindow::hide_match_3_clicked);
146 connect(ui->set_comment_btn, &QPushButton::clicked, this, &MainWindow::set_comment_clicked);
147 connect(ui->set_and_show_comment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_comment_clicked);
148 connect(ui->hide_comment_btn, &QPushButton::clicked, this, &MainWindow::hide_comment_clicked);
149 connect(ui->set_and_show_autocomment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_autocomment_clicked);
150 connect(ui->autoshow_autocomment, &QCheckBox::stateChanged, this, &MainWindow::autocomment_update);
152 connect(ui->show_lower_third_btn, &QPushButton::clicked, this, &MainWindow::show_lower_third_clicked);
153 connect(ui->hide_lower_third_btn, &QPushButton::clicked, this, &MainWindow::hide_lower_third_clicked);
155 connect(ui->quick_lower_third_edit, &QLineEdit::returnPressed, this, &MainWindow::quick_lower_third_activate);
156 connect(ui->show_quick_lower_third_btn, &QPushButton::clicked, this, &MainWindow::quick_lower_third_activate);
158 connect(ui->show_scorebug_btn, &QPushButton::clicked, this, &MainWindow::show_scorebug_clicked);
159 connect(ui->show_group_a_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group A"); });
160 connect(ui->show_group_b_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group B"); });
161 connect(ui->show_group_c_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group C"); });
162 connect(ui->show_schedule_btn, &QPushButton::clicked, this, &MainWindow::show_schedule_clicked);
163 connect(ui->show_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_carousel_clicked);
164 connect(ui->show_nothing_btn, &QPushButton::clicked, this, &MainWindow::show_nothing_clicked);
165 connect(ui->show_roster_1_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_1_edit->currentText().toStdString()); });
166 connect(ui->show_roster_2_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_2_edit->currentText().toStdString()); });
167 connect(ui->show_roster_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_roster_carousel_clicked);
169 connect(ui->initials_1_edit, QOverload<int>::of(&QComboBox::currentIndexChanged), [=](int index) {
170 ui->color_1_edit->setText(ui->initials_1_edit->itemData(index).toString());
172 connect(ui->initials_2_edit, QOverload<int>::of(&QComboBox::currentIndexChanged), [=](int index) {
173 ui->color_2_edit->setText(ui->initials_2_edit->itemData(index).toString());
176 autocomment_update();
178 const set<pair<unsigned, unsigned>> usb{{ 0x0e8f, 0x0041 }};
179 event_device = new EventDevice(usb, ui->quick_lower_third_edit);
180 event_device->start_thread();
183 MainWindow::~MainWindow()
188 void MainWindow::ws_disconnect_clicked()
190 ws->change_port(stoi(ui->ws_port_box->text().toStdString()));
193 void MainWindow::set_initials_clicked()
195 map<string, string> param;
196 param["team1"] = escape_html(ui->initials_1_edit->currentText().toStdString());
197 param["team2"] = escape_html(ui->initials_2_edit->currentText().toStdString());
198 ws->send_command("update " + serialize_as_json(param));
199 ws->send_command("eval setteams()");
202 void MainWindow::set_match_2_initials_clicked()
204 map<string, string> param;
205 param["team1"] = escape_html(ui->match_2_initials_1_edit->currentText().toStdString());
206 param["team2"] = escape_html(ui->match_2_initials_2_edit->currentText().toStdString());
207 ws->send_command("update " + serialize_as_json(param));
208 ws->send_command("eval setteams2()");
211 void MainWindow::set_match_3_initials_clicked()
213 map<string, string> param;
214 param["team1"] = escape_html(ui->match_3_initials_1_edit->currentText().toStdString());
215 param["team2"] = escape_html(ui->match_3_initials_2_edit->currentText().toStdString());
216 ws->send_command("update " + serialize_as_json(param));
217 ws->send_command("eval setteams3()");
220 void MainWindow::set_color_clicked()
222 map<string, string> param;
223 param["team1color"] = ui->color_1_edit->text().toStdString(); // Should maybe be escaped, but meh.
224 param["team2color"] = ui->color_2_edit->text().toStdString();
225 ws->send_command("update " + serialize_as_json(param));
226 ws->send_command("eval setcolors()");
229 void MainWindow::set_score_clicked()
231 map<string, string> param;
232 param["score1"] = to_string(ui->score_1_box->value());
233 param["score2"] = to_string(ui->score_2_box->value());
234 ws->send_command("update " + serialize_as_json(param));
235 ws->send_command("eval setscore()");
236 autocomment_update();
239 void MainWindow::set_all_scorebug_clicked()
241 set_initials_clicked();
246 void MainWindow::add_goal(QSpinBox *box, int delta)
248 box->setValue(box->value() + delta);
252 void MainWindow::set_clock_clicked()
254 map<string, string> param;
255 param["clock_min"] = to_string(ui->clock_min_box->value());
256 param["clock_sec"] = to_string(ui->clock_sec_box->value());
257 ws->send_command("update " + serialize_as_json(param));
258 ws->send_command("eval setclockfromstate()");
261 void MainWindow::set_clock_limit_clicked()
263 map<string, string> param;
264 param["clock_limit_min"] = to_string(ui->clock_limit_min_box->value());
265 param["clock_limit_sec"] = to_string(ui->clock_limit_sec_box->value());
266 ws->send_command("update " + serialize_as_json(param));
267 ws->send_command("eval setclocklimitfromstate()");
270 void MainWindow::start_and_show_clock_clicked()
272 ws->send_command("eval startclock(0)"); // Also shows.
275 void MainWindow::stop_clock_clicked()
277 ws->send_command("eval stopclock(0)");
280 void MainWindow::show_clock_clicked()
282 ws->send_command("eval showclock()");
285 void MainWindow::hide_clock_clicked()
287 ws->send_command("eval hideclock()");
290 void MainWindow::show_match_2_clicked()
292 ws->send_command("eval showmatch2()");
295 void MainWindow::hide_match_2_clicked()
297 ws->send_command("eval hidematch2()");
300 void MainWindow::show_match_3_clicked()
302 ws->send_command("eval showmatch3()");
305 void MainWindow::hide_match_3_clicked()
307 ws->send_command("eval hidematch3()");
310 void MainWindow::set_comment_clicked()
312 map<string, string> param;
313 param["comment"] = ui->comment_edit->text().toStdString();
314 ws->send_command("update " + serialize_as_json(param));
315 ws->send_command("eval setcomment()");
318 void MainWindow::set_and_show_comment_clicked()
320 set_comment_clicked();
321 ws->send_command("eval showcomment()");
324 void MainWindow::set_and_show_autocomment_clicked()
326 ui->comment_edit->setText(ui->autocomment_edit->text());
327 set_and_show_comment_clicked();
330 void MainWindow::hide_comment_clicked()
332 ws->send_command("eval hidecomment()");
335 void MainWindow::show_lower_third_clicked()
337 map<string, string> param;
338 param["text1"] = ui->lowerthird_heading_edit->text().toStdString();
339 param["text2"] = ui->lowerthird_subheading_edit->text().toStdString();
340 ws->send_command("update " + serialize_as_json(param));
341 ws->send_command("eval setandshowlowerthird()");
344 void MainWindow::hide_lower_third_clicked()
346 ws->send_command("eval hidelowerthird()");
349 void MainWindow::quick_lower_third_activate()
351 string code = ui->quick_lower_third_edit->text().toUpper().toStdString();
353 add_goal(ui->score_1_box, 1);
354 } else if (code == "B") {
355 add_goal(ui->score_2_box, 1);
356 } else if (code == "C") {
357 ws->send_command("eval hidelowerthird()");
359 map<string, string> param;
360 param["code"] = code;
361 ws->send_command("update " + serialize_as_json(param));
362 ws->send_command("eval quicklowerthird()");
364 ui->quick_lower_third_edit->clear();
367 void MainWindow::autocomment_update()
369 int score1 = ui->score_1_box->value();
370 int score2 = ui->score_2_box->value();
372 if (abs(score1 - score2) >= 3) {
373 msg = "Game ends after this point";
375 int cap = max(score1, score2) + 1;
376 if (score1 == score2) ++cap;
379 msg = "Point cap: First to 13";
382 snprintf(buf, sizeof(buf), "Pagacap: First to %d", cap);
386 ui->autocomment_edit->setText(QString::fromStdString(msg));
388 map<string, string> param;
389 param["autocomment_on_clock_limit"] = ui->autoshow_autocomment->isChecked() ? "1" : "0";
390 param["autocomment"] = msg;
391 ws->send_command("update " + serialize_as_json(param));
394 void MainWindow::show_scorebug_clicked()
396 ws->send_command("eval stopcarousel()");
397 ws->send_command("eval hidetable()");
398 ws->send_command("eval showscorebug()");
401 void MainWindow::show_group_clicked(const std::string &group_name)
403 map<string, string> param;
404 param["group_name"] = group_name;
405 ws->send_command("eval stopcarousel()");
406 ws->send_command("update " + serialize_as_json(param));
407 ws->send_command("eval showgroup_from_state()");
410 void MainWindow::show_roster_clicked(const std::string &team_code)
412 map<string, string> param;
413 param["team_code"] = team_code;
414 ws->send_command("eval stopcarousel()");
415 ws->send_command("update " + serialize_as_json(param));
416 ws->send_command("eval showroster_from_state()");
419 void MainWindow::show_schedule_clicked()
421 ws->send_command("eval stopcarousel()");
422 ws->send_command("eval showschedule()");
425 void MainWindow::show_carousel_clicked()
427 ws->send_command("eval stopcarousel()");
428 ws->send_command("eval showcarousel()");
431 void MainWindow::show_roster_carousel_clicked()
433 map<string, string> param;
434 param["team1"] = escape_html(ui->initials_1_edit->currentText().toStdString());
435 param["team2"] = escape_html(ui->initials_2_edit->currentText().toStdString());
436 ws->send_command("eval stopcarousel()");
437 ws->send_command("update " + serialize_as_json(param));
438 ws->send_command("eval showrostercarousel_from_state()");
441 void MainWindow::show_nothing_clicked()
443 ws->send_command("eval hidescorebug()");
444 ws->send_command("eval stopcarousel()");
445 ws->send_command("eval hidetable()");
448 void udp_thread_nat_func(int sock, int port)
451 memset(&saddr6, 0, sizeof(saddr6));
452 saddr6.sin6_family = AF_INET6;
453 inet_pton(AF_INET6, "::ffff:193.35.52.50", &saddr6.sin6_addr);
454 saddr6.sin6_port = htons(port);
458 sendto(sock, buf, 4, 0, (sockaddr *)&saddr6, sizeof(saddr6));
463 void MainWindow::udp_thread_func(int port)
465 int sock = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP);
472 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) == -1) {
473 perror("setsockopt");
478 memset(&saddr6, 0, sizeof(saddr6));
479 saddr6.sin6_family = AF_INET6;
480 inet_pton(AF_INET6, "::", &saddr6.sin6_addr);
481 saddr6.sin6_port = htons(port);
482 if (bind(sock, (sockaddr *)&saddr6, sizeof(saddr6)) == -1) {
487 std::thread(&udp_thread_nat_func, sock, port + 1000).detach();
491 int err = recv(sock, buf, sizeof(buf), 0);
497 post_to_main_thread([buf, err, port, this] {
498 bt6000_message_received(string(buf, err), port);
503 int parse_digit(char ch)
505 if (ch >= '0' && ch <= '9') {
511 int parse_clock(char ch1, char ch2)
513 int s1 = parse_digit(ch1);
514 int s2 = parse_digit(ch2);
518 int parse_score(char ch2, char ch3)
520 int s2 = parse_digit(ch2);
521 int s3 = parse_digit(ch3);
525 int parse_score(char ch1, char ch2, char ch3)
527 int s1 = parse_digit(ch1);
528 int s2 = parse_digit(ch2);
529 int s3 = parse_digit(ch3);
530 return s1 * 100 + s2 * 10 + s3;
533 int parse_score_weird(char ch1, char ch2, char ch3)
537 if (ch1 != ' ') *ptr++ = ch1;
538 if (ch2 != ' ') *ptr++ = ch2;
539 if (ch3 != ' ') *ptr++ = ch3;
546 void MainWindow::bt6000_message_received(const string &msg, int port)
548 fprintf(stderr, "BT6000 message: '%s' (port %d)\n", msg.c_str(), port);
550 if (!ui->bt6000_3_enable->isChecked()) {
553 } else if (port == 6001) {
554 if (!ui->bt6000_2_enable->isChecked()) {
558 if (!ui->bt6000_enable->isChecked()) {
563 if (msg.size() >= 9 && msg[0] == 'G' && msg[1] == '0' && msg[2] == '1') {
564 // G01: Game clock, period number, and number of time-outs.
565 bool clock_running = !(msg[3] & 0x02);
566 // bool klaxon = (msg[3] & 0x04);
567 int minutes = parse_clock(msg[5], msg[6]);
568 int seconds = parse_clock(msg[7], msg[8]);
570 map<string, string> param;
571 param["clock_min"] = to_string(minutes);
572 param["clock_sec"] = to_string(seconds);
573 ws->send_command("update " + serialize_as_json(param));
576 ws->send_command("eval adjustclockfromstate(2)");
578 ws->send_command("eval startclock(2)");
580 ws->send_command("eval stopclock(2)");
582 } else if (port == 6001) {
583 ws->send_command("eval adjustclockfromstate(1)");
585 ws->send_command("eval startclock(1)");
587 ws->send_command("eval stopclock(1)");
590 ws->send_command("eval adjustclockfromstate(0)");
592 ws->send_command("eval startclock(0)");
594 ws->send_command("eval stopclock(0)");
598 if (msg.size() >= 10 && msg[0] == 'G' && msg[1] == '0' && msg[2] == '2') {
599 int score1 = parse_score(msg[4], msg[5], msg[6]);
600 int score2 = parse_score(msg[7], msg[8], msg[9]);
602 map<string, string> param;
603 param["score1"] = to_string(score1);
604 param["score2"] = to_string(score2);
605 ws->send_command("update " + serialize_as_json(param));
606 ws->send_command("eval setscore3()");
607 } else if (port == 6001) {
608 map<string, string> param;
609 param["score1"] = to_string(score1);
610 param["score2"] = to_string(score2);
611 ws->send_command("update " + serialize_as_json(param));
612 ws->send_command("eval setscore2()");
614 ui->score_1_box->setValue(score1);
615 ui->score_2_box->setValue(score2);
619 if (msg.size() >= 9 && msg[0] == 'G' && msg[1] == '1' && msg[2] == '0') {
620 // G10: Game clock, score, period?
621 bool clock_running = !(msg[3] & 0x02);
622 // bool klaxon = (msg[3] & 0x04);
623 int minutes = parse_clock(msg[4], msg[5]);
624 int seconds = parse_clock(msg[6], msg[7]);
625 int score1 = parse_score_weird(msg[8], msg[9], msg[10]);
626 int score2 = parse_score_weird(msg[11], msg[12], msg[13]);
628 map<string, string> param;
629 ws->send_command("update " + serialize_as_json(param));
632 map<string, string> param;
633 param["score1"] = to_string(score1);
634 param["score2"] = to_string(score2);
635 ws->send_command("update " + serialize_as_json(param));
636 ws->send_command("eval setscore3()");
637 } else if (port == 6001) {
638 map<string, string> param;
639 param["score1"] = to_string(score1);
640 param["score2"] = to_string(score2);
641 ws->send_command("update " + serialize_as_json(param));
642 ws->send_command("eval setscore2()");
644 map<string, string> param;
645 param["clock_min"] = to_string(minutes);
646 param["clock_sec"] = to_string(seconds);
647 ws->send_command("update " + serialize_as_json(param));
648 ui->score_1_box->setValue(score1);
649 ui->score_2_box->setValue(score2);
653 ws->send_command("eval adjustclockfromstate(2)");
655 ws->send_command("eval startclock(2)");
657 ws->send_command("eval stopclock(2)");
659 } else if (port == 6001) {
660 ws->send_command("eval adjustclockfromstate(1)");
662 ws->send_command("eval startclock(1)");
664 ws->send_command("eval stopclock(1)");
667 ws->send_command("eval adjustclockfromstate(0)");
669 ws->send_command("eval startclock(0)");
671 ws->send_command("eval stopclock(0)");
676 // Ignore type 3 (penalties) and type 4 (timeouts).