]> git.sesse.net Git - ultimatescore/blob - client/mainwindow.cpp
baf103d988a8e122018bdc6ea08774ff8a6313bb
[ultimatescore] / client / mainwindow.cpp
1 #include "mainwindow.h"
2 #include "post_to_main_thread.h"
3 #include "ui_mainwindow.h"
4
5 #include <stdlib.h>
6
7 using namespace std;
8
9 string escape_html(const string &str)
10 {
11         string s = "";
12         for (char ch : str) {
13                 if (ch == '<') {
14                         s += "&lt;";
15                 } else if (ch == '>') {
16                         s += "&gt;";
17                 } else if (ch == '&') {
18                         s += "&amp;";
19                 } else {
20                         s += ch;
21                 }
22         }
23         return s;
24 }
25
26 string escape_unicode(const string &str)
27 {
28         string s = "";
29         for (size_t pos = 0; pos < str.size(); ) {
30                 wchar_t wc;
31                 int len = mbtowc(&wc, str.data() + pos, str.size() - pos);
32                 if (len == -1) {
33                         wc = '?';
34                         len = 1;
35                 }
36                 pos += len;
37
38                 if (wc == '\\') {
39                         s += "\\\\";
40                 } else if (isprint(wc)) {
41                         s += wc;
42                 } else {
43                         char buf[16];
44                         snprintf(buf, sizeof(buf), "\\u%04x", wc);
45                         s += buf;
46                 }
47         }
48         return s;
49 }
50
51 string escape_quotes(const string &str)
52 {
53         string s = "";
54         for (char ch : str) {
55                 if (ch == '"' || ch == '\\') {
56                         s += '\\';
57                 }
58                 s += ch;
59         }
60         return s;
61 }
62
63 string serialize_as_json(const map<string, string> &param)
64 {
65         string s = "{";
66
67         bool first = true;
68         for (const auto &key_value : param) {
69                 if (!first) s += ", ";
70                 first = false;
71
72                 s += '"';
73                 s += escape_quotes(escape_unicode(key_value.first));
74                 s += "\": \"";
75                 s += escape_quotes(escape_unicode(key_value.second));
76                 s += '"';
77         }
78         s += "}";
79         return s;       
80 }
81
82 MainWindow::MainWindow(QWidget *parent) :
83     QMainWindow(parent),
84     ui(new Ui::MainWindow)
85 {
86         ui->setupUi(this);
87         acmp = new ACMPClient("127.0.0.1", 5250);
88         acmp->add_init_command("MIXER 1 STRAIGHT_ALPHA_OUTPUT 1");
89         acmp->add_init_command("CG 1 ADD 0 \"score\" 1 \"<templateData></templateData>\"");
90         acmp->set_connection_callback([this](bool connected) {
91                 string msg = connected ? "Connected" : "Not connected";
92                 post_to_main_thread([this, msg]() {
93                         ui->casparcg_connected_label->setText(QString::fromStdString(msg));
94                 });
95         });
96
97         acmp->start();
98
99         connect(ui->casparcg_reconnect_btn, &QPushButton::clicked, this, &MainWindow::casparcg_reconnect_clicked);
100         connect(ui->set_initials_btn, &QPushButton::clicked, this, &MainWindow::set_initials_clicked);
101         connect(ui->set_color_btn, &QPushButton::clicked, this, &MainWindow::set_color_clicked);
102         connect(ui->set_score_btn, &QPushButton::clicked, this, &MainWindow::set_score_clicked);
103         connect(ui->set_all_scorebug_btn, &QPushButton::clicked, this, &MainWindow::set_all_scorebug_clicked);
104         connect(ui->goal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, 1); });
105         connect(ui->ungoal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, -1); });
106         connect(ui->goal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, 1); });
107         connect(ui->ungoal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, -1); });
108
109         connect(ui->set_clock_btn, &QPushButton::clicked, this, &MainWindow::set_clock_clicked);
110         connect(ui->start_and_show_clock_btn, &QPushButton::clicked, this, &MainWindow::start_and_show_clock_clicked);
111         connect(ui->stop_clock_btn, &QPushButton::clicked, this, &MainWindow::stop_clock_clicked);
112         connect(ui->show_clock_btn, &QPushButton::clicked, this, &MainWindow::show_clock_clicked);
113         connect(ui->hide_clock_btn, &QPushButton::clicked, this, &MainWindow::hide_clock_clicked);
114
115         connect(ui->set_comment_btn, &QPushButton::clicked, this, &MainWindow::set_comment_clicked);
116         connect(ui->set_and_show_comment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_comment_clicked);
117         connect(ui->hide_comment_btn, &QPushButton::clicked, this, &MainWindow::hide_comment_clicked);
118         connect(ui->set_and_show_autocomment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_autocomment_clicked);
119
120         connect(ui->show_lower_third_btn, &QPushButton::clicked, this, &MainWindow::show_lower_third_clicked);
121         connect(ui->hide_lower_third_btn, &QPushButton::clicked, this, &MainWindow::hide_lower_third_clicked);
122
123         connect(ui->quick_lower_third_edit, &QLineEdit::returnPressed, this, &MainWindow::quick_lower_third_activate);
124         connect(ui->show_quick_lower_third_btn, &QPushButton::clicked, this, &MainWindow::quick_lower_third_activate);
125
126         connect(ui->show_scorebug_btn, &QPushButton::clicked, this, &MainWindow::show_scorebug_clicked);
127         connect(ui->show_group_a_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group A"); });
128         connect(ui->show_group_b_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group B"); });
129         connect(ui->show_schedule_btn, &QPushButton::clicked, this, &MainWindow::show_schedule_clicked);
130         connect(ui->show_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_carousel_clicked);
131         connect(ui->show_nothing_btn, &QPushButton::clicked, this, &MainWindow::show_nothing_clicked);
132         connect(ui->show_roster_1_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_1_edit->text().toStdString()); });
133         connect(ui->show_roster_2_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_2_edit->text().toStdString()); });
134         connect(ui->show_roster_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_roster_carousel_clicked);
135
136         autocomment_update();
137
138         const set<pair<unsigned, unsigned>> usb{{ 0x0e8f, 0x0041 }};
139         event_device = new EventDevice(usb, ui->quick_lower_third_edit);
140         event_device->start_thread();
141 }
142
143 MainWindow::~MainWindow()
144 {
145         acmp->end();
146         delete ui;
147 }
148
149 void MainWindow::casparcg_reconnect_clicked()
150 {
151         acmp->change_server(ui->casparcg_host_box->text().toStdString(),
152                             stoi(ui->casparcg_port_box->text().toStdString()));
153 }
154
155 void MainWindow::set_initials_clicked()
156 {
157         map<string, string> param;
158         param["team1"] = escape_html(ui->initials_1_edit->text().toStdString());
159         param["team2"] = escape_html(ui->initials_2_edit->text().toStdString());
160         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
161         acmp->send_command("cg 1 invoke 1 setteams");
162 }
163
164 void MainWindow::set_color_clicked()
165 {
166         map<string, string> param;
167         param["team1color"] = ui->color_1_edit->text().toStdString();  // Should maybe be escaped, but meh.
168         param["team2color"] = ui->color_2_edit->text().toStdString();
169         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
170         acmp->send_command("cg 1 invoke 1 setcolors");
171 }
172
173 void MainWindow::set_score_clicked()
174 {
175         map<string, string> param;
176         param["score1"] = to_string(ui->score_1_box->value());
177         param["score2"] = to_string(ui->score_2_box->value());
178         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
179         acmp->send_command("cg 1 invoke 1 setscore");
180         autocomment_update();
181 }
182
183 void MainWindow::set_all_scorebug_clicked()
184 {
185         set_initials_clicked();
186         set_color_clicked();
187         set_score_clicked();
188 }
189
190 void MainWindow::add_goal(QSpinBox *box, int delta)
191 {
192         box->setValue(box->value() + delta);
193         set_score_clicked();
194 }
195
196 void MainWindow::set_clock_clicked()
197 {
198         map<string, string> param;
199         param["clock_min"] = to_string(ui->clock_min_box->value());
200         param["clock_sec"] = to_string(ui->clock_sec_box->value());
201         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
202         acmp->send_command("cg 1 invoke 1 setclockfromstate");
203 }
204
205 void MainWindow::start_and_show_clock_clicked()
206 {
207         acmp->send_command("cg 1 invoke 1 startclock");  // Also shows.
208 }
209
210 void MainWindow::stop_clock_clicked()
211 {
212         acmp->send_command("cg 1 invoke 1 stopclock");
213 }
214
215 void MainWindow::show_clock_clicked()
216 {
217         acmp->send_command("cg 1 invoke 1 showclock");
218 }
219
220 void MainWindow::hide_clock_clicked()
221 {
222         acmp->send_command("cg 1 invoke 1 hideclock");
223 }
224
225 void MainWindow::set_comment_clicked()
226 {
227         map<string, string> param;
228         param["comment"] = ui->comment_edit->text().toStdString();
229         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
230         acmp->send_command("cg 1 invoke 1 setcomment");
231 }
232
233 void MainWindow::set_and_show_comment_clicked()
234 {
235         set_comment_clicked();
236         acmp->send_command("cg 1 invoke 1 showcomment");
237 }
238
239 void MainWindow::set_and_show_autocomment_clicked()
240 {
241         ui->comment_edit->setText(ui->autocomment_edit->text());
242         set_and_show_comment_clicked();
243 }
244
245 void MainWindow::hide_comment_clicked()
246 {
247         acmp->send_command("cg 1 invoke 1 hidecomment");
248 }
249
250 void MainWindow::show_lower_third_clicked()
251 {
252         map<string, string> param;
253         param["text1"] = ui->lowerthird_heading_edit->text().toStdString();
254         param["text2"] = ui->lowerthird_subheading_edit->text().toStdString();
255         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
256         acmp->send_command("cg 1 invoke 1 setandshowlowerthird");
257 }
258
259 void MainWindow::hide_lower_third_clicked()
260 {
261         acmp->send_command("cg 1 invoke 1 hidelowerthird");
262 }
263
264 void MainWindow::quick_lower_third_activate()
265 {
266         string code = ui->quick_lower_third_edit->text().toUpper().toStdString();
267         if (code == "A") {
268                 add_goal(ui->score_1_box, 1);
269         } else if (code == "B") {
270                 add_goal(ui->score_2_box, 1);
271         } else if (code == "C") {
272                 acmp->send_command("cg 1 invoke 1 hidelowerthird");
273         } else {
274                 map<string, string> param;
275                 param["code"] = code;
276                 acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
277                 acmp->send_command("cg 1 invoke 1 quicklowerthird");
278         }
279         ui->quick_lower_third_edit->clear();
280 }
281
282 void MainWindow::autocomment_update()
283 {
284         int score1 = ui->score_1_box->value();
285         int score2 = ui->score_2_box->value();
286         string msg;
287         if (abs(score1 - score2) >= 3) {
288                 msg = "Game ends after this point";
289         } else if (score1 >= 12 || score2 >= 12) {
290                 msg = "Point cap: First to 13";
291         } else {
292                 char buf[32];
293                 snprintf(buf, sizeof(buf), "Pagacap: First to %d", max(score1, score2) + 1);
294                 msg = buf;
295         }
296         ui->autocomment_edit->setText(QString::fromStdString(msg));
297 }
298
299 void MainWindow::show_scorebug_clicked()
300 {
301         acmp->send_command("cg 1 invoke 1 stopcarousel");
302         acmp->send_command("cg 1 invoke 1 hidetable");
303         acmp->send_command("cg 1 invoke 1 showscorebug");
304 }
305
306 void MainWindow::show_group_clicked(const std::string &group_name)
307 {
308         map<string, string> param;
309         param["group_name"] = group_name;
310         acmp->send_command("cg 1 invoke 1 stopcarousel");
311         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
312         acmp->send_command("cg 1 invoke 1 showgroup_from_state");
313 }
314
315 void MainWindow::show_roster_clicked(const std::string &team_code)
316 {
317         map<string, string> param;
318         param["team_code"] = team_code;
319         acmp->send_command("cg 1 invoke 1 stopcarousel");
320         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
321         acmp->send_command("cg 1 invoke 1 showroster_from_state");
322 }
323
324 void MainWindow::show_schedule_clicked()
325 {
326         acmp->send_command("cg 1 invoke 1 stopcarousel");
327         acmp->send_command("cg 1 invoke 1 showschedule");
328 }
329
330 void MainWindow::show_carousel_clicked()
331 {
332         acmp->send_command("cg 1 invoke 1 stopcarousel");
333         acmp->send_command("cg 1 invoke 1 showcarousel");
334 }
335
336 void MainWindow::show_roster_carousel_clicked()
337 {
338         map<string, string> param;
339         param["team1"] = escape_html(ui->initials_1_edit->text().toStdString());
340         param["team2"] = escape_html(ui->initials_2_edit->text().toStdString());
341         acmp->send_command("cg 1 invoke 1 stopcarousel");
342         acmp->send_command("cg 1 update 1 \"" + escape_quotes(serialize_as_json(param)) + "\"");
343         acmp->send_command("cg 1 invoke 1 showrostercarousel_from_state");
344 }
345
346 void MainWindow::show_nothing_clicked()
347 {
348         acmp->send_command("cg 1 invoke 1 hidescorebug");
349         acmp->send_command("cg 1 invoke 1 stopcarousel");
350         acmp->send_command("cg 1 invoke 1 hidetable");
351 }