Welcome to mirror list, hosted at ThFree Co, Russian Federation.

git.blender.org/blender.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc
diff options
context:
space:
mode:
Diffstat (limited to 'source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc')
-rw-r--r--source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc453
1 files changed, 0 insertions, 453 deletions
diff --git a/source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc b/source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc
deleted file mode 100644
index 5e050517710..00000000000
--- a/source/blender/editors/space_spreadsheet/spreadsheet_from_geometry.cc
+++ /dev/null
@@ -1,453 +0,0 @@
-/*
- * 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.
- */
-
-#include "BKE_context.h"
-#include "BKE_editmesh.h"
-#include "BKE_lib_id.h"
-#include "BKE_mesh.h"
-#include "BKE_mesh_wrapper.h"
-#include "BKE_modifier.h"
-
-#include "DNA_ID.h"
-#include "DNA_mesh_types.h"
-#include "DNA_meshdata_types.h"
-#include "DNA_space_types.h"
-#include "DNA_userdef_types.h"
-
-#include "DEG_depsgraph_query.h"
-
-#include "bmesh.h"
-
-#include "spreadsheet_from_geometry.hh"
-#include "spreadsheet_intern.hh"
-
-namespace blender::ed::spreadsheet {
-
-using blender::bke::ReadAttribute;
-using blender::bke::ReadAttributePtr;
-
-static void add_columns_for_instances(const InstancesComponent &instances_component,
- SpreadsheetColumnLayout &column_layout,
- ResourceScope &scope)
-{
- Span<InstancedData> instance_data = instances_component.instanced_data();
- Span<float4x4> transforms = instances_component.transforms();
-
- Vector<std::unique_ptr<SpreadsheetColumn>> &columns =
- scope.construct<Vector<std::unique_ptr<SpreadsheetColumn>>>("columns");
-
- columns.append(spreadsheet_column_from_function(
- "Name", [instance_data](int index, CellValue &r_cell_value) {
- const InstancedData &data = instance_data[index];
- if (data.type == INSTANCE_DATA_TYPE_OBJECT) {
- if (data.data.object != nullptr) {
- r_cell_value.value_object = ObjectCellValue{data.data.object};
- }
- }
- else if (data.type == INSTANCE_DATA_TYPE_COLLECTION) {
- if (data.data.collection != nullptr) {
- r_cell_value.value_collection = CollectionCellValue{data.data.collection};
- }
- }
- }));
-
- columns.last()->default_width = 8.0f;
-
- static std::array<char, 3> axis_char = {'X', 'Y', 'Z'};
- for (const int i : {0, 1, 2}) {
- std::string name = std::string("Position ") + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [transforms, i](int index, CellValue &r_cell_value) {
- r_cell_value.value_float = transforms[index].translation()[i];
- }));
- }
-
- for (const int i : {0, 1, 2}) {
- std::string name = std::string("Rotation ") + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [transforms, i](int index, CellValue &r_cell_value) {
- r_cell_value.value_float = transforms[index].to_euler()[i];
- }));
- }
-
- for (const int i : {0, 1, 2}) {
- std::string name = std::string("Scale ") + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [transforms, i](int index, CellValue &r_cell_value) {
- r_cell_value.value_float = transforms[index].scale()[i];
- }));
- }
-
- for (std::unique_ptr<SpreadsheetColumn> &column : columns) {
- column_layout.columns.append(column.get());
- }
-
- column_layout.row_indices = instance_data.index_range().as_span();
- column_layout.tot_rows = instances_component.instances_amount();
-}
-
-static Vector<std::string> get_sorted_attribute_names_to_display(
- const GeometryComponent &component, const AttributeDomain domain)
-{
- Vector<std::string> attribute_names;
- component.attribute_foreach(
- [&](const StringRef attribute_name, const AttributeMetaData &meta_data) {
- if (meta_data.domain == domain) {
- attribute_names.append(attribute_name);
- }
- return true;
- });
- std::sort(attribute_names.begin(),
- attribute_names.end(),
- [](const std::string &a, const std::string &b) {
- return BLI_strcasecmp_natural(a.c_str(), b.c_str()) < 0;
- });
- return attribute_names;
-}
-
-static void add_columns_for_attribute(const ReadAttribute *attribute,
- const StringRefNull attribute_name,
- Vector<std::unique_ptr<SpreadsheetColumn>> &columns)
-{
- const CustomDataType data_type = attribute->custom_data_type();
- switch (data_type) {
- case CD_PROP_FLOAT: {
- columns.append(spreadsheet_column_from_function(
- attribute_name, [attribute](int index, CellValue &r_cell_value) {
- float value;
- attribute->get(index, &value);
- r_cell_value.value_float = value;
- }));
- break;
- }
- case CD_PROP_FLOAT2: {
- static std::array<char, 2> axis_char = {'X', 'Y'};
- for (const int i : {0, 1}) {
- std::string name = attribute_name + " " + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [attribute, i](int index, CellValue &r_cell_value) {
- float2 value;
- attribute->get(index, &value);
- r_cell_value.value_float = value[i];
- }));
- }
- break;
- }
- case CD_PROP_FLOAT3: {
- static std::array<char, 3> axis_char = {'X', 'Y', 'Z'};
- for (const int i : {0, 1, 2}) {
- std::string name = attribute_name + " " + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [attribute, i](int index, CellValue &r_cell_value) {
- float3 value;
- attribute->get(index, &value);
- r_cell_value.value_float = value[i];
- }));
- }
- break;
- }
- case CD_PROP_COLOR: {
- static std::array<char, 4> axis_char = {'R', 'G', 'B', 'A'};
- for (const int i : {0, 1, 2, 3}) {
- std::string name = attribute_name + " " + axis_char[i];
- columns.append(spreadsheet_column_from_function(
- name, [attribute, i](int index, CellValue &r_cell_value) {
- Color4f value;
- attribute->get(index, &value);
- r_cell_value.value_float = value[i];
- }));
- }
- break;
- }
- case CD_PROP_INT32: {
- columns.append(spreadsheet_column_from_function(
- attribute_name, [attribute](int index, CellValue &r_cell_value) {
- int value;
- attribute->get(index, &value);
- r_cell_value.value_int = value;
- }));
- break;
- }
- case CD_PROP_BOOL: {
- columns.append(spreadsheet_column_from_function(
- attribute_name, [attribute](int index, CellValue &r_cell_value) {
- bool value;
- attribute->get(index, &value);
- r_cell_value.value_bool = value;
- }));
- break;
- }
- default:
- break;
- }
-}
-
-static GeometrySet get_display_geometry_set(SpaceSpreadsheet *sspreadsheet,
- Object *object_eval,
- const GeometryComponentType used_component_type)
-{
- GeometrySet geometry_set;
- if (sspreadsheet->object_eval_state == SPREADSHEET_OBJECT_EVAL_STATE_ORIGINAL) {
- Object *object_orig = DEG_get_original_object(object_eval);
- if (object_orig->type == OB_MESH) {
- MeshComponent &mesh_component = geometry_set.get_component_for_write<MeshComponent>();
- if (object_orig->mode == OB_MODE_EDIT) {
- Mesh *mesh = (Mesh *)object_orig->data;
- BMEditMesh *em = mesh->edit_mesh;
- if (em != nullptr) {
- Mesh *new_mesh = (Mesh *)BKE_id_new_nomain(ID_ME, nullptr);
- /* This is a potentially heavy operation to do on every redraw. The best solution here is
- * to display the data directly from the bmesh without a conversion, which can be
- * implemented a bit later. */
- BM_mesh_bm_to_me_for_eval(em->bm, new_mesh, nullptr);
- mesh_component.replace(new_mesh, GeometryOwnershipType::Owned);
- }
- }
- else {
- Mesh *mesh = (Mesh *)object_orig->data;
- mesh_component.replace(mesh, GeometryOwnershipType::ReadOnly);
- }
- mesh_component.copy_vertex_group_names_from_object(*object_orig);
- }
- else if (object_orig->type == OB_POINTCLOUD) {
- PointCloud *pointcloud = (PointCloud *)object_orig->data;
- PointCloudComponent &pointcloud_component =
- geometry_set.get_component_for_write<PointCloudComponent>();
- pointcloud_component.replace(pointcloud, GeometryOwnershipType::ReadOnly);
- }
- }
- else {
- if (used_component_type == GEO_COMPONENT_TYPE_MESH && object_eval->mode == OB_MODE_EDIT) {
- Mesh *mesh = BKE_modifier_get_evaluated_mesh_from_evaluated_object(object_eval, false);
- if (mesh == nullptr) {
- return geometry_set;
- }
- BKE_mesh_wrapper_ensure_mdata(mesh);
- MeshComponent &mesh_component = geometry_set.get_component_for_write<MeshComponent>();
- mesh_component.replace(mesh, GeometryOwnershipType::ReadOnly);
- mesh_component.copy_vertex_group_names_from_object(*object_eval);
- }
- else {
- if (sspreadsheet->object_eval_state == SPREADSHEET_OBJECT_EVAL_STATE_NODE) {
- if (object_eval->runtime.geometry_set_preview != nullptr) {
- geometry_set = *object_eval->runtime.geometry_set_preview;
- }
- }
- else if (sspreadsheet->object_eval_state == SPREADSHEET_OBJECT_EVAL_STATE_FINAL) {
- if (object_eval->runtime.geometry_set_eval != nullptr) {
- geometry_set = *object_eval->runtime.geometry_set_eval;
- }
- }
- }
- }
- return geometry_set;
-}
-
-using IsVertexSelectedFn = FunctionRef<bool(int vertex_index)>;
-
-static void get_selected_vertex_indices(const Mesh &mesh,
- const IsVertexSelectedFn is_vertex_selected_fn,
- Vector<int64_t> &r_vertex_indices)
-{
- for (const int i : IndexRange(mesh.totvert)) {
- if (is_vertex_selected_fn(i)) {
- r_vertex_indices.append(i);
- }
- }
-}
-
-static void get_selected_corner_indices(const Mesh &mesh,
- const IsVertexSelectedFn is_vertex_selected_fn,
- Vector<int64_t> &r_corner_indices)
-{
- for (const int i : IndexRange(mesh.totloop)) {
- const MLoop &loop = mesh.mloop[i];
- if (is_vertex_selected_fn(loop.v)) {
- r_corner_indices.append(i);
- }
- }
-}
-
-static void get_selected_face_indices(const Mesh &mesh,
- const IsVertexSelectedFn is_vertex_selected_fn,
- Vector<int64_t> &r_face_indices)
-{
- for (const int poly_index : IndexRange(mesh.totpoly)) {
- const MPoly &poly = mesh.mpoly[poly_index];
- bool is_selected = true;
- for (const int loop_index : IndexRange(poly.loopstart, poly.totloop)) {
- const MLoop &loop = mesh.mloop[loop_index];
- if (!is_vertex_selected_fn(loop.v)) {
- is_selected = false;
- break;
- }
- }
- if (is_selected) {
- r_face_indices.append(poly_index);
- }
- }
-}
-
-static void get_selected_edge_indices(const Mesh &mesh,
- const IsVertexSelectedFn is_vertex_selected_fn,
- Vector<int64_t> &r_edge_indices)
-{
- for (const int i : IndexRange(mesh.totedge)) {
- const MEdge &edge = mesh.medge[i];
- if (is_vertex_selected_fn(edge.v1) && is_vertex_selected_fn(edge.v2)) {
- r_edge_indices.append(i);
- }
- }
-}
-
-static void get_selected_indices_on_domain(const Mesh &mesh,
- const AttributeDomain domain,
- const IsVertexSelectedFn is_vertex_selected_fn,
- Vector<int64_t> &r_indices)
-{
- switch (domain) {
- case ATTR_DOMAIN_POINT:
- return get_selected_vertex_indices(mesh, is_vertex_selected_fn, r_indices);
- case ATTR_DOMAIN_FACE:
- return get_selected_face_indices(mesh, is_vertex_selected_fn, r_indices);
- case ATTR_DOMAIN_CORNER:
- return get_selected_corner_indices(mesh, is_vertex_selected_fn, r_indices);
- case ATTR_DOMAIN_EDGE:
- return get_selected_edge_indices(mesh, is_vertex_selected_fn, r_indices);
- default:
- return;
- }
-}
-
-static Span<int64_t> filter_mesh_elements_by_selection(const bContext *C,
- Object *object_eval,
- const MeshComponent *component,
- const AttributeDomain domain,
- ResourceScope &scope)
-{
- SpaceSpreadsheet *sspreadsheet = CTX_wm_space_spreadsheet(C);
- const bool show_only_selected = sspreadsheet->filter_flag & SPREADSHEET_FILTER_SELECTED_ONLY;
- if (object_eval->mode == OB_MODE_EDIT && show_only_selected) {
- Object *object_orig = DEG_get_original_object(object_eval);
- Vector<int64_t> &visible_rows = scope.construct<Vector<int64_t>>("visible rows");
- const Mesh *mesh_eval = component->get_for_read();
- Mesh *mesh_orig = (Mesh *)object_orig->data;
- BMesh *bm = mesh_orig->edit_mesh->bm;
- BM_mesh_elem_table_ensure(bm, BM_VERT);
-
- int *orig_indices = (int *)CustomData_get_layer(&mesh_eval->vdata, CD_ORIGINDEX);
- if (orig_indices != nullptr) {
- /* Use CD_ORIGINDEX layer if it exists. */
- auto is_vertex_selected = [&](int vertex_index) -> bool {
- const int i_orig = orig_indices[vertex_index];
- if (i_orig < 0) {
- return false;
- }
- if (i_orig >= bm->totvert) {
- return false;
- }
- BMVert *vert = bm->vtable[i_orig];
- return BM_elem_flag_test(vert, BM_ELEM_SELECT);
- };
- get_selected_indices_on_domain(*mesh_eval, domain, is_vertex_selected, visible_rows);
- }
- else if (mesh_eval->totvert == bm->totvert) {
- /* Use a simple heuristic to match original vertices to evaluated ones. */
- auto is_vertex_selected = [&](int vertex_index) -> bool {
- BMVert *vert = bm->vtable[vertex_index];
- return BM_elem_flag_test(vert, BM_ELEM_SELECT);
- };
- get_selected_indices_on_domain(*mesh_eval, domain, is_vertex_selected, visible_rows);
- }
- /* This is safe, because the vector lives in the resource collector. */
- return visible_rows.as_span();
- }
- /* No filter is used. */
- const int domain_size = component->attribute_domain_size(domain);
- return IndexRange(domain_size).as_span();
-}
-
-static GeometryComponentType get_display_component_type(const bContext *C, Object *object_eval)
-{
- SpaceSpreadsheet *sspreadsheet = CTX_wm_space_spreadsheet(C);
- if (sspreadsheet->object_eval_state != SPREADSHEET_OBJECT_EVAL_STATE_ORIGINAL) {
- return (GeometryComponentType)sspreadsheet->geometry_component_type;
- }
- if (object_eval->type == OB_POINTCLOUD) {
- return GEO_COMPONENT_TYPE_POINT_CLOUD;
- }
- return GEO_COMPONENT_TYPE_MESH;
-}
-
-void spreadsheet_columns_from_geometry(const bContext *C,
- Object *object_eval,
- SpreadsheetColumnLayout &column_layout,
- ResourceScope &scope)
-{
- SpaceSpreadsheet *sspreadsheet = CTX_wm_space_spreadsheet(C);
- const AttributeDomain domain = (AttributeDomain)sspreadsheet->attribute_domain;
- const GeometryComponentType component_type = get_display_component_type(C, object_eval);
-
- /* Create a resource collector that owns stuff that needs to live until drawing is done. */
- GeometrySet &geometry_set = scope.add_value(
- get_display_geometry_set(sspreadsheet, object_eval, component_type), "geometry set");
-
- const GeometryComponent *component = geometry_set.get_component_for_read(component_type);
- if (component == nullptr) {
- return;
- }
- if (component_type == GEO_COMPONENT_TYPE_INSTANCES) {
- add_columns_for_instances(
- *static_cast<const InstancesComponent *>(component), column_layout, scope);
- return;
- }
-
- if (!component->attribute_domain_supported(domain)) {
- return;
- }
-
- Vector<std::string> attribute_names = get_sorted_attribute_names_to_display(*component, domain);
-
- Vector<std::unique_ptr<SpreadsheetColumn>> &columns =
- scope.construct<Vector<std::unique_ptr<SpreadsheetColumn>>>("columns");
-
- for (StringRefNull attribute_name : attribute_names) {
- ReadAttributePtr attribute_ptr = component->attribute_try_get_for_read(attribute_name);
- ReadAttribute &attribute = *attribute_ptr;
- scope.add(std::move(attribute_ptr), "attribute");
- add_columns_for_attribute(&attribute, attribute_name, columns);
- }
-
- for (std::unique_ptr<SpreadsheetColumn> &column : columns) {
- column_layout.columns.append(column.get());
- }
-
- /* The filter below only works for mesh vertices currently. */
- Span<int64_t> visible_rows;
- if (component_type == GEO_COMPONENT_TYPE_MESH) {
- visible_rows = filter_mesh_elements_by_selection(
- C, object_eval, static_cast<const MeshComponent *>(component), domain, scope);
- }
- else {
- visible_rows = IndexRange(component->attribute_domain_size(domain)).as_span();
- }
-
- const int domain_size = component->attribute_domain_size(domain);
- column_layout.row_indices = visible_rows;
- column_layout.tot_rows = domain_size;
-}
-
-} // namespace blender::ed::spreadsheet
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-16 02:19:39 +0300