oarconv.cpp

/** 
@brief   Convert OAR Files to Collada Files

@author  Fumi.Iseki and Austin Tate
@date    2024 9/15
@version 1.18.1
*/

#include "oarconv.h"


using namespace jbxl;


int main(int argc, char** argv)
{
    setlocale(LC_NUMERIC, "");      // 数字表記．システムから．

    Buffer inpdir  = init_Buffer();
    Buffer outdir  = init_Buffer();
    Buffer astdir  = init_Buffer();
    Buffer infile  = init_Buffer();
    Buffer oarfile = init_Buffer();
    Buffer cmmnd   = init_Buffer();
    Buffer tscale  = init_Buffer();

    float xshift   = 0.0;
    float yshift   = 0.0;
    float zshift   = 0.0;

    int   strtnum  = 0;
    int   stopnum  = -1;
    int   format   = JBXL_3D_FORMAT_DAE;
    int   texture  = JBXL_TEXTURE_PNG;
    int   engine   = JBXL_3D_ENGINE_UNITY;

    bool  degeneracy = false;
    bool  useBrep = false;      // required
    bool  procJoints = false;

    for (int i=1; i<argc; i++) {
        if      (!strcmp(argv[i], "-i")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &inpdir); i++;}}
        else if (!strcmp(argv[i], "-f")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &infile); i++;}}
        else if (!strcmp(argv[i], "-o")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &outdir); i++;}}
        else if (!strcmp(argv[i], "-a")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &astdir); i++;}}
        else if (!strcmp(argv[i], "-t")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &tscale); i++;}}
        else if (!strcmp(argv[i], "-c")) { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &cmmnd);  i++;}}
        else if (!strcmp(argv[i], "-s")) { if (i!=argc-1){ strtnum = atoi(argv[i+1]); i++;}}
        else if (!strcmp(argv[i], "-e")) { if (i!=argc-1){ stopnum = atoi(argv[i+1]); i++;}}
        else if (!strcmp(argv[i], "-x")) { if (i!=argc-1){ xshift = (float)atof(argv[i+1]); i++;}}
        else if (!strcmp(argv[i], "-y")) { if (i!=argc-1){ yshift = (float)atof(argv[i+1]); i++;}}
        else if (!strcmp(argv[i], "-z")) { if (i!=argc-1){ zshift = (float)atof(argv[i+1]); i++;}}
        //
        else if (!strcmp(argv[i], "-d")) { DebugMode = ON;}
        else if (!strcmp(argv[i], "-h")) { oarconv_help(stdout); exit(0);}
        else if (!strcmp(argv[i], "-v")) { fprintf(stdout, " Version is %s\n", OARCONV_VERSION); exit(0);}
        //
        else if (!strcmp(argv[i], "--oar"))  { if (i!=argc-1){ copy_s2Buffer(argv[i+1], &oarfile); i++;}}

        else if (!strcmp(argv[i], "--unity")){ engine  = JBXL_3D_ENGINE_UNITY;}  // for UNITY
        else if (!strcmp(argv[i], "--ue"))   { engine  = JBXL_3D_ENGINE_UE;}     // for UE
        else if (!strcmp(argv[i], "--dae"))  { format  = JBXL_3D_FORMAT_DAE;}    // DAEデータを出力
        else if (!strcmp(argv[i], "--obj"))  { format  = JBXL_3D_FORMAT_OBJ;}    // OBJデータを出力
        else if (!strcmp(argv[i], "--gltf")) { format  = JBXL_3D_FORMAT_GLTF;}   // glTFデータを出力
        else if (!strcmp(argv[i], "--glb"))  { format  = JBXL_3D_FORMAT_GLB;}    // glbデータを出力 // 未サポート
        else if (!strcmp(argv[i], "--fbx"))  { format  = JBXL_3D_FORMAT_FBX;}    // FBXデータを出力 // 未サポート
        else if (!strcmp(argv[i], "--stl"))  { format  = JBXL_3D_FORMAT_STL_A;}  // STLデータを出力
        else if (!strcmp(argv[i], "--tga"))  { texture = JBXL_TEXTURE_TGA;}      // テクスチャとして TGA ファイルを使う
        else if (!strcmp(argv[i], "--png"))  { texture = JBXL_TEXTURE_PNG;}      // テクスチャとして PNG ファイルを使う
        else if (!strcmp(argv[i], "--dg"))   { degeneracy = true;}
        else if (!strcmp(argv[i], "--nooffset")) { degeneracy = true;}
        else if (!strcmp(argv[i], "--joint")){ procJoints = true;}
        else if (!strcmp(argv[i], "--help")) { oarconv_help(stdout); exit(0);}
        else  {
            PRINT_MESG("%s : unknown option [%s]\n", argv[0], argv[i]);
            exit(1);
        }
    }

    OARTool oar;
    if (inpdir.buf==NULL && oarfile.buf!=NULL) {
        inpdir = oar.ExtractOAR(oarfile, 0750);
    }
    if (inpdir.buf==NULL) {
        inpdir = make_Buffer_bystr(".");
    }
    //
    float scale = (float)TRNT_DEFAULT_TEX_SCALE;
    if (tscale.buf!=NULL) {
        scale = (float)atof((char*)tscale.buf);
        if (scale<=0.0f) scale = (float)TRNT_DEFAULT_TEX_SCALE;
    }
    if (format==JBXL_3D_FORMAT_OBJ   ||
        format==JBXL_3D_FORMAT_STL_A || format==JBXL_3D_FORMAT_STL_B) procJoints = false;

    init_rand();

    //////////////////////////////////////////////////////////////////////////////////////////////
    // main process
    oar.SetEngine(engine);
    oar.SetDataFormat(format);
    oar.SetTextureFormat(texture);
    oar.SetNoOffset(degeneracy);
    oar.SetProcJoints(procJoints);
    oar.SetPathInfo((char*)inpdir.buf, (char*)outdir.buf, (char*)astdir.buf);
    oar.GetDataInfo();  // -f オプション（個別ファイル指定）があるので，成否は無視．
    oar.MakeOutputFolder();
    oar.SetTerrainShift(xshift, yshift, zshift);
 
#ifndef WIN32
    memory_check_start();
#endif

    if (infile.buf!=NULL) {
        oar.objectsNum = 1;
        oar.GenerateObjectFromDataFile((char*)infile.buf, useBrep, (char*)cmmnd.buf);
    }
    else {
        if (strtnum==0) {
            oar.ReadTerrainData();
            oar.SetTerrainTextureScale(scale);
            oar.GenerateTerrainDataFile();
            strtnum = 1;
        }
        if (stopnum!=0) {
            oar.GenerateObjectFromDataIndex(strtnum, stopnum, useBrep, (char*)cmmnd.buf);
        }
    }
    if (format==JBXL_3D_FORMAT_GLB) oar.RmTextureFolder();
    oar.free();
    
    //
    free_Buffer(&inpdir);
    free_Buffer(&outdir);
    free_Buffer(&astdir);
    free_Buffer(&tscale);
    free_Buffer(&cmmnd);
    free_Buffer(&infile);
    free_Buffer(&oarfile);

    return 0;
}