#define WHITE_SPRITE_ID 511

static String ui_shader_vert_win32 = lit_str(
                                             "#version 330 core\n"
                                             "layout (location = 0) in vec3 a_pos;\n"
                                             "layout (location = 1) in vec2 a_uv;\n"
                                             "layout (location = 2) in vec4 a_color;\n"
                                             "out vec2 uv;\n"
                                             "out vec4 color;\n"
                                             "uniform mat4 proj;\n"
                                             "void main(void) {\n"
                                             "  gl_Position = proj * vec4(a_pos, 1.0);\n"
                                             "  uv = a_uv;\n"
                                             "  color = a_color;\n"
                                             "}"
                                             );

static String ui_shader_vert_wasm = lit_str(
                                            "precision highp float;\n"
                                            "attribute vec3 a_pos;\n"
                                            "attribute vec2 a_uv;\n"
                                            "attribute vec4 a_color;\n"
                                            "varying vec2 uv;\n"
                                            "varying vec4 color;\n"
                                            "uniform mat4 proj;\n"
                                            "void main(void) {\n"
                                            "  gl_Position = proj * vec4(a_pos, 1.0);\n"
                                            "  uv = a_uv;\n"
                                            "  color = a_color;\n"
                                            "}"
                                            );

static String ui_shader_frag_win32 = lit_str(
                                             "#version 330 core\n"
                                             "in vec2 uv;\n"
                                             "in vec4 color;\n"
                                             "out vec4 FragColor;\n"
                                             "uniform sampler2D texture;\n"
                                             "void main(void) {\n"
                                             "  FragColor = texture(texture, uv) * color;\n"
                                             "}"
                                             );

static String ui_shader_frag_wasm = lit_str(
                                            "precision highp float;\n"
                                            "varying vec2 uv;\n"
                                            "varying vec4 color;\n"
                                            "uniform sampler2D texture;\n"
                                            "void main(void) {\n"
                                            "  gl_FragColor = texture2D(texture, uv) * color;\n"
                                            "}"
                                            );

internal UI
ui_create(u32 widget_pool_cap, u32 verts_cap, Input_State* input, Allocator* a)
{
  UI result = {};
  zero_struct(result);
  result.input = input;
  
  // Widgets
  result.widgets.free = allocator_alloc_array(a, UI_Widget, widget_pool_cap);
  result.widgets.free_cap = widget_pool_cap;
  result.widgets.states_cap = 3 * widget_pool_cap;
  result.widgets.states = allocator_alloc_array(a, UI_Widget_State, result.widgets.states_cap);
  result.widgets.states_hash = allocator_alloc_array(a, u32, result.widgets.states_cap);
  
  result.panels = bsp_create(a, 32);
  result.panels.root = bsp_push(&result.panels, {0}, {v2{},v2{1400, 800}}, 1);
  
  // Per Frame Vertex Buffer 
  Geo_Vertex_Buffer_Storage storage = (
                                       GeoVertexBufferStorage_Position |
                                       GeoVertexBufferStorage_TexCoord |
                                       GeoVertexBufferStorage_Color
                                       );
  result.geo = geo_quad_buffer_builder_create(a, verts_cap, storage, verts_cap * 2);
  
  result.per_frame_buffer = platform_geometry_buffer_create(
                                                            result.geo.buffer_vertex.values, 
                                                            result.geo.buffer_vertex.cap, 
                                                            result.geo.buffer_index.values, 
                                                            result.geo.buffer_index.cap
                                                            );
  
#if defined(PLATFORM_win32)
  String vert = ui_shader_vert_win32;
  String frag = ui_shader_frag_win32;
#elif defined(PLATFORM_wasm)
  String vert = ui_shader_vert_wasm;
  String frag = ui_shader_frag_wasm;
#endif
  
  String attrs[] = { lit_str("a_pos"), lit_str("a_uv"), lit_str("a_color") };
  String uniforms[] = { lit_str("proj") };
  result.shader = platform_shader_create(
                                         vert, frag, attrs, 3, uniforms, 1
                                         );
  
  platform_vertex_attrib_pointer(
                                 result.per_frame_buffer, result.shader, 3, result.shader.attrs[0], 9, 0
                                 );
  platform_vertex_attrib_pointer(
                                 result.per_frame_buffer, result.shader, 2, result.shader.attrs[1], 9, 3
                                 );
  platform_vertex_attrib_pointer(
                                 result.per_frame_buffer, result.shader, 4, result.shader.attrs[2], 9, 5
                                 );
  
  // Texture Atlas
  result.atlas = texture_atlas_create(1024, 1024, 512, permanent);
  result.atlas_texture = platform_texture_create(result.atlas.pixels, 1024, 1024, 1024);
  
  u32 white_sprite[] = { 
    0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF,
    0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF,
    0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF,
    0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF,
  };
  ui_sprite_register(&result, (u8*)white_sprite, 4, 4, WHITE_SPRITE_ID);
  
  ui_create_default_style_sheet();
  return result;
}

