mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 14:04:27 +01:00
81 lines
2.9 KiB
Dart
81 lines
2.9 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 = [];
|
|
_selectedNode = null;
|
|
|
|
graph.nodes.clear();
|
|
graph.edges.clear();
|
|
|
|
// 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;
|
|
}
|