From 96b6c688470b4f99e1db1f8cd4ebf8526a9721c0 Mon Sep 17 00:00:00 2001
From: leowrites <leo@liuwork.ca>
Date: Wed, 16 Oct 2024 21:28:46 -0400
Subject: [PATCH] account for quotations and text styles when calculating text
 length

---
 memory-viz/src/memory_model.ts | 38 +++++++++++++++++++++-------------
 1 file changed, 24 insertions(+), 14 deletions(-)

diff --git a/memory-viz/src/memory_model.ts b/memory-viz/src/memory_model.ts
index d5fba9c..867260f 100644
--- a/memory-viz/src/memory_model.ts
+++ b/memory-viz/src/memory_model.ts
@@ -225,7 +225,8 @@ export class MemoryModel {
     ): Rect {
         let box_width = Math.max(
             this.obj_min_width,
-            this.getTextLength(String(value)) + this.obj_x_padding
+            this.getTextLength(`"${String(value)}"`, style.text_value) +
+                this.obj_x_padding
         );
         this.drawRect(
             x,
@@ -303,12 +304,12 @@ export class MemoryModel {
     ) {
         let id_box = Math.max(
             this.prop_min_width,
-            this.getTextLength(`id${id}`) + 10
+            this.getTextLength(`id${id}`, style.text_id) + 10
         );
 
         let type_box = Math.max(
             this.prop_min_width,
-            this.getTextLength(type) + 10
+            this.getTextLength(type, style.text_type) + 10
         );
 
         this.drawRect(x, y, id_box, this.prop_min_height, style.box_id);
@@ -380,7 +381,8 @@ export class MemoryModel {
         element_ids.forEach((v) => {
             box_width += Math.max(
                 this.item_min_width,
-                this.getTextLength(v === null ? "" : `id${v}`) + 10
+                this.getTextLength(v === null ? "" : `id${v}`, style.text_id) +
+                    10
             );
         });
 
@@ -419,7 +421,7 @@ export class MemoryModel {
             const idv = v === null ? "" : `id${v}`;
             const item_length = Math.max(
                 this.item_min_width,
-                this.getTextLength(idv) + 10
+                this.getTextLength(idv, style.text_id) + 10
             );
             this.drawRect(curr_x, item_y, item_length, this.item_min_height);
             this.drawText(
@@ -482,7 +484,8 @@ export class MemoryModel {
         element_ids.forEach((v) => {
             box_width += Math.max(
                 this.item_min_width,
-                this.getTextLength(v === null ? "" : `id${v}`) + 10
+                this.getTextLength(v === null ? "" : `id${v}`, style.text_id) +
+                    10
             );
         });
         box_width = Math.max(this.obj_min_width, box_width);
@@ -518,7 +521,7 @@ export class MemoryModel {
             const idv = v === null ? "" : `id${v}`;
             const item_length = Math.max(
                 this.item_min_width,
-                this.getTextLength(idv) + 10
+                this.getTextLength(idv, style.text_id) + 10
             );
             this.drawRect(curr_x, item_y, item_length, this.item_min_height);
             this.drawText(
@@ -587,11 +590,11 @@ export class MemoryModel {
 
             let key_box = Math.max(
                 this.item_min_width,
-                this.getTextLength(idk + 5)
+                this.getTextLength(idk + 5, style.text_id)
             );
             let value_box = Math.max(
                 this.item_min_width,
-                this.getTextLength(idv + 5)
+                this.getTextLength(idv + 5, style.text_value)
             );
 
             // Draw the rectangles representing the keys.
@@ -632,7 +635,7 @@ export class MemoryModel {
 
             let value_box = Math.max(
                 this.item_min_width,
-                this.getTextLength(idv + 5)
+                this.getTextLength(idv + 5, style.text_value)
             );
 
             // Draw the rectangle for values.
@@ -692,14 +695,17 @@ export class MemoryModel {
         let box_width = this.obj_min_width;
         let longest = 0;
         for (const attribute in attributes) {
-            longest = Math.max(longest, this.getTextLength(attribute));
+            longest = Math.max(
+                longest,
+                this.getTextLength(attribute, style.text_value)
+            );
         }
         if (longest > 0) {
             box_width = longest + this.item_min_width * 3;
         }
         box_width = Math.max(
             box_width,
-            this.prop_min_width + this.getTextLength(name) + 10
+            this.prop_min_width + this.getTextLength(name, style.text_type) + 10
         );
 
         let box_height = 0;
@@ -750,7 +756,7 @@ export class MemoryModel {
         }
 
         if (stack_frame) {
-            let text_length = this.getTextLength(name);
+            let text_length = this.getTextLength(name, style.text_type);
             this.drawRect(
                 x,
                 y,
@@ -847,7 +853,11 @@ export class MemoryModel {
      * Return the length of this text.
      * @param s - The given text.
      */
-    getTextLength(s: string): number {
+    getTextLength(s: string, textStyle?: CSS.PropertiesHyphen): number {
+        if (textStyle?.["font-size"]) {
+            // Note: this assumes font size is in px
+            return s.length * parseInt(textStyle["font-size"]) * 0.6;
+        }
         return s.length * 12;
     }