internal void
ui_quad_push(UI* ui, v3 pmin, v3 pmax, v2 tmin, v2 tmax, v4 c)
{
  v3 p0 = pmin;
  v3 p1 = v3{pmax.x, pmin.y, pmin.z};
  v3 p2 = pmax;
  v3 p3 = v3{pmin.x, pmax.y, pmin.z};
  v2 t0 = tmin;
  v2 t1 = v2{tmax.x,tmin.y};
  v2 t2 = tmax;
  v2 t3 = v2{tmin.x,tmax.y};
  geo_quad_buffer_builder_push(&ui->geo, p0, p1, p2, p3, t0, t1, t2, t3, c);
}

internal void
ui_sprite_register(UI* ui, u8* pixels, u32 w, u32 h, u32 id)
{
  texture_atlas_register(&ui->atlas, pixels, w, h, id, v2{0,0}, TextureAtlasRegistration_PixelFormat_RGBA);
  platform_texture_update(ui->atlas_texture, ui->atlas.pixels, ui->atlas.width, ui->atlas.height, ui->atlas.width);
}

internal void 
ui_sprite_push(UI* ui, v3 pmin, v3 pmax, u32 id, v4 color)
{
  Texture_Atlas_Sprite sprite = texture_atlas_sprite_get(&ui->atlas, id);
  v4 uv = texture_atlas_sprite_get_uvs(&ui->atlas, sprite);
  pmin.XY += sprite.draw_offset;
  pmax.XY += sprite.draw_offset;
  ui_quad_push(ui, pmin, pmax, uv.xy, uv.zw, color);
}

internal void 
ui_sprite_push(UI* ui, v3 pmin, v3 pmax, u32 id)
{
  ui_sprite_push(ui, pmin, pmax, id, v4{1,1,1,1});
}

struct UI_Char_Draw_Cmd
{
  v4 uv;
  v3 pmin;
  v3 pmax;
  v3 baseline_after;
};

internal UI_Char_Draw_Cmd
ui_sprite_char_get_draw_cmd(UI* ui, v3 at, u32 codepoint)
{
  UI_Char_Draw_Cmd result = {};
  zero_struct(result);
  
  Texture_Atlas_Sprite sprite = texture_atlas_sprite_get(&ui->atlas, codepoint);
  result.uv = texture_atlas_sprite_get_uvs(&ui->atlas, sprite);
  
  v3 dim = v3{ 
    (r32)(sprite.max_x - sprite.min_x), 
    (r32)(sprite.max_y - sprite.min_y),
    0,
  };
  result.pmin = at;
  result.pmin.XY += sprite.draw_offset;
  result.pmin.XY = v2_floor(result.pmin.XY);
  result.pmax = result.pmin + dim;
  
  result.baseline_after = v3{ result.pmax.x, at.y, at.z };
  
  return result;
}

