4 files changed, 365 insertions, 85 deletions

diff --git a/CruUI/ui/controls/text_block.cpp b/CruUI/ui/controls/text_block.cpp
index 8800fd91..93d66ba6 100644
--- a/CruUI/ui/controls/text_block.cpp
+++ b/CruUI/ui/controls/text_block.cpp
@@ -20,9 +20,11 @@ namespace cru
             TextBlock::TextBlock(const Microsoft::WRL::ComPtr<IDWriteTextFormat>& init_text_format,
                 const Microsoft::WRL::ComPtr<ID2D1Brush>& init_brush) : Control(false)
             {
-                text_format_ = init_text_format;
-                if (init_brush == nullptr)
-                    brush_ = CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::Black));
+                text_format_ = init_text_format == nullptr ? graph::CreateDefaultTextFormat() : init_text_format;
+
+                RecreateTextLayout();
+
+                brush_ = init_brush == nullptr ? CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::Black)) : init_brush;
 
                 selection_brush_ = CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::LightSkyBlue));
             }
@@ -60,7 +62,7 @@ namespace cru
             void TextBlock::RemoveTextLayoutHandler(const TextLayoutHandlerPtr& handler)
             {
                 const auto find_result = std::find(text_layout_handlers_.cbegin(), text_layout_handlers_.cend(),
-                                                   handler);
+                    handler);
                 if (find_result != text_layout_handlers_.cend())
                     text_layout_handlers_.erase(find_result);
             }
@@ -96,42 +98,42 @@ namespace cru
             void TextBlock::OnDraw(ID2D1DeviceContext* device_context)
             {
                 Control::OnDraw(device_context);
-                if (text_layout_ != nullptr)
+                if (selected_range_.has_value())
                 {
-                    if (selected_range_.has_value())
+                    DWRITE_TEXT_METRICS text_metrics{};
+                    ThrowIfFailed(text_layout_->GetMetrics(&text_metrics));
+                    const auto metrics_count = text_metrics.lineCount * text_metrics.maxBidiReorderingDepth;
+
+                    Vector<DWRITE_HIT_TEST_METRICS> hit_test_metrics(metrics_count);
+                    UINT32 actual_count;
+                    text_layout_->HitTestTextRange(
+                        selected_range_.value().position, selected_range_.value().count,
+                        0, 0,
+                        hit_test_metrics.data(), metrics_count, &actual_count
+                    );
+
+                    hit_test_metrics.erase(hit_test_metrics.cbegin() + actual_count, hit_test_metrics.cend());
+
+                    for (const auto& metrics : hit_test_metrics)
                     {
-                        DWRITE_TEXT_METRICS text_metrics{};
-                        ThrowIfFailed(text_layout_->GetMetrics(&text_metrics));
-                        const auto metrics_count = text_metrics.lineCount * text_metrics.maxBidiReorderingDepth;
-
-                        Vector<DWRITE_HIT_TEST_METRICS> hit_test_metrics(metrics_count);
-                        UINT32 actual_count;
-                        text_layout_->HitTestTextRange(
-                            selected_range_.value().position, selected_range_.value().count,
-                            0, 0,
-                            hit_test_metrics.data(), metrics_count, &actual_count
-                        );
-
-                        hit_test_metrics.erase(hit_test_metrics.cbegin() + actual_count, hit_test_metrics.cend());
-
-                        for (const auto& metrics : hit_test_metrics)
-                        {
-                            device_context->FillRoundedRectangle(D2D1::RoundedRect(D2D1::RectF(metrics.left, metrics.top, metrics.left + metrics.width, metrics.top + metrics.height), 3, 3), selection_brush_.Get());
-                        }
+                        device_context->FillRoundedRectangle(D2D1::RoundedRect(D2D1::RectF(metrics.left, metrics.top, metrics.left + metrics.width, metrics.top + metrics.height), 3, 3), selection_brush_.Get());
                     }
-                    device_context->DrawTextLayout(D2D1::Point2F(), text_layout_.Get(), brush_.Get());
                 }
+                device_context->DrawTextLayout(D2D1::Point2F(), text_layout_.Get(), brush_.Get());
             }
 
