static int
_nrrdFieldCheck_kinds(const Nrrd *nrrd, int useBiff) {
static const char me[]="_nrrdFieldCheck_kinds";
int val[NRRD_DIM_MAX];
unsigned int wantLen, ai;
nrrdAxisInfoGet_nva(nrrd, nrrdAxisInfoKind, val);
for (ai=0; ai<nrrd->dim; ai++) {
if (!( nrrdKindUnknown == val[ai]
|| !airEnumValCheck(nrrdKind, val[ai]) )) {
biffMaybeAddf(useBiff, NRRD,
"%s: axis %d kind %d invalid", me, ai, val[ai]);
return 1;
}
wantLen = nrrdKindSize(val[ai]);
if (wantLen && wantLen != nrrd->axis[ai].size) {
char stmp[AIR_STRLEN_SMALL];
biffMaybeAddf(useBiff, NRRD,
"%s: axis %d kind %s requires size %u, but have %s", me,
ai, airEnumStr(nrrdKind, val[ai]), wantLen,
airSprintSize_t(stmp, nrrd->axis[ai].size));
return 1;
}
}
return 0;
}
/*
** _nrrdKindAltered:
**
** implements logic for how kind should be updated when samples
** along the axis are altered
*/
int
_nrrdKindAltered(int kindIn, int resampling) {
int kindOut;
if (nrrdStateKindNoop) {
kindOut = nrrdKindUnknown;
/* HEY: setting the kindOut to unknown is arguably not a no-op.
It is more like pointedly and stubbornly simplistic. So maybe
nrrdStateKindNoop could be renamed .. */
} else {
if (nrrdKindIsDomain(kindIn)
|| (0 == nrrdKindSize(kindIn) && !resampling)) {
kindOut = kindIn;
} else {
kindOut = nrrdKindUnknown;
}
}
return kindOut;
}
int
_nrrdFieldCheck_kinds(const Nrrd *nrrd, int useBiff) {
char me[]="_nrrdFieldCheck_kinds", err[BIFF_STRLEN];
int val[NRRD_DIM_MAX];
unsigned int wantLen, ai;
nrrdAxisInfoGet_nva(nrrd, nrrdAxisInfoKind, val);
for (ai=0; ai<nrrd->dim; ai++) {
if (!( nrrdKindUnknown == val[ai]
|| !airEnumValCheck(nrrdKind, val[ai]) )) {
sprintf(err, "%s: axis %d kind %d invalid", me, ai, val[ai]);
biffMaybeAdd(NRRD, err, useBiff); return 1;
}
wantLen = nrrdKindSize(val[ai]);
if (wantLen && wantLen != nrrd->axis[ai].size) {
sprintf(err, "%s: axis %d kind %s requires size %d, but have "
_AIR_SIZE_T_CNV, me,
ai, airEnumStr(nrrdKind, val[ai]), wantLen, nrrd->axis[ai].size);
biffMaybeAdd(NRRD, err, useBiff); return 1;
}
}
return 0;
}
/*
******** nrrdShuffle
**
** rearranges hyperslices of a nrrd along a given axis according to
** given permutation. This could be used to on a 4D array,
** representing a 3D volume of vectors, to re-order the vector
** components.
**
** the given permutation array must allocated for at least as long as
** the input nrrd along the chosen axis. perm[j] = i means that the
** value at position j in the _new_ array should come from position i
** in the _old_array. The standpoint is from the new, looking at
** where to find the values amid the old array (perm answers "what do
** I put here", not "where do I put this"). This allows multiple
** positions in the new array to copy from the same old position, and
** insures that there is an source for all positions along the new
** array.
*/
int
nrrdShuffle(Nrrd *nout, const Nrrd *nin, unsigned int axis,
const size_t *perm) {
static const char me[]="nrrdShuffle", func[]="shuffle";
char buff2[AIR_STRLEN_SMALL];
/* Sun Feb 8 13:13:58 CST 2009: There was a memory bug here caused
by using the same buff1[NRRD_DIM_MAX*30] declaration that had
worked fine for nrrdAxesPermute and nrrdReshape, but does NOT
work here because now samples along an axes are re-ordered, not
axes, so its often not allocated for long enough to hold the
string that's printed to it. Ideally there'd be another argument
that says whether to document the shuffle in the content string,
which would mean an API change. Or, we can use a secret
heuristic (or maybe later a nrrdState variable) for determining
when an axis is short enough to make documenting the shuffle
interesting. This is useful since functions like nrrdFlip()
probably do *not* need the shuffle (the sample reversal) to be
documented for long axes */
#define LONGEST_INTERESTING_AXIS 42
char buff1[LONGEST_INTERESTING_AXIS*30];
unsigned int ai, ldim, len;
size_t idxInB=0, idxOut, lineSize, numLines, size[NRRD_DIM_MAX], *lsize,
cIn[NRRD_DIM_MAX+1], cOut[NRRD_DIM_MAX+1];
char *dataIn, *dataOut;
if (!(nin && nout && perm)) {
biffAddf(NRRD, "%s: got NULL pointer", me);
return 1;
}
if (nout == nin) {
biffAddf(NRRD, "%s: nout==nin disallowed", me);
return 1;
}
if (!( axis < nin->dim )) {
biffAddf(NRRD, "%s: axis %d outside valid range [0,%d]",
me, axis, nin->dim-1);
return 1;
}
len = AIR_CAST(unsigned int, nin->axis[axis].size);
for (ai=0; ai<len; ai++) {
if (!( perm[ai] < len )) {
char stmp[AIR_STRLEN_SMALL];
biffAddf(NRRD, "%s: perm[%d] (%s) outside valid range [0,%d]", me, ai,
airSprintSize_t(stmp, perm[ai]), len-1);
return 1;
}
}
/* this shouldn't actually be necessary .. */
if (!nrrdElementSize(nin)) {
biffAddf(NRRD, "%s: nrrd reports zero element size!", me);
return 1;
}
/* set information in new volume */
nout->blockSize = nin->blockSize;
nrrdAxisInfoGet_nva(nin, nrrdAxisInfoSize, size);
if (nrrdMaybeAlloc_nva(nout, nin->type, nin->dim, size)) {
biffAddf(NRRD, "%s: failed to allocate output", me);
return 1;
}
if (nrrdAxisInfoCopy(nout, nin, NULL, NRRD_AXIS_INFO_NONE)) {
biffAddf(NRRD, "%s:", me);
return 1;
}
/* the min and max along the shuffled axis are now meaningless */
nout->axis[axis].min = nout->axis[axis].max = AIR_NAN;
/* do the safe thing first */
nout->axis[axis].kind = _nrrdKindAltered(nin->axis[axis].kind, AIR_FALSE);
/* try cleverness */
if (!nrrdStateKindNoop) {
if (0 == nrrdKindSize(nin->axis[axis].kind)
|| nrrdKindStub == nin->axis[axis].kind
|| nrrdKindScalar == nin->axis[axis].kind
|| nrrdKind2Vector == nin->axis[axis].kind
|| nrrdKind3Color == nin->axis[axis].kind
|| nrrdKind4Color == nin->axis[axis].kind
|| nrrdKind3Vector == nin->axis[axis].kind
|| nrrdKind3Gradient == nin->axis[axis].kind
|| nrrdKind3Normal == nin->axis[axis].kind
|| nrrdKind4Vector == nin->axis[axis].kind) {
/* these kinds have no intrinsic ordering */
nout->axis[axis].kind = nin->axis[axis].kind;
}
}
/* the skinny */
lineSize = 1;
for (ai=0; ai<axis; ai++) {
lineSize *= nin->axis[ai].size;
}
numLines = nrrdElementNumber(nin)/lineSize;
lineSize *= nrrdElementSize(nin);
lsize = size + axis;
ldim = nin->dim - axis;
dataIn = AIR_CAST(char *, nin->data);
dataOut = AIR_CAST(char *, nout->data);
memset(cIn, 0, sizeof(cIn));
memset(cOut, 0, sizeof(cOut));
for (idxOut=0; idxOut<numLines; idxOut++) {
memcpy(cIn, cOut, sizeof(cIn));
cIn[0] = perm[cOut[0]];
NRRD_INDEX_GEN(idxInB, cIn, lsize, ldim);
NRRD_INDEX_GEN(idxOut, cOut, lsize, ldim);
memcpy(dataOut + idxOut*lineSize, dataIn + idxInB*lineSize, lineSize);
NRRD_COORD_INCR(cOut, lsize, ldim, 0);
}
/* Set content. The LONGEST_INTERESTING_AXIS hack avoids the
previous array out-of-bounds bug */
if (len <= LONGEST_INTERESTING_AXIS) {
strcpy(buff1, "");
for (ai=0; ai<len; ai++) {
char stmp[AIR_STRLEN_SMALL];
sprintf(buff2, "%s%s", (ai ? "," : ""), airSprintSize_t(stmp, perm[ai]));
strcat(buff1, buff2);
}
if (nrrdContentSet_va(nout, func, nin, "%s", buff1)) {
biffAddf(NRRD, "%s:", me);
return 1;
}
} else {
if (nrrdContentSet_va(nout, func, nin, "")) {
biffAddf(NRRD, "%s:", me);
return 1;
}
}
if (nrrdBasicInfoCopy(nout, nin,
NRRD_BASIC_INFO_DATA_BIT
| NRRD_BASIC_INFO_TYPE_BIT
| NRRD_BASIC_INFO_BLOCKSIZE_BIT
| NRRD_BASIC_INFO_DIMENSION_BIT
| NRRD_BASIC_INFO_CONTENT_BIT
| NRRD_BASIC_INFO_COMMENTS_BIT
| (nrrdStateKeyValuePairsPropagate
? 0
: NRRD_BASIC_INFO_KEYVALUEPAIRS_BIT))) {
biffAddf(NRRD, "%s:", me);
return 1;
}
return 0;
#undef LONGEST_INTERESTING_AXIS
}