internal void
ui_frame_prepare(UI* ui, v2 window_dim)
{
  ui->geo.buffer_vertex.len = 0;
  ui->geo.buffer_index.len = 0;
  
  ui->widgets.free_len = 0;
  ui->widgets.active_parent = 0;
  ui->widgets.root = ui_widget_pool_push(&ui->widgets, lit_str("root"));
  ui->widgets.active_parent = ui->widgets.root;
  
  BSP_Node* panel_root = bsp_get(&ui->panels, ui->panels.root);
  if (window_dim.x != 0 && window_dim.y != 0 && window_dim != panel_root->area.max)
  {
    BSP_Area area = {};
    area.min = v2{0,0};
    area.max = window_dim;
    bsp_node_area_update(&ui->panels, ui->panels.root, area);
  }
  
  v2 half_d = window_dim * 0.5f;
  ui->proj = HMM_Orthographic(0, window_dim.x, window_dim.y, 0, 0.01f, 100);
  
  if (ui->widget_next_hot.value != 0)
  {
    ui->widget_next_hot_frames += 1;
    if (ui->widget_next_hot_frames > 1) ui_widget_next_hot_set(ui, 0);
  }
  if (ui->widget_hot.value != 0)
  {
    ui->widget_hot_frames += 1;
    if (ui->widget_hot_frames > 1) ui_widget_hot_set(ui, 0);
  }
}

global bool show = false;

internal void
ui_draw_panel(BSP* tree, BSP_Node_Id id, BSP_Node* node, u8* user_data)
{
  if (node->split.kind != BSPSplit_None) return;
  UI* ui = (UI*)user_data;
  BSP_Area area = node->area;
  
  if (ui->draw_panel_cb) ui->draw_panel_cb(ui->draw_panel_cb_data, id, *node, area);
  
  r32 z = -1;
  v3 l0p0 = v3{ area.min.x,     area.min.y, z }; // left side
  v3 l0p1 = v3{ area.min.x + 1, area.max.y, z }; 
  v3 l1p0 = v3{ area.max.x - 1, area.min.y, z }; // right side
  v3 l1p1 = v3{ area.max.x,     area.max.y, z };
  v3 l2p0 = v3{ area.min.x, area.min.y    , z }; // bottom side
  v3 l2p1 = v3{ area.max.x, area.min.y + 1, z };
  v3 l3p0 = v3{ area.min.x, area.max.y    , z }; // top side
  v3 l3p1 = v3{ area.max.x, area.max.y + 1, z };
  u32 sid = WHITE_SPRITE_ID;
  v4 c = WHITE_V4;
  if (rect2_contains(area.min, area.max, ui->input->frame_hot->mouse_pos))
  {
    c = PINK_V4;
  }
  
  ui_sprite_push(ui, l0p0, l0p1, sid, c);
  ui_sprite_push(ui, l1p0, l1p1, sid, c);
  ui_sprite_push(ui, l2p0, l2p1, sid, c);
  ui_sprite_push(ui, l3p0, l3p1, sid, c);
}

internal void
ui_draw(UI* ui)
{
  bsp_walk_inorder(&ui->panels, ui->panels.root, ui_draw_panel, (u8*)ui);
  
  u32 widget_count = ui->widgets.free_len;
  r32 range_min = -10;
  r32 range_max = -1;
  r32 range_step = (range_max - range_min) / (r32)(widget_count * 4);
  ui_widgets_to_geometry_recursive(ui, ui->widgets.root, -10, range_step);
  
  platform_geometry_buffer_update(
                                  &ui->per_frame_buffer, 
                                  (r32*)ui->geo.buffer_vertex.values,
                                  0,
                                  ui->geo.buffer_vertex.len * ui->geo.buffer_vertex.stride,
                                  ui->geo.buffer_index.values,
                                  0,
                                  ui->geo.buffer_index.len
                                  );
  platform_shader_bind(ui->shader);
  platform_set_uniform(ui->shader, 0, ui->proj);
  platform_texture_bind(ui->atlas_texture);
  platform_geometry_bind(ui->per_frame_buffer);
  platform_geometry_draw(ui->per_frame_buffer, ui->geo.buffer_index.len);
}

////////////////////////////////////////////
// Widgets

internal UI_Widget_Id
ui_widget_id_create(String string, u32 index)
{
  assert(string.len != 0 && string.str != 0);
  UI_Widget_Id result = {};
  zero_struct(result);
  result.value = hash_djb2_to_u32(string);
  result.index = index;
  return result;
}

internal UI_Widget_State*
ui_widget_state_get(UI_Widget_Pool* pool, UI_Widget_Id id)
{
  u32 index = hash_table_find(pool->states_hash, pool->states_cap, id.value);
  assert(index != pool->states_cap);
  UI_Widget_State* result = pool->states + index;
  return result;
}
internal UI_Widget_State*
ui_widget_state_get(UI* ui, UI_Widget_Id id)
{
  return ui_widget_state_get(&ui->widgets, id);
}

