summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibGUI/PersistentModelIndex.h
blob: 0690ebd0de4553030f03c3c71a8b7646ad22a33e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
 * Copyright (c) 2021, sin-ack <sin-ack@protonmail.com>
 * Copyright (c) 2022, the SerenityOS developers.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/RefCounted.h>
#include <AK/WeakPtr.h>
#include <LibGUI/Model.h>
#include <LibGUI/ModelIndex.h>

namespace GUI {

/// A PersistentHandle is an internal data structure used to keep track of the
/// target of multiple PersistentModelIndex instances.
class PersistentHandle : public Weakable<PersistentHandle> {
    friend Model;
    friend PersistentModelIndex;
    friend AK::Traits<GUI::PersistentModelIndex>;

    PersistentHandle(ModelIndex const& index)
        : m_index(index)
    {
    }

    ModelIndex m_index;
};

class PersistentModelIndex {
public:
    PersistentModelIndex() = default;
    PersistentModelIndex(ModelIndex const&);
    PersistentModelIndex(PersistentModelIndex const&) = default;
    PersistentModelIndex(PersistentModelIndex&&) = default;

    PersistentModelIndex& operator=(PersistentModelIndex const&) = default;
    PersistentModelIndex& operator=(PersistentModelIndex&&) = default;

    bool is_valid() const { return has_valid_handle() && m_handle->m_index.is_valid(); }
    bool has_valid_handle() const { return !m_handle.is_null(); }

    int row() const;
    int column() const;
    PersistentModelIndex parent() const;
    PersistentModelIndex sibling_at_column(int column) const;
    Variant data(ModelRole = ModelRole::Display) const;

    void* internal_data() const
    {
        if (has_valid_handle())
            return m_handle->m_index.internal_data();
        else
            return nullptr;
    }

    operator ModelIndex() const;
    bool operator==(PersistentModelIndex const&) const;
    bool operator!=(PersistentModelIndex const&) const;
    bool operator==(ModelIndex const&) const;
    bool operator!=(ModelIndex const&) const;

private:
    friend AK::Traits<GUI::PersistentModelIndex>;

    WeakPtr<PersistentHandle> m_handle;
};

}

namespace AK {

template<>
struct Formatter<GUI::PersistentModelIndex> : Formatter<FormatString> {
    ErrorOr<void> format(FormatBuilder& builder, GUI::PersistentModelIndex const& value)
    {
        return Formatter<FormatString>::format(builder, "PersistentModelIndex({},{},{})"sv, value.row(), value.column(), value.internal_data());
    }
};

template<>
struct Traits<GUI::PersistentModelIndex> : public GenericTraits<GUI::PersistentModelIndex> {
    static unsigned hash(const GUI::PersistentModelIndex& index)
    {
        if (index.has_valid_handle())
            return Traits<GUI::ModelIndex>::hash(index.m_handle->m_index);
        return 0;
    }
};

}