pke: static-ui add flex_padding

4 files changed, 33 insertions, 7 deletions

diff --git a/src/serialization-static-ui.cpp b/src/serialization-static-ui.cpp
index 8c21d7c..ae948ab 100644
--- a/src/serialization-static-ui.cpp
+++ b/src/serialization-static-ui.cpp
@@ -309,6 +309,13 @@ pk_handle pke_serialize_ui_box(srlztn_serialize_helper *h, pke_ui_box *box) {
 		kve.val = s;
 		kve.end = SRLZTN_KVE_END;
 		pk_arr_append_t(&kvec.arr, kve);
+		kve.key = SRLZTN_UI_BOX_FLEX_PADDING;
+		len = snprintf(NULL, 0, "%f", box->flex_padding);
+		s = pk_new_arr<char>(len+1, h->bkt);
+		sprintf(s, "%f", box->flex_padding);
+		kve.val = s;
+		kve.end = SRLZTN_KVE_END;
+		pk_arr_append_t(&kvec.arr, kve);
 	}
 	{
 		kve.key = SRLZTN_UI_BOX_TYPE;
@@ -462,6 +469,14 @@ void pke_deserialize_ui_box(srlztn_deserialize_helper *h, pke_kve_container *kve
 			}
 			continue;
 		}
+		if (strstr(SRLZTN_UI_BOX_FLEX_PADDING, kve->key)) {
+			res = pk_stn(&bx.flex_padding, kve->val, &stn_end);
+			if (res != PK_STN_RES_SUCCESS) {
+				fprintf(stderr, "[pke_deserialize_ui_box] Failed to parse value for: '%s', err: %i\n", SRLZTN_UI_BOX_FLEX_PADDING, res);
+				continue;
+			}
+			continue;
+		}
 		if (strstr(SRLZTN_UI_BOX_FLEX_DIRECTION, kve->key)) {
 			res = pk_stn(&bx.flex_direction, kve->val, &stn_end);
 			if (res != PK_STN_RES_SUCCESS) {
@@ -564,6 +579,7 @@ void pke_deserialize_ui_box(srlztn_deserialize_helper *h, pke_kve_container *kve
 	box->min_size = bx.min_size;
 	box->max_size = bx.max_size;
 	box->flex_weight = bx.flex_weight;
+	box->flex_padding = bx.flex_padding;
 	box->type = bx.type;
 	box->flex_direction = bx.flex_direction;
 	box->layer = bx.layer;
diff --git a/src/serialization.hpp b/src/serialization.hpp
index a4ea7e4..777976b 100644
--- a/src/serialization.hpp
+++ b/src/serialization.hpp
@@ -55,6 +55,7 @@ iccsc SRLZTN_UI_BOX_POS_TOP_LEFT                 = "PosTopLeft:";
 iccsc SRLZTN_UI_BOX_MIN_SIZE                     = "MinSize:";
 iccsc SRLZTN_UI_BOX_MAX_SIZE                     = "MaxSize:";
 iccsc SRLZTN_UI_BOX_FLEX_WEIGHT                  = "FlexWeight:";
+iccsc SRLZTN_UI_BOX_FLEX_PADDING                 = "FlexPadding:";
 iccsc SRLZTN_UI_BOX_FLEX_DIRECTION               = "FlexDirection:";
 iccsc SRLZTN_UI_BOX_LAYER                        = "Layer:";
 iccsc SRLZTN_UI_BOX_TYPE                         = "Type:";
diff --git a/src/static-ui.cpp b/src/static-ui.cpp
index 2682de1..72bd298 100644
--- a/src/static-ui.cpp
+++ b/src/static-ui.cpp
@@ -256,9 +256,10 @@ struct pke_ui_flex_params {
 	float used_units;
 };
 
-bool pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_flex_params *flex_params, pke_ui_box *box) {
+bool pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_flex_params *flex_params, pke_ui_box *box, uint8_t child_index) {
 	assert(box != nullptr);
 	pke_ui_box_count_T i;
+	float flex_padding;
 	glm::vec2 px_size;
 	glm::vec2 px_min_size;
 	glm::vec2 px_max_size;
@@ -333,14 +334,17 @@ bool pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<
 	if (PK_HAS_FLAG(box->flags, PKE_UI_BOX_FLAG_POSITION_TYPE_FLEX)) {
 		assert(flex_params != nullptr);
 		assert(box->internal.parent != nullptr);
+		flex_padding = box->internal.parent->flex_padding * float(child_index + 1);
 		if (box->internal.parent->flex_direction == 0) {
 			px_size.x = flex_params->px_per_unit * box->flex_weight;
 			px_size.y = parent_size_padded.y;
 			box->internal.px_corner.x += flex_params->used_units * flex_params->px_per_unit;
+			box->internal.px_corner.x += flex_padding;
 		} else {
 			px_size.x = parent_size_padded.x;
 			px_size.y = flex_params->px_per_unit * box->flex_weight;
 			box->internal.px_corner.y += flex_params->used_units * flex_params->px_per_unit;
+			box->internal.px_corner.y += flex_padding;
 		}
 		flex_params->used_units += box->flex_weight;
 	} else if (PK_HAS_FLAG(box->flags, PKE_UI_BOX_FLAG_POSITION_TYPE_STATIC)) {
@@ -482,6 +486,7 @@ bool pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<
 void pke_ui_recalc_sizes_recursive(pk_arr_t<pke_ui_box_instance_buffer_item> &arr, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_box *box) {
 	PKE_UI_BOX_FLAG_T flags_masked;
 	uint8_t flex_count = 0;
+	float available_size = 0;
 	pke_ui_flex_params flex_params{};
 	flex_params.px_per_unit = 0;
 	flex_params.unit_total = 0;
@@ -507,14 +512,17 @@ void pke_ui_recalc_sizes_recursive(pk_arr_t<pke_ui_box_instance_buffer_item> &ar
 		return;
 	}
 	if (flex_count != 0) {
-		flex_params.px_per_unit =
+		assert(box->flex_padding == box->flex_padding); // NaN
+		available_size =
 			box->flex_direction == 0
-			? (box->internal.px_size.x - box->internal.px_padding_l - box->internal.px_padding_r) / flex_params.unit_total
-			: (box->internal.px_size.y - box->internal.px_padding_t - box->internal.px_padding_b) / flex_params.unit_total;
+			? (box->internal.px_size.x - box->internal.px_padding_l - box->internal.px_padding_r)
+			: (box->internal.px_size.y - box->internal.px_padding_t - box->internal.px_padding_b);
+		available_size -= (float)(flex_count+1) * (box->flex_padding);
+		flex_params.px_per_unit = available_size / flex_params.unit_total;
 	}
 
 	for (pke_ui_box_count_T i = 0; i < box->internal.h_children; ++i) {
-		if (!pke_ui_calc_px(arr, tmp_txtr_buffer, &flex_params, box->internal.children[i])) {
+		if (!pke_ui_calc_px(arr, tmp_txtr_buffer, &flex_params, box->internal.children[i], i)) {
 			i -= 1;
 			continue;
 		}
@@ -670,7 +678,7 @@ void pke_ui_tick(double delta) {
 			continue;
 		}
 		// root boxes marked for removal were already skipped
-		pke_ui_calc_px(arr, tmp_txtr_arr, nullptr, box);
+		pke_ui_calc_px(arr, tmp_txtr_arr, nullptr, box, 0);
 		pke_ui_recalc_sizes_recursive(arr, tmp_txtr_arr, box);
 	}
 
diff --git a/src/static-ui.hpp b/src/static-ui.hpp
index ec99456..e4c86af 100644
--- a/src/static-ui.hpp
+++ b/src/static-ui.hpp
@@ -113,6 +113,7 @@ struct pke_ui_box : public Entity_Base {
 	glm::vec2 min_size;
 	glm::vec2 max_size;
 	float flex_weight;
+	float flex_padding;
 	PKE_UI_BOX_TYPE type;
 	uint8_t flex_direction;
 	uint8_t layer;
@@ -194,7 +195,7 @@ pke_ui_box *pke_ui_box_new_child(pke_ui_box *parent, const PKE_UI_BOX_TYPE type
 void pke_ui_box_update_textures(pke_ui_box *box);
 
 #ifdef PKE_TEST_EXPOSE
-void pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_flex_params *flex_params, pke_ui_box *box);
+void pke_ui_calc_px(pk_arr_t<pke_ui_box_instance_buffer_item> &buffer, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_flex_params *flex_params, pke_ui_box *box, uint8_t child_index);
 void pke_ui_recalc_sizes_recursive(pk_arr_t<pke_ui_box_instance_buffer_item> &arr, pk_arr_t<std::pair<pke_ui_box*,pke_ui_box_instance_buffer_item>> &tmp_txtr_buffer, pke_ui_box *box);
 #endif