mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-24 10:22:05 -05:00
Utilities: Add a disk space usage analyzation program.
SpaceAnalyzer: Partially address code review changes. - Use GUI::CommonActions::make_about_action(). - Pass large arguments by const reference instead of by value. - Mark const functions as such. - Add newline at end of SpaceAnalyzer.af - Use full words instead of abbreviations in variable names. - Use application's namespace instead of 'TreeMap'. - move() certain assignments. - Use member declaration initialization. - Initialize TreeNode* member of QueueEntry. - Rewrite find_mount_for_path to return MountInfo* instead. - Rename ITreeMap and ITreeMapNode to TreeMap and TreeMapNode. - Replace ext suffix with rect suffix for rectangles. SpaceAnalyzer: Further address code review and coding style. - Remove get prefix from accessor functions. - Layout algorithm in its own function, with callback. - Remove nullptr comparisons. - Store lstat errors in error_accumulator. - Use Rect::shatter. - Use Rect's orientation based functions. SpaceAnalyzer: Make sort_children_by_area const qualified.
This commit is contained in:
parent
56ff0f89af
commit
e575d3fd3d
9 changed files with 801 additions and 0 deletions
|
@ -17,6 +17,7 @@ add_subdirectory(Piano)
|
|||
add_subdirectory(PixelPaint)
|
||||
add_subdirectory(QuickShow)
|
||||
add_subdirectory(SoundPlayer)
|
||||
add_subdirectory(SpaceAnalyzer)
|
||||
add_subdirectory(Spreadsheet)
|
||||
add_subdirectory(SystemMonitor)
|
||||
add_subdirectory(ThemeEditor)
|
||||
|
|
10
Applications/SpaceAnalyzer/CMakeLists.txt
Normal file
10
Applications/SpaceAnalyzer/CMakeLists.txt
Normal file
|
@ -0,0 +1,10 @@
|
|||
compile_gml(SpaceAnalyzer.gml SpaceAnalyzerGML.h space_analyzer_gml)
|
||||
|
||||
set(SOURCES
|
||||
main.cpp
|
||||
TreeMapWidget.cpp
|
||||
SpaceAnalyzerGML.h
|
||||
)
|
||||
|
||||
serenity_app(SpaceAnalyzer ICON app-space-analyzer)
|
||||
target_link_libraries(SpaceAnalyzer LibGfx LibGUI)
|
20
Applications/SpaceAnalyzer/SpaceAnalyzer.gml
Normal file
20
Applications/SpaceAnalyzer/SpaceAnalyzer.gml
Normal file
|
@ -0,0 +1,20 @@
|
|||
@GUI::Widget {
|
||||
layout: @GUI::VerticalBoxLayout {
|
||||
spacing: 0
|
||||
}
|
||||
|
||||
@GUI::ToolBarContainer {
|
||||
@GUI::BreadcrumbBar {
|
||||
fixed_height: 25
|
||||
name: "breadcrumb_bar"
|
||||
}
|
||||
}
|
||||
|
||||
@SpaceAnalyzer::TreeMapWidget {
|
||||
name: "tree_map"
|
||||
}
|
||||
|
||||
@GUI::StatusBar {
|
||||
name: "status_bar"
|
||||
}
|
||||
}
|
376
Applications/SpaceAnalyzer/TreeMapWidget.cpp
Normal file
376
Applications/SpaceAnalyzer/TreeMapWidget.cpp
Normal file
|
@ -0,0 +1,376 @@
|
|||
/*
|
||||
* Copyright (c) 2021, the SerenityOS developers.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright notice, this
|
||||
* list of conditions and the following disclaimer.
|
||||
*
|
||||
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "TreeMapWidget.h"
|
||||
#include <AK/NumberFormat.h>
|
||||
#include <LibGUI/Painter.h>
|
||||
#include <LibGUI/WindowServerConnection.h>
|
||||
#include <LibGfx/Font.h>
|
||||
#include <WindowServer/WindowManager.h>
|
||||
|
||||
namespace SpaceAnalyzer {
|
||||
|
||||
REGISTER_WIDGET(SpaceAnalyzer, TreeMapWidget)
|
||||
|
||||
TreeMapWidget::TreeMapWidget()
|
||||
: m_viewpoint(0)
|
||||
{
|
||||
}
|
||||
|
||||
TreeMapWidget::~TreeMapWidget()
|
||||
{
|
||||
}
|
||||
|
||||
static const Color colors[] = {
|
||||
Color(253, 231, 37),
|
||||
Color(148, 216, 64),
|
||||
Color(60, 188, 117),
|
||||
Color(31, 150, 139),
|
||||
Color(45, 112, 142),
|
||||
Color(63, 71, 136),
|
||||
Color(85, 121, 104),
|
||||
};
|
||||
|
||||
static float get_normalized_aspect_ratio(float a, float b)
|
||||
{
|
||||
if (a < b) {
|
||||
return a / b;
|
||||
} else {
|
||||
return b / a;
|
||||
}
|
||||
}
|
||||
|
||||
static bool node_is_leaf(const TreeMapNode& node)
|
||||
{
|
||||
return node.num_children() == 0;
|
||||
}
|
||||
|
||||
bool TreeMapWidget::rect_can_contain_label(const Gfx::IntRect& rect) const
|
||||
{
|
||||
return rect.height() > font().presentation_size() && rect.width() > 20;
|
||||
}
|
||||
|
||||
bool TreeMapWidget::rect_can_contain_children(const Gfx::IntRect& rect) const
|
||||
{
|
||||
return rect.height() > 10 && rect.width() > 10;
|
||||
}
|
||||
|
||||
Gfx::IntRect TreeMapWidget::inner_rect_for_frame(const Gfx::IntRect& rect) const
|
||||
{
|
||||
const int margin = 5;
|
||||
Gfx::IntRect tmp_rect = rect;
|
||||
tmp_rect.shrink(2, 2); // border
|
||||
tmp_rect.shrink(2, 2); // shading
|
||||
if (rect_can_contain_label(rect)) {
|
||||
tmp_rect.set_y(tmp_rect.y() + font().presentation_size() + margin);
|
||||
tmp_rect.set_height(tmp_rect.height() - (font().presentation_size() + margin * 2));
|
||||
tmp_rect.set_x(tmp_rect.x() + margin);
|
||||
tmp_rect.set_width(tmp_rect.width() - margin * 2);
|
||||
}
|
||||
return tmp_rect;
|
||||
}
|
||||
|
||||
void TreeMapWidget::paint_cell_frame(GUI::Painter& painter, const TreeMapNode& node, const Gfx::IntRect& cell_rect, int depth, bool fill_frame) const
|
||||
{
|
||||
const Gfx::IntRect border_rect = cell_rect.shrunken(2, 2);
|
||||
const Gfx::IntRect outer_rect = border_rect.shrunken(2, 2);
|
||||
const Gfx::IntRect inner_rect = inner_rect_for_frame(cell_rect);
|
||||
|
||||
painter.clear_clip_rect();
|
||||
painter.add_clip_rect(cell_rect);
|
||||
Color color = colors[depth % (sizeof(colors) / sizeof(colors[0]))];
|
||||
if (m_selected_node_cache == &node) {
|
||||
color = color.darkened(0.8f);
|
||||
}
|
||||
|
||||
// Draw borders.
|
||||
painter.draw_rect(cell_rect, Color::Black, false);
|
||||
painter.draw_line(border_rect.bottom_left(), border_rect.top_left(), color.lightened());
|
||||
painter.draw_line(border_rect.top_left(), border_rect.top_right(), color.lightened());
|
||||
painter.draw_line(border_rect.top_right(), border_rect.bottom_right(), color.darkened());
|
||||
painter.draw_line(border_rect.bottom_left(), border_rect.bottom_right(), color.darkened());
|
||||
|
||||
// Paint the background.
|
||||
if (fill_frame) {
|
||||
painter.fill_rect(outer_rect, color);
|
||||
} else {
|
||||
for (auto& shard : outer_rect.shatter(inner_rect)) {
|
||||
painter.fill_rect(shard, color);
|
||||
}
|
||||
}
|
||||
|
||||
// Paint text.
|
||||
if (rect_can_contain_label(outer_rect)) {
|
||||
Gfx::IntRect text_rect = outer_rect;
|
||||
text_rect.move_by(2, 2);
|
||||
painter.draw_text(text_rect, node.name(), font(), Gfx::TextAlignment::TopLeft, Color::Black);
|
||||
if (node_is_leaf(node)) {
|
||||
text_rect.move_by(0, font().presentation_size() + 1);
|
||||
painter.draw_text(text_rect, human_readable_size(node.area()), font(), Gfx::TextAlignment::TopLeft, Color::Black);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template<typename Function>
|
||||
void TreeMapWidget::lay_out_children(const TreeMapNode& node, const Gfx::IntRect& rect, int depth, Function callback)
|
||||
{
|
||||
if (node.num_children() == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Check if the children are sorted yet, if not do that now.
|
||||
for (size_t k = 0; k < node.num_children() - 1; k++) {
|
||||
if (node.child_at(k).area() < node.child_at(k + 1).area()) {
|
||||
node.sort_children_by_area();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
int total_area = node.area();
|
||||
Gfx::IntRect canvas = rect;
|
||||
bool remaining_nodes_are_too_small = false;
|
||||
for (size_t i = 0; !remaining_nodes_are_too_small && i < node.num_children(); i++) {
|
||||
const int i_node_area = node.child_at(i).area();
|
||||
if (i_node_area == 0)
|
||||
break;
|
||||
|
||||
const int long_side_size = max(canvas.width(), canvas.height());
|
||||
const int short_side_size = min(canvas.width(), canvas.height());
|
||||
|
||||
int row_or_column_size = (long long int)long_side_size * i_node_area / total_area;
|
||||
int node_area_sum = i_node_area;
|
||||
size_t k = i + 1;
|
||||
|
||||
// Try to add nodes to this row or column so long as the worst aspect ratio of
|
||||
// the new set of nodes is better than the worst aspect ratio of the current set.
|
||||
{
|
||||
float best_worst_aspect_ratio_so_far = get_normalized_aspect_ratio(row_or_column_size, short_side_size);
|
||||
for (; k < node.num_children(); k++) {
|
||||
// Do a preliminary calculation of the worst aspect ratio of the nodes at index i and k
|
||||
// if that aspect ratio is better than the 'best_worst_aspect_ratio_so_far' we keep it,
|
||||
// otherwise it is discarded.
|
||||
int k_node_area = node.child_at(k).area();
|
||||
if (k_node_area == 0) {
|
||||
break;
|
||||
}
|
||||
int new_node_area_sum = node_area_sum + k_node_area;
|
||||
int new_row_or_column_size = (long long int)long_side_size * new_node_area_sum / total_area;
|
||||
int i_node_size = (long long int)short_side_size * i_node_area / new_node_area_sum;
|
||||
int k_node_size = (long long int)short_side_size * k_node_area / new_node_area_sum;
|
||||
float i_node_aspect_ratio = get_normalized_aspect_ratio(new_row_or_column_size, i_node_size);
|
||||
float k_node_aspect_ratio = get_normalized_aspect_ratio(new_row_or_column_size, k_node_size);
|
||||
float new_worst_aspect_ratio = min(i_node_aspect_ratio, k_node_aspect_ratio);
|
||||
if (new_worst_aspect_ratio < best_worst_aspect_ratio_so_far) {
|
||||
break;
|
||||
}
|
||||
best_worst_aspect_ratio_so_far = new_worst_aspect_ratio;
|
||||
node_area_sum = new_node_area_sum;
|
||||
row_or_column_size = new_row_or_column_size;
|
||||
}
|
||||
}
|
||||
|
||||
// Paint the elements from 'i' up to and including 'k-1'.
|
||||
{
|
||||
const int fixed_side_size = row_or_column_size;
|
||||
int placement_area = node_area_sum;
|
||||
int main_dim = short_side_size;
|
||||
|
||||
// Lay out nodes in a row or column.
|
||||
Orientation orientation = canvas.width() > canvas.height() ? Orientation::Horizontal : Orientation::Vertical;
|
||||
Gfx::IntRect layout_rect = canvas;
|
||||
layout_rect.set_primary_size_for_orientation(orientation, fixed_side_size);
|
||||
for (size_t q = i; q < k; q++) {
|
||||
auto& child = node.child_at(q);
|
||||
int node_size = (long long int)main_dim * child.area() / placement_area;
|
||||
Gfx::IntRect cell_rect = layout_rect;
|
||||
cell_rect.set_secondary_size_for_orientation(orientation, node_size);
|
||||
Gfx::IntRect inner_rect = inner_rect_for_frame(cell_rect);
|
||||
bool is_visual_leaf = child.num_children() == 0 || !rect_can_contain_children(inner_rect);
|
||||
callback(child, q, cell_rect, depth, is_visual_leaf ? IsVisualLeaf::Yes : IsVisualLeaf::No, IsRemainder::No);
|
||||
if (cell_rect.width() * cell_rect.height() < 16) {
|
||||
remaining_nodes_are_too_small = true;
|
||||
} else {
|
||||
lay_out_children(child, inner_rect, depth + 1, callback);
|
||||
}
|
||||
layout_rect.set_secondary_offset_for_orientation(orientation, layout_rect.secondary_offset_for_orientation(orientation) + node_size);
|
||||
main_dim -= node_size;
|
||||
placement_area -= child.area();
|
||||
}
|
||||
canvas.set_primary_offset_for_orientation(orientation, canvas.primary_offset_for_orientation(orientation) + fixed_side_size);
|
||||
canvas.set_primary_size_for_orientation(orientation, canvas.primary_size_for_orientation(orientation) - fixed_side_size);
|
||||
}
|
||||
|
||||
// Consume nodes that were added to this row or column.
|
||||
i = k - 1;
|
||||
total_area -= node_area_sum;
|
||||
}
|
||||
|
||||
// If not the entire canvas was filled with nodes, fill the remaining area with a dither pattern.
|
||||
if (!canvas.is_empty()) {
|
||||
callback(node, 0, canvas, depth, IsVisualLeaf::No, IsRemainder::Yes);
|
||||
}
|
||||
}
|
||||
|
||||
const TreeMapNode* TreeMapWidget::path_node(size_t n) const
|
||||
{
|
||||
if (!m_tree.ptr())
|
||||
return nullptr;
|
||||
const TreeMapNode* iter = &m_tree->root();
|
||||
size_t path_index = 0;
|
||||
while (iter && path_index < m_path.size() && path_index < n) {
|
||||
size_t child_index = m_path[path_index];
|
||||
if (child_index >= iter->num_children()) {
|
||||
return nullptr;
|
||||
}
|
||||
iter = &iter->child_at(child_index);
|
||||
path_index++;
|
||||
}
|
||||
return iter;
|
||||
}
|
||||
|
||||
void TreeMapWidget::paint_event(GUI::PaintEvent& event)
|
||||
{
|
||||
GUI::Frame::paint_event(event);
|
||||
GUI::Painter painter(*this);
|
||||
|
||||
m_selected_node_cache = path_node(m_path.size());
|
||||
|
||||
const TreeMapNode* node = path_node(m_viewpoint);
|
||||
if (!node) {
|
||||
painter.fill_rect(frame_inner_rect(), Color::MidGray);
|
||||
} else if (node_is_leaf(*node)) {
|
||||
paint_cell_frame(painter, *node, frame_inner_rect(), m_viewpoint - 1, true);
|
||||
} else {
|
||||
lay_out_children(*node, frame_inner_rect(), m_viewpoint, [&](const TreeMapNode& node, int, const Gfx::IntRect& rect, int depth, IsVisualLeaf visual_leaf, IsRemainder remainder) {
|
||||
if (remainder == IsRemainder::No) {
|
||||
bool fill = visual_leaf == IsVisualLeaf::Yes ? true : false;
|
||||
paint_cell_frame(painter, node, rect, depth, fill);
|
||||
} else {
|
||||
Color color = colors[depth % (sizeof(colors) / sizeof(colors[0]))];
|
||||
painter.clear_clip_rect();
|
||||
painter.add_clip_rect(rect);
|
||||
painter.draw_rect(rect, Color::Black);
|
||||
painter.fill_rect_with_dither_pattern(rect.shrunken(2, 2), color, Color::Black);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
Vector<int> TreeMapWidget::path_to_position(const Gfx::IntPoint& position)
|
||||
{
|
||||
const TreeMapNode* node = path_node(m_viewpoint);
|
||||
if (!node) {
|
||||
return {};
|
||||
}
|
||||
Vector<int> path;
|
||||
lay_out_children(*node, frame_inner_rect(), m_viewpoint, [&](const TreeMapNode&, int index, const Gfx::IntRect& rect, int, IsVisualLeaf, IsRemainder is_remainder) {
|
||||
if (is_remainder == IsRemainder::No && rect.contains(position)) {
|
||||
path.append(index);
|
||||
}
|
||||
});
|
||||
return path;
|
||||
}
|
||||
|
||||
void TreeMapWidget::mousedown_event(GUI::MouseEvent& event)
|
||||
{
|
||||
const TreeMapNode* node = path_node(m_viewpoint);
|
||||
if (node && !node_is_leaf(*node)) {
|
||||
Vector<int> path = path_to_position(event.position());
|
||||
if (!path.is_empty()) {
|
||||
m_path.shrink(m_viewpoint);
|
||||
m_path.append(path);
|
||||
if (on_path_change) {
|
||||
on_path_change();
|
||||
}
|
||||
update();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void TreeMapWidget::doubleclick_event(GUI::MouseEvent& event)
|
||||
{
|
||||
const TreeMapNode* node = path_node(m_viewpoint);
|
||||
if (node && !node_is_leaf(*node)) {
|
||||
Vector<int> path = path_to_position(event.position());
|
||||
m_path.shrink(m_viewpoint);
|
||||
m_path.append(path);
|
||||
m_viewpoint = m_path.size();
|
||||
if (on_path_change) {
|
||||
on_path_change();
|
||||
}
|
||||
update();
|
||||
}
|
||||
}
|
||||
|
||||
void TreeMapWidget::mousewheel_event(GUI::MouseEvent& event)
|
||||
{
|
||||
int delta = event.wheel_delta();
|
||||
// FIXME: The wheel_delta is premultiplied in the window server, we actually want a raw value here.
|
||||
int step_size = GUI::WindowServerConnection::the().send_sync<Messages::WindowServer::GetScrollStepSize>()->step_size();
|
||||
if (delta > 0) {
|
||||
size_t step_back = delta / step_size;
|
||||
if (step_back > m_viewpoint)
|
||||
step_back = m_viewpoint;
|
||||
set_viewpoint(m_viewpoint - step_back);
|
||||
} else {
|
||||
size_t step_up = (-delta) / step_size;
|
||||
set_viewpoint(m_viewpoint + step_up);
|
||||
}
|
||||
}
|
||||
|
||||
void TreeMapWidget::set_tree(RefPtr<TreeMap> tree)
|
||||
{
|
||||
m_tree = tree;
|
||||
m_path.clear();
|
||||
m_viewpoint = 0;
|
||||
if (on_path_change) {
|
||||
on_path_change();
|
||||
}
|
||||
update();
|
||||
}
|
||||
|
||||
void TreeMapWidget::set_viewpoint(size_t viewpoint)
|
||||
{
|
||||
if (viewpoint > m_path.size())
|
||||
viewpoint = m_path.size();
|
||||
m_viewpoint = viewpoint;
|
||||
if (on_path_change) {
|
||||
on_path_change();
|
||||
}
|
||||
update();
|
||||
}
|
||||
|
||||
size_t TreeMapWidget::path_size() const
|
||||
{
|
||||
return m_path.size() + 1;
|
||||
}
|
||||
|
||||
size_t TreeMapWidget::viewpoint() const
|
||||
{
|
||||
return m_viewpoint;
|
||||
}
|
||||
|
||||
}
|
90
Applications/SpaceAnalyzer/TreeMapWidget.h
Normal file
90
Applications/SpaceAnalyzer/TreeMapWidget.h
Normal file
|
@ -0,0 +1,90 @@
|
|||
/*
|
||||
* Copyright (c) 2021, the SerenityOS developers.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright notice, this
|
||||
* list of conditions and the following disclaimer.
|
||||
*
|
||||
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <LibGUI/Frame.h>
|
||||
#include <LibGfx/Rect.h>
|
||||
|
||||
namespace SpaceAnalyzer {
|
||||
|
||||
struct TreeMapNode {
|
||||
virtual String name() const = 0;
|
||||
virtual int64_t area() const = 0;
|
||||
virtual size_t num_children() const = 0;
|
||||
virtual const TreeMapNode& child_at(size_t i) const = 0;
|
||||
virtual void sort_children_by_area() const = 0;
|
||||
};
|
||||
|
||||
struct TreeMap : public RefCounted<TreeMap> {
|
||||
virtual ~TreeMap() { }
|
||||
virtual const TreeMapNode& root() const = 0;
|
||||
};
|
||||
|
||||
class TreeMapWidget final : public GUI::Frame {
|
||||
C_OBJECT(TreeMapWidget)
|
||||
|
||||
public:
|
||||
virtual ~TreeMapWidget() override;
|
||||
Function<void()> on_path_change;
|
||||
size_t path_size() const;
|
||||
const TreeMapNode* path_node(size_t n) const;
|
||||
size_t viewpoint() const;
|
||||
void set_viewpoint(size_t);
|
||||
void set_tree(RefPtr<TreeMap> tree);
|
||||
|
||||
private:
|
||||
TreeMapWidget();
|
||||
virtual void paint_event(GUI::PaintEvent&) override;
|
||||
virtual void mousedown_event(GUI::MouseEvent&) override;
|
||||
virtual void doubleclick_event(GUI::MouseEvent&) override;
|
||||
virtual void mousewheel_event(GUI::MouseEvent&) override;
|
||||
|
||||
bool rect_can_contain_children(const Gfx::IntRect& rect) const;
|
||||
bool rect_can_contain_label(const Gfx::IntRect& rect) const;
|
||||
Gfx::IntRect inner_rect_for_frame(const Gfx::IntRect& rect) const;
|
||||
|
||||
enum class IsVisualLeaf {
|
||||
Yes,
|
||||
No
|
||||
};
|
||||
enum class IsRemainder {
|
||||
Yes,
|
||||
No
|
||||
};
|
||||
|
||||
template<typename Function>
|
||||
void lay_out_children(const TreeMapNode&, const Gfx::IntRect&, int depth, Function);
|
||||
void paint_cell_frame(GUI::Painter&, const TreeMapNode&, const Gfx::IntRect&, int depth, bool fill) const;
|
||||
Vector<int> path_to_position(const Gfx::IntPoint&);
|
||||
|
||||
RefPtr<TreeMap> m_tree;
|
||||
Vector<int> m_path;
|
||||
size_t m_viewpoint;
|
||||
const void* m_selected_node_cache;
|
||||
};
|
||||
|
||||
}
|
300
Applications/SpaceAnalyzer/main.cpp
Normal file
300
Applications/SpaceAnalyzer/main.cpp
Normal file
|
@ -0,0 +1,300 @@
|
|||
/*
|
||||
* Copyright (c) 2021, the SerenityOS developers.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright notice, this
|
||||
* list of conditions and the following disclaimer.
|
||||
*
|
||||
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
#include "TreeMapWidget.h"
|
||||
#include <AK/Queue.h>
|
||||
#include <AK/QuickSort.h>
|
||||
#include <AK/RefCounted.h>
|
||||
#include <Applications/SpaceAnalyzer/SpaceAnalyzerGML.h>
|
||||
#include <LibCore/DirIterator.h>
|
||||
#include <LibCore/File.h>
|
||||
#include <LibGUI/AboutDialog.h>
|
||||
#include <LibGUI/Application.h>
|
||||
#include <LibGUI/BreadcrumbBar.h>
|
||||
#include <LibGUI/Icon.h>
|
||||
#include <LibGUI/Menu.h>
|
||||
#include <LibGUI/MenuBar.h>
|
||||
#include <LibGUI/StatusBar.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
static const char* APP_NAME = "SpaceAnalyzer";
|
||||
|
||||
struct TreeNode : public SpaceAnalyzer::TreeMapNode {
|
||||
TreeNode(String name)
|
||||
: m_name(move(name)) {};
|
||||
|
||||
virtual String name() const { return m_name; }
|
||||
virtual int64_t area() const { return m_area; }
|
||||
virtual size_t num_children() const
|
||||
{
|
||||
if (m_children) {
|
||||
return m_children->size();
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
virtual const TreeNode& child_at(size_t i) const { return m_children->at(i); }
|
||||
virtual void sort_children_by_area() const
|
||||
{
|
||||
if (m_children) {
|
||||
Vector<TreeNode>* children = const_cast<Vector<TreeNode>*>(m_children.ptr());
|
||||
quick_sort(*children, [](auto& a, auto& b) { return b.m_area < a.m_area; });
|
||||
}
|
||||
}
|
||||
|
||||
String m_name;
|
||||
int64_t m_area { 0 };
|
||||
OwnPtr<Vector<TreeNode>> m_children;
|
||||
};
|
||||
|
||||
struct Tree : public SpaceAnalyzer::TreeMap {
|
||||
Tree(String root_name)
|
||||
: m_root(move(root_name)) {};
|
||||
virtual ~Tree() {};
|
||||
TreeNode m_root;
|
||||
virtual const SpaceAnalyzer::TreeMapNode& root() const override
|
||||
{
|
||||
return m_root;
|
||||
};
|
||||
};
|
||||
|
||||
struct MountInfo {
|
||||
String mount_point;
|
||||
String source;
|
||||
};
|
||||
|
||||
static void fill_mounts(Vector<MountInfo>& output)
|
||||
{
|
||||
// Output info about currently mounted filesystems.
|
||||
auto df = Core::File::construct("/proc/df");
|
||||
if (!df->open(Core::IODevice::ReadOnly)) {
|
||||
fprintf(stderr, "Failed to open /proc/df: %s\n", df->error_string());
|
||||
return;
|
||||
}
|
||||
|
||||
auto content = df->read_all();
|
||||
auto json = JsonValue::from_string(content);
|
||||
ASSERT(json.has_value());
|
||||
|
||||
json.value().as_array().for_each([&output](auto& value) {
|
||||
auto filesystem_object = value.as_object();
|
||||
MountInfo mount_info;
|
||||
mount_info.mount_point = filesystem_object.get("mount_point").to_string();
|
||||
mount_info.source = filesystem_object.get("source").as_string_or("none");
|
||||
output.append(mount_info);
|
||||
});
|
||||
}
|
||||
|
||||
static MountInfo* find_mount_for_path(String path, Vector<MountInfo>& mounts)
|
||||
{
|
||||
MountInfo* result = nullptr;
|
||||
size_t length = 0;
|
||||
for (auto& mount_info : mounts) {
|
||||
String& mount_point = mount_info.mount_point;
|
||||
if (path.starts_with(mount_point)) {
|
||||
if (!result || mount_point.length() > length) {
|
||||
result = &mount_info;
|
||||
length = mount_point.length();
|
||||
}
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
static long long int update_totals(TreeNode& node)
|
||||
{
|
||||
long long int result = 0;
|
||||
if (node.m_children) {
|
||||
for (auto& child : *node.m_children) {
|
||||
result += update_totals(child);
|
||||
}
|
||||
node.m_area = result;
|
||||
} else {
|
||||
result = node.m_area;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
struct QueueEntry {
|
||||
QueueEntry(String path, TreeNode* node)
|
||||
: path(move(path))
|
||||
, node(node) {};
|
||||
String path;
|
||||
TreeNode* node { nullptr };
|
||||
};
|
||||
|
||||
static void populate_filesize_tree(TreeNode& root, Vector<MountInfo>& mounts, HashMap<int, int>& error_accumulator)
|
||||
{
|
||||
ASSERT(!root.m_name.ends_with("/"));
|
||||
|
||||
Queue<QueueEntry> queue;
|
||||
queue.enqueue(QueueEntry(root.m_name, &root));
|
||||
|
||||
StringBuilder builder = StringBuilder();
|
||||
builder.append(root.m_name);
|
||||
builder.append("/");
|
||||
MountInfo* root_mount_info = find_mount_for_path(builder.to_string(), mounts);
|
||||
if (!root_mount_info) {
|
||||
return;
|
||||
}
|
||||
while (!queue.is_empty()) {
|
||||
QueueEntry queue_entry = queue.dequeue();
|
||||
|
||||
builder.clear();
|
||||
builder.append(queue_entry.path);
|
||||
builder.append("/");
|
||||
|
||||
MountInfo* mount_info = find_mount_for_path(builder.to_string(), mounts);
|
||||
if (!mount_info || (mount_info != root_mount_info && mount_info->source != root_mount_info->source)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
Core::DirIterator dir_iterator(builder.to_string(), Core::DirIterator::SkipParentAndBaseDir);
|
||||
if (dir_iterator.has_error()) {
|
||||
int error_sum = error_accumulator.get(dir_iterator.error()).value_or(0);
|
||||
error_accumulator.set(dir_iterator.error(), error_sum + 1);
|
||||
} else {
|
||||
queue_entry.node->m_children = make<Vector<TreeNode>>();
|
||||
while (dir_iterator.has_next()) {
|
||||
queue_entry.node->m_children->append(TreeNode(dir_iterator.next_path()));
|
||||
}
|
||||
for (auto& child : *queue_entry.node->m_children) {
|
||||
String& name = child.m_name;
|
||||
int name_len = name.length();
|
||||
builder.append(name);
|
||||
struct stat st;
|
||||
int stat_result = lstat(builder.to_string().characters(), &st);
|
||||
if (stat_result < 0) {
|
||||
int error_sum = error_accumulator.get(errno).value_or(0);
|
||||
error_accumulator.set(errno, error_sum + 1);
|
||||
} else {
|
||||
if (S_ISDIR(st.st_mode)) {
|
||||
queue.enqueue(QueueEntry(builder.to_string(), &child));
|
||||
} else {
|
||||
child.m_area = st.st_size;
|
||||
}
|
||||
}
|
||||
builder.trim(name_len);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
update_totals(root);
|
||||
}
|
||||
|
||||
static void analyze(RefPtr<Tree> tree, SpaceAnalyzer::TreeMapWidget& treemapwidget, GUI::StatusBar& statusbar)
|
||||
{
|
||||
// Build an in-memory tree mirroring the filesystem and for each node
|
||||
// calculate the sum of the file size for all its descendants.
|
||||
TreeNode* root = &tree->m_root;
|
||||
Vector<MountInfo> mounts;
|
||||
fill_mounts(mounts);
|
||||
HashMap<int, int> error_accumulator;
|
||||
populate_filesize_tree(*root, mounts, error_accumulator);
|
||||
|
||||
// Display an error summary in the statusbar.
|
||||
if (!error_accumulator.is_empty()) {
|
||||
StringBuilder builder;
|
||||
bool first = true;
|
||||
builder.append("Some directories were not analyzed: ");
|
||||
for (auto& key : error_accumulator.keys()) {
|
||||
if (!first) {
|
||||
builder.append(", ");
|
||||
}
|
||||
builder.append(strerror(key));
|
||||
builder.append(" (");
|
||||
int value = error_accumulator.get(key).value();
|
||||
builder.append(String::number(value));
|
||||
if (value == 1) {
|
||||
builder.append(" time");
|
||||
} else {
|
||||
builder.append(" times");
|
||||
}
|
||||
builder.append(")");
|
||||
first = false;
|
||||
}
|
||||
statusbar.set_text(builder.to_string());
|
||||
} else {
|
||||
statusbar.set_text("No errors");
|
||||
}
|
||||
treemapwidget.set_tree(tree);
|
||||
}
|
||||
|
||||
int main(int argc, char* argv[])
|
||||
{
|
||||
auto app = GUI::Application::construct(argc, argv);
|
||||
|
||||
RefPtr<Tree> tree = adopt(*new Tree(""));
|
||||
|
||||
// Configure application window.
|
||||
auto app_icon = GUI::Icon::default_icon("app-space-analyzer");
|
||||
auto window = GUI::Window::construct();
|
||||
window->set_title(APP_NAME);
|
||||
window->resize(640, 480);
|
||||
window->set_icon(app_icon.bitmap_for_size(16));
|
||||
|
||||
// Load widgets.
|
||||
auto& mainwidget = window->set_main_widget<GUI::Widget>();
|
||||
mainwidget.load_from_gml(space_analyzer_gml);
|
||||
auto& breadcrumbbar = *mainwidget.find_descendant_of_type_named<GUI::BreadcrumbBar>("breadcrumb_bar");
|
||||
auto& treemapwidget = *mainwidget.find_descendant_of_type_named<SpaceAnalyzer::TreeMapWidget>("tree_map");
|
||||
auto& statusbar = *mainwidget.find_descendant_of_type_named<GUI::StatusBar>("status_bar");
|
||||
|
||||
// Configure the menubar.
|
||||
auto menubar = GUI::MenuBar::construct();
|
||||
auto& app_menu = menubar->add_menu(APP_NAME);
|
||||
app_menu.add_action(GUI::Action::create("Analyze", [&](auto&) {
|
||||
analyze(tree, treemapwidget, statusbar);
|
||||
}));
|
||||
app_menu.add_action(GUI::CommonActions::make_quit_action([&](auto&) {
|
||||
app->quit();
|
||||
}));
|
||||
auto& help_menu = menubar->add_menu("Help");
|
||||
help_menu.add_action(GUI::CommonActions::make_about_action(APP_NAME, app_icon, window));
|
||||
app->set_menubar(move(menubar));
|
||||
|
||||
// Configure event handlers.
|
||||
breadcrumbbar.on_segment_click = [&](size_t index) {
|
||||
ASSERT(index < treemapwidget.path_size());
|
||||
treemapwidget.set_viewpoint(index);
|
||||
};
|
||||
treemapwidget.on_path_change = [&]() {
|
||||
breadcrumbbar.clear_segments();
|
||||
for (size_t k = 0; k < treemapwidget.path_size(); k++) {
|
||||
if (k == 0) {
|
||||
breadcrumbbar.append_segment("/");
|
||||
} else {
|
||||
const SpaceAnalyzer::TreeMapNode* node = treemapwidget.path_node(k);
|
||||
breadcrumbbar.append_segment(node->name());
|
||||
}
|
||||
}
|
||||
breadcrumbbar.set_selected_segment(treemapwidget.viewpoint());
|
||||
};
|
||||
|
||||
// At startup automatically do an analysis of root.
|
||||
analyze(tree, treemapwidget, statusbar);
|
||||
|
||||
window->show();
|
||||
return app->exec();
|
||||
}
|
4
Base/res/apps/SpaceAnalyzer.af
Normal file
4
Base/res/apps/SpaceAnalyzer.af
Normal file
|
@ -0,0 +1,4 @@
|
|||
[App]
|
||||
Name=SpaceAnalyzer
|
||||
Executable=/bin/SpaceAnalyzer
|
||||
Category=Utilities
|
BIN
Base/res/icons/16x16/app-space-analyzer.png
Normal file
BIN
Base/res/icons/16x16/app-space-analyzer.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 5.4 KiB |
BIN
Base/res/icons/32x32/app-space-analyzer.png
Normal file
BIN
Base/res/icons/32x32/app-space-analyzer.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 952 B |
Loading…
Add table
Reference in a new issue