path: root/src/ui/render/RenderObject.cpp

#include "cru/ui/render/RenderObject.h"

#include "cru/common/Logger.h"
#include "cru/platform/GraphicsBase.h"
#include "cru/platform/graphics/util/Painter.h"
#include "cru/ui/Base.h"
#include "cru/ui/DebugFlags.h"
#include "cru/ui/host/WindowHost.h"

#include <algorithm>
#include <string>
#include <string_view>
#include <vector>

namespace cru::ui::render {
void RenderObject::SetAttachedControl(controls::Control* new_control) {
  control_ = new_control;
  OnAttachedControlChanged(new_control);
}

void RenderObject::AddChild(RenderObject* render_object, const Index position) {
  Expects(child_mode_ != ChildMode::None);
  Expects(!(child_mode_ == ChildMode::Single && children_.size() > 0));

  Expects(render_object->GetParent() ==
          nullptr);        // Render object already has a parent.
  Expects(position >= 0);  // Position index is less than 0.
  Expects(
      position <=
      static_cast<Index>(children_.size()));  // Position index is out of bound.

  children_.insert(children_.cbegin() + position, render_object);
  render_object->SetParent(this);
  render_object->SetWindowHostRecursive(GetWindowHost());
  OnAddChild(render_object, position);
}

void RenderObject::RemoveChild(const Index position) {
  Expects(position >= 0);  // Position index is less than 0.
  Expects(position < static_cast<Index>(
                         children_.size()));  // Position index is out of bound.

  const auto i = children_.cbegin() + position;
  const auto removed_child = *i;
  children_.erase(i);
  removed_child->SetParent(nullptr);
  removed_child->SetWindowHostRecursive(nullptr);
  OnRemoveChild(removed_child, position);
}

RenderObject* RenderObject::GetFirstChild() const {
  const auto& children = GetChildren();
  if (children.empty()) {
    return nullptr;
  } else {
    return children.front();
  }
}

void RenderObject::TraverseDescendants(
    const std::function<void(RenderObject*)>& action) {
  action(this);
  for (auto child : children_) child->TraverseDescendants(action);
}

Point RenderObject::GetTotalOffset() const {
  Point result{};
  const RenderObject* render_object = this;

  while (render_object != nullptr) {
    const auto o = render_object->GetOffset();
    result.x += o.x;
    result.y += o.y;
    render_object = render_object->GetParent();
  }

  return result;
}

Point RenderObject::FromRootToContent(const Point& point) const {
  const auto offset = GetTotalOffset();
  const auto rect = GetContentRect();
  return Point{point.x - (offset.x + rect.left),
               point.y - (offset.y + rect.top)};
}

void RenderObject::Measure(const MeasureRequirement& requirement,
                           const MeasureSize& preferred_size) {
  MeasureRequirement merged_requirement =
      MeasureRequirement::Merge(requirement, custom_measure_requirement_)
          .Normalize();
  MeasureSize merged_preferred_size =
      preferred_size.OverrideBy(preferred_size_);

  if constexpr (cru::ui::debug_flags::layout) {
    log::Debug(u"{} Measure begins :\nrequirement: {}\npreferred size: {}",
               this->GetDebugPathInTree(), requirement.ToDebugString(),
               preferred_size.ToDebugString());
  }

  size_ = OnMeasureCore(merged_requirement, merged_preferred_size);

  if constexpr (cru::ui::debug_flags::layout) {
    log::Debug(u"{} Measure ends :\nresult size: {}",
               this->GetDebugPathInTree(), size_);
  }

  Ensures(size_.width >= 0);
  Ensures(size_.height >= 0);
}

void RenderObject::Layout(const Point& offset) {
  if constexpr (cru::ui::debug_flags::layout) {
    log::Debug(u"{} Layout :\noffset: {} size: {}", this->GetDebugPathInTree(),
               offset, GetSize());
  }
  offset_ = offset;
  OnLayoutCore();
}

Thickness RenderObject::GetOuterSpaceThickness() const {
  return margin_ + padding_;
}

void RenderObject::Draw(platform::graphics::IPainter* painter) {
  OnDrawCore(painter);
}

RenderObject* RenderObject::GetSingleChild() const {
  Expects(child_mode_ == ChildMode::Single);
  const auto& children = GetChildren();
  if (children.empty())
    return nullptr;
  else
    return children.front();
}

void RenderObject::OnParentChanged(RenderObject* old_parent,
                                   RenderObject* new_parent) {
  CRU_UNUSED(old_parent)
  CRU_UNUSED(new_parent)
}

void RenderObject::OnAddChild(RenderObject* new_child, Index position) {
  CRU_UNUSED(new_child)
  CRU_UNUSED(position)

  InvalidateLayout();
  InvalidatePaint();
}

void RenderObject::OnRemoveChild(RenderObject* removed_child, Index position) {
  CRU_UNUSED(removed_child)
  CRU_UNUSED(position)

  InvalidateLayout();
  InvalidatePaint();
}

void RenderObject::DefaultDrawChildren(platform::graphics::IPainter* painter) {
  for (const auto child : GetChildren()) {
    auto offset = child->GetOffset();
    platform::graphics::util::WithTransform(
        painter, platform::Matrix::Translation(offset.x, offset.y),
        [child](auto p) { child->Draw(p); });
  }
}

void RenderObject::DefaultDrawContent(platform::graphics::IPainter* painter) {
  const auto content_rect = GetContentRect();

  platform::graphics::util::WithTransform(
      painter, Matrix::Translation(content_rect.left, content_rect.top),
      [this](auto p) { this->OnDrawContent(p); });
}

void RenderObject::OnDrawCore(platform::graphics::IPainter* painter) {
  DefaultDrawContent(painter);
  DefaultDrawChildren(painter);
}

void RenderObject::OnDrawContent(platform::graphics::IPainter* painter) {
  CRU_UNUSED(painter);
}

Size RenderObject::OnMeasureCore(const MeasureRequirement& requirement,
                                 const MeasureSize& preferred_size) {
  const Thickness outer_space = GetOuterSpaceThickness();
  const Size space_size{outer_space.GetHorizontalTotal(),
                        outer_space.GetVerticalTotal()};

  MeasureRequirement content_requirement = requirement;

  content_requirement.max = content_requirement.max.Minus(space_size);
  content_requirement.min = content_requirement.min.Minus(space_size);

  MeasureSize content_preferred_size =
      content_requirement.Coerce(preferred_size.Minus(space_size));

  const auto content_size =
      OnMeasureContent(content_requirement, content_preferred_size);

  return space_size + content_size;
}

void RenderObject::OnLayoutCore() {
  Size total_size = GetSize();
  const Thickness outer_space = GetOuterSpaceThickness();
  const Size space_size{outer_space.GetHorizontalTotal(),
                        outer_space.GetVerticalTotal()};

  auto content_size = total_size - space_size;

  if (content_size.width < 0) {
    content_size.width = 0;
  }
  if (content_size.height < 0) {
    content_size.height = 0;
  }

  Point lefttop{outer_space.left, outer_space.top};
  if (lefttop.x > total_size.width) {
    lefttop.x = total_size.width;
  }
  if (lefttop.y > total_size.height) {
    lefttop.y = total_size.height;
  }

  const Rect content_rect{lefttop, content_size};

  OnLayoutContent(content_rect);
}

void RenderObject::OnAfterLayout() {}

Rect RenderObject::GetPaddingRect() const {
  const auto size = GetSize();
  Rect rect{Point{}, size};
  rect = rect.Shrink(GetMargin());
  rect.left = std::min(rect.left, size.width);
  rect.top = std::min(rect.top, size.height);
  rect.width = std::max(rect.width, 0.0f);
  rect.height = std::max(rect.height, 0.0f);
  return rect;
}

Rect RenderObject::GetContentRect() const {
  const auto size = GetSize();
  Rect rect{Point{}, size};
  rect = rect.Shrink(GetMargin());
  rect = rect.Shrink(GetPadding());
  rect.left = std::min(rect.left, size.width);
  rect.top = std::min(rect.top, size.height);
  rect.width = std::max(rect.width, 0.0f);
  rect.height = std::max(rect.height, 0.0f);
  return rect;
}

void RenderObject::SetParent(RenderObject* new_parent) {
  const auto old_parent = parent_;
  parent_ = new_parent;
  OnParentChanged(old_parent, new_parent);
}

void RenderObject::InvalidateLayout() {
  if (window_host_ != nullptr) window_host_->InvalidateLayout();
}

void RenderObject::InvalidatePaint() {
  if (window_host_ != nullptr) window_host_->InvalidatePaint();
}

constexpr std::u16string_view kUnamedName(u"UNNAMED");

std::u16string_view RenderObject::GetName() const { return kUnamedName; }

std::u16string RenderObject::GetDebugPathInTree() const {
  std::vector<std::u16string_view> chain;
  const RenderObject* parent = this;
  while (parent != nullptr) {
    chain.push_back(parent->GetName());
    parent = parent->GetParent();
  }

  std::u16string result(chain.back());
  for (auto iter = chain.crbegin() + 1; iter != chain.crend(); ++iter) {
    result += u" -> ";
    result += *iter;
  }

  return result;
}

void RenderObject::SetWindowHostRecursive(host::WindowHost* host) {
  if (window_host_ != nullptr) {
    detach_from_host_event_.Raise(nullptr);
  }
  window_host_ = host;
  if (host != nullptr) {
    attach_to_host_event_.Raise(nullptr);
  }
  for (const auto child : GetChildren()) {
    child->SetWindowHostRecursive(host);
  }
}
}  // namespace cru::ui::render