3 #include "game/camera.h"
5 #include "system/stacktrace.h"
6 #include "system/nth_alloc.h"
7 #include "system/log.h"
10 #include "rect_layer.h"
12 #include "system/line_stream.h"
13 #include "color_picker.h"
14 #include "system/str.h"
15 #include "ui/edit_field.h"
16 #include "undo_history.h"
17 #include "game/level/action.h"
18 #include "action_picker.h"
20 #define RECT_LAYER_SELECTION_THICCNESS 10.0f
21 #define RECT_LAYER_ID_LABEL_SIZE vec(3.0f, 3.0f)
22 #define CREATE_AREA_THRESHOLD 10.0
23 #define RECT_LAYER_GRID_ROWS 3
24 #define RECT_LAYER_GRID_COLUMNS 4
26 static int clipboard = 0;
27 static Rect clipboard_rect;
28 static Color clipboard_color;
33 // TODO(#955): Rectangles in Level Editor have only one resize anchor to work with
34 // TODO: different cursor image in resize mode
49 ColorPicker color_picker;
50 ActionPicker action_picker;
54 Vec2f move_anchor; // The mouse offset from the left-top
55 // corner of the rect during moving it
56 Edit_field *id_edit_field;
60 const char *id_name_prefix;
79 char id[ENTITY_MAX_ID_SIZE];
100 UndoElementContext element;
101 UndoSwapContext swap;
105 UndoContext create_undo_add_context(RectLayer *layer, size_t index)
108 trace_assert(index < dynarray_count(layer->rects));
110 UndoContext undo_context;
111 undo_context.add.type = UNDO_ADD;
112 undo_context.add.layer = layer;
113 undo_context.add.index = index;
118 UndoContext create_undo_element_context(RectLayer *layer)
121 size_t index = (size_t) layer->selection;
122 trace_assert(index < dynarray_count(layer->rects));
124 UndoContext undo_context;
125 undo_context.element.layer = layer;
126 undo_context.element.index = index;
127 dynarray_copy_to(layer->rects, &undo_context.element.rect, index);
128 dynarray_copy_to(layer->colors, &undo_context.element.color, index);
129 dynarray_copy_to(layer->ids, undo_context.element.id, index);
130 dynarray_copy_to(layer->actions, &undo_context.element.action, index);
135 UndoContext create_undo_update_context(RectLayer *rect_layer)
137 UndoContext undo_context = create_undo_element_context(rect_layer);
138 undo_context.type = UNDO_UPDATE;
143 UndoContext create_undo_delete_context(RectLayer *rect_layer)
145 UndoContext undo_context = create_undo_element_context(rect_layer);
146 undo_context.type = UNDO_DELETE;
151 UndoContext create_undo_swap_context(RectLayer *rect_layer, size_t index1, size_t index2)
153 UndoContext undo_context;
154 undo_context.swap.type = UNDO_SWAP;
155 undo_context.swap.layer = rect_layer;
156 undo_context.swap.index1 = index1;
157 undo_context.swap.index2 = index2;
162 void rect_layer_undo(void *context, size_t context_size)
164 trace_assert(context);
165 trace_assert(sizeof(UndoContext) == context_size);
167 UndoContext *undo_context = context;
169 switch (undo_context->type) {
171 RectLayer *layer = undo_context->add.layer;
172 dynarray_delete_at(layer->rects, undo_context->add.index);
173 dynarray_delete_at(layer->colors, undo_context->add.index);
174 dynarray_delete_at(layer->ids, undo_context->add.index);
175 dynarray_delete_at(layer->actions, undo_context->add.index);
176 layer->selection = -1;
180 RectLayer *layer = undo_context->element.layer;
181 dynarray_insert_before(layer->rects, undo_context->element.index, &undo_context->element.rect);
182 dynarray_insert_before(layer->colors, undo_context->element.index, &undo_context->element.color);
183 dynarray_insert_before(layer->ids, undo_context->element.index, &undo_context->element.id);
184 dynarray_insert_before(layer->actions, undo_context->element.index, &undo_context->element.action);
185 layer->selection = -1;
189 RectLayer *layer = undo_context->element.layer;
190 dynarray_replace_at(layer->rects, undo_context->element.index, &undo_context->element.rect);
191 dynarray_replace_at(layer->colors, undo_context->element.index, &undo_context->element.color);
192 dynarray_replace_at(layer->ids, undo_context->element.index, &undo_context->element.id);
193 dynarray_replace_at(layer->actions, undo_context->element.index, &undo_context->element.action);
197 RectLayer *layer = undo_context->element.layer;
198 dynarray_swap(layer->rects, undo_context->swap.index1, undo_context->swap.index2);
199 dynarray_swap(layer->colors, undo_context->swap.index1, undo_context->swap.index2);
200 dynarray_swap(layer->ids, undo_context->swap.index1, undo_context->swap.index2);
201 dynarray_swap(layer->actions, undo_context->swap.index1, undo_context->swap.index2);
206 #define UNDO_PUSH(HISTORY, CONTEXT) \
208 UndoContext context = (CONTEXT); \
216 static int rect_layer_add_rect(RectLayer *layer,
219 UndoHistory *undo_history)
223 if (dynarray_push(layer->rects, &rect) < 0) {
227 if (dynarray_push(layer->colors, &color) < 0) {
231 char id[ENTITY_MAX_ID_SIZE];
232 snprintf(id, ENTITY_MAX_ID_SIZE, "%s_%d",
233 layer->id_name_prefix,
234 layer->id_name_counter++);
235 if (dynarray_push(layer->ids, id)) {
239 dynarray_push_empty(layer->actions);
243 create_undo_add_context(
245 dynarray_count(layer->rects) - 1));
250 static int rect_layer_rect_at(RectLayer *layer, Vec2f position)
254 int n = (int) dynarray_count(layer->rects);
255 Rect *rects = dynarray_data(layer->rects);
257 for (int i = n - 1; i >= 0; --i) {
258 if (rect_contains_point(rects[i], position)) {
266 static void rect_layer_swap_elements(RectLayer *layer, size_t a, size_t b,
267 UndoHistory *undo_history)
270 trace_assert(a < dynarray_count(layer->rects));
271 trace_assert(b < dynarray_count(layer->rects));
273 dynarray_swap(layer->rects, a, b);
274 dynarray_swap(layer->colors, a, b);
275 dynarray_swap(layer->ids, a, b);
276 dynarray_swap(layer->actions, a, b);
278 UNDO_PUSH(undo_history, create_undo_swap_context(layer, a, b));
281 static int rect_layer_delete_rect_at(RectLayer *layer,
283 UndoHistory *undo_history)
287 UNDO_PUSH(undo_history, create_undo_delete_context(layer));
289 dynarray_delete_at(layer->rects, i);
290 dynarray_delete_at(layer->colors, i);
291 dynarray_delete_at(layer->ids, i);
292 dynarray_delete_at(layer->actions, i);
297 // TODO: rect in calc_resize_mask is not mapped to the screen
298 static int calc_resize_mask(Vec2f point, Rect rect)
301 for (Rect_side side = 0; side < RECT_SIDE_N; ++side) {
302 if (rect_side_distance(rect, point, side) < RECT_LAYER_SELECTION_THICCNESS) {
303 mask = mask | (1 << side);
309 static int rect_layer_event_idle(RectLayer *layer,
310 const SDL_Event *event,
311 const Camera *camera,
312 UndoHistory *undo_history)
316 trace_assert(camera);
318 int color_changed = 0;
319 if (color_picker_event(&layer->color_picker, event, camera, &color_changed) < 0) {
324 if (layer->selection >= 0) {
325 dynarray_copy_to(layer->colors, &layer->inter_color, (size_t)layer->selection);
326 layer->state = RECT_LAYER_RECOLOR;
331 switch (event->type) {
332 case SDL_MOUSEBUTTONDOWN: {
333 switch (event->button.button) {
334 case SDL_BUTTON_LEFT: {
335 Vec2f position = camera_map_screen(
339 int rect_at_position =
340 rect_layer_rect_at(layer, position);
342 Rect *rects = dynarray_data(layer->rects);
343 Color *colors = dynarray_data(layer->colors);
345 if (layer->selection >= 0 &&
346 layer->selection == rect_at_position &&
347 (layer->resize_mask = calc_resize_mask(position, rects[layer->selection]))) {
348 layer->state = RECT_LAYER_RESIZE;
349 dynarray_copy_to(layer->rects, &layer->inter_rect, (size_t) layer->selection);
350 } else if (rect_at_position >= 0) {
351 layer->selection = rect_at_position;
352 layer->state = RECT_LAYER_MOVE;
357 rects[layer->selection].x,
358 rects[layer->selection].y));
359 layer->color_picker =
360 create_color_picker_from_rgba(colors[rect_at_position]);
362 dynarray_copy_to(layer->rects, &layer->inter_rect, (size_t) rect_at_position);
364 layer->selection = rect_at_position;
366 if (layer->selection < 0) {
367 layer->state = RECT_LAYER_CREATE;
368 layer->create_begin = position;
369 layer->create_end = position;
377 switch (event->key.keysym.sym) {
379 if ((event->key.keysym.mod & KMOD_SHIFT)
380 && (layer->selection >= 0)
381 && ((size_t)(layer->selection + 1) < dynarray_count(layer->rects))) {
382 rect_layer_swap_elements(
384 (size_t) layer->selection,
385 (size_t) layer->selection + 1,
392 if ((event->key.keysym.mod & KMOD_SHIFT)
393 && (layer->selection > 0)
394 && ((size_t) layer->selection < dynarray_count(layer->rects))) {
395 rect_layer_swap_elements(
397 (size_t) layer->selection,
398 (size_t) layer->selection - 1,
405 if (layer->selection >= 0) {
406 rect_layer_delete_rect_at(layer, (size_t) layer->selection, undo_history);
407 layer->selection = -1;
412 if (layer->selection >= 0) {
413 const char *ids = dynarray_data(layer->ids);
414 Color *colors = dynarray_data(layer->colors);
417 layer->id_edit_field,
418 RECT_LAYER_ID_LABEL_SIZE,
419 color_invert(colors[layer->selection]));
421 layer->state = RECT_LAYER_ID_RENAME;
423 layer->id_edit_field,
424 ids + layer->selection * ENTITY_MAX_ID_SIZE);
425 SDL_StartTextInput();
430 if ((event->key.keysym.mod & KMOD_LCTRL) && layer->selection >= 0) {
432 dynarray_copy_to(layer->rects, &clipboard_rect, (size_t)layer->selection);
433 dynarray_copy_to(layer->colors, &clipboard_color, (size_t)layer->selection);
438 if ((event->key.keysym.mod & KMOD_LCTRL) && clipboard) {
440 SDL_GetMouseState(&x, &y);
441 Vec2f position = camera_map_screen(camera, x, y);
445 rect(position.x, position.y,
446 clipboard_rect.w, clipboard_rect.h),
458 static int rect_layer_event_create(RectLayer *layer,
459 const SDL_Event *event,
460 const Camera *camera,
461 UndoHistory *undo_history)
465 trace_assert(camera);
467 switch (event->type) {
468 case SDL_MOUSEBUTTONUP: {
469 switch (event->button.button) {
470 case SDL_BUTTON_LEFT: {
471 const Rect real_rect =
475 const float area = real_rect.w * real_rect.h;
477 if (area >= CREATE_AREA_THRESHOLD) {
481 color_picker_rgba(&layer->color_picker),
484 log_info("The area is too small %f. Such small box won't be created.\n", area);
486 layer->state = RECT_LAYER_IDLE;
491 case SDL_MOUSEMOTION: {
492 layer->create_end = camera_map_screen(
501 static int rect_layer_event_resize(RectLayer *layer,
502 const SDL_Event *event,
503 const Camera *camera,
504 UndoHistory *undo_history)
508 trace_assert(camera);
509 trace_assert(layer->selection >= 0);
511 Rect *rects = dynarray_data(layer->rects);
513 switch (event->type) {
514 case SDL_MOUSEMOTION: {
515 Vec2f position = camera_map_screen(
520 switch (layer->resize_mask) {
522 layer->inter_rect = rect_from_points(
523 vec(rects[layer->selection].x, position.y),
524 rect_position2(rects[layer->selection]));
528 layer->inter_rect = rect_from_points(
529 vec(position.x, rects[layer->selection].y),
530 rect_position2(rects[layer->selection]));
534 layer->inter_rect = rect_from_points(
535 rect_position(rects[layer->selection]),
536 vec(rects[layer->selection].x + rects[layer->selection].w,
541 layer->inter_rect = rect_from_points(
542 rect_position(rects[layer->selection]),
544 rects[layer->selection].y + rects[layer->selection].h));
549 case SDL_MOUSEBUTTONUP: {
550 layer->state = RECT_LAYER_IDLE;
551 UNDO_PUSH(undo_history, create_undo_update_context(layer));
552 dynarray_replace_at(layer->rects, (size_t) layer->selection, &layer->inter_rect);
559 static int rect_layer_event_move(RectLayer *layer,
560 const SDL_Event *event,
561 const Camera *camera,
562 UndoHistory *undo_history)
566 trace_assert(camera);
567 trace_assert(layer->selection >= 0);
569 Rect *rects = dynarray_data(layer->rects);
571 switch (event->type) {
572 case SDL_MOUSEMOTION: {
573 const Uint8 *state = SDL_GetKeyboardState(NULL);
574 const Vec2f mouse_pos = vec_sub(
581 if (!(state[SDL_SCANCODE_LCTRL] || state[SDL_SCANCODE_RCTRL])) {
582 layer->inter_rect.x = mouse_pos.x;
583 layer->inter_rect.y = mouse_pos.y;
585 const Vec2f rect_pos = rect_position(rects[layer->selection]);
587 const float dx = fabsf(rect_pos.x - mouse_pos.x);
588 const float dy = fabsf(rect_pos.y - mouse_pos.y);
591 layer->inter_rect.x = mouse_pos.x;
592 layer->inter_rect.y = rect_pos.y;
594 layer->inter_rect.x = rect_pos.x;
595 layer->inter_rect.y = mouse_pos.y;
600 case SDL_MOUSEBUTTONUP: {
601 layer->state = RECT_LAYER_IDLE;
603 float distance = vec_length(
604 vec_sub(rect_position(layer->inter_rect),
605 rect_position(rects[layer->selection])));
607 if (distance > 1e-6) {
608 UNDO_PUSH(undo_history, create_undo_update_context(layer));
609 dynarray_replace_at(layer->rects, (size_t) layer->selection, &layer->inter_rect);
616 static int rect_layer_event_id_rename(RectLayer *layer,
617 const SDL_Event *event,
618 const Camera *camera,
619 UndoHistory *undo_history)
623 trace_assert(camera);
624 trace_assert(layer->selection >= 0);
626 switch (event->type) {
628 switch (event->key.keysym.sym) {
630 UNDO_PUSH(undo_history, create_undo_update_context(layer));
632 char *id = dynarray_pointer_at(layer->ids, (size_t)layer->selection);
633 memset(id, 0, ENTITY_MAX_ID_SIZE);
634 memcpy(id, edit_field_as_text(layer->id_edit_field), ENTITY_MAX_ID_SIZE - 1);
635 layer->state = RECT_LAYER_IDLE;
640 layer->state = RECT_LAYER_IDLE;
647 return edit_field_event(layer->id_edit_field, event);
650 LayerPtr rect_layer_as_layer(RectLayer *rect_layer)
659 RectLayer *create_rect_layer(const char *id_name_prefix)
661 Lt *lt = create_lt();
663 RectLayer *layer = PUSH_LT(lt, nth_calloc(1, sizeof(RectLayer)), free);
669 layer->ids = PUSH_LT(
671 create_dynarray(sizeof(char) * ENTITY_MAX_ID_SIZE),
673 if (layer->ids == NULL) {
677 layer->rects = PUSH_LT(
679 create_dynarray(sizeof(Rect)),
681 if (layer->rects == NULL) {
685 layer->colors = PUSH_LT(
687 create_dynarray(sizeof(Color)),
689 if (layer->colors == NULL) {
693 layer->actions = PUSH_LT(
695 create_dynarray(sizeof(Action)),
697 if (layer->actions == NULL) {
701 layer->id_edit_field = PUSH_LT(
704 RECT_LAYER_ID_LABEL_SIZE,
707 if (layer->id_edit_field == NULL) {
716 sizeof(Grid) + sizeof(Widget*) * RECT_LAYER_GRID_ROWS * RECT_LAYER_GRID_COLUMNS),
718 if (layer->grid == NULL) {
721 layer->grid->rows = RECT_LAYER_GRID_ROWS;
722 layer->grid->columns = RECT_LAYER_GRID_COLUMNS;
723 grid_put_widget(layer->grid, &layer->action_picker.widget, 0, RECT_LAYER_GRID_COLUMNS - 1);
725 layer->color_picker = create_color_picker_from_rgba(rgba(1.0f, 0.0f, 0.0f, 1.0f));
726 layer->selection = -1;
727 layer->id_name_prefix = id_name_prefix;
732 RectLayer *create_rect_layer_from_line_stream(LineStream *line_stream, const char *id_name_prefix)
734 trace_assert(line_stream);
736 RectLayer *layer = create_rect_layer(id_name_prefix);
741 const char *line = line_stream_next(line_stream);
743 RETURN_LT(layer->lt, NULL);
747 if (sscanf(line, "%zu", &count) < 0) {
748 RETURN_LT(layer->lt, NULL);
751 for (size_t i = 0; i < count; ++i) {
752 line = line_stream_next(line_stream);
754 RETURN_LT(layer->lt, NULL);
759 char id[ENTITY_MAX_ID_SIZE];
763 "%"STRINGIFY(ENTITY_MAX_ID_SIZE)"s%f%f%f%f%6s%n",
768 log_fail("%s\n", strerror(errno));
769 RETURN_LT(layer->lt, NULL);
773 Color color = hexstr(hex);
774 dynarray_push(layer->rects, &rect);
775 dynarray_push(layer->ids, id);
776 dynarray_push(layer->colors, &color);
783 if (sscanf(line, "%d%n", (int*)&action.type, &n) > 0) {
785 switch (action.type) {
786 case ACTION_NONE: break;
788 case ACTION_TOGGLE_GOAL:
789 case ACTION_HIDE_LABEL: {
790 if (sscanf(line, "%"STRINGIFY(ENTITY_MAX_ID_SIZE)"s", action.entity_id) <= 0) {
791 log_fail("%s\n", strerror(errno));
792 RETURN_LT(layer->lt, NULL);
796 case ACTION_N: break;
800 dynarray_push(layer->actions, &action);
806 void destroy_rect_layer(RectLayer *layer)
809 RETURN_LT0(layer->lt);
812 int rect_layer_render(const RectLayer *layer, const Camera *camera, int active)
815 trace_assert(camera);
817 const size_t n = dynarray_count(layer->rects);
818 Rect *rects = dynarray_data(layer->rects);
819 Color *colors = dynarray_data(layer->colors);
820 const char *ids = dynarray_data(layer->ids);
823 for (size_t i = 0; i < n; ++i) {
824 Rect rect = rects[i];
825 Color color = colors[i];
827 if (layer->selection == (int) i) {
828 if (layer->state == RECT_LAYER_RESIZE || layer->state == RECT_LAYER_MOVE) {
829 rect = layer->inter_rect;
832 if (layer->state == RECT_LAYER_RECOLOR) {
833 color = layer->inter_color;
838 if (camera_fill_rect(
843 rgba(1.0f, 1.0f, 1.0f, active ? 1.0f : 0.5f))) < 0) {
849 if (active && layer->selection >= 0) {
850 Rect rect = rects[layer->selection];
851 Color color = colors[layer->selection];
853 if (layer->state == RECT_LAYER_RESIZE || layer->state == RECT_LAYER_MOVE) {
854 rect = layer->inter_rect;
857 if (layer->state == RECT_LAYER_RECOLOR) {
858 color = layer->inter_color;
861 const Rect overlay_rect =
863 camera_rect(camera, rect),
864 -RECT_LAYER_SELECTION_THICCNESS * 0.5f);
865 const Color overlay_color = color_invert(color);
868 if (camera_draw_thicc_rect_screen(
872 RECT_LAYER_SELECTION_THICCNESS) < 0) {
876 const Vec2f rect_id_pos = vec_sub(
879 RECT_LAYER_ID_LABEL_SIZE,
880 vec(0.0f, FONT_CHAR_HEIGHT)));
883 if (layer->state == RECT_LAYER_ID_RENAME) {
884 // ID renaming Edit Field
885 if (edit_field_render_world(
886 layer->id_edit_field,
893 if (camera_render_text(
895 ids + layer->selection * ENTITY_MAX_ID_SIZE,
896 RECT_LAYER_ID_LABEL_SIZE,
905 const Color color = color_picker_rgba(&layer->color_picker);
906 if (layer->state == RECT_LAYER_CREATE) {
907 if (camera_fill_rect(camera, rect_from_points(layer->create_begin, layer->create_end), color) < 0) {
912 if (active && color_picker_render(&layer->color_picker, camera) < 0) {
920 int rect_layer_event_recolor(RectLayer *layer,
921 const SDL_Event *event,
922 const Camera *camera,
923 UndoHistory *undo_history)
927 trace_assert(camera);
928 trace_assert(undo_history);
929 trace_assert(layer->selection >= 0);
931 int color_changed = 0;
932 if (color_picker_event(&layer->color_picker, event, camera, &color_changed) < 0) {
937 layer->inter_color = color_picker_rgba(&layer->color_picker);
939 if (!color_picker_drag(&layer->color_picker)) {
940 UNDO_PUSH(undo_history, create_undo_update_context(layer));
941 dynarray_replace_at(layer->colors, (size_t) layer->selection, &layer->inter_color);
942 layer->state = RECT_LAYER_IDLE;
949 int rect_layer_event(RectLayer *layer,
950 const SDL_Event *event,
951 const Camera *camera,
952 UndoHistory *undo_history)
956 trace_assert(undo_history);
958 switch (event->type) {
959 case SDL_WINDOWEVENT: {
960 switch (event->window.event) {
961 case SDL_WINDOWEVENT_RESIZED: {
962 grid_relayout(layer->grid, rect(0.0f, 0.0f,
963 (float) event->window.data1,
964 (float) event->window.data2));
970 switch (layer->state) {
971 case RECT_LAYER_IDLE:
972 return rect_layer_event_idle(layer, event, camera, undo_history);
974 case RECT_LAYER_CREATE:
975 return rect_layer_event_create(layer, event, camera, undo_history);
977 case RECT_LAYER_RESIZE:
978 return rect_layer_event_resize(layer, event, camera, undo_history);
980 case RECT_LAYER_MOVE:
981 return rect_layer_event_move(layer, event, camera, undo_history);
983 case RECT_LAYER_ID_RENAME:
984 return rect_layer_event_id_rename(layer, event, camera, undo_history);
986 case RECT_LAYER_RECOLOR:
987 return rect_layer_event_recolor(layer, event, camera, undo_history);
993 size_t rect_layer_count(const RectLayer *layer)
995 return dynarray_count(layer->rects);
998 const Rect *rect_layer_rects(const RectLayer *layer)
1000 return dynarray_data(layer->rects);
1003 const Color *rect_layer_colors(const RectLayer *layer)
1005 return dynarray_data(layer->colors);
1008 const char *rect_layer_ids(const RectLayer *layer)
1010 return dynarray_data(layer->ids);
1013 int rect_layer_dump_stream(const RectLayer *layer, FILE *filedump)
1015 trace_assert(layer);
1016 trace_assert(filedump);
1018 size_t n = dynarray_count(layer->ids);
1019 char *ids = dynarray_data(layer->ids);
1020 Rect *rects = dynarray_data(layer->rects);
1021 Color *colors = dynarray_data(layer->colors);
1022 Action *actions = dynarray_data(layer->actions);
1024 fprintf(filedump, "%zd\n", n);
1025 for (size_t i = 0; i < n; ++i) {
1026 fprintf(filedump, "%s %f %f %f %f ",
1027 ids + ENTITY_MAX_ID_SIZE * i,
1028 rects[i].x, rects[i].y, rects[i].w, rects[i].h);
1029 color_hex_to_stream(colors[i], filedump);
1031 switch (actions[i].type) {
1032 case ACTION_NONE: {} break;
1034 case ACTION_TOGGLE_GOAL:
1035 case ACTION_HIDE_LABEL: {
1036 fprintf(filedump, " %d %.*s",
1037 (int)actions[i].type,
1038 ENTITY_MAX_ID_SIZE, actions[i].entity_id);
1040 case ACTION_N: break;
1043 fprintf(filedump, "\n");
1049 const Action *rect_layer_actions(const RectLayer *layer)
1051 return dynarray_data(layer->actions);