collada_internal.cpp

Go to the documentation of this file.

/*
 * ***** 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 *****
 */

/* COLLADABU_ASSERT, may be able to remove later */
#include "COLLADABUPlatform.h"

#include "collada_internal.h"

UnitConverter::UnitConverter() : unit(), up_axis(COLLADAFW::FileInfo::Z_UP) {}

void UnitConverter::read_asset(const COLLADAFW::FileInfo* asset)
{
    unit = asset->getUnit();
    up_axis = asset->getUpAxisType();
}

UnitConverter::UnitSystem UnitConverter::isMetricSystem()
{
    switch(unit.getLinearUnitUnit()) {
        case COLLADAFW::FileInfo::Unit::MILLIMETER:
        case COLLADAFW::FileInfo::Unit::CENTIMETER:
        case COLLADAFW::FileInfo::Unit::DECIMETER:
        case COLLADAFW::FileInfo::Unit::METER:
        case COLLADAFW::FileInfo::Unit::KILOMETER:
            return UnitConverter::Metric;
        case COLLADAFW::FileInfo::Unit::INCH:
        case COLLADAFW::FileInfo::Unit::FOOT:
        case COLLADAFW::FileInfo::Unit::YARD:
            return UnitConverter::Imperial;
        default:
            return UnitConverter::None;
    }
}

float UnitConverter::getLinearMeter()
{
    return (float)unit.getLinearUnitMeter();
}

void UnitConverter::convertVector3(COLLADABU::Math::Vector3 &vec, float *v)
{
    v[0] = vec.x;
    v[1] = vec.y;
    v[2] = vec.z;
}

// TODO need also for angle conversion, time conversion...

void UnitConverter::dae_matrix_to_mat4_(float out[][4], const COLLADABU::Math::Matrix4& in)
{
    // in DAE, matrices use columns vectors, (see comments in COLLADABUMathMatrix4.h)
    // so here, to make a blender matrix, we swap columns and rows
    for (int i = 0; i < 4; i++) {
        for (int j = 0; j < 4; j++) {
            out[i][j] = in[j][i];
        }
    }
}

void UnitConverter::mat4_to_dae(float out[][4], float in[][4])
{
    copy_m4_m4(out, in);
    transpose_m4(out);
}

void UnitConverter::mat4_to_dae_double(double out[][4], float in[][4])
{
    float mat[4][4];

    mat4_to_dae(mat, in);

    for (int i = 0; i < 4; i++)
        for (int j = 0; j < 4; j++)
            out[i][j] = mat[i][j];
}

void TransformBase::decompose(float mat[][4], float *loc, float eul[3], float quat[4], float *size)
{
    mat4_to_size(size, mat);
    if (eul) {
        mat4_to_eul(eul, mat);
    }
    if (quat) {
        mat4_to_quat(quat, mat);
    }
    copy_v3_v3(loc, mat[3]);
}

const unsigned char translate_start_name_map[256] = {
95,  95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
65,  66,  67,  68,  69,  70,  71,  72,
73,  74,  75,  76,  77,  78,  79,  80,
81,  82,  83,  84,  85,  86,  87,  88,
89,  90,  95,  95,  95,  95,  95,  95,
97,  98,  99,  100,  101,  102,  103,  104,
105,  106,  107,  108,  109,  110,  111,  112,
113,  114,  115,  116,  117,  118,  119,  120,
121,  122,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  192,
193,  194,  195,  196,  197,  198,  199,  200,
201,  202,  203,  204,  205,  206,  207,  208,
209,  210,  211,  212,  213,  214,  95,  216,
217,  218,  219,  220,  221,  222,  223,  224,
225,  226,  227,  228,  229,  230,  231,  232,
233,  234,  235,  236,  237,  238,  239,  240,
241,  242,  243,  244,  245,  246,  95,  248,
249,  250,  251,  252,  253,  254,  255};

const unsigned char translate_name_map[256] = {
95,  95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  45,  95,  95,  48,
49,  50,  51,  52,  53,  54,  55,  56,
57,  95,  95,  95,  95,  95,  95,  95,
65,  66,  67,  68,  69,  70,  71,  72,
73,  74,  75,  76,  77,  78,  79,  80,
81,  82,  83,  84,  85,  86,  87,  88,
89,  90,  95,  95,  95,  95,  95,  95,
97,  98,  99,  100,  101,  102,  103,  104,
105,  106,  107,  108,  109,  110,  111,  112,
113,  114,  115,  116,  117,  118,  119,  120,
121,  122,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  95,  95,
95,  95,  95,  95,  95,  95,  183,  95,
95,  95,  95,  95,  95,  95,  95,  192,
193,  194,  195,  196,  197,  198,  199,  200,
201,  202,  203,  204,  205,  206,  207,  208,
209,  210,  211,  212,  213,  214,  95,  216,
217,  218,  219,  220,  221,  222,  223,  224,
225,  226,  227,  228,  229,  230,  231,  232,
233,  234,  235,  236,  237,  238,  239,  240,
241,  242,  243,  244,  245,  246,  95,  248,
249,  250,  251,  252,  253,  254,  255};

typedef std::map< std::string, std::vector<std::string> > map_string_list;
map_string_list global_id_map;

void clear_global_id_map()
{
    global_id_map.clear();
}

std::string translate_id(const std::string &id)
{
    if (id.size() == 0)
    { return id; }
    std::string id_translated = id;
    id_translated[0] = translate_start_name_map[(unsigned int)id_translated[0]];
    for (unsigned int i=1; i < id_translated.size(); i++)
    {
        id_translated[i] = translate_name_map[(unsigned int)id_translated[i]];
    }
    // It's so much workload now, the if() should speed up things.
    if (id_translated != id)
    {
        // Search duplicates
        map_string_list::iterator iter = global_id_map.find(id_translated);
        if (iter != global_id_map.end())
        {
            unsigned int i = 0;
            bool found = false;
            for (i=0; i < iter->second.size(); i++)
            {
                if (id == iter->second[i])
                { 
                    found = true;
                    break;
                }
            }
            bool convert = false;
            if (found)
            {
              if (i > 0)
              { convert = true; }
            }
            else
            { 
                convert = true;
                global_id_map[id_translated].push_back(id);
            }
            if (convert)
            {
                std::stringstream out;
                out << ++i;
                id_translated += out.str();
            }
        }
        else { global_id_map[id_translated].push_back(id); }
    }
    return id_translated;
}

std::string id_name(void *id)
{
    return ((ID*)id)->name + 2;
}

std::string get_geometry_id(Object *ob)
{
    return translate_id(id_name(ob->data)) + "-mesh";
}

std::string get_light_id(Object *ob)
{
    return translate_id(id_name(ob)) + "-light";
}

std::string get_joint_id(Bone *bone, Object *ob_arm)
{
    return translate_id(/*id_name(ob_arm) + "_" +*/ bone->name);
}

std::string get_camera_id(Object *ob)
{
    return translate_id(id_name(ob)) + "-camera";
}

std::string get_material_id(Material *mat)
{
    return translate_id(id_name(mat)) + "-material";
}

bool has_object_type(Scene *sce, short obtype)
{
    Base *base= (Base*) sce->base.first;
    while(base) {
        Object *ob = base->object;
            
        if (ob->type == obtype && ob->data) {
            return true;
        }
        base= base->next;
    }
    return false;
}