internal UI_Widget*
ui_widget_pool_push(UI_Widget_Pool* pool, String string)
{
  assert(pool->free_len < pool->free_cap);
  UI_Widget* result = pool->free + pool->free_len++;
  
  result->id = ui_widget_id_create(string, pool->free_len);
  result->parent = 0;
  result->next = 0;
  result->child_first = 0;
  result->child_last = 0;
  
  u32 index = hash_table_find(pool->states_hash, pool->states_cap, result->id.value);
  if (index == pool->states_cap)
  {
    index = hash_table_register(pool->states_hash, pool->states_cap, result->id.value);
  }
  assert(index != pool->states_cap);
  UI_Widget_State* state = pool->states + index;
  zero_struct(*state);
  
  if (pool->active_parent)
  {
    result->parent = pool->active_parent;
    sll_push(
             pool->active_parent->child_first, 
             pool->active_parent->child_last, 
             result
             );
  }
  pool->active_parent = result;
  
  return result;
}

internal void
ui_widget_pool_pop(UI_Widget_Pool* pool)
{
  if (pool->active_parent->parent)
  {
    pool->active_parent = pool->active_parent->parent;
  }
}

internal bool
ui_widget_id_equals(UI_Widget_Id a, UI_Widget_Id b)
{
  return (a.value == b.value);
}

internal bool
ui_widget_id_is_valid(UI_Widget_Id h)
{
  return h.value != 0;
}

internal void
ui_widget_next_hot_set(UI* ui, UI_Widget* w)
{
  if (w) {
    ui->widget_next_hot = w->id;
  } else {
    ui->widget_next_hot = UI_Widget_Id{0};
  }
  ui->widget_next_hot_frames = 0;
}

internal void
ui_widget_hot_set(UI* ui, UI_Widget* w)
{
  if (w) {
    ui->widget_hot = w->id;
  } else {
    ui->widget_hot = UI_Widget_Id{0};
  }
  ui->widget_hot_frames = 0;
}

internal UI_Widget_Result
ui_widget_push(UI* ui, UI_Widget_Desc desc)
{
  UI_Widget_Result result = {};
  zero_struct(result);
  v2 dim = desc.p_max - desc.p_min;
  if (dim.x == 0 || dim.y == 0) return result;
  
  UI_Widget* w = ui_widget_pool_push(&ui->widgets, desc.string);
  w->desc = desc;
  result.id = w->id;
  
  UI_Widget_State* state = ui_widget_state_get(ui, w->id);
  w->desc.fill_pct = state->scroll;
  
  v2 mouse_p = ui->input->frame_hot->mouse_pos;
  bool mouse_over = (
                     mouse_p.x >= desc.p_min.x && mouse_p.x <= desc.p_max.x &&
                     mouse_p.y >= desc.p_min.y && mouse_p.y <= desc.p_max.y
                     );
  
  UI_Widget_Style_Flags flags = desc.style.flags;
  UI_Widget_Style_Flags mask_drag = (UIWidgetStyle_MouseDragH | UIWidgetStyle_MouseDragV);
  UI_Widget_Style_Flags mask_hover = (mask_drag | UIWidgetStyle_MouseClick);
  if (has_flag(flags, mask_hover))
  {
    // CASES:
    // Mouse Over | Mouse Clicked | Is Next Hot | Response
    //      f     |      f        |      t      | clear next hot
    //      f     |      f        |      f      | do nothing
    //      f     |      t        |      f      | do nothing
    //      t     |      f        |      f      | beome next hot
    //      t     |      t        |      f      | become next hot
    //      t     |      t        |      t      | become hot
    
    if (mouse_over)
    {
      if (ui_widget_id_equals(w->id, ui->widget_next_hot))
      {
        if (input_key_is_down(ui->input, KeyCode_MouseLeftButton))
        {
          ui_widget_hot_set(ui, w);
          result.flags |= UIWidgetResult_MouseLeft_IsDown;
        }
        if (input_key_went_up(ui->input, KeyCode_MouseLeftButton))
        {
          result.flags |= UIWidgetResult_MouseLeft_WentUp;
          ui_widget_hot_set(ui, 0);
        }
      }
      
      if ((w->id.index >= ui->widget_next_hot.index) && ui->widget_hot.value == 0)
      {
        ui_widget_next_hot_set(ui, w);
      }
    }
    else
    {
      if (ui_widget_id_equals(w->id, ui->widget_next_hot))
      {
        ui_widget_next_hot_set(ui, 0);
      }
    }
  }
  
  if(ui_widget_id_equals(w->id, ui->widget_hot))
  {
    if (input_key_is_down(ui->input, KeyCode_MouseLeftButton))
    {
      ui_widget_next_hot_set(ui, w);
      ui_widget_hot_set(ui, w);
      result.flags |= UIWidgetResult_MouseLeft_IsDown;
    }
    
    if (has_flag(flags, mask_drag))
    {
      v2 drag_pct_mask = { 
        has_flag(flags, UIWidgetStyle_MouseDragH) ? 1.0f : 0.0f,
        has_flag(flags, UIWidgetStyle_MouseDragV) ? 1.0f : 0.0f
      };
      v2 drag = ui->input->frame_hot->mouse_pos - w->desc.p_min;
      drag = v2{ clamp(0, drag.x, w->desc.p_max.x), clamp(0, drag.y, w->desc.p_max.y) };
      drag *= drag_pct_mask;
      v2 drag_pct = drag / dim;
      drag_pct = v2{ clamp(0, drag_pct.x, 1), clamp(0, drag_pct.y, 1) };
      result.drag = drag_pct;
      
      state->scroll = drag_pct;
    }
  }
  
  return result;
}

