usd_reader_curve.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 * Adapted from the Blender Alembic importer implementation. Copyright 2016 Kévin Dietrich.
 * Modifications Copyright 2021 Tangent Animation. All rights reserved. */
#pragma once
 
#include "usd.hh"
#include "usd_reader_geom.hh"
 
#include <pxr/usd/usd/prim.h>
#include <pxr/usd/usdGeom/basisCurves.h>
#include <pxr/usd/usdGeom/curves.h>
 
struct Curves;
struct Main;
 
namespace blender::bke {
struct GeometrySet;
class CurvesGeometry;
}  // namespace blender::bke
 
namespace blender::io::usd {
 
class USDCurvesReader : public USDGeomReader {
 public:
  USDCurvesReader(const pxr::UsdPrim &prim,
                  const USDImportParams &import_params,
                  const ImportSettings &settings)
      : USDGeomReader(prim, import_params, settings)
  {
  }
 
  void create_object(Main *bmain) override;
  void read_object_data(Main *bmain, pxr::UsdTimeCode time) override;
 
  void read_geometry(bke::GeometrySet &geometry_set,
                     USDMeshReadParams params,
                     const char **r_err_str) override;
 
  void read_velocities(bke::CurvesGeometry &curves,
                       const pxr::UsdGeomCurves &usd_curves,
                       const pxr::UsdTimeCode time) const;
  void read_custom_data(bke::CurvesGeometry &curves, const pxr::UsdTimeCode time) const;
 
  virtual bool is_animated() const = 0;
  virtual void read_curve_sample(Curves *curves_id, pxr::UsdTimeCode time) = 0;
};
 
class USDBasisCurvesReader : public USDCurvesReader {
 private:
  pxr::UsdGeomBasisCurves curve_prim_;
 
 public:
  USDBasisCurvesReader(const pxr::UsdPrim &prim,
                       const USDImportParams &import_params,
                       const ImportSettings &settings)
      : USDCurvesReader(prim, import_params, settings), curve_prim_(prim)
  {
  }
 
  bool valid() const override
  {
    return bool(curve_prim_);
  }
 
  bool is_animated() const override;
  void read_curve_sample(Curves *curves_id, pxr::UsdTimeCode time) override;
};
 
}  // namespace blender::io::usd