source: mainline/uspace/app/viewer/viewer.c

Last change on this file was 4c25f86b, checked in by Jiří Zárevúcky <zarevucky.jiri@…>, 10 months ago

viewer: Fix deallocation loop
  • Property mode set to 100644
File size: 11.9 KB
Line 
1/*
2 * Copyright (c) 2024 Jiri Svoboda
3 * Copyright (c) 2013 Martin Decky
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * - Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * - Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * - The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30/** @addtogroup viewer
31 * @{
32 */
33/** @file
34 */
35
36#include <errno.h>
37#include <gfximage/tga.h>
38#include <stdbool.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <str.h>
42#include <ui/filedialog.h>
43#include <ui/image.h>
44#include <ui/ui.h>
45#include <ui/wdecor.h>
46#include <ui/window.h>
47#include <vfs/vfs.h>
48
49#define NAME "viewer"
50
51typedef struct {
52 ui_t *ui;
53
54 size_t imgs_count;
55 size_t imgs_current;
56 char **imgs;
57
58 bool fullscreen;
59 ui_window_t *window;
60 gfx_bitmap_t *bitmap;
61 ui_image_t *image;
62 gfx_context_t *window_gc;
63 ui_file_dialog_t *dialog;
64
65 gfx_rect_t img_rect;
66} viewer_t;
67
68static bool viewer_img_load(viewer_t *, const char *, gfx_bitmap_t **,
69 gfx_rect_t *);
70static bool viewer_img_setup(viewer_t *, gfx_bitmap_t *, gfx_rect_t *);
71static errno_t viewer_window_create(viewer_t *);
72static void viewer_window_destroy(viewer_t *);
73
74static void wnd_close(ui_window_t *, void *);
75static void wnd_kbd_event(ui_window_t *, void *, kbd_event_t *);
76
77static ui_window_cb_t window_cb = {
78 .close = wnd_close,
79 .kbd = wnd_kbd_event
80};
81
82static void file_dialog_bok(ui_file_dialog_t *, void *, const char *);
83static void file_dialog_bcancel(ui_file_dialog_t *, void *);
84static void file_dialog_close(ui_file_dialog_t *, void *);
85
86static ui_file_dialog_cb_t file_dialog_cb = {
87 .bok = file_dialog_bok,
88 .bcancel = file_dialog_bcancel,
89 .close = file_dialog_close
90};
91
92/** Window close request
93 *
94 * @param window Window
95 * @param arg Argument (calc_t *)
96 */
97static void wnd_close(ui_window_t *window, void *arg)
98{
99 viewer_t *viewer = (viewer_t *) arg;
100
101 ui_quit(viewer->ui);
102}
103
104/** Viewer unmodified key press.
105 *
106 * @param viewer Viewer
107 * @param event Keyboard event
108 */
109static void viewer_kbd_event_unmod(viewer_t *viewer, kbd_event_t *event)
110{
111 bool update = false;
112
113 if (event->key == KC_Q || event->key == KC_ESCAPE)
114 ui_quit(viewer->ui);
115
116 if (event->key == KC_PAGE_DOWN) {
117 if (viewer->imgs_current == viewer->imgs_count - 1)
118 viewer->imgs_current = 0;
119 else
120 viewer->imgs_current++;
121
122 update = true;
123 }
124
125 if (event->key == KC_PAGE_UP) {
126 if (viewer->imgs_current == 0)
127 viewer->imgs_current = viewer->imgs_count - 1;
128 else
129 viewer->imgs_current--;
130
131 update = true;
132 }
133
134 if (update) {
135 gfx_bitmap_t *lbitmap;
136 gfx_rect_t lrect;
137
138 if (!viewer_img_load(viewer, viewer->imgs[viewer->imgs_current],
139 &lbitmap, &lrect)) {
140 printf("Cannot load image \"%s\".\n",
141 viewer->imgs[viewer->imgs_current]);
142 exit(4);
143 }
144 if (!viewer_img_setup(viewer, lbitmap, &lrect)) {
145 printf("Cannot setup image \"%s\".\n",
146 viewer->imgs[viewer->imgs_current]);
147 exit(6);
148 }
149 }
150}
151
152/** Viewer ctrl-key key press.
153 *
154 * @param viewer Viewer
155 * @param event Keyboard event
156 */
157static void viewer_kbd_event_ctrl(viewer_t *viewer, kbd_event_t *event)
158{
159 if (event->key == KC_Q)
160 ui_quit(viewer->ui);
161}
162
163/** Viewer window keyboard event.
164 *
165 * @param window UI window
166 * @param arg Argument (viewer_t *)
167 * @param event Keyboard event
168 */
169static void wnd_kbd_event(ui_window_t *window, void *arg,
170 kbd_event_t *event)
171{
172 viewer_t *viewer = (viewer_t *)arg;
173
174 if (event->type != KEY_PRESS)
175 return;
176
177 if ((event->mods & (KM_CTRL | KM_ALT | KM_SHIFT)) == 0)
178 viewer_kbd_event_unmod(viewer, event);
179
180 if ((event->mods & KM_CTRL) != 0 &&
181 (event->mods & (KM_ALT | KM_SHIFT)) == 0)
182 viewer_kbd_event_ctrl(viewer, event);
183
184 ui_window_def_kbd(window, event);
185}
186
187/** File dialog OK button press.
188 *
189 * @param dialog File dialog
190 * @param arg Argument (viewer_t *)
191 * @param fname File name
192 */
193static void file_dialog_bok(ui_file_dialog_t *dialog, void *arg,
194 const char *fname)
195{
196 viewer_t *viewer = (viewer_t *) arg;
197 errno_t rc;
198
199 viewer->imgs_count = 1;
200 viewer->imgs = calloc(viewer->imgs_count, sizeof(char *));
201 if (viewer->imgs == NULL) {
202 printf("Out of memory.\n");
203 ui_quit(viewer->ui);
204 return;
205 }
206
207 viewer->imgs[0] = str_dup(fname);
208 if (viewer->imgs[0] == NULL) {
209 printf("Out of memory.\n");
210 ui_quit(viewer->ui);
211 return;
212 }
213
214 rc = viewer_window_create(viewer);
215 if (rc != EOK)
216 ui_quit(viewer->ui);
217
218 ui_file_dialog_destroy(dialog);
219 viewer->dialog = NULL;
220}
221
222/** File dialog cancel button press.
223 *
224 * @param dialog File dialog
225 * @param arg Argument (viewer_t *)
226 */
227static void file_dialog_bcancel(ui_file_dialog_t *dialog, void *arg)
228{
229 viewer_t *viewer = (viewer_t *) arg;
230
231 ui_file_dialog_destroy(dialog);
232 ui_quit(viewer->ui);
233}
234
235/** File dialog close request.
236 *
237 * @param dialog File dialog
238 * @param arg Argument (viewer_t *)
239 */
240static void file_dialog_close(ui_file_dialog_t *dialog, void *arg)
241{
242 viewer_t *viewer = (viewer_t *) arg;
243
244 ui_file_dialog_destroy(dialog);
245 ui_quit(viewer->ui);
246}
247
248static bool viewer_img_load(viewer_t *viewer, const char *fname,
249 gfx_bitmap_t **rbitmap, gfx_rect_t *rect)
250{
251 int fd;
252 errno_t rc = vfs_lookup_open(fname, WALK_REGULAR, MODE_READ, &fd);
253 if (rc != EOK)
254 return false;
255
256 vfs_stat_t stat;
257 rc = vfs_stat(fd, &stat);
258 if (rc != EOK) {
259 vfs_put(fd);
260 return false;
261 }
262
263 void *tga = malloc(stat.size);
264 if (tga == NULL) {
265 vfs_put(fd);
266 return false;
267 }
268
269 size_t nread;
270 rc = vfs_read(fd, (aoff64_t []) { 0 }, tga, stat.size, &nread);
271 if (rc != EOK || nread != stat.size) {
272 free(tga);
273 vfs_put(fd);
274 return false;
275 }
276
277 vfs_put(fd);
278
279 rc = decode_tga(viewer->window_gc, tga, stat.size, rbitmap, rect);
280 if (rc != EOK) {
281 free(tga);
282 return false;
283 }
284
285 free(tga);
286
287 viewer->img_rect = *rect;
288 return true;
289}
290
291static bool viewer_img_setup(viewer_t *viewer, gfx_bitmap_t *bmp,
292 gfx_rect_t *rect)
293{
294 gfx_rect_t arect;
295 gfx_rect_t irect;
296 ui_resource_t *ui_res;
297 errno_t rc;
298
299 ui_res = ui_window_get_res(viewer->window);
300
301 ui_window_get_app_rect(viewer->window, &arect);
302
303 /* Center image on application area */
304 gfx_rect_ctr_on_rect(rect, &arect, &irect);
305
306 if (viewer->image != NULL) {
307 ui_image_set_bmp(viewer->image, bmp, rect);
308 (void) ui_image_paint(viewer->image);
309 ui_image_set_rect(viewer->image, &irect);
310 } else {
311 rc = ui_image_create(ui_res, bmp, rect, &viewer->image);
312 if (rc != EOK) {
313 gfx_bitmap_destroy(bmp);
314 return false;
315 }
316
317 ui_image_set_rect(viewer->image, &irect);
318 ui_window_add(viewer->window, ui_image_ctl(viewer->image));
319 }
320
321 if (viewer->bitmap != NULL)
322 gfx_bitmap_destroy(viewer->bitmap);
323
324 viewer->bitmap = bmp;
325 return true;
326}
327
328static void print_syntax(void)
329{
330 printf("Syntax: %s [<options] <image-file>...\n", NAME);
331 printf("\t-d <display-spec> Use the specified display\n");
332 printf("\t-f Full-screen mode\n");
333}
334
335static errno_t viewer_window_create(viewer_t *viewer)
336{
337 ui_wnd_params_t params;
338 gfx_bitmap_t *lbitmap;
339 gfx_rect_t lrect;
340 gfx_rect_t wrect;
341 gfx_coord2_t off;
342 gfx_rect_t rect;
343 errno_t rc;
344
345 /*
346 * We don't know the image size yet, so create tiny window and resize
347 * later.
348 */
349 ui_wnd_params_init(&params);
350 params.caption = "Viewer";
351 params.rect.p0.x = 0;
352 params.rect.p0.y = 0;
353 params.rect.p1.x = 1;
354 params.rect.p1.y = 1;
355
356 if (viewer->fullscreen) {
357 params.style &= ~ui_wds_decorated;
358 params.placement = ui_wnd_place_full_screen;
359 }
360
361 rc = ui_window_create(viewer->ui, &params, &viewer->window);
362 if (rc != EOK) {
363 printf("Error creating window.\n");
364 goto error;
365 }
366
367 viewer->window_gc = ui_window_get_gc(viewer->window);
368
369 ui_window_set_cb(viewer->window, &window_cb, (void *)viewer);
370
371 if (!viewer_img_load(viewer, viewer->imgs[viewer->imgs_current],
372 &lbitmap, &lrect)) {
373 printf("Cannot load image \"%s\".\n",
374 viewer->imgs[viewer->imgs_current]);
375 goto error;
376 }
377
378 /*
379 * Compute window rectangle such that application area corresponds
380 * to rect
381 */
382 ui_wdecor_rect_from_app(viewer->ui, params.style, &lrect, &wrect);
383 off = wrect.p0;
384 gfx_rect_rtranslate(&off, &wrect, &rect);
385
386 if (!viewer->fullscreen) {
387 rc = ui_window_resize(viewer->window, &rect);
388 if (rc != EOK) {
389 printf("Error resizing window.\n");
390 goto error;
391 }
392 }
393
394 if (!viewer_img_setup(viewer, lbitmap, &lrect)) {
395 printf("Cannot setup image \"%s\".\n",
396 viewer->imgs[viewer->imgs_current]);
397 goto error;
398 }
399
400 rc = ui_window_paint(viewer->window);
401 if (rc != EOK) {
402 printf("Error painting window.\n");
403 goto error;
404 }
405
406 return EOK;
407error:
408 viewer_window_destroy(viewer);
409 ui_quit(viewer->ui);
410 return rc;
411}
412
413static void viewer_window_destroy(viewer_t *viewer)
414{
415 if (viewer->window != NULL)
416 ui_window_destroy(viewer->window);
417 viewer->window = NULL;
418}
419
420int main(int argc, char *argv[])
421{
422 const char *display_spec = UI_ANY_DEFAULT;
423 ui_t *ui = NULL;
424 viewer_t *viewer;
425 errno_t rc;
426 int i;
427 unsigned u;
428 ui_file_dialog_params_t fdparams;
429
430 viewer = calloc(1, sizeof(viewer_t));
431 if (viewer == NULL) {
432 printf("Out of memory.\n");
433 goto error;
434 }
435
436 i = 1;
437 while (i < argc && argv[i][0] == '-') {
438 if (str_cmp(argv[i], "-d") == 0) {
439 ++i;
440 if (i >= argc) {
441 printf("Argument missing.\n");
442 print_syntax();
443 goto error;
444 }
445
446 display_spec = argv[i++];
447 } else if (str_cmp(argv[i], "-f") == 0) {
448 ++i;
449 viewer->fullscreen = true;
450 } else {
451 printf("Invalid option '%s'.\n", argv[i]);
452 print_syntax();
453 goto error;
454 }
455 }
456
457 /* Images specified? */
458 if (i < argc) {
459 viewer->imgs_count = argc - i;
460 viewer->imgs = calloc(viewer->imgs_count, sizeof(char *));
461 if (viewer->imgs == NULL) {
462 printf("Out of memory.\n");
463 goto error;
464 }
465
466 for (int j = 0; j < argc - i; j++) {
467 viewer->imgs[j] = str_dup(argv[i + j]);
468 if (viewer->imgs[j] == NULL) {
469 printf("Out of memory.\n");
470 goto error;
471 }
472 }
473 }
474
475 rc = ui_create(display_spec, &ui);
476 if (rc != EOK) {
477 printf("Error creating UI on display %s.\n", display_spec);
478 goto error;
479 }
480
481 if (ui_is_fullscreen(ui))
482 viewer->fullscreen = true;
483
484 viewer->ui = ui;
485
486 if (viewer->imgs != NULL) {
487 /* We have images, create viewer window. */
488 rc = viewer_window_create(viewer);
489 if (rc != EOK)
490 goto error;
491 } else {
492 /* No images specified, browse for one. */
493 ui_file_dialog_params_init(&fdparams);
494 fdparams.caption = "Open Image";
495
496 rc = ui_file_dialog_create(viewer->ui, &fdparams,
497 &viewer->dialog);
498 if (rc != EOK) {
499 printf("Error creating file dialog.\n");
500 goto error;
501 }
502
503 ui_file_dialog_set_cb(viewer->dialog, &file_dialog_cb,
504 (void *)viewer);
505 }
506
507 ui_run(ui);
508
509 ui_window_destroy(viewer->window);
510 ui_destroy(ui);
511 free(viewer);
512
513 return 0;
514error:
515 if (viewer != NULL && viewer->dialog != NULL)
516 ui_file_dialog_destroy(viewer->dialog);
517 if (viewer != NULL && viewer->imgs != NULL) {
518 for (u = 0; u < viewer->imgs_count; u++) {
519 if (viewer->imgs[u] != NULL)
520 free(viewer->imgs[u]);
521 }
522 free(viewer->imgs);
523 }
524 if (viewer != NULL)
525 viewer_window_destroy(viewer);
526 if (ui != NULL)
527 ui_destroy(ui);
528 if (viewer != NULL)
529 free(viewer);
530 return 1;
531}
532
533/** @}
534 */
Note: See TracBrowser for help on using the repository browser.