#include <onidev.h>

string fileGetPath(string file) {
    int pos = file.find_last_of('\\');
    if(pos == -1) pos = file.find_last_of('/');
    return file.substr(0, pos);
}

string fileGetName(string file) {
    int pos = file.find_last_of('\\');
    if(pos == -1) pos = file.find_last_of('/');
    return file.substr(pos+1, file.length());
}

void voxelize(string name, string path);

int main(int argc, char **argv)
{
    if(argc <= 1)
        return 0;
    voxelize(argv[1], fileGetPath(argv[0]) + "\\");
    /*voxelize("bomb.png", "");
    system("pause");
    return 0;*/
}

void voxelize(string name, string path)
{
    // read png file
    uint wid, hei, *data, *bump;
    data = (uint*)imageLoadPng(name, wid, hei);
   
    if(data == NULL) {
        printf("Fail to load image. End program.\n");
        return;
    }
   
    string name2 = name.substr(0, name.length()-4)+"_bump.png";
    uint wid2, hei2;
    if(fileExists(name2)) {
        bump = (uint*)imageLoadPng(name2, wid2, hei2);
        if(wid!=wid2 || hei!=hei2)
        {
            delete[] bump;
            delete[] data;
            printf("Bump map size error.\n");
            return;
        }
    }
    else
    {
        bump = new uint[wid*hei];
        for(uint i=0; i<wid*hei; i++)
            bump[i] = 0;
    }
   
    // create indexed image
    vector<uint> palette;
    short *dat = new short[wid * hei];
   
    for(uint i=0; i<wid; i++)
    for(uint j=0; j<hei; j++) {
        uint col = data[i+j*wid];
       
        // transparent color
        if((col&0xFF000000) == 0) {
            dat[i+j*wid] = 0;
            continue;
        }
           
        // check color
        int find=0;
        for(uint n=0; n<palette.size(); n++) {
            if(palette.at(n) == col) {
                find = n+1;
                break;
            }
        }
       
        // find new color
        if(find == 0) {
            palette.push_back(col);
            find = palette.size();
        }
       
        dat[i+j*wid] = find;
    }
    delete[] data;
   
    // Write mtl file
    FILE *f = fopen( (path + "material.mtl").c_str(), "w");
    if(f == NULL) {
        printf("Fail to create material.mtl\nEnd program.\n");
        delete[] dat;
        return;
    }
   
    // write infos
    fprintf(f, "# Generated by Voxelizer 0.1\n");
   
    // write material
    for(uint i=0; i<palette.size(); i++) {
        float r = ( palette.at(i)&0xFF )/255.0;
        float g = ( (palette.at(i)&0xFF00)>>8 )/255.0;
        float b = ( (palette.at(i)&0xFF0000)>>16 )/255.0;
        float a = ( (palette.at(i)&0xFF000000)>>24 )/255.0;
        r = pow(r, 2.38);
        g = pow(g, 2.38);
        b = pow(b, 2.38);
       
        fprintf(f, "newmtl Material.%d\n", i+1);
        fprintf(f, "Ns %f\n", 100.0);
        fprintf(f, "Ka %f %f %f\n", 0.0, 0.0, 0.0);
        fprintf(f, "Kd %f %f %f\n", r, g, b);
        fprintf(f, "Ks %f %f %f\n", 0.0, 0.0, 0.0);
        fprintf(f, "Ni %f\n", 0.0);
        fprintf(f, "d %f\n", a);
        fprintf(f, "illum 1\n\n\n");
    }
    fclose(f);
   
    // Write obj file
    f = fopen( (path + "out.obj").c_str(), "w");
    if(f == NULL) {
        printf("Fail to create out.obj\nEnd program.\n");
        delete[] dat;
        return;
    }
   
    // infos
    fprintf(f, "# Generated by Voxelizer 0.1\n");
    fprintf(f, "mtllib material.mtl\n");
   
    for(float y=0; y<hei; y++)
    for(float x=0; x<wid; x++)
    {
        // object name
        fprintf(f, "o Cube%d\n", int(x + y*wid));
       
        float z = bump[int(x + y*wid)]&255;
        if(z >= 40.0) z = (z-40.0)/200.0;
        else z=0;
       
        // vertices
        fprintf(f, "v %f %f %f\n", x/10.0    , z    , y/10.0    );
        fprintf(f, "v %f %f %f\n", (x+1)/10.0, z    , y/10.0    );
        fprintf(f, "v %f %f %f\n", (x+1)/10.0, z    , (y+1)/10.0);
        fprintf(f, "v %f %f %f\n", x/10.0    , z    , (y+1)/10.0);
        fprintf(f, "v %f %f %f\n", x/10.0    , z+0.1, y/10.0    );
        fprintf(f, "v %f %f %f\n", (x+1)/10.0, z+0.1, y/10.0    );
        fprintf(f, "v %f %f %f\n", (x+1)/10.0, z+0.1, (y+1)/10.0);
        fprintf(f, "v %f %f %f\n", x/10.0    , z+0.1, (y+1)/10.0);
       
        // material
        if(dat[int(x+y*wid)])
        fprintf(f, "usemtl Material.%d\n", dat[int(x + y*wid)]);
       
        // no segments
        fprintf(f, "s off\n");
       
        // faces
        if(!dat[int(x + y*wid)])
            continue;
       
        int i = 8*(x + y*wid);
        fprintf(f, "f %d %d %d %d\n", i+1, i+2, i+3, i+4); // back face
        fprintf(f, "f %d %d %d %d\n", i+5, i+6, i+7, i+8); // front face
        fprintf(f, "f %d %d %d %d\n", i+1, i+5, i+8, i+4); // left face
        fprintf(f, "f %d %d %d %d\n", i+2, i+6, i+7, i+3); // right face
        fprintf(f, "f %d %d %d %d\n", i+1, i+2, i+6, i+5); // top face
        fprintf(f, "f %d %d %d %d\n", i+4, i+3, i+7, i+8); // bottom face
    }
   
    fclose(f);
    delete[] dat;
   
    printf("Voxelization sucess !\n");
}