TreeView.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2023 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <functional>
 
#include "open3d/visualization/gui/Widget.h"
 
namespace open3d {
namespace visualization {
namespace gui {
 
class Checkbox;
class ColorEdit;
class Label;
class NumberEdit;
 
class CheckableTextTreeCell : public Widget {
public:
    CheckableTextTreeCell(const char* text,
                          bool is_checked,
                          std::function<void(bool)> on_toggled);
    ~CheckableTextTreeCell();
 
    std::shared_ptr<Checkbox> GetCheckbox();
    std::shared_ptr<Label> GetLabel();
 
    Size CalcPreferredSize(const LayoutContext& context,
                           const Constraints& constraints) const override;
    void Layout(const LayoutContext& context) override;
 
private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
class LUTTreeCell : public Widget {
public:
    LUTTreeCell(const char* text,
                bool is_checked,
                const Color& color,
                std::function<void(bool)> on_enabled,
                std::function<void(const Color&)> on_color_changed);
    ~LUTTreeCell();
 
    std::shared_ptr<Checkbox> GetCheckbox();
    std::shared_ptr<Label> GetLabel();
    std::shared_ptr<ColorEdit> GetColorEdit();
 
    Size CalcPreferredSize(const LayoutContext& context,
                           const Constraints& constraints) const override;
    void Layout(const LayoutContext& context) override;
 
private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
class ColormapTreeCell : public Widget {
public:
    ColormapTreeCell(double value,
                     const Color& color,
                     std::function<void(double)> on_value_changed,
                     std::function<void(const Color&)> on_color_changed);
    ~ColormapTreeCell();
 
    std::shared_ptr<NumberEdit> GetNumberEdit();
    std::shared_ptr<ColorEdit> GetColorEdit();
 
    Size CalcPreferredSize(const LayoutContext& context,
                           const Constraints& constraints) const override;
    void Layout(const LayoutContext& context) override;
 
private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
class TreeView : public Widget {
    using Super = Widget;
 
public:
    using ItemId = int;
 
    TreeView();
    ~TreeView();
 
    ItemId GetRootItem() const;
    ItemId AddItem(ItemId parent_id, std::shared_ptr<Widget> item);
    ItemId AddTextItem(ItemId parent_id, const char* text);
    void RemoveItem(ItemId item_id);
    void Clear();
    std::shared_ptr<Widget> GetItem(ItemId item_id) const;
    std::vector<ItemId> GetItemChildren(ItemId parent_id) const;
 
    bool GetCanSelectItemsWithChildren() const;
    void SetCanSelectItemsWithChildren(bool can_select);
 
    ItemId GetSelectedItemId() const;
    void SetSelectedItemId(ItemId item_id);
 
    Size CalcPreferredSize(const LayoutContext& context,
                           const Constraints& constraints) const override;
    Size CalcMinimumSize(const LayoutContext& context) const override;
 
    void Layout(const LayoutContext& context) override;
 
    DrawResult Draw(const DrawContext& context) override;
 
    void SetOnSelectionChanged(
            std::function<void(ItemId)> on_selection_changed);
 
private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
}  // namespace gui
}  // namespace visualization
}  // namespace open3d