DocumentImporter.h

Go to the documentation of this file.

/*
 * $Id: DocumentImporter.h 37663 2011-06-20 10:50:17Z jesterking $
 *
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * 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.
 *
 * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov.
 *
 * ***** END GPL LICENSE BLOCK *****
 */

#ifndef __DOCUMENTIMPORTER_H__
#define __DOCUMENTIMPORTER_H__

#include "COLLADAFWIWriter.h"
#include "COLLADAFWMaterial.h"
#include "COLLADAFWEffect.h"
#include "COLLADAFWColor.h"
#include "COLLADAFWImage.h"
#include "COLLADAFWInstanceGeometry.h"
#include "COLLADAFWController.h"
#include "COLLADAFWMorphController.h"
#include "COLLADAFWSkinController.h"

#include "BKE_object.h"

#include "TransformReader.h"
#include "AnimationImporter.h"
#include "ArmatureImporter.h"
#include "MeshImporter.h"
#include "ExtraTags.h"


struct Main;
struct bContext;

class DocumentImporter : COLLADAFW::IWriter
{
public:
        enum ImportStage {
                General,                
                Controller,             
        };
        DocumentImporter(bContext *C, const char *filename);

        ~DocumentImporter();

        bool import();

        Object* create_camera_object(COLLADAFW::InstanceCamera*, Scene*);
        Object* create_lamp_object(COLLADAFW::InstanceLight*, Scene*);
        Object* create_instance_node(Object*, COLLADAFW::Node*, COLLADAFW::Node*, Scene*, Object*, bool);
        void write_node(COLLADAFW::Node*, COLLADAFW::Node*, Scene*, Object*, bool);
        MTex* create_texture(COLLADAFW::EffectCommon*, COLLADAFW::Texture&, Material*, int, TexIndexTextureArrayMap&);
        void write_profile_COMMON(COLLADAFW::EffectCommon*, Material*);
        void translate_anim_recursive(COLLADAFW::Node*, COLLADAFW::Node*, Object*);

        void cancel(const COLLADAFW::String& errorMessage);

        void start();

        void finish();

        bool writeGlobalAsset(const COLLADAFW::FileInfo*);

        bool writeScene(const COLLADAFW::Scene*);

        bool writeVisualScene(const COLLADAFW::VisualScene*);

        bool writeLibraryNodes(const COLLADAFW::LibraryNodes*);

        bool writeAnimation(const COLLADAFW::Animation*);

        bool writeAnimationList(const COLLADAFW::AnimationList*);

        bool writeGeometry(const COLLADAFW::Geometry*);

        bool writeMaterial(const COLLADAFW::Material*);

        bool writeEffect(const COLLADAFW::Effect*);

        bool writeCamera(const COLLADAFW::Camera*);

        bool writeImage(const COLLADAFW::Image*);

        bool writeLight(const COLLADAFW::Light*);

        bool writeSkinControllerData(const COLLADAFW::SkinControllerData*);

        bool writeController(const COLLADAFW::Controller*);

        bool writeFormulas(const COLLADAFW::Formulas*);

        bool writeKinematicsScene(const COLLADAFW::KinematicsScene*);

        bool addExtraTags(const COLLADAFW::UniqueId &uid, ExtraTags *extra_tags);
        ExtraTags* getExtraTags(const COLLADAFW::UniqueId &uid);

private:

        ImportStage mImportStage;
        std::string mFilename;

        bContext *mContext;

        UnitConverter unit_converter;
        ArmatureImporter armature_importer;
        MeshImporter mesh_importer;
        AnimationImporter anim_importer;
        
        typedef std::map<std::string, ExtraTags*> TagsMap;
        TagsMap uid_tags_map;

        std::map<COLLADAFW::UniqueId, Image*> uid_image_map;
        std::map<COLLADAFW::UniqueId, Material*> uid_material_map;
        std::map<COLLADAFW::UniqueId, Material*> uid_effect_map;
        std::map<COLLADAFW::UniqueId, Camera*> uid_camera_map;
        std::map<COLLADAFW::UniqueId, Lamp*> uid_lamp_map;
        std::map<Material*, TexIndexTextureArrayMap> material_texture_mapping_map;
        std::map<COLLADAFW::UniqueId, Object*> object_map;
        std::map<COLLADAFW::UniqueId, COLLADAFW::Node*> node_map;
        std::vector<const COLLADAFW::VisualScene*> vscenes;
        std::vector<Object*> libnode_ob;

        std::map<COLLADAFW::UniqueId, COLLADAFW::Node*> root_map; // find root joint by child joint uid, for bone tree evaluation during resampling

};

#endif