internal void
ui_widget_pop(UI* ui, UI_Widget_Id widget_id)
{
  if (!ui_widget_id_is_valid(widget_id)) return;
  assert(ui_widget_id_equals(widget_id, ui->widgets.active_parent->id));
  ui_widget_pool_pop(&ui->widgets);
}

internal r32
ui_widgets_to_geometry_recursive(UI* ui, UI_Widget* widget, r32 z_start, r32 z_step)
{
  r32 z_at = z_start;
  for (UI_Widget* child = widget->child_first; child != 0; child = child->next)
  {
    UI_Widget_Desc desc = child->desc;
    v3 bg_min = v2_to_v3(desc.p_min, z_at); 
    v3 bg_max = v2_to_v3(desc.p_max, z_at); 
    
    v4 color_fg = desc.style.color_fg;
    v4 color_bg = desc.style.color_bg;
    if (ui_widget_id_equals(ui->widget_next_hot, child->id))
    {
      color_fg = desc.style.color_bg;
      color_bg = desc.style.color_fg;
    }
    if (ui_widget_id_equals(ui->widget_hot, child->id))
    {
      //color_fg = desc.style.color_fg;
      //color_bg = desc.style.color_bg;
    }
    
    if (has_flag(child->desc.style.flags, UIWidgetStyle_Outline))
    {
      ui_sprite_push(ui, bg_min, bg_max, WHITE_SPRITE_ID, color_fg);
      z_at += z_step;
      bg_min += v3{ 1, 1, 0};
      bg_max -= v3{ 1, 1, 0};
    }
    
    if (has_flag(child->desc.style.flags, UIWidgetStyle_Bg))
    {
      bg_min.z = z_at;
      bg_max.z = z_at;
      ui_sprite_push(ui, bg_min, bg_max, desc.style.sprite, color_bg);
      z_at += z_step;
    }
    
    if (has_flag(child->desc.style.flags, (UIWidgetStyle_FillH | UIWidgetStyle_FillV)))
    {
      v3 fill_min = {};
      zero_struct(fill_min);
      v3 fill_max = {};
      zero_struct(fill_max);
      if (has_flag(child->desc.style.flags, UIWidgetStyle_FillH))
      {
        r32 fill_x = HMM_Lerp(bg_min.x, child->desc.fill_pct.x, bg_max.x);
        
        if (has_flag(child->desc.style.flags, UIWidgetStyle_LineInsteadOfFill))
        {
          fill_min = v3{ fill_x,     bg_min.y, z_at };
          fill_max = v3{ fill_x + 1, bg_max.y, z_at };
        }
        else
        {
          fill_min = bg_min;
          fill_max = v3{ fill_x, bg_max.y, z_at };
        }
      }
      else if (has_flag(child->desc.style.flags, UIWidgetStyle_FillV))
      {
        r32 fill_y = HMM_Lerp(bg_min.y, child->desc.fill_pct.y, bg_max.y);
        
        if (has_flag(child->desc.style.flags, UIWidgetStyle_LineInsteadOfFill))
        {
          fill_min = v3{ bg_min.x, fill_y,     z_at };
          fill_max = v3{ bg_max.x, fill_y + 1, z_at };
        }
        else
        {
          fill_min = bg_min;
          fill_max = v3{ bg_max.x, fill_y, z_at };
        }
      }
      
      ui_sprite_push(ui, fill_min, fill_max, WHITE_SPRITE_ID, color_fg);
      z_at += z_step;
    }
    
    if (has_flag(child->desc.style.flags, UIWidgetStyle_TextWrap | UIWidgetStyle_TextClip))
    {
      r32 space_width = ui->font_space_width;
      r32 to_baseline = ui->font_line_gap + ui->font_ascent;
      v3 line_offset = { 5, 3 + to_baseline, 0 };
      r32 baseline_x_start = desc.p_min.x + line_offset.x;
      r32 baseline_y_start = desc.p_min.y + line_offset.y;
      v3 baseline = { baseline_x_start, baseline_y_start, z_at };
      for (u64 i = 0; i < child->desc.string.len; i++)
      {
        u8 at = child->desc.string.str[i];
        UI_Char_Draw_Cmd cmd = {};
        zero_struct(cmd);
        if (!char_is_space(at))
        {
          cmd = ui_sprite_char_get_draw_cmd(ui, baseline, (u32)at);
        }
        else
        {
          cmd.baseline_after = baseline;
          cmd.baseline_after.x += space_width;
        }
        
        if (cmd.baseline_after.x >= desc.p_max.x - 5)
        {
          if (has_flag(child->desc.style.flags, UIWidgetStyle_TextClip)) break;
          
          baseline.x = baseline_x_start;
          baseline.y += ui->font_ascent + ui->font_descent + ui->font_line_gap;
          cmd = ui_sprite_char_get_draw_cmd(ui, baseline, (u32)at);
        }
        
        if (!char_is_space(at))
        {
          ui_quad_push(ui, cmd.pmin, cmd.pmax, cmd.uv.xy, cmd.uv.zw, color_fg);
        }
        baseline = cmd.baseline_after;
      }
    }
    
    if (child->child_first)
    {
      z_at = ui_widgets_to_geometry_recursive(ui, child, z_at + z_step, z_step);
    }
    
    z_at += z_step;
  }
  return z_at;
}

