ArmatureExporter.h

Go to the documentation of this file.

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

 
#pragma once
 
#include <string>
 
#include <COLLADASWInputList.h>
#include <COLLADASWInstanceController.h>
#include <COLLADASWLibraryControllers.h>
#include <COLLADASWNode.h>
#include <COLLADASWStreamWriter.h>
 
#include "DNA_armature_types.h"
#include "DNA_object_types.h"
 
#include "InstanceWriter.h"
#include "TransformWriter.h"
 
#include "ExportSettings.h"
 
class SceneExporter;
 
/* XXX exporter writes wrong data for shared armatures.  A separate
 * controller should be written for each armature-mesh binding how do
 * we make controller ids then? */
class ArmatureExporter : public COLLADASW::LibraryControllers,
                         protected TransformWriter,
                         protected InstanceWriter {
 public:
  /* XXX exporter writes wrong data for shared armatures.  A separate
   * controller should be written for each armature-mesh binding how do
   * we make controller ids then? */
  ArmatureExporter(BlenderContext &blender_context,
                   COLLADASW::StreamWriter *sw,
                   BCExportSettings &export_settings)
      : COLLADASW::LibraryControllers(sw),
        blender_context(blender_context),
        export_settings(export_settings)
  {
  }
 
  void add_bone_collections(Object *ob_arm, COLLADASW::Node &node);
 
  /* write bone nodes */
  void add_armature_bones(Object *ob_arm,
                          ViewLayer *view_layer,
                          SceneExporter *se,
                          std::vector<Object *> &child_objects);
 
  bool add_instance_controller(Object *ob);
 
 private:
  BlenderContext &blender_context;
  BCExportSettings &export_settings;
 
#if 0
  std::vector<Object *> written_armatures;
 
  bool already_written(Object *ob_arm);
 
  void wrote(Object *ob_arm);
 
  void find_objects_using_armature(Object *ob_arm, std::vector<Object *> &objects, Scene *sce);
#endif

  void add_bone_node(Bone *bone,
                     Object *ob_arm,
                     SceneExporter *se,
                     std::vector<Object *> &child_objects);
 
  bool can_export(Bone *bone)
  {
    return !(export_settings.get_deform_bones_only() && bone->flag & BONE_NO_DEFORM);
  }
 
  bool is_export_root(Bone *bone);
  void add_bone_transform(Object *ob_arm, Bone *bone, COLLADASW::Node &node);
 
  std::string get_controller_id(Object *ob_arm, Object *ob);
 
  void write_bone_URLs(COLLADASW::InstanceController &ins, Object *ob_arm, Bone *bone);
};