jedi-outcast/utils/Assimilate/xsiimp.cpp
2013-04-22 15:25:59 +10:00

901 lines
16 KiB
C++

/**********************************************************************
*<
FILE: aiimp.cpp
DESCRIPTION: .AI file import module
CREATED BY: Tom Hudson
HISTORY: created 28 June 1996
*> Copyright (c) 1996, All Rights Reserved.
**********************************************************************/
#include "Stdafx.h"
#include "Includes.h"
/*
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <windows.h>
#include <direct.h>
*/
#include "matrix4.h"
#include "xsiimp.h"
#define M_PI ( 3.14159265359f)
void
split_fn(char *path,char *file,char *pf)
{
int ix,jx,bs_loc,fn_loc;
if(strlen(pf)==0) {
if(path) *path=0;
if(file) *file=0;
return;
}
bs_loc=strlen(pf);
for(ix=bs_loc-1; ix>=0; --ix) {
if(pf[ix]=='\\') {
bs_loc=ix;
fn_loc=ix+1;
goto do_split;
}
if(pf[ix]==':') {
bs_loc=ix+1;
fn_loc=ix+1;
goto do_split;
}
}
bs_loc= -1;
fn_loc=0;
do_split:
if(file)
strcpy(file,&pf[fn_loc]);
if(path) {
if(bs_loc>0) {
for(jx=0; jx<bs_loc; ++jx)
path[jx]=pf[jx];
path[jx]=0;
}
else path[0]=0;
}
}
int mystrncmp(char *nt,char *nont,int len)
{
int i;
for (i=0;i<len;i++)
{
if (nt[i]!=nont[i])
return 1;
}
if (nt[len]!=0)
return 1;
return 0;
}
void mystrncpy(char *nt,char *nont,int len)
{
memcpy(nt,nont,len);
nt[len]=0;
}
float myatof(char *nont,int len)
{
static char tmp[100];
memcpy(tmp,nont,len);
tmp[len]=0;
return (float)atof(tmp);
}
int myatoi(char *nont,int len)
{
static char tmp[100];
memcpy(tmp,nont,len);
tmp[len]=0;
return atoi(tmp);
}
void PrintIdent(int ident)
{
static char mess[10000];
int i;
for (i=0;i<ident;i++)
mess[i]=' ';
mess[i]=0;
OutputDebugString(mess);
}
struct TxtNode
{
char *text;
int length;
TxtNode *child;
TxtNode *sibling;
TxtNode()
{
text=0;
length=0;
child=0;
sibling=0;
}
~TxtNode()
{
if (child)
delete child;
child=0;
if (sibling)
delete sibling;
sibling=0;
}
void Print(int ident)
{
if (text&&length)
{
PrintIdent(ident);
static char mess[10000];
static char str[10000];
mystrncpy(str,text,length);
sprintf(mess,"|%s|\n",str);
OutputDebugString(mess);
}
if (child)
{
PrintIdent(ident);
OutputDebugString("{\n");
TxtNode *sib=child;
while (sib)
{
sib->Print(ident+1);
sib=sib->sibling;
}
PrintIdent(ident);
OutputDebugString("}\n");
}
}
};
#define PDEB (0)
char *ReadToken(char *at,char *end,int &len,char **out)
{
char *ret=0;
len=0;
while (at<end&&(isspace(*at)||*at==';'||*at==','))
at++;
if (at==end)
{
*out=end;
return 0;
}
if ((*at=='\"'||*at=='<')&&at+1<end)
{
char endchar='\"';
if (*at=='<')
endchar='>';
at++;
ret=at;
while (at<end&&*at!=endchar)
{
at++;
len++;
}
if (at<end)
at++;
}
else if ((*at=='{'||*at=='}')&&at+1<end)
{
ret=at;
len=1;
at++;
}
else
{
ret=at;
while (at<end&&!(isspace(*at)||*at==';'||*at==','||*at=='}'||*at=='>'||*at=='{'||*at=='<'||*at=='\"'))
{
len++;
at++;
}
}
*out=at;
#if 0
static char mess[10000];
static char str[10000];
mystrncpy(str,ret,len);
sprintf(mess,"Token |%s|\n",str);
OutputDebugString(mess);
#endif
return ret;
}
TxtNode *ReadANode(char *at,char *end,char **atout)
{
TxtNode *ret=0;
int len;
char *retat;
char *start;
//OutputDebugString("***");
start=ReadToken(at,end,len,&retat);
at=retat;
*atout=at;
if (!start)
{
//OutputDebugString("Eof\n");
return 0;
}
if (*start=='}')
{
//OutputDebugString("End}\n");
return 0;
}
if (*start=='{')
{
//OutputDebugString("Start{\n");
ret=new TxtNode;
TxtNode **last=&ret->child;
while (1)
{
TxtNode *t=ReadANode(at,end,&retat);
at=retat;
if (!t)
break;
*last=t;
last=&t->sibling;
}
}
else
{
ret=new TxtNode;
ret->text=start;
ret->length=len;
}
*atout=at;
return ret;
}
struct SceneNodes
{
char name[100];
SceneNodes *parent;
Matrix4 mat;
Matrix4 BaseMat;
Matrix4 *animat;
int nanimat;
};
static SceneNodes sNodes[5000];
static nsNodes=0;
int FindNodes(SceneNodes *parent,TxtNode *t)
{
static char tmp[1000];
if (t->child)
{
TxtNode *sib=t->child;
while (sib)
{
if (sib->text&&!mystrncmp("Frame",sib->text,sib->length))
{
SceneNodes tmpNode;
Matrix4 tm,tmBase;
tm.Identity();
sib=sib->sibling;
if (!sib)
return 1;
assert(sib->length<100);
if (!strncmp("frm-",sib->text,4))
mystrncpy(tmpNode.name,sib->text+4,sib->length-4);
else
mystrncpy(tmpNode.name,sib->text,sib->length);
//OutputDebugString(tmpNode.name);
//OutputDebugString("\n");
sib=sib->sibling;
if (!sib)
return 2;
{
TxtNode *FrameLook=sib;
TxtNode *sub=sib->child;
if (!sub)
return 3;
if (!sub->text)
return 4;
if (mystrncmp("FrameTransformMatrix",sub->text,sub->length))
return 5;
sub=sub->sibling;
{
TxtNode *xf=sub->child;
int r,c;
#if 0
OutputDebugString("matrix\n");
#endif
for (r=0;r<4;r++)
{
int rr=r;
float sign=1.0f;
if (r==1)
rr=2;
if (r==2)
{
sign=-1.0f;
rr=1;
}
for (c=0;c<4;c++)
{
if (!xf||!xf->text)
return 6;
if (c==0)
tm[rr][0]=sign*(float)myatof(xf->text,xf->length);
else if (c==1)
tm[rr][2]=sign*(float)myatof(xf->text,xf->length);
else if (c==2)
tm[rr][1]=-sign*(float)myatof(xf->text,xf->length);
xf=xf->sibling;
}
}
#if 0
for (r=0;r<4;r++)
{
sprintf(tmp,"%6f %6f %6f\n",tm[r][0],tm[r][1],tm[r][2]);
OutputDebugString(tmp);
}
#endif
tm.CalcFlags();
}
tmBase=tm;
sub=sub->sibling;
while (sub)
{
if (sub->text&&!mystrncmp("SI_FrameBasePoseMatrix",sub->text,sub->length))
{
sub=sub->sibling;
{
TxtNode *xf=sub->child;
int r,c;
#if 0
OutputDebugString("pose matrix\n");
#endif
for (r=0;r<4;r++)
{
int rr=r;
float sign=1.0f;
if (r==1)
rr=2;
if (r==2)
{
sign=-1.0f;
rr=1;
}
for (c=0;c<4;c++)
{
if (!xf||!xf->text)
return 6;
if (c==0)
tmBase[rr][0]=sign*(float)myatof(xf->text,xf->length);
else if (c==1)
tmBase[rr][2]=sign*(float)myatof(xf->text,xf->length);
else if (c==2)
tmBase[rr][1]=-sign*(float)myatof(xf->text,xf->length);
xf=xf->sibling;
}
}
tmBase.CalcFlags();
#if 0
for (r=0;r<4;r++)
{
sprintf(tmp,"%6f %6f %6f\n",tmBase[r][0],tmBase[r][1],tmBase[r][2]);
OutputDebugString(tmp);
}
#endif
}
}
sub=sub->sibling;
}
tmpNode.parent=parent;
tmpNode.mat=tm;
Matrix4 par,tmat;
if (parent)
tmat.Concat(tmBase,parent->BaseMat);
else
tmat=tmBase;
tmpNode.BaseMat=tmat;
SceneNodes *newparent=sNodes+nsNodes;
tmpNode.animat=0;
tmpNode.nanimat=0;
sNodes[nsNodes++]=tmpNode;
int err=FindNodes(newparent,FrameLook);
if (err)
return err;
}
}
sib=sib->sibling;
}
}
return 0;
}
static int maxframe=-10000;
static int minframe=10000;
static int startframe=10000;
static int endframe=-10000;
static int CacheNodes[10000];
static int nCacheNodes=0;
static char CacheName[100];
static int maxmatframe;
static Matrix4 mats[1000];
static Vect3 scales[1000];
int ProcNKeys(TxtNode *t,int& foundkeys)
{
if (t->child)
{
TxtNode *sib=t->child;
if (!sib->text)
return 207;
int tp=myatoi(sib->text,sib->length);
sib=sib->sibling;
if (!sib||!sib->text)
return 208;
int num=myatoi(sib->text,sib->length);
sib=sib->sibling;
int i;
int lastframe=startframe-1;
TxtNode *s=sib;
for (i=0;i<num;i++)
{
if (!s||!s->text)
return 110;
int frame=myatoi(s->text,s->length);
if (frame>33000)
frame=frame-65536;
if (frame<startframe)
{
assert(!i);
lastframe=frame-1;
startframe=frame;
}
if (frame>endframe)
endframe=frame;
if (frame-startframe>maxmatframe)
maxmatframe=frame-startframe;
s=s->sibling;
if (!s||!s->text)
return 212;
int junk=myatoi(s->text,s->length);
if (junk!=3)
return 213;
s=s->sibling;
if (!s||!s->text)
return 216;
float x=(float)myatof(s->text,s->length);
s=s->sibling;
if (!s||!s->text)
return 216;
float y=(float)myatof(s->text,s->length);
s=s->sibling;
if (!s||!s->text)
return 216;
float z=(float)myatof(s->text,s->length);
s=s->sibling;
int k;
if (tp==1) //scale
{
Vect3 v=Vect3(x,z,y);
for (k=lastframe+1;k<=frame;k++)
scales[k-startframe]=v;
foundkeys|=4;
}
else if (tp==2) //translation
{
Vect3 v(x,-z,y);
for (k=lastframe+1;k<=frame;k++)
{
mats[k-startframe].SetRow(3,v);
mats[k-startframe].CalcFlags();
}
foundkeys|=1;
}
else if (tp==3) //rotation
{
Matrix4 tmp;
tmp.Rotate(-x*M_PI/180.0f,-y*M_PI/180.0f,-z*M_PI/180.0f);
Vect3 t,tt;
tmp.GetRow(2,t);
t*=-1.0f;
tmp.GetRow(1,tt);
tmp.SetRow(2,tt);
tmp.SetRow(1,t);
for (k=0;k<3;k++)
{
float tt=-tmp[k][2];
tmp[k][2]=tmp[k][1];
tmp[k][1]=tt;
}
for (k=lastframe+1;k<=frame;k++)
{
int l;
for (l=0;l<3;l++)
{
tmp.GetRow(l,t);
mats[k-startframe].SetRow(l,t);
mats[k-startframe].CalcFlags();
}
}
foundkeys|=2;
}
lastframe=frame;
}
}
return 0;
}
int FindPKeys(TxtNode *t)
{
if (t->child)
{
TxtNode *sib=t->child;
if (!sib)
return 200;
char framename[100];
{
if (!sib->child)
return 201;
if (!sib->child->text)
return 202;
if (!strncmp("frm-",sib->child->text,4))
mystrncpy(framename,sib->child->text+4,sib->child->length-4);
else
{
if (!strncmp("SCENE",sib->child->text,4))
return 0;
mystrncpy(framename,sib->child->text,sib->child->length);
}
sib=sib->sibling;
}
int foundkeys=0;
maxmatframe=0;
while (sib)
{
if (sib->text&&!mystrncmp("SI_AnimationKey",sib->text,sib->length))
{
if (!sib->sibling)
return 103;
sib=sib->sibling;
int err=ProcNKeys(sib,foundkeys);
if (err)
return err;
}
sib=sib->sibling;
}
// if (foundkeys)
{
if (maxmatframe>maxframe)
maxframe=maxmatframe;
int i;
for (i=0;i<nsNodes;i++)
{
if (!strcmp(sNodes[i].name,framename))
{
int j;
if (sNodes[i].nanimat<maxmatframe+1)
{
delete[] sNodes[i].animat;
sNodes[i].nanimat=maxmatframe+1;
sNodes[i].animat=new Matrix4[maxmatframe+1];
for (j=0;j<=maxmatframe;j++)
sNodes[i].animat[j].Identity();
}
for (j=0;j<=maxmatframe;j++)
{
Matrix4 xf;
xf=sNodes[i].mat;
Vect3 tt,t2;
if (foundkeys&1)
{
mats[j].GetRow(3,tt);
xf.SetRow(3,tt);
}
if (foundkeys&2)
{
float l;
xf.GetRow(0,t2);
l=t2.Len();
mats[j].GetRow(0,tt);
tt*=l;
xf.SetRow(0,tt);
xf.GetRow(1,t2);
l=t2.Len();
mats[j].GetRow(1,tt);
tt*=l;
xf.SetRow(1,tt);
xf.GetRow(2,t2);
l=t2.Len();
mats[j].GetRow(2,tt);
tt*=l;
xf.SetRow(2,tt);
}
if (foundkeys&4)
{
xf.GetRow(0,tt);
tt.Norm();
tt*=scales[j].x();
xf.SetRow(0,tt);
xf.GetRow(1,tt);
tt.Norm();
tt*=scales[j].x();
xf.SetRow(1,tt);
xf.GetRow(2,tt);
tt.Norm();
tt*=scales[j].x();
xf.SetRow(2,tt);
}
xf.CalcFlags();
Matrix4 par,tmat;
if (sNodes[i].parent)
{
if (j<sNodes[i].parent->nanimat)
par=sNodes[i].parent->animat[j];
else
par=sNodes[i].parent->BaseMat;
tmat.Concat(xf,par);
}
else
tmat=xf;
sNodes[i].animat[j]=tmat;
#if 0
par.Inverse(sNodes[i].BaseMat);
tmat.Concat(par,sNodes[i].animat[j]);
char tmp[1000];
int r;
sprintf(tmp,"final frame %d %s\n",j,sNodes[i].name);
OutputDebugString(tmp);
for (r=0;r<4;r++)
{
sprintf(tmp,"%6f %6f %6f\n",tmat[r][0],tmat[r][1],tmat[r][2]);
OutputDebugString(tmp);
}
sprintf(tmp,"baseinv frame %d %s\n",j,sNodes[i].name);
OutputDebugString(tmp);
for (r=0;r<4;r++)
{
sprintf(tmp,"%6f %6f %6f\n",par[r][0],par[r][1],par[r][2]);
OutputDebugString(tmp);
}
sprintf(tmp,"bone frame %d %s\n",j,sNodes[i].name);
OutputDebugString(tmp);
for (r=0;r<4;r++)
{
sprintf(tmp,"%6f %6f %6f\n",sNodes[i].animat[j][r][0],sNodes[i].animat[j][r][1],sNodes[i].animat[j][r][2]);
OutputDebugString(tmp);
}
#endif
}
}
}
}
}
return 0;
}
int FindSets(TxtNode *t)
{
if (t->child)
{
TxtNode *sib=t->child;
while (sib)
{
if (sib->text&&!mystrncmp("Animation",sib->text,sib->length))
{
if (!sib->sibling)
return 101;
sib=sib->sibling;
if (!sib->sibling)
return 102;
sib=sib->sibling;
int err=FindPKeys(sib);
if (err)
return err;
}
sib=sib->sibling;
}
}
return 0;
}
int FindAnims(TxtNode *t)
{
int i;
for (i=0;i<nsNodes;i++)
{
sNodes[i].animat=0;
sNodes[i].nanimat=0;
}
if (t->child)
{
TxtNode *sib=t->child;
while (sib)
{
if (sib->text&&!mystrncmp("AnimationSet",sib->text,sib->length))
{
if (!sib->sibling)
return 100;
sib=sib->sibling;
int err=FindSets(sib);
if (err)
return err;
}
sib=sib->sibling;
}
}
return 0;
}
/*
** returns the number of frames loaded
*/
int XSI_LoadFile(const char *filename)
{
char *filebin=0;
TxtNode *root;
FILE *fp=fopen(filename,"ra");
if (!fp){
InfoBox(va("File not found: \"%s\"",filename));
return 0;
}
fseek(fp,0,SEEK_END);
int len=ftell(fp);
filebin=new char[len];
fseek(fp,0,SEEK_SET);
if (fread(filebin,1,len,fp)!=(size_t)len)
{
fclose(fp);
InfoBox(va("Bad XSI file\n",filename));
delete(filebin);
return 0;
}
fclose(fp);
char *at=filebin;
char *end=filebin+len;
root=new TxtNode;
TxtNode **last=&root->child;
char *retat;
while (1)
{
TxtNode *t=ReadANode(at,end,&retat);
if (!t)
{
at=end;
break;
}
at=retat;
*last=t;
last=&t->sibling;
}
if (!root->child)
{
printf("Bad XSI file\n");
delete(filebin);
delete root;
return 0;
}
#if PDEB
root->Print(0);
#endif
nsNodes=0;
maxframe=-10000;
minframe=10000;
startframe=10000;
endframe=-10000;
CacheName[0]=0;
int fn=FindNodes(0,root);
if (!fn)
{
fn=FindAnims(root);
}
delete(filebin);
delete root;
if (fn==2000)
{
printf("XSI import failed, Missing Keys\n",fn);
return 0;
}
if (fn)
{
printf("XSI import failed, code = %d",fn);
return 0;
}
return maxframe+1;
}
int XSI_GetNumFrames()
{
if (maxframe+1>0)
return maxframe+1;
return 0;
}
int XSI_FindBone(char *bonename)
{
int j;
for (j=0;j<nsNodes;j++)
{
if (!strcmpi(bonename,sNodes[j].name))
return j;
}
return -1;
}
void XSI_GetBoneMatrix(int bone,int frame,float mat[3][4])
{
assert(bone>=0&&bone<nsNodes);
assert(frame>=0&&frame<maxframe+1);
int i,j;
for (i=0;i<4;i++)
{
for (j=0;j<3;j++)
{
mat[j][i]=sNodes[bone].animat[frame][i][j];
}
}
}
void XSI_GetBoneMatrix4(int bone,int frame,Matrix4 &m)
{
assert(bone>=0&&bone<nsNodes);
assert(frame>=0&&frame<maxframe+1);
m=sNodes[bone].animat[frame];
}
int XSI_GetNumBones()
{
return nsNodes;
}
void XSI_GetBoneName(int bone,char *bonename)
{
assert(bone>=0&&bone<nsNodes);
strcpy(bonename,sNodes[bone].name);
}
void XSI_Cleanup()
{
int j;
for (j=0;j<nsNodes;j++)
{
delete[] sNodes[j].animat;
}
nsNodes=0;
maxframe=-10000;
minframe=10000;
startframe=10000;
endframe=-10000;
CacheName[0]=0;
}