-            std::optional<unsigned> TextLayoutHitTest(IDWriteTextLayout* text_layout, const Point& point, bool test_inside = true)
+            namespace
             {
-                BOOL is_trailing, is_inside;
-                DWRITE_HIT_TEST_METRICS metrics{};
-                text_layout->HitTestPoint(point.x, point.y, &is_trailing, &is_inside, &metrics);
-                if (!test_inside || is_inside)
-                    return is_trailing == 0 ? metrics.textPosition : metrics.textPosition + 1;
-                else
-                    return std::nullopt;
+                std::optional<unsigned> TextLayoutHitTest(IDWriteTextLayout* text_layout, const Point& point, const bool test_inside = true)
+                {
+                    BOOL is_trailing, is_inside;
+                    DWRITE_HIT_TEST_METRICS metrics{};
+                    text_layout->HitTestPoint(point.x, point.y, &is_trailing, &is_inside, &metrics);
+                    if (!test_inside || is_inside)
+                        return is_trailing == 0 ? metrics.textPosition : metrics.textPosition + 1;
+                    else
+                        return std::nullopt;
+                }
             }
 
             void TextBlock::OnMouseDownCore(events::MouseButtonEventArgs& args)
@@ -197,9 +199,6 @@ namespace cru
 
             Size TextBlock::OnMeasure(const Size& available_size)
             {
-                if (text_.empty())
-                    return Size::Zero();
-
                 const auto layout_params = GetLayoutParams();
 
                 if (layout_params->width.mode == MeasureMode::Stretch && layout_params->height.mode == MeasureMode::Stretch)
@@ -259,36 +258,12 @@ namespace cru
                 Repaint();
             }
 
