blender-archive/source/blender/alembic/intern/abc_object.h

/*
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

/** \file
 * \ingroup balembic
 */

#ifndef __ABC_OBJECT_H__
#define __ABC_OBJECT_H__

#include <Alembic/Abc/All.h>
#include <Alembic/AbcGeom/All.h>

#include "abc_exporter.h"

extern "C" {
#include "DNA_ID.h"
}

class AbcTransformWriter;

struct Main;
struct Object;

/* ************************************************************************** */

class AbcObjectWriter {
protected:
	Object *m_object;
	ExportSettings &m_settings;

	uint32_t m_time_sampling;

	Imath::Box3d m_bounds;
	std::vector<AbcObjectWriter *> m_children;

	std::vector<std::pair<std::string, IDProperty *>> m_props;

	bool m_first_frame;
	std::string m_name;

public:
	AbcObjectWriter(Object *ob,
	                uint32_t time_sampling,
	                ExportSettings &settings,
	                AbcObjectWriter *parent = NULL);

	virtual ~AbcObjectWriter();

	void addChild(AbcObjectWriter *child);

	virtual Imath::Box3d bounds();

	void write();

private:
	virtual void do_write() = 0;
};

/* ************************************************************************** */

struct CacheFile;

struct ImportSettings {
	bool do_convert_mat;
	float conversion_mat[4][4];

	int from_up;
	int from_forward;
	float scale;
	bool is_sequence;
	bool set_frame_range;

	/* Length and frame offset of file sequences. */
	int sequence_len;
	int sequence_offset;

	/* From MeshSeqCacheModifierData.read_flag */
	int read_flag;

	bool validate_meshes;

	CacheFile *cache_file;

	ImportSettings()
	    : do_convert_mat(false)
	    , from_up(0)
	    , from_forward(0)
	    , scale(1.0f)
	    , is_sequence(false)
	    , set_frame_range(false)
	    , sequence_len(1)
	    , sequence_offset(0)
	    , read_flag(0)
	    , validate_meshes(false)
	    , cache_file(NULL)
	{}
};

template <typename Schema>
static bool has_animations(Schema &schema, ImportSettings *settings)
{
	return settings->is_sequence || !schema.isConstant();
}

/* ************************************************************************** */

struct Mesh;

using Alembic::AbcCoreAbstract::chrono_t;

class AbcObjectReader {
protected:
	std::string m_name;
	std::string m_object_name;
	std::string m_data_name;
	Object *m_object;
	Alembic::Abc::IObject m_iobject;

	ImportSettings *m_settings;

	chrono_t m_min_time;
	chrono_t m_max_time;

	/* Use reference counting since the same reader may be used by multiple
	 * modifiers and/or constraints. */
	int m_refcount;

	bool m_inherits_xform;

public:
	AbcObjectReader *parent_reader;

public:
	explicit AbcObjectReader(const Alembic::Abc::IObject &object, ImportSettings &settings);

	virtual ~AbcObjectReader();

	const Alembic::Abc::IObject &iobject() const;

	typedef std::vector<AbcObjectReader *> ptr_vector;

	/**
	 * Returns the transform of this object. This can be the Alembic object
	 * itself (in case of an Empty) or it can be the parent Alembic object.
	 */
	virtual Alembic::AbcGeom::IXform xform();

	Object *object() const;
	void object(Object *ob);

	const std::string & name() const { return m_name; }
	const std::string & object_name() const { return m_object_name; }
	const std::string & data_name() const { return m_data_name; }
	bool inherits_xform() const { return m_inherits_xform; }

	virtual bool valid() const = 0;
	virtual bool accepts_object_type(const Alembic::AbcCoreAbstract::ObjectHeader &alembic_header,
	                                 const Object *const ob,
	                                 const char **err_str) const = 0;

	virtual void readObjectData(Main *bmain, const Alembic::Abc::ISampleSelector &sample_sel) = 0;

	virtual struct Mesh *read_mesh(struct Mesh *mesh,
	                               const Alembic::Abc::ISampleSelector &sample_sel,
	                               int read_flag,
	                               const char **err_str);

	/** Reads the object matrix and sets up an object transform if animated. */
	void setupObjectTransform(const float time);

	void addCacheModifier();

	chrono_t minTime() const;
	chrono_t maxTime() const;

	int refcount() const;
	void incref();
	void decref();

	void read_matrix(float r_mat[4][4], const float time,
	                 const float scale, bool &is_constant);

protected:
	void determine_inherits_xform();
};

Imath::M44d get_matrix(const Alembic::AbcGeom::IXformSchema &schema, const float time);

#endif  /* __ABC_OBJECT_H__ */