usd_reader_xform.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2021 Tangent Animation. All rights reserved.
 * SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 *
 * Adapted from the Blender Alembic importer implementation. */
 
#pragma once
 
#include "usd.hh"
#include "usd_reader_prim.hh"
 
/* For #UsdGeomXformable. */
#include <pxr/usd/usdGeom/xformable.h>
 
struct Main;
 
namespace blender::io::usd {

using XformResult = std::tuple<pxr::GfMatrix4f, bool>;
 
class USDXformReader : public USDPrimReader {
 private:
  bool use_parent_xform_ = false;
 
  /* Indicates if the created object is the root of a
   * transform hierarchy. */
  bool is_root_xform_;
 
 public:
  USDXformReader(const pxr::UsdPrim &prim,
                 const USDImportParams &import_params,
                 const ImportSettings &settings)
      : USDPrimReader(prim, import_params, settings), is_root_xform_(is_root_xform_prim())
  {
  }
 
  void create_object(Main *bmain) override;
  void read_object_data(Main *bmain, pxr::UsdTimeCode time) override;
 
  pxr::SdfPath object_prim_path() const override;
 
  void read_matrix(float r_mat[4][4],
                   pxr::UsdTimeCode time,
                   float scale,
                   bool *r_is_constant) const;
 
  bool use_parent_xform() const
  {
    return use_parent_xform_;
  }
  void set_use_parent_xform(bool flag)
  {
    use_parent_xform_ = flag;
    is_root_xform_ = is_root_xform_prim();
  }
 
  bool prim_has_xform_ops() const;
 
 protected:
  /* Returns true if the contained USD prim is the root of a transform hierarchy. */
  bool is_root_xform_prim() const;

  virtual std::optional<XformResult> get_local_usd_xform(pxr::UsdTimeCode time) const;
 
 private:
  pxr::UsdGeomXformable get_xformable() const;
};
 
}  // namespace blender::io::usd