///////////////////////////////////////////
// Layout Manager

internal void
ui_layout_set_row_info(UI* ui, UI_Layout* l)
{
  l->row_height = (ui->font_ascent + ui->font_descent + ui->font_line_gap + 15);
  l->row_gap = 2;
  l->col_gap = 2;
}

internal void
ui_layout_push(UI* ui, UI_Layout* layout)
{
  if (ui->layout)
  {
    layout->parent = ui->layout;
    ui->layout = layout;
  }
  else
  {
    ui->layout = layout;
    layout->parent = 0;
  }
}

internal void
ui_layout_pop(UI* ui)
{
  if (ui->layout) 
  {
    ui->layout = ui->layout->parent;
  }
}

internal void
ui_layout_row_begin(UI_Layout* layout, u32 cols)
{
  layout->mode = UILayout_Rows;
  layout->cols = cols;
}
internal void
ui_layout_row_begin(UI* ui, u32 cols) { ui_layout_row_begin(ui->layout, cols); }

internal void
ui_layout_row_end(UI_Layout* layout)
{
  layout->mode = UILayout_Columns;
}
internal void
ui_layout_row_end(UI* ui) { ui_layout_row_end(ui->layout); }

internal UI_Layout_Bounds
ui_layout_get_next(UI_Layout* layout)
{
  UI_Layout_Bounds result = {};
  zero_struct(result);
  if (layout->at.x >= layout->bounds_max.x || layout->at.y >= layout->bounds_max.y ||
      layout->at.y + layout->row_height > layout->bounds_max.y)
  {
    return result;
  }
  
  switch (layout->mode)
  {
    case UILayout_Columns:
    {
      result.min = layout->at;
      result.max = v2{ layout->bounds_max.x, layout->at.y + layout->row_height };
      layout->at = v2{ layout->bounds_min.x, result.max.y + layout->row_gap};
    } break;
    
    case UILayout_Rows:
    {
      r32 col_width = (layout->bounds_max.x - layout->bounds_min.x) / layout->cols;
      col_width -= (layout->cols - 1) * layout->col_gap;
      result.min = layout->at;
      result.max = v2{ layout->at.x + col_width, layout->at.y + layout->row_height };
      layout->at = v2{ result.max.x + layout->col_gap, layout->at.y };
      if (layout->at.x >= layout->bounds_max.x)
      {
        layout->at = v2{ 
          layout->bounds_min.x, 
          layout->at.y + layout->row_height + layout->row_gap
        };
      }
    } break;
    
    invalid_default_case;
  }
  
  if (result.min.x < layout->bounds_min.x || result.min.y < layout->bounds_min.y ||
      result.max.x < layout->bounds_min.x || result.max.y < layout->bounds_min.y)
  {
    zero_struct(result.min);
    zero_struct(result.max);
  }
  
  return result;
}
internal UI_Layout_Bounds 
ui_layout_get_next(UI* ui)  { return ui_layout_get_next(ui->layout); }


