BKE_bake_items.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "BLI_memory_counter_fwd.hh"
 
#include "BKE_bake_data_block_map.hh"
#include "BKE_geometry_set.hh"
#include "BKE_volume_grid_fwd.hh"
 
#include "NOD_socket_interface_key.hh"
 
namespace blender::bke::bake {

class BakeItem {
 public:
  std::string name;
 
  virtual ~BakeItem() = default;
 
  virtual void count_memory(MemoryCounter &memory) const;
};
 
struct BakeState {
  Map<int, std::unique_ptr<BakeItem>> items_by_id;
 
  void count_memory(MemoryCounter &memory) const;
};

struct BakeStateRef {
  Map<int, const BakeItem *> items_by_id;
 
  BakeStateRef() = default;
  BakeStateRef(const BakeState &bake_state);
};
 
class GeometryBakeItem : public BakeItem {
 public:
  GeometrySet geometry;
 
  GeometryBakeItem(GeometrySet geometry);
 
  void count_memory(MemoryCounter &memory) const override;

  static void prepare_geometry_for_bake(GeometrySet &geometry, BakeDataBlockMap *data_block_map);

  static void try_restore_data_blocks(GeometrySet &geometry, BakeDataBlockMap *data_block_map);
};

class AttributeBakeItem : public BakeItem {
 private:
  std::string name_;
 
 public:
  AttributeBakeItem(std::string name) : name_(std::move(name)) {}
 
  StringRefNull name() const
  {
    return name_;
  }
};
 
#ifdef WITH_OPENVDB
class VolumeGridBakeItem : public BakeItem {
 public:
  std::unique_ptr<GVolumeGrid> grid;
 
  VolumeGridBakeItem(std::unique_ptr<GVolumeGrid> grid);
  ~VolumeGridBakeItem() override;
 
  void count_memory(MemoryCounter &memory) const override;
};
#endif

class PrimitiveBakeItem : public BakeItem {
 private:
  const CPPType &type_;
  void *value_;
 
 public:
  PrimitiveBakeItem(const CPPType &type, const void *value);
  ~PrimitiveBakeItem() override;
 
  const void *value() const
  {
    return value_;
  }
 
  const CPPType &type() const
  {
    return type_;
  }
};
 
class StringBakeItem : public BakeItem {
 private:
  std::string value_;
 
 public:
  StringBakeItem(std::string value);
 
  StringRefNull value() const
  {
    return value_;
  }
 
  void count_memory(MemoryCounter &memory) const override;
};

class BundleBakeItem : public BakeItem {
 public:
  struct Item {
    nodes::SocketInterfaceKey key;
    std::string socket_idname;
    std::unique_ptr<BakeItem> value;
  };
 
  Vector<Item> items;
};
 
}  // namespace blender::bke::bake