Serious-Engine/Sources/LWSkaExporter/CopyWeightMaps.cpp
2016-03-11 15:57:17 +02:00

122 lines
3.6 KiB
C++

/* Copyright (c) 2002-2012 Croteam Ltd. All rights reserved. */
#include "base.h"
#define LAYER_BACKGROUND 1
#define LAYER_FOREGROUND 2
// weightmapnames
extern const char **_astrWeightMapNames;
extern int _ctWeightMapNames;
extern int _ctUsedWeightMapNames;
extern EDStateRef _state;
// points
int _ctPointsInBackground;
EDPointInfo *_ppiPoint;
int _ctGlobalCounter;
struct BackgroundPointInfo {
double position[3];
int *aiWmapIndices;
float *aiWmapWeights;
int iNumWmaps;
};
BackgroundPointInfo *_aPointsInBackground;
void ListWeightMaps() {
// list all the weightmaps used
_ctWeightMapNames = _objfunc->numVMaps(LWVMAP_WGHT);
_astrWeightMapNames = (const char**) malloc(_ctWeightMapNames*sizeof(char*));
memset(_astrWeightMapNames, 0, _ctWeightMapNames*sizeof(char*));
_ctUsedWeightMapNames = 0;
for(int iWeightMap=0; iWeightMap<_ctWeightMapNames; iWeightMap++) {
const char *strName = _objfunc->vmapName(LWVMAP_WGHT, iWeightMap);
_astrWeightMapNames[iWeightMap] = strName;
}
}
EDError ScanBackgroundPoints (void *userdata, const EDPointInfo *ppiPointInfo) {
static float fwmapval;
_aPointsInBackground[_ctGlobalCounter].position[0] = ppiPointInfo->position[0];
_aPointsInBackground[_ctGlobalCounter].position[1] = ppiPointInfo->position[1];
_aPointsInBackground[_ctGlobalCounter].position[2] = ppiPointInfo->position[2];
_aPointsInBackground[_ctGlobalCounter].aiWmapIndices = new int[_ctWeightMapNames];
_aPointsInBackground[_ctGlobalCounter].aiWmapWeights = new float[_ctWeightMapNames];
_aPointsInBackground[_ctGlobalCounter].iNumWmaps = 0;
for (int iwmap=0;iwmap<_ctWeightMapNames;iwmap++) {
_meshEditOperations->pointVSet(_state,NULL,LWVMAP_WGHT,_astrWeightMapNames[iwmap]);
if (_meshEditOperations->pointVGet(_state,ppiPointInfo->pnt,&fwmapval)) {
_aPointsInBackground[_ctGlobalCounter].aiWmapIndices[_aPointsInBackground[_ctGlobalCounter].iNumWmaps] = iwmap;
_aPointsInBackground[_ctGlobalCounter].aiWmapWeights[_aPointsInBackground[_ctGlobalCounter].iNumWmaps] = fwmapval;
_aPointsInBackground[_ctGlobalCounter].iNumWmaps++;
}
}
_ctGlobalCounter++;
return EDERR_NONE;
};
void ScanBackground() {
_ctPointsInBackground = _meshEditOperations->pointCount(_state,OPLYR_BG,EDCOUNT_ALL);
_aPointsInBackground = new BackgroundPointInfo [_ctPointsInBackground];
_ctGlobalCounter = 0;
_meshEditOperations->pointScan(_state,ScanBackgroundPoints,NULL,OPLYR_BG);
}
void FreeMem() {
if (_aPointsInBackground != NULL) {
for (int i=0;i<_ctPointsInBackground;i++) {
if (_aPointsInBackground[i].aiWmapIndices != NULL) {
delete []_aPointsInBackground[i].aiWmapIndices;
}
if (_aPointsInBackground[i].aiWmapWeights != NULL) {
delete []_aPointsInBackground[i].aiWmapWeights;
}
}
delete []_aPointsInBackground;
}
if (_astrWeightMapNames != NULL) {
delete []_astrWeightMapNames;
}
}
EDError CopyWeightMaps (void *userdata, const EDPointInfo *ppiPointInfo)
{
for (int ctpnt=0;ctpnt<_ctPointsInBackground;ctpnt++) {
if ((fabs(ppiPointInfo->position[0] - _aPointsInBackground[ctpnt].position[0]) < 0.001f) &&
(fabs(ppiPointInfo->position[1] - _aPointsInBackground[ctpnt].position[1]) < 0.001f) &&
(fabs(ppiPointInfo->position[2] - _aPointsInBackground[ctpnt].position[2]) < 0.001f)) {
for (int iwmaps=0;iwmaps < _aPointsInBackground[ctpnt].iNumWmaps;iwmaps++) {
_meshEditOperations->pntVMap(_state,ppiPointInfo->pnt,LWVMAP_WGHT,_astrWeightMapNames[_aPointsInBackground[ctpnt].aiWmapIndices[iwmaps]],1,&_aPointsInBackground[ctpnt].aiWmapWeights[iwmaps]);
}
}
}
return EDERR_NONE;
};