///////////////////////////////////////////
// Specific Widget Implementations
//
// These all rely on a layout manager to make calling them simpler

global UI_Style_Sheet ui_default_style_sheet = {};

internal void
ui_create_default_style_sheet()
{
  ui_default_style_sheet.styles[UIWidget_Text] = {
    (UIWidgetStyle_TextWrap), v4{0,0,0,0}, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_Button] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_Outline | UIWidgetStyle_MouseClick), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_Toggle] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_MouseClick), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_Menu] = ui_default_style_sheet.styles[UIWidget_Toggle];
  ui_default_style_sheet.styles[UIWidget_Dropdown] = ui_default_style_sheet.styles[UIWidget_Toggle];
  
  ui_default_style_sheet.styles[UIWidget_HSlider] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_Outline | UIWidgetStyle_MouseDragH | UIWidgetStyle_FillH ), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_VSlider] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_Outline | UIWidgetStyle_MouseDragV | UIWidgetStyle_FillV ), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_HScroll] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_Outline | UIWidgetStyle_MouseDragH | UIWidgetStyle_FillH | UIWidgetStyle_LineInsteadOfFill ), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  ui_default_style_sheet.styles[UIWidget_VScroll] = {
    (UIWidgetStyle_TextClip | UIWidgetStyle_Bg | UIWidgetStyle_Outline | UIWidgetStyle_MouseDragV | UIWidgetStyle_FillV | UIWidgetStyle_LineInsteadOfFill ), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  
  ui_default_style_sheet.styles[UIWidget_Window] = {
    (UIWidgetStyle_TextWrap), BLACK_V4, WHITE_V4, WHITE_SPRITE_ID
  };
  
};

internal UI_Widget_Style
ui_get_style(UI* ui, UI_Widget_Kind kind)
{
  if (ui->style_sheet) return ui->style_sheet->styles[kind];
  return ui_default_style_sheet.styles[kind];
}

internal UI_Widget_Desc
ui_layout_next_widget(UI* ui, UI_Widget_Kind kind)
{
  UI_Layout_Bounds b = ui_layout_get_next(ui);
  UI_Widget_Desc d = {};
  zero_struct(d);
  d.p_min = b.min;
  d.p_max = b.max;
  d.style = ui_get_style(ui, kind);
  return d;
}

internal void
ui_text(UI* ui, String string, v4 color)
{
  UI_Widget_Desc d = ui_layout_next_widget(ui, UIWidget_Text);
  d.string = string;
  d.style.color_fg = color;
  UI_Widget_Result r = ui_widget_push(ui, d);
  ui_widget_pop(ui, r.id);
}

internal void
ui_text(UI* ui, String string)
{
  UI_Widget_Desc d = ui_layout_next_widget(ui, UIWidget_Text);
  d.string = string;
  UI_Widget_Result r = ui_widget_push(ui, d);
  ui_widget_pop(ui, r.id);
}

