mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 22:14:28 +01:00
d6b0894c6b
The SkillTreeViewModel class serves as the view model for the skill tree and extends Flutter's ChangeNotifier for state management. Features include: - Storing and managing the list of SkillTreeNodes and SkillTreeEdges. - Managing the state of the selected node. - Initializing the skill tree with predefined nodes and edges. - Methods for toggling node selection, allowing for only a single node to be selected at any given time. The view model utilizes the GraphView package for visualization and layout.
77 lines
2.8 KiB
Dart
77 lines
2.8 KiB
Dart
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_edge.dart';
|
|
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_node.dart';
|
|
import 'package:flutter/foundation.dart';
|
|
import 'package:graphview/GraphView.dart';
|
|
|
|
class SkillTreeViewModel extends ChangeNotifier {
|
|
List<SkillTreeNode> _skillTreeNodes = [];
|
|
List<SkillTreeEdge> _skillTreeEdges = [];
|
|
SkillTreeNode? _selectedNode;
|
|
|
|
SkillTreeNode? get selectedNode => _selectedNode;
|
|
|
|
final Graph graph = Graph()..isTree = true;
|
|
BuchheimWalkerConfiguration builder = BuchheimWalkerConfiguration();
|
|
|
|
void initializeSkillTree() {
|
|
_skillTreeNodes = [];
|
|
_skillTreeEdges = [];
|
|
|
|
// Add nodes to _skillTreeNodes
|
|
_skillTreeNodes.addAll([
|
|
SkillTreeNode(color: 'red', id: 1),
|
|
SkillTreeNode(color: 'blue', id: 2),
|
|
SkillTreeNode(color: 'green', id: 3),
|
|
SkillTreeNode(color: 'yellow', id: 4),
|
|
SkillTreeNode(color: 'orange', id: 5),
|
|
SkillTreeNode(color: 'purple', id: 6),
|
|
SkillTreeNode(color: 'brown', id: 7),
|
|
SkillTreeNode(color: 'pink', id: 8),
|
|
SkillTreeNode(color: 'grey', id: 9),
|
|
SkillTreeNode(color: 'cyan', id: 10),
|
|
SkillTreeNode(color: 'magenta', id: 11),
|
|
SkillTreeNode(color: 'lime', id: 12)
|
|
]);
|
|
|
|
// Add edges to _skillTreeEdges
|
|
_skillTreeEdges.addAll([
|
|
SkillTreeEdge(id: '1_to_2', from: '1', to: '2', arrows: 'to'),
|
|
SkillTreeEdge(id: '1_to_3', from: '1', to: '3', arrows: 'to'),
|
|
SkillTreeEdge(id: '1_to_4', from: '1', to: '4', arrows: 'to'),
|
|
SkillTreeEdge(id: '2_to_5', from: '2', to: '5', arrows: 'to'),
|
|
SkillTreeEdge(id: '2_to_6', from: '2', to: '6', arrows: 'to'),
|
|
SkillTreeEdge(id: '6_to_7', from: '6', to: '7', arrows: 'to'),
|
|
SkillTreeEdge(id: '6_to_8', from: '6', to: '8', arrows: 'to'),
|
|
SkillTreeEdge(id: '4_to_9', from: '4', to: '9', arrows: 'to'),
|
|
SkillTreeEdge(id: '4_to_10', from: '4', to: '10', arrows: 'to'),
|
|
SkillTreeEdge(id: '4_to_11', from: '4', to: '11', arrows: 'to'),
|
|
SkillTreeEdge(id: '11_to_12', from: '11', to: '12', arrows: 'to')
|
|
]);
|
|
|
|
builder
|
|
..siblingSeparation = (100)
|
|
..levelSeparation = (150)
|
|
..subtreeSeparation = (150)
|
|
..orientation = (BuchheimWalkerConfiguration.ORIENTATION_LEFT_RIGHT);
|
|
|
|
notifyListeners();
|
|
}
|
|
|
|
void toggleNodeSelection(int nodeId) {
|
|
if (_selectedNode?.id == nodeId) {
|
|
// Unselect the node if it's already selected
|
|
_selectedNode = null;
|
|
} else {
|
|
// Select the new node
|
|
_selectedNode = _skillTreeNodes.firstWhere((node) => node.id == nodeId);
|
|
}
|
|
notifyListeners();
|
|
}
|
|
|
|
// Getter to expose nodes for the View
|
|
List<SkillTreeNode> get skillTreeNodes => _skillTreeNodes;
|
|
|
|
// Getter to expose edges for the View
|
|
List<SkillTreeEdge> get skillTreeEdges => _skillTreeEdges;
|
|
}
|