Cornus/TreeData.cpp at main · f35f22fan/Cornus · GitHub

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
#include "TreeData.hpp"

#include "gui/TreeItem.hpp"

Q_DECLARE_METATYPE(QVector<cornus::gui::TreeItem *>);

namespace cornus {

TreeData::TreeData() {}

TreeData::~TreeData()
{
	// .roots deleted by hand in App destructor.
}

gui::TreeItem* TreeData::GetBookmarksRoot(int *index) const
{
	int i = -1;
	for (gui::TreeItem *next: roots)
	{
		i++;
		if (next->is_bookmarks_root()) {
			if (index)
				*index = i;
			return next;
		}
	}

	return nullptr;
}

gui::TreeItem* TreeData::GetCurrentPartition() const
{
	for (gui::TreeItem *root: roots)
	{
		if (!root->is_disk())
			continue;
		for (gui::TreeItem *child: root->children()) {
			if (child->current_partition())
				return child;
		}
	}

	return nullptr;
}

gui::TreeItem* TreeData::GetPartitionByMountPath(const QString &full_path)
{
	int largest = 0;
	gui::TreeItem *current = nullptr;

	for (gui::TreeItem *root: roots)
	{
		if (!root->is_disk())
			continue;
		for (gui::TreeItem *child: root->children()) {
			if (child->mounted() && full_path.startsWith(child->mount_path())) {
				if (largest < child->mount_path().size()) {
					current = child;
					largest = child->mount_path().size();
					//mtl_printq2("Found: ", child->mount_path());
				}
			}
		}
	}

	return current;
}

void TreeData::MarkCurrentPartition(const QString &full_path)
{
	gui::TreeItem *current = GetPartitionByMountPath(full_path);
	if (current != nullptr)
		current->current_partition(true);

	for (gui::TreeItem *root: roots)
	{
		if (!root->is_disk())
			continue;
		for (gui::TreeItem *child: root->children()) {
			if (child != current && child->current_partition()) {
				child->current_partition(false);
			}
		}
	}
}

MutexGuard TreeData::try_guard()
{
	return MutexGuard(&mutex, LockType::TryLock);
}

}