internal void
ui_text_f(UI* ui, char* fmt, ...)
{
  scratch_get(scratch);
  
  va_list args;
  va_start(args, fmt);
  String string = string_fv(scratch.a, fmt, args);
  va_end(args);
  
  return ui_text(ui, string);
}

internal bool
ui_button(UI* ui, String string)
{
  UI_Widget_Desc d = ui_layout_next_widget(ui, UIWidget_Button);
  d.string = string;
  UI_Widget_Result r = ui_widget_push(ui, d);
  ui_widget_pop(ui, r.id);
  return has_flag(r.flags, UIWidgetResult_MouseLeft_WentUp);
}

internal bool
ui_toggle(UI* ui, String string, bool value)
{
  UI_Widget_Desc d = ui_layout_next_widget(ui, UIWidget_Button);
  if (value) {
    v4 t = d.style.color_fg;
    d.style.color_fg = d.style.color_bg;
    d.style.color_bg = t;
  }
  d.string = string;
  UI_Widget_Result r = ui_widget_push(ui, d);
  ui_widget_pop(ui, r.id);
  bool result = value;
  if (has_flag(r.flags, UIWidgetResult_MouseLeft_WentUp)) result = !result;
  return result;
}

internal UI_Layout*
ui_scroll_view_begin(UI* ui, String string, v2 bounds_min, v2 bounds_max, u32 rows)
{
  scratch_get(scratch);
  
  r32 scroll_bar_dim = 15;
  v2 scroll_bars_area = v2{0, 0};
  v2 scroll_area_min = bounds_min;
  v2 scroll_area_max = bounds_max - scroll_bars_area;
  v2 scroll_area_dim = scroll_area_max - scroll_area_min;
  
  v2 scroll_offset = {};
  zero_struct(scroll_offset);
  r32 rows_avail = floorf(scroll_area_dim.y / ui->layout->row_height);
  if (rows > rows_avail)
  {
    scroll_bars_area = v2{ scroll_bar_dim, 0};
    scroll_area_min = bounds_min;
    scroll_area_max = bounds_max - scroll_bars_area;
    scroll_area_dim = scroll_area_max - scroll_area_min;
    
    UI_Widget_Desc vscroll_d = {};
    zero_struct(vscroll_d);
    vscroll_d.p_min = { bounds_max.x - scroll_bar_dim, bounds_min.y };
    vscroll_d.p_max = { bounds_max.x, bounds_max.y };
    vscroll_d.style = ui_get_style(ui, UIWidget_VScroll);
    vscroll_d.string = string_f(scratch.a, "%.*s_vscroll", str_varg(string));
    UI_Widget_Result r = ui_widget_push(ui, vscroll_d);
    ui_widget_pop(ui, r.id);
    
    UI_Widget_State* vscroll_state = ui_widget_state_get(ui, r.id);
    scroll_offset.y = vscroll_state->scroll.y;
  }
  
  r32 rows_scroll_to = max(0, rows - (rows_avail - 1));
  r32 y_scroll_dist = rows_scroll_to * ui->layout->row_height;
  
  scroll_offset *= v2{ 0, y_scroll_dist };
  
  UI_Layout* layout = allocator_alloc_struct(scratch.a, UI_Layout);
  layout->mode = UILayout_Columns;
  layout->bounds_min = scroll_area_min;
  layout->bounds_max = scroll_area_max;
  ui_layout_set_row_info(ui, layout);
  layout->at = bounds_min - scroll_offset;
  ui_layout_push(ui, layout);
  
  return layout;
}

internal void
ui_scroll_view_end(UI* ui)
{
  ui_layout_pop(ui);
}

#if 0
internal bool
ui_dropdown_begin(UI* ui, String string, bool state)
{
  bool result = ui_toggle(ui, string, state);
  UI_Layout* layout = allocator_alloc_struct(scratch, UI_Layout);
  zero_struct(*layout);
  if (result)
  {
    ui_scroll_view_begin(ui, layout);
  }
  return result;
}

internal void
ui_dropdown_end(UI* ui, bool state)
{
  if (state)
  {
    ui_scroll_view_end(ui);
  }
}
#endif