Removed legacy editor, started working on editing embedded graphs.
parent
e8af30d10c
commit
44087d7a7c
|
@ -13,8 +13,13 @@
|
|||
#include "imnodes.h"
|
||||
#include "misc/cpp/imgui_stdlib.h"
|
||||
|
||||
static AnimGraphResource sGraphGresource = AnimGraphResource();
|
||||
static bool sGraphLoadedThisFrame = false;
|
||||
struct EditorState {
|
||||
AnimGraphResource* rootGraphResource = nullptr;
|
||||
AnimGraphResource* currentGraphResource = nullptr;
|
||||
bool isGraphLoadedThisFrame = false;
|
||||
};
|
||||
|
||||
static EditorState sEditorState;
|
||||
|
||||
ImNodesPinShape sGetSocketShapeFromSocketType(const SocketType& socket_type) {
|
||||
switch (socket_type) {
|
||||
|
@ -164,7 +169,10 @@ void AnimGraphEditorRenderSidebar(
|
|||
node_resource->m_name = node_name_buffer;
|
||||
}
|
||||
|
||||
int num_properties = node_resource->m_socket_accessor->m_properties.size();
|
||||
int num_properties = 0;
|
||||
if (node_resource->m_socket_accessor != nullptr) {
|
||||
num_properties = node_resource->m_socket_accessor->m_properties.size();
|
||||
}
|
||||
for (int i = 0; i < num_properties; i++) {
|
||||
Socket& property = node_resource->m_socket_accessor->m_properties[i];
|
||||
if (property.m_type == SocketType::SocketTypeInt) {
|
||||
|
@ -251,9 +259,13 @@ void AnimGraphEditorRenderSidebar(
|
|||
}
|
||||
|
||||
void AnimGraphEditorClear() {
|
||||
sGraphGresource.Clear();
|
||||
sGraphGresource.m_blend_tree_resource.InitGraphConnectors();
|
||||
sGraphGresource.m_graph_type_name = "BlendTree";
|
||||
sEditorState.currentGraphResource = nullptr;
|
||||
delete sEditorState.rootGraphResource;
|
||||
|
||||
sEditorState.rootGraphResource = new AnimGraphResource();
|
||||
sEditorState.rootGraphResource->m_graph_type_name = "BlendTree";
|
||||
sEditorState.rootGraphResource->m_blend_tree_resource.InitGraphConnectors();
|
||||
sEditorState.currentGraphResource = sEditorState.rootGraphResource;
|
||||
}
|
||||
|
||||
void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
||||
|
@ -261,11 +273,11 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
|
||||
ImGui::BeginMenuBar();
|
||||
if (ImGui::Button("Save")) {
|
||||
sGraphGresource.SaveToFile("editor_graph.json");
|
||||
sEditorState.rootGraphResource->SaveToFile("editor_graph.json");
|
||||
}
|
||||
if (ImGui::Button("Load")) {
|
||||
sGraphGresource.LoadFromFile("editor_graph.json");
|
||||
sGraphLoadedThisFrame = true;
|
||||
sEditorState.rootGraphResource->LoadFromFile("editor_graph.json");
|
||||
sEditorState.isGraphLoadedThisFrame = true;
|
||||
}
|
||||
if (ImGui::Button("Clear")) {
|
||||
AnimGraphEditorClear();
|
||||
|
@ -277,10 +289,10 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
memset(graph_name_buffer, 0, sizeof(graph_name_buffer));
|
||||
strncpy(
|
||||
graph_name_buffer,
|
||||
sGraphGresource.m_name.c_str(),
|
||||
sEditorState.currentGraphResource->m_name.c_str(),
|
||||
sizeof(graph_name_buffer));
|
||||
if (ImGui::InputText("Name", graph_name_buffer, sizeof(graph_name_buffer))) {
|
||||
sGraphGresource.m_name = graph_name_buffer;
|
||||
sEditorState.currentGraphResource->m_name = graph_name_buffer;
|
||||
}
|
||||
|
||||
ImGui::EndMenuBar();
|
||||
|
@ -292,19 +304,20 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
//
|
||||
ax::NodeEditor::Begin("Graph Editor");
|
||||
|
||||
#if 1
|
||||
for (size_t node_id = 0,
|
||||
n = sGraphGresource.m_blend_tree_resource.GetNumNodes();
|
||||
n = sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNumNodes();
|
||||
node_id < n;
|
||||
node_id++) {
|
||||
AnimNodeResource* node_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(node_id);
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
node_id);
|
||||
|
||||
if (node_id == 0 || node_id == 1) {
|
||||
// continue;
|
||||
}
|
||||
|
||||
if (sGraphLoadedThisFrame) {
|
||||
if (sEditorState.isGraphLoadedThisFrame) {
|
||||
ax::NodeEditor::SetNodePosition(
|
||||
node_id,
|
||||
ImVec2(node_resource->m_position[0], node_resource->m_position[1]));
|
||||
|
@ -313,8 +326,9 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
ImGui::Text("%s", node_resource->m_node_type_name.c_str());
|
||||
|
||||
// Inputs
|
||||
std::vector<Socket>& node_inputs =
|
||||
node_resource->m_socket_accessor->m_inputs;
|
||||
std::vector<Socket> node_inputs =
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNodeInputSockets(node_resource);
|
||||
for (size_t j = 0, ni = node_inputs.size(); j < ni; j++) {
|
||||
Socket& socket = node_inputs[j];
|
||||
ax::NodeEditor::BeginPin(
|
||||
|
@ -327,8 +341,9 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
}
|
||||
|
||||
// Outputs
|
||||
std::vector<Socket>& node_outputs =
|
||||
node_resource->m_socket_accessor->m_outputs;
|
||||
std::vector<Socket> node_outputs =
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNodeOutputSockets(node_resource);
|
||||
for (size_t j = 0, ni = node_outputs.size(); j < ni; j++) {
|
||||
Socket& socket = node_outputs[j];
|
||||
ax::NodeEditor::BeginPin(
|
||||
|
@ -349,21 +364,23 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
|
||||
int link_id = 0;
|
||||
for (size_t connection_id = 0,
|
||||
n = sGraphGresource.m_blend_tree_resource.GetNumConnections();
|
||||
n = sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNumConnections();
|
||||
connection_id < n;
|
||||
connection_id++) {
|
||||
const BlendTreeConnectionResource* connection_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetConnection(connection_id);
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetConnection(
|
||||
connection_id);
|
||||
|
||||
const AnimNodeResource* source_node_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
connection_resource->source_node_index);
|
||||
int source_socket_index =
|
||||
source_node_resource->m_socket_accessor->GetOutputIndex(
|
||||
connection_resource->source_socket_name.c_str());
|
||||
|
||||
const AnimNodeResource* target_node_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
connection_resource->target_node_index);
|
||||
int target_socket_index =
|
||||
target_node_resource->m_socket_accessor->GetInputIndex(
|
||||
|
@ -379,9 +396,6 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
ax::NodeEditor::Link(link_id++, source_socket_pin_id, target_socket_pin_id);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
#if 1
|
||||
// Create Connections
|
||||
if (ax::NodeEditor::BeginCreate()) {
|
||||
ax::NodeEditor::PinId input_pin_id, output_pin_id;
|
||||
|
@ -397,7 +411,8 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
&source_node_socket_index);
|
||||
|
||||
const AnimNodeResource* source_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(source_node_index);
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
source_node_index);
|
||||
if (source_node->m_socket_accessor->m_outputs.size()
|
||||
< source_node_socket_index) {
|
||||
source_node_socket_index = -1;
|
||||
|
@ -412,14 +427,25 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
&target_node_socket_index);
|
||||
|
||||
const AnimNodeResource* target_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(target_node_index);
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
target_node_index);
|
||||
if (target_node->m_socket_accessor->m_inputs.size()
|
||||
< target_node_socket_index) {
|
||||
target_node_socket_index = -1;
|
||||
}
|
||||
|
||||
if (source_node_socket_index == -1
|
||||
|| target_node_socket_index == -1) {
|
||||
const Socket* source_socket =
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNodeOutputSocketByIndex(
|
||||
source_node,
|
||||
source_node_socket_index);
|
||||
const Socket* target_socket =
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.GetNodeInputSocketByIndex(
|
||||
target_node,
|
||||
target_node_socket_index);
|
||||
|
||||
if (source_socket == nullptr || target_socket == nullptr) {
|
||||
ax::NodeEditor::RejectNewItem();
|
||||
} else {
|
||||
const std::string& source_socket_name =
|
||||
|
@ -432,7 +458,8 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
->m_inputs[target_node_socket_index]
|
||||
.m_name;
|
||||
|
||||
sGraphGresource.m_blend_tree_resource.ConnectSockets(
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource
|
||||
.ConnectSockets(
|
||||
source_node,
|
||||
source_socket_name,
|
||||
target_node,
|
||||
|
@ -443,7 +470,6 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
}
|
||||
}
|
||||
ax::NodeEditor::EndCreate();
|
||||
#endif
|
||||
|
||||
// Popup menu
|
||||
{
|
||||
|
@ -489,12 +515,18 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
node_type_name = "ConstScalarNode";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("BlendTree")) {
|
||||
node_type_name = "BlendTree";
|
||||
}
|
||||
|
||||
if (!node_type_name.empty()) {
|
||||
AnimNodeResource* node_resource =
|
||||
AnimNodeResourceFactory(node_type_name);
|
||||
size_t node_id = sGraphGresource.m_blend_tree_resource.GetNumNodes();
|
||||
size_t node_id = sEditorState.currentGraphResource
|
||||
->m_blend_tree_resource.GetNumNodes();
|
||||
ax::NodeEditor::SetNodePosition(node_id, popup_mouse_position);
|
||||
sGraphGresource.m_blend_tree_resource.AddNode(node_resource);
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.AddNode(
|
||||
node_resource);
|
||||
}
|
||||
|
||||
ImGui::EndPopup();
|
||||
|
@ -517,353 +549,31 @@ void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context) {
|
|||
|
||||
if (selected_node_id.Get() != 0) {
|
||||
AnimGraphEditorRenderSidebar(
|
||||
sGraphGresource.m_blend_tree_resource,
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource,
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
selected_node_id.Get()));
|
||||
}
|
||||
}
|
||||
|
||||
ImGui::Columns(1);
|
||||
|
||||
ax::NodeEditor::NodeId double_clicked_node_id =
|
||||
ax::NodeEditor::GetDoubleClickedNode();
|
||||
if (!double_clicked_node_id.Invalid) {
|
||||
AnimNodeResource* clicked_node_resource =
|
||||
sEditorState.currentGraphResource->m_blend_tree_resource.GetNode(
|
||||
double_clicked_node_id.Get());
|
||||
if (clicked_node_resource->m_node_type_name == "BlendTree") {
|
||||
sEditorState.currentGraphResource =
|
||||
dynamic_cast<AnimGraphResource*>(clicked_node_resource);
|
||||
ax::NodeEditor::ClearSelection();
|
||||
}
|
||||
}
|
||||
|
||||
ax::NodeEditor::SetCurrentEditor(nullptr);
|
||||
sGraphLoadedThisFrame = false;
|
||||
}
|
||||
|
||||
void LegacyAnimGraphEditorUpdate() {
|
||||
ImGui::BeginMenuBar();
|
||||
if (ImGui::Button("Save")) {
|
||||
sGraphGresource.SaveToFile("editor_graph.json");
|
||||
}
|
||||
if (ImGui::Button("Load")) {
|
||||
sGraphGresource.LoadFromFile("editor_graph.json");
|
||||
|
||||
for (size_t i = 0, n = sGraphGresource.m_blend_tree_resource.GetNumNodes();
|
||||
i < n;
|
||||
i++) {
|
||||
const AnimNodeResource* node_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(i);
|
||||
ImNodes::SetNodeGridSpacePos(
|
||||
i,
|
||||
ImVec2(node_resource->m_position[0], node_resource->m_position[1]));
|
||||
}
|
||||
}
|
||||
if (ImGui::Button("Clear")) {
|
||||
sGraphGresource.Clear();
|
||||
}
|
||||
char graph_name_buffer[256];
|
||||
memset(graph_name_buffer, 0, sizeof(graph_name_buffer));
|
||||
strncpy(
|
||||
graph_name_buffer,
|
||||
sGraphGresource.m_name.c_str(),
|
||||
sizeof(graph_name_buffer));
|
||||
if (ImGui::InputText("Name", graph_name_buffer, sizeof(graph_name_buffer))) {
|
||||
sGraphGresource.m_name = graph_name_buffer;
|
||||
}
|
||||
|
||||
ImGui::EndMenuBar();
|
||||
|
||||
ImGui::Columns(2);
|
||||
|
||||
//
|
||||
// Node editor canvas
|
||||
//
|
||||
ImNodes::BeginNodeEditor();
|
||||
|
||||
// Popup menu
|
||||
{
|
||||
const bool open_popup =
|
||||
ImGui::IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows)
|
||||
&& ImNodes::IsEditorHovered()
|
||||
&& ImGui::IsMouseReleased(ImGuiMouseButton_Right);
|
||||
|
||||
ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, ImVec2(8.f, 8.f));
|
||||
if (!ImGui::IsAnyItemHovered() && open_popup) {
|
||||
ImGui::OpenPopup("add node");
|
||||
}
|
||||
|
||||
if (ImGui::BeginPopup("add node")) {
|
||||
const ImVec2 click_pos = ImGui::GetMousePosOnOpeningCurrentPopup();
|
||||
std::string node_type_name = "";
|
||||
if (ImGui::MenuItem("AnimSampler")) {
|
||||
node_type_name = "AnimSampler";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("Blend2")) {
|
||||
node_type_name = "Blend2";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("SpeedScale")) {
|
||||
node_type_name = "SpeedScale";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("LockTranslationNode")) {
|
||||
node_type_name = "LockTranslationNode";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("MathAddNode")) {
|
||||
node_type_name = "MathAddNode";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("MathFloatToVec3Node")) {
|
||||
node_type_name = "MathFloatToVec3Node";
|
||||
}
|
||||
|
||||
if (ImGui::MenuItem("ConstScalarNode")) {
|
||||
node_type_name = "ConstScalarNode";
|
||||
}
|
||||
|
||||
if (node_type_name != "") {
|
||||
AnimNodeResource* node_resource =
|
||||
AnimNodeResourceFactory(node_type_name);
|
||||
size_t node_id = sGraphGresource.m_blend_tree_resource.GetNumNodes();
|
||||
ImNodes::SetNodeScreenSpacePos(node_id, ImGui::GetMousePos());
|
||||
sGraphGresource.m_blend_tree_resource.AddNode(node_resource);
|
||||
}
|
||||
|
||||
ImGui::EndPopup();
|
||||
}
|
||||
|
||||
ImGui::PopStyleVar(ImGuiStyleVar_WindowPadding);
|
||||
}
|
||||
|
||||
for (size_t i = 0, n = sGraphGresource.m_blend_tree_resource.GetNumNodes();
|
||||
i < n;
|
||||
i++) {
|
||||
AnimNodeResource* node_resource =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(i);
|
||||
|
||||
ImNodes::BeginNode(i);
|
||||
ImGui::PushItemWidth(110.0f);
|
||||
|
||||
// Header
|
||||
ImNodes::BeginNodeTitleBar();
|
||||
if (node_resource
|
||||
== sGraphGresource.m_blend_tree_resource.GetGraphOutputNode()) {
|
||||
ImGui::TextUnformatted("Graph Outputs");
|
||||
} else if (
|
||||
node_resource
|
||||
== sGraphGresource.m_blend_tree_resource.GetGraphInputNode()) {
|
||||
ImGui::TextUnformatted("Graph Inputs");
|
||||
} else {
|
||||
ImGui::TextUnformatted(node_resource->m_node_type_name.c_str());
|
||||
}
|
||||
ImNodes::EndNodeTitleBar();
|
||||
|
||||
// Inputs
|
||||
std::vector<Socket>& node_inputs =
|
||||
node_resource->m_socket_accessor->m_inputs;
|
||||
for (size_t j = 0, ni = node_inputs.size(); j < ni; j++) {
|
||||
Socket& socket = node_inputs[j];
|
||||
|
||||
ImColor socket_color = ImColor(255, 255, 255, 255);
|
||||
if (socket.m_flags & SocketFlagAffectsTime) {
|
||||
socket_color = ImColor(255, 128, 128, 255);
|
||||
}
|
||||
|
||||
ImNodes::BeginInputAttribute(
|
||||
GenerateInputAttributeId(i, j),
|
||||
sGetSocketShapeFromSocketType(socket.m_type),
|
||||
socket_color);
|
||||
ImGui::TextUnformatted(socket.m_name.c_str());
|
||||
|
||||
bool socket_connected =
|
||||
sGraphGresource.m_blend_tree_resource.IsSocketConnected(
|
||||
node_resource,
|
||||
socket.m_name);
|
||||
if (!socket_connected && (socket.m_type == SocketType::SocketTypeFloat)) {
|
||||
ImGui::SameLine();
|
||||
float socket_value = socket.m_value.float_value;
|
||||
ImGui::PushItemWidth(
|
||||
130.0f - ImGui::CalcTextSize(socket.m_name.c_str()).x);
|
||||
if (ImGui::DragFloat("##hidelabel", &socket_value, 0.01f)) {
|
||||
socket.SetValue(socket_value);
|
||||
}
|
||||
ImGui::PopItemWidth();
|
||||
}
|
||||
|
||||
if (!socket_connected && (socket.m_type == SocketType::SocketTypeInt)) {
|
||||
ImGui::SameLine();
|
||||
int socket_value = socket.m_value.int_value;
|
||||
ImGui::PushItemWidth(
|
||||
130.0f - ImGui::CalcTextSize(socket.m_name.c_str()).x);
|
||||
if (ImGui::InputInt("##hidelabel", &socket_value, 1)) {
|
||||
socket.SetValue(socket_value);
|
||||
}
|
||||
ImGui::PopItemWidth();
|
||||
}
|
||||
|
||||
ImNodes::PushAttributeFlag(
|
||||
ImNodesAttributeFlags_EnableLinkDetachWithDragClick);
|
||||
ImNodes::EndInputAttribute();
|
||||
}
|
||||
|
||||
// Outputs
|
||||
const std::vector<Socket>& node_outputs =
|
||||
node_resource->m_socket_accessor->m_outputs;
|
||||
for (size_t j = 0, ni = node_outputs.size(); j < ni; j++) {
|
||||
const Socket& socket = node_outputs[j];
|
||||
ImNodes::BeginOutputAttribute(
|
||||
GenerateOutputAttributeId(i, j),
|
||||
sGetSocketShapeFromSocketType(socket.m_type),
|
||||
ImColor(255, 255, 255, 255));
|
||||
ImGui::TextUnformatted(socket.m_name.c_str());
|
||||
ImNodes::PushAttributeFlag(
|
||||
ImNodesAttributeFlags_EnableLinkDetachWithDragClick);
|
||||
ImNodes::EndInputAttribute();
|
||||
}
|
||||
|
||||
// Graph output node
|
||||
if (i == 0) {
|
||||
if (ImGui::Button("+Output")) {
|
||||
AnimNodeResource* graph_output_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetGraphOutputNode();
|
||||
|
||||
static float bla = 0.f;
|
||||
std::string socket_name = "Output";
|
||||
socket_name += std::to_string(
|
||||
graph_output_node->m_socket_accessor->m_inputs.size());
|
||||
graph_output_node->m_socket_accessor->RegisterInput<float>(
|
||||
socket_name.c_str(),
|
||||
nullptr);
|
||||
}
|
||||
} else if (i == 1) {
|
||||
if (ImGui::Button("+Input")) {
|
||||
AnimNodeResource* graph_input_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetGraphInputNode();
|
||||
|
||||
static float bla = 0.f;
|
||||
std::string socket_name = "Input";
|
||||
socket_name += std::to_string(
|
||||
graph_input_node->m_socket_accessor->m_outputs.size());
|
||||
graph_input_node->m_socket_accessor->RegisterOutput<float>(
|
||||
socket_name.c_str(),
|
||||
nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
// Save state in node resource
|
||||
ImVec2 node_pos = ImNodes::GetNodeGridSpacePos(i);
|
||||
node_resource->m_position[0] = node_pos[0];
|
||||
node_resource->m_position[1] = node_pos[1];
|
||||
ImGui::PopItemWidth();
|
||||
|
||||
ImNodes::EndNode();
|
||||
|
||||
// Ensure flags such as SocketFlagAffectsTime are properly set.
|
||||
node_resource->m_socket_accessor->UpdateFlags();
|
||||
}
|
||||
|
||||
for (size_t i = 0,
|
||||
n = sGraphGresource.m_blend_tree_resource.GetNumConnections();
|
||||
i < n;
|
||||
i++) {
|
||||
const BlendTreeConnectionResource* connection =
|
||||
sGraphGresource.m_blend_tree_resource.GetConnection(i);
|
||||
int start_attr, end_attr;
|
||||
|
||||
const AnimNodeResource* source_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
connection->source_node_index);
|
||||
int source_socket_index = source_node->m_socket_accessor->GetOutputIndex(
|
||||
connection->source_socket_name.c_str());
|
||||
|
||||
const AnimNodeResource* target_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
connection->target_node_index);
|
||||
int target_socket_index = target_node->m_socket_accessor->GetInputIndex(
|
||||
connection->target_socket_name.c_str());
|
||||
|
||||
start_attr = GenerateOutputAttributeId(
|
||||
connection->source_node_index,
|
||||
source_socket_index);
|
||||
end_attr = GenerateInputAttributeId(
|
||||
connection->target_node_index,
|
||||
target_socket_index);
|
||||
|
||||
ImNodes::Link(i, start_attr, end_attr);
|
||||
}
|
||||
|
||||
ImNodes::EndNodeEditor();
|
||||
|
||||
// Handle newly created links.
|
||||
int start_attr, end_attr;
|
||||
if (ImNodes::IsLinkCreated(&start_attr, &end_attr)) {
|
||||
int node_start_id;
|
||||
int node_start_output_index;
|
||||
SplitOutputAttributeId(
|
||||
start_attr,
|
||||
&node_start_id,
|
||||
&node_start_output_index);
|
||||
|
||||
int node_end_id;
|
||||
int node_end_input_index;
|
||||
SplitInputAttributeId(end_attr, &node_end_id, &node_end_input_index);
|
||||
|
||||
BlendTreeConnectionResource connection;
|
||||
connection.source_node_index = node_start_id;
|
||||
const AnimNodeResource* source_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(node_start_id);
|
||||
connection.source_socket_name =
|
||||
source_node->m_socket_accessor->m_outputs[node_start_output_index]
|
||||
.m_name;
|
||||
|
||||
connection.target_node_index = node_end_id;
|
||||
const AnimNodeResource* target_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(node_end_id);
|
||||
connection.target_socket_name =
|
||||
target_node->m_socket_accessor->m_inputs[node_end_input_index].m_name;
|
||||
|
||||
sGraphGresource.m_blend_tree_resource.ConnectSockets(
|
||||
source_node,
|
||||
connection.source_socket_name,
|
||||
target_node,
|
||||
connection.target_socket_name);
|
||||
}
|
||||
|
||||
if (ImGui::IsKeyPressed(ImGuiKey_Delete, false)) {
|
||||
std::cerr << "Delete key!" << std::endl;
|
||||
}
|
||||
|
||||
// Handle link detachements.
|
||||
int link_id = 0;
|
||||
if (ImNodes::IsLinkDestroyed(&link_id)) {
|
||||
BlendTreeConnectionResource* connection =
|
||||
sGraphGresource.m_blend_tree_resource.GetConnection(link_id);
|
||||
AnimNodeResource* source_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
connection->source_node_index);
|
||||
AnimNodeResource* target_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(
|
||||
connection->target_node_index);
|
||||
sGraphGresource.m_blend_tree_resource.DisconnectSockets(
|
||||
source_node,
|
||||
connection->source_socket_name,
|
||||
target_node,
|
||||
connection->target_socket_name);
|
||||
}
|
||||
|
||||
int selected_nodes[ImNodes::NumSelectedNodes()];
|
||||
ImNodes::GetSelectedNodes(selected_nodes);
|
||||
|
||||
//
|
||||
// Sidebar
|
||||
//
|
||||
ImGui::NextColumn();
|
||||
|
||||
if (ImNodes::NumSelectedNodes() == 1) {
|
||||
if (selected_nodes[0]
|
||||
< sGraphGresource.m_blend_tree_resource.GetNumNodes()) {
|
||||
AnimNodeResource* selected_node =
|
||||
sGraphGresource.m_blend_tree_resource.GetNode(selected_nodes[0]);
|
||||
AnimGraphEditorRenderSidebar(
|
||||
sGraphGresource.m_blend_tree_resource,
|
||||
selected_node);
|
||||
}
|
||||
}
|
||||
|
||||
ImGui::Columns(1);
|
||||
sEditorState.isGraphLoadedThisFrame = false;
|
||||
}
|
||||
|
||||
void AnimGraphEditorGetRuntimeGraph(AnimGraphBlendTree& blend_tree) {
|
||||
sGraphGresource.CreateBlendTreeInstance(blend_tree);
|
||||
sEditorState.rootGraphResource->CreateBlendTreeInstance(blend_tree);
|
||||
}
|
|
@ -69,8 +69,6 @@ void AnimGraphEditorClear();
|
|||
|
||||
void AnimGraphEditorUpdate(ax::NodeEditor::EditorContext* context);
|
||||
|
||||
void LegacyAnimGraphEditorUpdate();
|
||||
|
||||
void AnimGraphEditorGetRuntimeGraph(AnimGraphBlendTree& anim_graph);
|
||||
|
||||
#endif //ANIMTESTBED_ANIMGRAPHEDITOR_H
|
||||
|
|
|
@ -343,6 +343,7 @@ static bool sAnimGraphResourceBlendTreeFromJson(
|
|||
}
|
||||
|
||||
// Load connections
|
||||
if (json_data.contains("connections")) {
|
||||
for (const auto& json_connection : json_data["connections"]) {
|
||||
if (json_connection["type"] != "AnimGraphConnectionResource") {
|
||||
std::cerr << "Invalid json object. Expected type "
|
||||
|
@ -361,6 +362,7 @@ static bool sAnimGraphResourceBlendTreeFromJson(
|
|||
blend_tree_resource.GetNode(connection.target_node_index),
|
||||
connection.target_socket_name);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
@ -503,6 +505,31 @@ Socket* BlendTreeResource::GetNodeOutputSocket(
|
|||
return output_socket;
|
||||
}
|
||||
|
||||
const Socket* BlendTreeResource::GetNodeOutputSocketByIndex(
|
||||
const AnimNodeResource* node,
|
||||
const size_t socket_output_index) const {
|
||||
const std::vector<Socket>* output_sockets = nullptr;
|
||||
|
||||
if (node->m_socket_accessor) {
|
||||
output_sockets = &node->m_socket_accessor->m_outputs;
|
||||
} else if (node->m_node_type_name == "BlendTree") {
|
||||
const AnimGraphResource* graph_resource =
|
||||
dynamic_cast<const AnimGraphResource*>(node);
|
||||
const BlendTreeResource& blend_tree_resource =
|
||||
graph_resource->m_blend_tree_resource;
|
||||
|
||||
output_sockets =
|
||||
&blend_tree_resource.GetGraphOutputNode()->m_socket_accessor->m_outputs;
|
||||
}
|
||||
|
||||
if (output_sockets != nullptr
|
||||
&& output_sockets->size() > socket_output_index) {
|
||||
return &output_sockets->operator[](socket_output_index);
|
||||
}
|
||||
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
Socket* BlendTreeResource::GetNodeInputSocket(
|
||||
const AnimNodeResource* node,
|
||||
const std::string& input_socket_name) const {
|
||||
|
@ -526,6 +553,67 @@ Socket* BlendTreeResource::GetNodeInputSocket(
|
|||
return input_socket;
|
||||
}
|
||||
|
||||
const Socket* BlendTreeResource::GetNodeInputSocketByIndex(
|
||||
const AnimNodeResource* node,
|
||||
const size_t socket_input_index) const {
|
||||
const std::vector<Socket>* output_sockets = nullptr;
|
||||
|
||||
if (node->m_socket_accessor) {
|
||||
output_sockets = &node->m_socket_accessor->m_inputs;
|
||||
} else if (node->m_node_type_name == "BlendTree") {
|
||||
const AnimGraphResource* graph_resource =
|
||||
dynamic_cast<const AnimGraphResource*>(node);
|
||||
const BlendTreeResource& blend_tree_resource =
|
||||
graph_resource->m_blend_tree_resource;
|
||||
|
||||
output_sockets =
|
||||
&blend_tree_resource.GetGraphOutputNode()->m_socket_accessor->m_outputs;
|
||||
}
|
||||
|
||||
if (output_sockets != nullptr
|
||||
&& output_sockets->size() > socket_input_index) {
|
||||
return &output_sockets->operator[](socket_input_index);
|
||||
}
|
||||
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
std::vector<Socket> BlendTreeResource::GetNodeOutputSockets(
|
||||
const AnimNodeResource* node) const {
|
||||
if (node->m_socket_accessor) {
|
||||
return node->m_socket_accessor->m_outputs;
|
||||
}
|
||||
|
||||
if (node->m_node_type_name == "BlendTree") {
|
||||
const AnimGraphResource* graph_resource =
|
||||
dynamic_cast<const AnimGraphResource*>(node);
|
||||
const BlendTreeResource& blend_tree_resource =
|
||||
graph_resource->m_blend_tree_resource;
|
||||
return blend_tree_resource.GetGraphOutputNode()
|
||||
->m_socket_accessor->m_inputs;
|
||||
}
|
||||
|
||||
return std::vector<Socket>();
|
||||
}
|
||||
|
||||
std::vector<Socket> BlendTreeResource::GetNodeInputSockets(
|
||||
const AnimNodeResource* node) const {
|
||||
if (node->m_socket_accessor) {
|
||||
return node->m_socket_accessor->m_inputs;
|
||||
}
|
||||
|
||||
if (node->m_node_type_name == "BlendTree") {
|
||||
const AnimGraphResource* graph_resource =
|
||||
dynamic_cast<const AnimGraphResource*>(node);
|
||||
const BlendTreeResource& blend_tree_resource =
|
||||
graph_resource->m_blend_tree_resource;
|
||||
return blend_tree_resource.GetGraphInputNode()
|
||||
->m_socket_accessor->m_outputs;
|
||||
}
|
||||
|
||||
return std::vector<Socket>();
|
||||
}
|
||||
|
||||
bool BlendTreeResource::IsConnectionValid(
|
||||
const AnimNodeResource* source_node,
|
||||
const std::string& source_socket_name,
|
||||
|
|
|
@ -132,10 +132,21 @@ struct BlendTreeResource {
|
|||
const AnimNodeResource* node,
|
||||
const std::string& output_socket_name) const;
|
||||
|
||||
const Socket* GetNodeOutputSocketByIndex(
|
||||
const AnimNodeResource* node,
|
||||
const size_t socket_output_index) const;
|
||||
|
||||
Socket* GetNodeInputSocket(
|
||||
const AnimNodeResource* node,
|
||||
const std::string& input_socket_name) const;
|
||||
|
||||
const Socket* GetNodeInputSocketByIndex(
|
||||
const AnimNodeResource* node,
|
||||
const size_t socket_input_index) const;
|
||||
|
||||
std::vector<Socket> GetNodeOutputSockets(const AnimNodeResource* node) const;
|
||||
std::vector<Socket> GetNodeInputSockets(const AnimNodeResource* node) const;
|
||||
|
||||
bool ConnectSockets(
|
||||
const AnimNodeResource* source_node,
|
||||
const std::string& source_socket_name,
|
||||
|
|
73
src/main.cc
73
src/main.cc
|
@ -170,13 +170,6 @@ struct ApplicationConfig {
|
|||
int window_position[2] = {100, 30};
|
||||
int window_size[2] = {1000, 600};
|
||||
|
||||
struct LegacyGraphEditor {
|
||||
bool visible = false;
|
||||
int position[2] = {20, 20};
|
||||
int size[2] = {800, 500};
|
||||
};
|
||||
LegacyGraphEditor legacy_graph_editor;
|
||||
|
||||
struct GraphEditor {
|
||||
bool visible = false;
|
||||
int position[2] = {20, 20};
|
||||
|
@ -222,14 +215,6 @@ void to_json(nlohmann::json& j, const ApplicationConfig& config) {
|
|||
j["main_window"]["size"][0] = config.window_size[0];
|
||||
j["main_window"]["size"][1] = config.window_size[1];
|
||||
|
||||
j["legacy_graph_editor"]["visible"] = config.legacy_graph_editor.visible;
|
||||
j["legacy_graph_editor"]["position"][0] =
|
||||
config.legacy_graph_editor.position[0];
|
||||
j["legacy_graph_editor"]["position"][1] =
|
||||
config.legacy_graph_editor.position[1];
|
||||
j["legacy_graph_editor"]["size"][0] = config.legacy_graph_editor.size[0];
|
||||
j["legacy_graph_editor"]["size"][1] = config.legacy_graph_editor.size[1];
|
||||
|
||||
j["graph_editor"]["visible"] = config.graph_editor.visible;
|
||||
j["graph_editor"]["position"][0] = config.graph_editor.position[0];
|
||||
j["graph_editor"]["position"][1] = config.graph_editor.position[1];
|
||||
|
@ -281,28 +266,6 @@ void from_json(const nlohmann::json& j, ApplicationConfig& config) {
|
|||
}
|
||||
}
|
||||
|
||||
if (j.contains("legacy_graph_editor")) {
|
||||
if (j["legacy_graph_editor"].contains("visible")) {
|
||||
config.legacy_graph_editor.visible = j["legacy_graph_editor"]["visible"];
|
||||
}
|
||||
|
||||
if (j["legacy_graph_editor"].contains("position")
|
||||
and j["legacy_graph_editor"]["position"].size() == 2) {
|
||||
config.legacy_graph_editor.position[0] =
|
||||
j["legacy_graph_editor"]["position"].at(0);
|
||||
config.legacy_graph_editor.position[1] =
|
||||
j["legacy_graph_editor"]["position"].at(1);
|
||||
}
|
||||
|
||||
if (j["legacy_graph_editor"].contains("size")
|
||||
and j["legacy_graph_editor"]["size"].size() == 2) {
|
||||
config.legacy_graph_editor.size[0] =
|
||||
j["legacy_graph_editor"]["size"].at(0);
|
||||
config.legacy_graph_editor.size[1] =
|
||||
j["legacy_graph_editor"]["size"].at(1);
|
||||
}
|
||||
}
|
||||
|
||||
if (j.contains("graph_editor")) {
|
||||
if (j["graph_editor"].contains("visible")) {
|
||||
config.graph_editor.visible = j["graph_editor"]["visible"];
|
||||
|
@ -817,9 +780,6 @@ int main() {
|
|||
&gApplicationConfig.animation_player_widget.visible);
|
||||
|
||||
ImGui::Separator();
|
||||
ImGui::Checkbox(
|
||||
"Legacy Graph Editor",
|
||||
&gApplicationConfig.legacy_graph_editor.visible);
|
||||
|
||||
ImGui::Checkbox(
|
||||
"ImGui Demo",
|
||||
|
@ -1118,39 +1078,6 @@ int main() {
|
|||
ImGui::End();
|
||||
}
|
||||
|
||||
// Legacy Animation Graph Editor
|
||||
if (gApplicationConfig.legacy_graph_editor.visible) {
|
||||
ImGui::SetNextWindowPos(
|
||||
ImVec2(
|
||||
gApplicationConfig.legacy_graph_editor.position[0],
|
||||
gApplicationConfig.legacy_graph_editor.position[1]),
|
||||
ImGuiCond_FirstUseEver);
|
||||
ImGui::SetNextWindowSize(
|
||||
ImVec2(
|
||||
gApplicationConfig.legacy_graph_editor.size[0],
|
||||
gApplicationConfig.legacy_graph_editor.size[1]),
|
||||
ImGuiCond_FirstUseEver);
|
||||
|
||||
ImGui::Begin(
|
||||
"Legacy Graph Editor",
|
||||
&gApplicationConfig.legacy_graph_editor.visible,
|
||||
ImGuiWindowFlags_MenuBar);
|
||||
|
||||
ImVec2 graph_editor_position = ImGui::GetWindowPos();
|
||||
gApplicationConfig.legacy_graph_editor.position[0] =
|
||||
graph_editor_position.x;
|
||||
gApplicationConfig.legacy_graph_editor.position[1] =
|
||||
graph_editor_position.y;
|
||||
|
||||
ImVec2 graph_editor_size = ImGui::GetWindowSize();
|
||||
gApplicationConfig.legacy_graph_editor.size[0] = graph_editor_size.x;
|
||||
gApplicationConfig.legacy_graph_editor.size[1] = graph_editor_size.y;
|
||||
|
||||
LegacyAnimGraphEditorUpdate();
|
||||
|
||||
ImGui::End();
|
||||
}
|
||||
|
||||
// 3. Show the ImGui test window. Most of the sample code is in ImGui::ShowDemoWindow()
|
||||
if (gApplicationConfig.show_imgui_demo_window) {
|
||||
ImGui::SetNextWindowPos(ImVec2(460, 20), ImGuiCond_FirstUseEver);
|
||||
|
|
Loading…
Reference in New Issue