-            void TextBlock::CreateDefaultTextFormat()
-            {
-                const auto dwrite_factory = GetDWriteFactory();
-
-                ThrowIfFailed(dwrite_factory->CreateTextFormat(
-                    L"΅ΘΟί", nullptr,
-                    DWRITE_FONT_WEIGHT_NORMAL,
-                    DWRITE_FONT_STYLE_NORMAL,
-                    DWRITE_FONT_STRETCH_NORMAL,
-                    24.0, L"zh-cn",
-                    &text_format_
-                ));
-
-                ThrowIfFailed(text_format_->SetTextAlignment(DWRITE_TEXT_ALIGNMENT_CENTER));
-                ThrowIfFailed(text_format_->SetParagraphAlignment(DWRITE_PARAGRAPH_ALIGNMENT_CENTER));
-            }
-
             void TextBlock::RecreateTextLayout()
             {
-                if (text_.empty())
-                {
-                    text_layout_ = nullptr;
-                    return;
-                }
+                assert(text_format_ != nullptr);
 
                 const auto dwrite_factory = GetDWriteFactory();
 
-                if (text_format_ == nullptr)
-                    CreateDefaultTextFormat();
-
                 const auto&& size = GetSize();
 
                 ThrowIfFailed(dwrite_factory->CreateTextLayout(
diff --git a/CruUI/ui/controls/text_block.h b/CruUI/ui/controls/text_block.h
index b05e7ff2..c87ffc51 100644
--- a/CruUI/ui/controls/text_block.h
+++ b/CruUI/ui/controls/text_block.h
@@ -11,29 +11,6 @@ namespace cru
     {
         namespace controls
         {
-            struct TextRange
-            {
-                TextRange() = default;
-                TextRange(const int position, const int count)
-                    : position(position), count(count)
-                {
-                    
-                }
-                TextRange(const TextRange& other) = default;
-                TextRange(TextRange&& other) = default;
-                TextRange& operator=(const TextRange& other) = default;
-                TextRange& operator=(TextRange&& other) = default;
-                ~TextRange() = default;
-
-                unsigned position;
-                unsigned count;
-
-                explicit operator DWRITE_TEXT_RANGE() const
-                {
-                    return DWRITE_TEXT_RANGE { position, count };
-                }
-            };
-
             class TextBlock : public Control
             {
             public:
@@ -116,7 +93,6 @@ namespace cru
             private:
                 void OnTextChangedCore(const String& old_text, const String& new_text);
 
-                void CreateDefaultTextFormat();
                 void RecreateTextLayout();
 
             private:
diff --git a/CruUI/ui/controls/text_box.cpp b/CruUI/ui/controls/text_box.cpp
new file mode 100644
index 00000000..a8d78398
--- /dev/null
+++ b/CruUI/ui/controls/text_box.cpp
@@ -0,0 +1,244 @@
+#include "text_box.h"
+
+#include <cwctype>
+
+#include "graph/graph.h"
+#include "exception.h"
+
+namespace cru::ui::controls
+{
+    using graph::CreateSolidBrush;
+
+    inline Microsoft::WRL::ComPtr<IDWriteFactory> GetDWriteFactory()
+    {
+        return graph::GraphManager::GetInstance()->GetDWriteFactory();
+    }
+
+    TextBox::TextBox(const Microsoft::WRL::ComPtr<IDWriteTextFormat>& init_text_format,
+        const Microsoft::WRL::ComPtr<ID2D1Brush>& init_brush) : Control(false)
+    {
+        text_format_ = init_text_format == nullptr ? graph::CreateDefaultTextFormat() : init_text_format;
+
+        RecreateTextLayout();
+
+        brush_ = init_brush == nullptr ? CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::Black)) : init_brush;
+
+        caret_brush_ = CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::Black));
+
+        caret_action_ = CreateActionPtr([this]
+        {
+            is_caret_show_ = !is_caret_show_;
+            Repaint();
+        });
+
+        //selection_brush_ = CreateSolidBrush(D2D1::ColorF(D2D1::ColorF::LightSkyBlue));
+    }
+
+    TextBox::~TextBox() = default;
+
+    void TextBox::SetText(const String& text)
+    {
+        if (text_ != text)
+        {
+            const auto old_text = text_;
+            text_ = text;
+            OnTextChangedCore(old_text, text);
+        }
+    }
+
+    void TextBox::SetBrush(const Microsoft::WRL::ComPtr<ID2D1Brush>& brush)
+    {
+        brush_ = brush;
+        Repaint();
+    }
+
+    void TextBox::SetTextFormat(const Microsoft::WRL::ComPtr<IDWriteTextFormat>& text_format)
+    {
+        text_format_ = text_format;
+        RecreateTextLayout();
+        Repaint();
+    }
+
+    void TextBox::OnSizeChangedCore(events::SizeChangedEventArgs& args)
+    {
+        Control::OnSizeChangedCore(args);
+        text_layout_->SetMaxWidth(args.GetNewSize().width);
+        text_layout_->SetMaxHeight(args.GetNewSize().height);
+        Repaint();
+    }
+
+    void TextBox::OnDraw(ID2D1DeviceContext* device_context)
+    {
+        Control::OnDraw(device_context);
+        if (text_layout_ != nullptr)
+        {
+            device_context->DrawTextLayout(D2D1::Point2F(), text_layout_.Get(), brush_.Get());
+            if (is_caret_show_)
+            {
+                const auto caret_half_width = Application::GetInstance()->GetCaretInfo().half_caret_width;
+                FLOAT x, y;
+                DWRITE_HIT_TEST_METRICS metrics{};
+                ThrowIfFailed(text_layout_->HitTestTextPosition(position_, FALSE, &x, &y, &metrics));
+                device_context->FillRectangle(D2D1::RectF(metrics.left - caret_half_width, metrics.top, metrics.left + caret_half_width, metrics.top + metrics.height), caret_brush_.Get());
+            }
+        }
+    }
+
+    namespace
+    {
+        std::optional<unsigned> TextLayoutHitTest(IDWriteTextLayout* text_layout, const Point& point, bool test_inside = true)
+        {
+            BOOL is_trailing, is_inside;
+            DWRITE_HIT_TEST_METRICS metrics{};
+            text_layout->HitTestPoint(point.x, point.y, &is_trailing, &is_inside, &metrics);
+            if (!test_inside || is_inside)
+                return is_trailing == 0 ? metrics.textPosition : metrics.textPosition + 1;
+            else
+                return std::nullopt;
+        }
+    }
+
+    void TextBox::OnMouseDownCore(events::MouseButtonEventArgs& args)
+    {
+        Control::OnMouseDownCore(args);
+        if (args.GetMouseButton() == MouseButton::Left)
+        {
+            position_ = TextLayoutHitTest(text_layout_.Get(), args.GetPoint(this), false).value();
+
+            Repaint();
+        }
+    }
+
+    void TextBox::OnGetFocusCore(events::FocusChangeEventArgs& args)
+    {
+        Control::OnGetFocusCore(args);
+        assert(caret_timer_ == nullptr);
+        is_caret_show_ = true;
+        caret_timer_ = SetInterval(Application::GetInstance()->GetCaretInfo().caret_blink_duration, caret_action_);
+    }
+
+    void TextBox::OnLoseFocusCore(events::FocusChangeEventArgs& args)
+    {
+        Control::OnLoseFocusCore(args);
+        assert(caret_timer_ != nullptr);
+        caret_timer_->Cancel();
+        is_caret_show_ = false;
+    }
+
+    void TextBox::OnKeyDownCore(events::KeyEventArgs& args)
+    {
+        Control::OnKeyDownCore(args);
+        if (args.GetVirtualCode() == VK_LEFT && position_ > 0)
+        {
+            position_--;
+            Repaint();
+        }
+
+        if (args.GetVirtualCode() == VK_RIGHT && position_ < GetText().size())
+        {
+            position_++;
+            Repaint();
+        }
+    }
+
+    void TextBox::OnCharCore(events::CharEventArgs& args)
+    {
+        Control::OnCharCore(args);
+        if (args.GetChar() == L'\b')
+        {
+            auto text = GetText();
+            if (!text.empty() && position_ > 0)
+            {
+                const auto position = --position_;
+                text.erase(position);
+                SetText(text);
+            }
+            return;
+        }
+
+        if (std::iswprint(args.GetChar()))
+        {
+            const auto position = position_++;
+            auto text = GetText();
+            text.insert(text.cbegin() + position, { args.GetChar() });
+            SetText(text);
+        }
+    }
+
+    Size TextBox::OnMeasure(const Size& available_size)
+    {
+        const auto layout_params = GetLayoutParams();
+
+        if (layout_params->width.mode == MeasureMode::Stretch && layout_params->height.mode == MeasureMode::Stretch)
+            return available_size;
+
+        auto&& get_measure_length = [](const LayoutSideParams& layout_length, const float available_length) -> float
+        {
+            switch (layout_length.mode)
+            {
+            case MeasureMode::Exactly:
+            {
+                return std::min(layout_length.length, available_length);
+            }
+            case MeasureMode::Stretch:
+            case MeasureMode::Content:
+            {
+                return available_length;
+            }
+            default:
+                UnreachableCode();
+            }
+        };
+
+        const Size measure_size(get_measure_length(layout_params->width, available_size.width),
+            get_measure_length(layout_params->height, available_size.height));
+
+        ThrowIfFailed(text_layout_->SetMaxWidth(measure_size.width));
+        ThrowIfFailed(text_layout_->SetMaxHeight(measure_size.height));
+
+        DWRITE_TEXT_METRICS metrics{};
+
+        ThrowIfFailed(text_layout_->GetMetrics(&metrics));
+
+        const Size measure_result(metrics.width, metrics.height);
+
+        auto&& calculate_final_length = [](const LayoutSideParams& layout_length, const float measure_length, const float measure_result_length) -> float
+        {
+            if ((layout_length.mode == MeasureMode::Stretch ||
+                layout_length.mode == MeasureMode::Exactly)
+                && measure_result_length < measure_length)
+                return measure_length;
+            else
+                return measure_result_length;
+        };
+
+        const Size result_size(
+            calculate_final_length(layout_params->width, measure_size.width, measure_result.width),
+            calculate_final_length(layout_params->height, measure_size.height, measure_result.height)
+        );
+
+        return result_size;
+    }
+
+    void TextBox::OnTextChangedCore(const String& old_text, const String& new_text)
+    {
+        RecreateTextLayout();
+        Repaint();
+    }
+
+    void TextBox::RecreateTextLayout()
+    {
+        assert(text_format_ != nullptr);
+
+        const auto dwrite_factory = GetDWriteFactory();
+
+        const auto&& size = GetSize();
+
+        ThrowIfFailed(dwrite_factory->CreateTextLayout(
+            text_.c_str(), static_cast<UINT32>(text_.size()),
+            text_format_.Get(),
+            size.width, size.height,
+            &text_layout_
+        ));
+    }
+}
diff --git a/CruUI/ui/controls/text_box.h b/CruUI/ui/controls/text_box.h
new file mode 100644
index 00000000..b815ed1f
--- /dev/null
+++ b/CruUI/ui/controls/text_box.h
@@ -0,0 +1,85 @@
+#pragma once
+
+#include "ui/control.h"
+#include "timer.h"
+
+namespace cru::ui::controls
+{
+    class TextBox : public Control
+    {
+    public:
+        static TextBox* Create(
+            const Microsoft::WRL::ComPtr<IDWriteTextFormat>& init_text_format = nullptr,
+            const Microsoft::WRL::ComPtr<ID2D1Brush>& init_brush = nullptr)
+        {
+            return new TextBox(init_text_format, init_brush);
+        }
+
+    protected:
+        explicit TextBox(
+            const Microsoft::WRL::ComPtr<IDWriteTextFormat>& init_text_format = nullptr,
+            const Microsoft::WRL::ComPtr<ID2D1Brush>& init_brush = nullptr
+        );
+    public:
+        TextBox(const TextBox& other) = delete;
+        TextBox(TextBox&& other) = delete;
+        TextBox& operator=(const TextBox& other) = delete;
+        TextBox& operator=(TextBox&& other) = delete;
+        ~TextBox() override;
+
+        String GetText() const
+        {
+            return text_;
+        }
+
+        void SetText(const String& text);
+
+        Microsoft::WRL::ComPtr<ID2D1Brush> GetBrush() const
+        {
+            return brush_;
+        }
+
+        void SetBrush(const Microsoft::WRL::ComPtr<ID2D1Brush>& brush);
+
+        Microsoft::WRL::ComPtr<IDWriteTextFormat> GetTextFormat() const
+        {
+            return text_format_;
+        }
+
+        void SetTextFormat(const Microsoft::WRL::ComPtr<IDWriteTextFormat>& text_format);
+
+    protected:
+        void OnSizeChangedCore(events::SizeChangedEventArgs& args) override final;
+        void OnDraw(ID2D1DeviceContext* device_context) override;
+
+        void OnMouseDownCore(events::MouseButtonEventArgs& args) override final;
+
+        void OnGetFocusCore(events::FocusChangeEventArgs& args) override final;
+        void OnLoseFocusCore(events::FocusChangeEventArgs& args) override final;
+
+        void OnKeyDownCore(events::KeyEventArgs& args) override final;
+        void OnCharCore(events::CharEventArgs& args) override final;
+
+        Size OnMeasure(const Size& available_size) override final;
+
+    private:
+        void OnTextChangedCore(const String& old_text, const String& new_text);
+
+        void RecreateTextLayout();
+
+    private:
+        String text_;
+
+        Microsoft::WRL::ComPtr<ID2D1Brush> brush_;
+        Microsoft::WRL::ComPtr<ID2D1Brush> caret_brush_;
+        //Microsoft::WRL::ComPtr<ID2D1Brush> selection_brush_;
+        Microsoft::WRL::ComPtr<IDWriteTextFormat> text_format_;
+        Microsoft::WRL::ComPtr<IDWriteTextLayout> text_layout_;
+
+        unsigned position_ = 0;
+
+        TimerTask caret_timer_;
+        ActionPtr caret_action_;
+        bool is_caret_show_;
+    };
+}