test2/source/blender/io/usd/hydra/volume.hh

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

#pragma once

#include <pxr/imaging/hd/sceneDelegate.h>

#include "object.hh"

namespace blender::io::hydra {

class VolumeData : public ObjectData {
 protected:
  std::string filepath_;
  pxr::HdVolumeFieldDescriptorVector field_descriptors_;
  MaterialData *mat_data_ = nullptr;

 public:
  VolumeData(HydraSceneDelegate *scene_delegate,
             const Object *object,
             pxr::SdfPath const &prim_id);

  void init() override;
  void insert() override;
  void remove() override;
  void update() override;

  pxr::VtValue get_data(pxr::TfToken const &key) const override;
  pxr::VtValue get_data(pxr::SdfPath const &id, pxr::TfToken const &key) const override;
  pxr::SdfPath material_id() const override;
  void available_materials(Set<pxr::SdfPath> &paths) const override;

  pxr::HdVolumeFieldDescriptorVector field_descriptors() const;

 protected:
  void write_materials() override;
};

}  // namespace blender::io::hydra