Hans Goudey
11be151d58
This commit adds a search menu when links are dragged above empty space. When releasing the drag, a menu displays all compatible sockets with the source link. The "main" sockets (usually the first) are weighted above other sockets in the search, so they appear first when you type the name of the node. A few special operators for creating a reroute or a group input node are also added to the search. Translation is started after choosing a node so it can be placed quickly, since users would likely adjust the position after anyway. A small "+" is displayed next to the cursor to give a hint about this. Further improvements are possible after this first iteration: - Support custom node trees. - Better drawing of items in the search menu. - Potential tweaks to filtering of items, depending on user feedback. Thanks to Juanfran Matheu for developing an initial patch. Differential Revision: https://developer.blender.org/D8286
140 lines
5.4 KiB
C++
140 lines
5.4 KiB
C++
/*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <string.h>
|
|
|
|
#include "BLI_float3.hh"
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "DNA_node_types.h"
|
|
|
|
#include "BKE_node.h"
|
|
|
|
#include "BLT_translation.h"
|
|
|
|
#include "NOD_geometry.h"
|
|
#include "NOD_geometry_exec.hh"
|
|
#include "NOD_socket_declarations.hh"
|
|
#include "NOD_socket_declarations_geometry.hh"
|
|
|
|
#include "node_util.h"
|
|
|
|
void geo_node_type_base(
|
|
struct bNodeType *ntype, int type, const char *name, short nclass, short flag);
|
|
bool geo_node_poll_default(struct bNodeType *ntype,
|
|
struct bNodeTree *ntree,
|
|
const char **r_disabled_hint);
|
|
|
|
namespace blender::nodes {
|
|
/**
|
|
* Update the availability of a group of input sockets with the same name,
|
|
* used for switching between attribute inputs or single values.
|
|
*
|
|
* \param mode: Controls which socket of the group to make available.
|
|
* \param name_is_available: If false, make all sockets with this name unavailable.
|
|
*/
|
|
void update_attribute_input_socket_availabilities(bNodeTree &ntree,
|
|
bNode &node,
|
|
const StringRef name,
|
|
const GeometryNodeAttributeInputMode mode,
|
|
const bool name_is_available = true);
|
|
|
|
Array<uint32_t> get_geometry_element_ids_as_uints(const GeometryComponent &component,
|
|
const AttributeDomain domain);
|
|
|
|
void transform_mesh(Mesh &mesh,
|
|
const float3 translation,
|
|
const float3 rotation,
|
|
const float3 scale);
|
|
|
|
void transform_geometry_set(GeometrySet &geometry,
|
|
const float4x4 &transform,
|
|
const Depsgraph &depsgraph);
|
|
|
|
Mesh *create_line_mesh(const float3 start, const float3 delta, const int count);
|
|
|
|
Mesh *create_grid_mesh(const int verts_x,
|
|
const int verts_y,
|
|
const float size_x,
|
|
const float size_y);
|
|
|
|
struct ConeAttributeOutputs {
|
|
StrongAnonymousAttributeID top_id;
|
|
StrongAnonymousAttributeID bottom_id;
|
|
StrongAnonymousAttributeID side_id;
|
|
};
|
|
|
|
Mesh *create_cylinder_or_cone_mesh(const float radius_top,
|
|
const float radius_bottom,
|
|
const float depth,
|
|
const int circle_segments,
|
|
const int side_segments,
|
|
const int fill_segments,
|
|
const GeometryNodeMeshCircleFillType fill_type,
|
|
ConeAttributeOutputs &attribute_outputs);
|
|
|
|
Mesh *create_cuboid_mesh(float3 size, int verts_x, int verts_y, int verts_z);
|
|
|
|
/**
|
|
* Copies the point domain attributes from `in_component` that are in the mask to `out_component`.
|
|
*/
|
|
void copy_point_attributes_based_on_mask(const GeometryComponent &in_component,
|
|
GeometryComponent &result_component,
|
|
Span<bool> masks,
|
|
const bool invert);
|
|
/**
|
|
* Returns the parts of the geometry that are on the selection for the given domain. If the domain
|
|
* is not applicable for the component, e.g. face domain for point cloud, nothing happens to that
|
|
* component. If no component can work with the domain, then `error_message` is set to true.
|
|
*/
|
|
void separate_geometry(GeometrySet &geometry_set,
|
|
const AttributeDomain domain,
|
|
const GeometryNodeDeleteGeometryMode mode,
|
|
const Field<bool> &selection_field,
|
|
const bool invert,
|
|
bool &r_is_error);
|
|
|
|
struct CurveToPointsResults {
|
|
int result_size;
|
|
MutableSpan<float3> positions;
|
|
MutableSpan<float> radii;
|
|
MutableSpan<float> tilts;
|
|
|
|
Map<AttributeIDRef, GMutableSpan> point_attributes;
|
|
|
|
MutableSpan<float3> tangents;
|
|
MutableSpan<float3> normals;
|
|
MutableSpan<float3> rotations;
|
|
};
|
|
/**
|
|
* Create references for all result point cloud attributes to simplify accessing them later on.
|
|
*/
|
|
CurveToPointsResults curve_to_points_create_result_attributes(PointCloudComponent &points,
|
|
const CurveEval &curve);
|
|
|
|
void curve_create_default_rotation_attribute(Span<float3> tangents,
|
|
Span<float3> normals,
|
|
MutableSpan<float3> rotations);
|
|
|
|
std::optional<CustomDataType> node_data_type_to_custom_data_type(eNodeSocketDatatype type);
|
|
std::optional<CustomDataType> node_socket_to_custom_data_type(const bNodeSocket &socket);
|
|
|
|
} // namespace blender::nodes
|