spreadsheet_data_source_geometry.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include <mutex>
 
#include "BLI_resource_scope.hh"
 
#include "BKE_geometry_set.hh"
#include "BKE_instances.hh"
 
#include "spreadsheet_data_source.hh"
 
struct bContext;
 
namespace blender::ed::spreadsheet {
 
class ExtraColumns {
 private:
  Map<std::string, GSpan> columns_;
 
 public:
  void add(std::string name, GSpan data)
  {
    columns_.add(std::move(name), data);
  }
 
  void foreach_default_column_ids(
      FunctionRef<void(const SpreadsheetColumnID &, bool is_extra)> fn) const;
 
  std::unique_ptr<ColumnValues> get_column_values(const SpreadsheetColumnID &column_id) const;
};
 
class GeometryDataSource : public DataSource {
 private:
  Object *object_orig_;
  const bke::GeometrySet geometry_set_;
  const bke::GeometryComponent *component_;
  bke::AttrDomain domain_;
  /* Layer index for grease pencil component. */
  int layer_index_;
  ExtraColumns extra_columns_;
 
  /* Some data is computed on the fly only when it is requested. Computing it does not change the
   * logical state of this data source. Therefore, the corresponding methods are const and need to
   * be protected with a mutex. */
  mutable std::mutex mutex_;
  mutable ResourceScope scope_;
 
 public:
  GeometryDataSource(Object *object_orig,
                     bke::GeometrySet geometry_set,
                     const bke::GeometryComponent::Type component_type,
                     const bke::AttrDomain domain,
                     const int layer_index = -1,
                     ExtraColumns extra_columns = {})
      : object_orig_(object_orig),
        geometry_set_(std::move(geometry_set)),
        component_(geometry_set_.get_component(component_type)),
        domain_(domain),
        layer_index_(layer_index),
        extra_columns_(std::move(extra_columns))
  {
  }
 
  bool has_selection_filter() const override;
  IndexMask apply_selection_filter(IndexMaskMemory &memory) const;
 
  void foreach_default_column_ids(
      FunctionRef<void(const SpreadsheetColumnID &, bool is_extra)> fn) const override;
 
  std::unique_ptr<ColumnValues> get_column_values(
      const SpreadsheetColumnID &column_id) const override;
 
  int tot_rows() const override;
 
 private:
  std::optional<const bke::AttributeAccessor> get_component_attributes() const;
};
 
class VolumeDataSource : public DataSource {
  const bke::GeometrySet geometry_set_;
  const bke::VolumeComponent *component_;
 
 public:
  VolumeDataSource(bke::GeometrySet geometry_set)
      : geometry_set_(std::move(geometry_set)),
        component_(geometry_set_.get_component<bke::VolumeComponent>())
  {
  }
 
  void foreach_default_column_ids(
      FunctionRef<void(const SpreadsheetColumnID &, bool is_extra)> fn) const override;
 
  std::unique_ptr<ColumnValues> get_column_values(
      const SpreadsheetColumnID &column_id) const override;
 
  int tot_rows() const override;
};
 
int get_instance_reference_icon(const bke::InstanceReference &reference);
 
std::unique_ptr<DataSource> data_source_from_geometry(const bContext *C, Object *object_eval);
 
bke::GeometrySet get_geometry_set_for_instance_ids(const bke::GeometrySet &root_geometry,
                                                   const Span<SpreadsheetInstanceID> instance_ids);
 
}  // namespace blender::ed::spreadsheet