| File: | src/pull/trace.c |
| Location: | line 1171, column 7 |
| Description: | Potential leak of memory pointed to by 'trc' |
| 1 | /* | |||
| 2 | Teem: Tools to process and visualize scientific data and images . | |||
| 3 | Copyright (C) 2013, 2012, 2011, 2010, 2009 University of Chicago | |||
| 4 | Copyright (C) 2008, 2007, 2006, 2005 Gordon Kindlmann | |||
| 5 | Copyright (C) 2004, 2003, 2002, 2001, 2000, 1999, 1998 University of Utah | |||
| 6 | ||||
| 7 | This library is free software; you can redistribute it and/or | |||
| 8 | modify it under the terms of the GNU Lesser General Public License | |||
| 9 | (LGPL) as published by the Free Software Foundation; either | |||
| 10 | version 2.1 of the License, or (at your option) any later version. | |||
| 11 | The terms of redistributing and/or modifying this software also | |||
| 12 | include exceptions to the LGPL that facilitate static linking. | |||
| 13 | ||||
| 14 | This library is distributed in the hope that it will be useful, | |||
| 15 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |||
| 16 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |||
| 17 | Lesser General Public License for more details. | |||
| 18 | ||||
| 19 | You should have received a copy of the GNU Lesser General Public License | |||
| 20 | along with this library; if not, write to Free Software Foundation, Inc., | |||
| 21 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
| 22 | */ | |||
| 23 | ||||
| 24 | ||||
| 25 | #include "pull.h" | |||
| 26 | #include "privatePull.h" | |||
| 27 | ||||
| 28 | pullTrace * | |||
| 29 | pullTraceNew(void) { | |||
| 30 | pullTrace *ret; | |||
| 31 | ||||
| 32 | ret = AIR_CALLOC(1, pullTrace)(pullTrace*)(calloc((1), sizeof(pullTrace))); | |||
| 33 | if (ret) { | |||
| 34 | ret->seedPos[0] = ret->seedPos[1] = AIR_NAN(airFloatQNaN.f); | |||
| 35 | ret->seedPos[2] = ret->seedPos[3] = AIR_NAN(airFloatQNaN.f); | |||
| 36 | ret->nvert = nrrdNew(); | |||
| 37 | ret->nstrn = nrrdNew(); | |||
| 38 | ret->nstab = nrrdNew(); | |||
| 39 | ret->norin = nrrdNew(); | |||
| 40 | ret->seedIdx = 0; | |||
| 41 | ret->whyStop[0] = ret->whyStop[1] = pullTraceStopUnknown; | |||
| 42 | ret->whyStopCFail[0] = ret->whyStopCFail[1] = pullConstraintFailUnknown; | |||
| 43 | ret->whyNowhere = pullTraceStopUnknown; | |||
| 44 | ret->whyNowhereCFail = pullConstraintFailUnknown; | |||
| 45 | } | |||
| 46 | return ret; | |||
| 47 | } | |||
| 48 | ||||
| 49 | pullTrace * | |||
| 50 | pullTraceNix(pullTrace *pts) { | |||
| 51 | ||||
| 52 | if (pts) { | |||
| 53 | nrrdNuke(pts->nvert); | |||
| 54 | nrrdNuke(pts->nstrn); | |||
| 55 | nrrdNuke(pts->nstab); | |||
| 56 | nrrdNuke(pts->norin); | |||
| 57 | free(pts); | |||
| 58 | } | |||
| 59 | return NULL((void*)0); | |||
| 60 | } | |||
| 61 | ||||
| 62 | void | |||
| 63 | pullTraceStability(double *spcStab, | |||
| 64 | double *oriStab, | |||
| 65 | const double pos0[4], | |||
| 66 | const double pos1[4], | |||
| 67 | const double ori0[3], | |||
| 68 | const double ori1[3], | |||
| 69 | double sigma0, | |||
| 70 | const pullContext *pctx) { | |||
| 71 | double sc, stb, dx, ds, diff[4]; | |||
| 72 | ||||
| 73 | ELL_4V_SUB(diff, pos1, pos0)((diff)[0] = (pos1)[0] - (pos0)[0], (diff)[1] = (pos1)[1] - ( pos0)[1], (diff)[2] = (pos1)[2] - (pos0)[2], (diff)[3] = (pos1 )[3] - (pos0)[3]); | |||
| 74 | dx = ELL_3V_LEN(diff)(sqrt((((diff))[0]*((diff))[0] + ((diff))[1]*((diff))[1] + (( diff))[2]*((diff))[2])))/(pctx->voxelSizeSpace); | |||
| 75 | sc = (pos0[3] + pos1[3])/2; | |||
| 76 | if (pctx->flag.scaleIsTau) { | |||
| 77 | sc = gageSigOfTau(sc)airSigmaOfTau(sc); | |||
| 78 | } | |||
| 79 | sc += sigma0; | |||
| 80 | dx /= sc; | |||
| 81 | ds = diff[3]; | |||
| 82 | stb = atan2(ds, dx)/(AIR_PI3.14159265358979323846/2); /* [-1,1]: 0 means least stable */ | |||
| 83 | stb = AIR_ABS(stb)((stb) > 0.0f ? (stb) : -(stb)); /* [0,1]: 0 means least stable */ | |||
| 84 | *spcStab = AIR_CLAMP(0, stb, 1)((stb) < (0) ? (0) : ((stb) > (1) ? (1) : (stb))); | |||
| 85 | /* | |||
| 86 | static double maxvelo = 0; | |||
| 87 | if (1 && vv > maxvelo) { | |||
| 88 | char me[]="pullTraceStability"; | |||
| 89 | printf("\n%s: [%g,%g,%g,%g] - [%g,%g,%g,%g] = [%g,%g,%g,%g]\n", me, | |||
| 90 | pos0[0], pos0[1], pos0[2], pos0[3], | |||
| 91 | pos1[0], pos1[1], pos1[2], pos1[3], | |||
| 92 | diff[0], diff[1], diff[2], diff[3]); | |||
| 93 | printf("%s: dx = %g -> %g; ds = %g\n", me, dx0, dx, ds); | |||
| 94 | printf("%s: vv = atan2(%g,%g)/pi = %g -> %g > %g\n", me, ds, dx, vv0, vv, maxvelo); | |||
| 95 | maxvelo = vv; | |||
| 96 | } | |||
| 97 | */ | |||
| 98 | if (ori0 && ori1) { | |||
| 99 | /* dori = delta in orientation */ | |||
| 100 | double dori = ell_3v_angle_d(ori0, ori1); | |||
| 101 | /* dori in [0,pi]; 0 and pi mean no change */ | |||
| 102 | if (dori > AIR_PI3.14159265358979323846/2) { | |||
| 103 | dori -= AIR_PI3.14159265358979323846; | |||
| 104 | } | |||
| 105 | /* dori in [-pi/2,pi/2]; 0 means no change; +-pi/2 means most change */ | |||
| 106 | dori = AIR_ABS(dori)((dori) > 0.0f ? (dori) : -(dori)); | |||
| 107 | /* dori in [0,pi/2]; 0 means no change; pi/2 means most change */ | |||
| 108 | *oriStab = atan2(ds, dori)/(AIR_PI3.14159265358979323846/2); | |||
| 109 | /* *oriStab in [0,1]; 0 means stable, 1 means not stable */ | |||
| 110 | } else { | |||
| 111 | *oriStab = 0; | |||
| 112 | } | |||
| 113 | return; | |||
| 114 | } | |||
| 115 | ||||
| 116 | int | |||
| 117 | _pullConstrTanSlap(pullContext *pctx, pullPoint *point, | |||
| 118 | double tlen, | |||
| 119 | /* input+output */ double toff[3], | |||
| 120 | /* output */ int *constrFailP) { | |||
| 121 | static const char me[]="_pullConstrTanSlap"; | |||
| 122 | double pos0[4], tt; | |||
| 123 | ||||
| 124 | if (!(pctx && point && toff && constrFailP)) { | |||
| 125 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 126 | return 1; | |||
| 127 | } | |||
| 128 | if (pctx->flag.zeroZ) { | |||
| 129 | toff[2] = 0; | |||
| 130 | } | |||
| 131 | ELL_3V_NORM(toff, toff, tt)(tt = (sqrt((((toff))[0]*((toff))[0] + ((toff))[1]*((toff))[1 ] + ((toff))[2]*((toff))[2]))), ((toff)[0] = (1.0/tt)*(toff)[ 0], (toff)[1] = (1.0/tt)*(toff)[1], (toff)[2] = (1.0/tt)*(toff )[2])); | |||
| 132 | if (!(tlen > 0 && tt > 0)) { | |||
| 133 | biffAddf(PULLpullBiffKey, "%s: tlen %g or |toff| %g not positive", me, tlen, tt); | |||
| 134 | return 1; | |||
| 135 | } | |||
| 136 | ELL_4V_COPY(pos0, point->pos)((pos0)[0] = (point->pos)[0], (pos0)[1] = (point->pos)[ 1], (pos0)[2] = (point->pos)[2], (pos0)[3] = (point->pos )[3]); /* pos0[3] should stay put; will test at end */ | |||
| 137 | ||||
| 138 | ELL_3V_SCALE_ADD2(point->pos, 1, pos0, tlen, toff)((point->pos)[0] = (1)*(pos0)[0] + (tlen)*(toff)[0], (point ->pos)[1] = (1)*(pos0)[1] + (tlen)*(toff)[1], (point->pos )[2] = (1)*(pos0)[2] + (tlen)*(toff)[2]); | |||
| 139 | if (_pullConstraintSatisfy(pctx->task[0], point, 0 /* travmax */, constrFailP)) { | |||
| 140 | biffAddf(PULLpullBiffKey, "%s: trouble", me); | |||
| 141 | ELL_4V_COPY(point->pos, pos0)((point->pos)[0] = (pos0)[0], (point->pos)[1] = (pos0)[ 1], (point->pos)[2] = (pos0)[2], (point->pos)[3] = (pos0 )[3]); return 1; | |||
| 142 | } | |||
| 143 | /* save offset to new position */ | |||
| 144 | ELL_3V_SUB(toff, point->pos, pos0)((toff)[0] = (point->pos)[0] - (pos0)[0], (toff)[1] = (point ->pos)[1] - (pos0)[1], (toff)[2] = (point->pos)[2] - (pos0 )[2]); | |||
| 145 | ||||
| 146 | if (pos0[3] != point->pos[3]) { | |||
| 147 | biffAddf(PULLpullBiffKey, "%s: point->pos[3] %g was changed (from %g)", | |||
| 148 | me, point->pos[3], pos0[3]); | |||
| 149 | ELL_4V_COPY(point->pos, pos0)((point->pos)[0] = (pos0)[0], (point->pos)[1] = (pos0)[ 1], (point->pos)[2] = (pos0)[2], (point->pos)[3] = (pos0 )[3]); return 1; | |||
| 150 | } | |||
| 151 | ELL_4V_COPY(point->pos, pos0)((point->pos)[0] = (pos0)[0], (point->pos)[1] = (pos0)[ 1], (point->pos)[2] = (pos0)[2], (point->pos)[3] = (pos0 )[3]); | |||
| 152 | return 0; | |||
| 153 | } | |||
| 154 | ||||
| 155 | int | |||
| 156 | _pullConstrOrientFind(pullContext *pctx, pullPoint *point, | |||
| 157 | int normalfind, /* find normal two 2D surface, | |||
| 158 | else find tangent to 1D curve */ | |||
| 159 | double tlen, | |||
| 160 | const double tdir0[3], /* if non-NULL, try using this direction */ | |||
| 161 | /* output */ | |||
| 162 | double dir[3], | |||
| 163 | int *constrFailP) { | |||
| 164 | static const char me[]="_pullConstrOrientFind"; | |||
| 165 | double tt; | |||
| 166 | ||||
| 167 | #define SLAP(LEN, DIR)if (_pullConstrTanSlap(pctx, point, (LEN), (DIR), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (DIR)[0], (DIR)[1], (DIR)[2]); return 1; } if (*constrFailP ) { return 0; } \ | |||
| 168 | /* fprintf(stderr, "!%s: SLAP %g %g %g -->", me, (DIR)[0], (DIR)[1], (DIR)[2]); */ \ | |||
| 169 | if (_pullConstrTanSlap(pctx, point, (LEN), (DIR), constrFailP)) { \ | |||
| 170 | biffAddf(PULLpullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)", \ | |||
| 171 | me, (DIR)[0], (DIR)[1], (DIR)[2]); \ | |||
| 172 | return 1; \ | |||
| 173 | } \ | |||
| 174 | if (*constrFailP) { \ | |||
| 175 | /* unsuccessful in finding tangent, but not a biff error */ \ | |||
| 176 | return 0; \ | |||
| 177 | } \ | |||
| 178 | /* fprintf(stderr, " %g %g %g\n", (DIR)[0], (DIR)[1], (DIR)[2]); */ | |||
| 179 | ||||
| 180 | if (!(pctx && point && constrFailP)) { | |||
| 181 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 182 | return 1; | |||
| 183 | } | |||
| 184 | *constrFailP = pullConstraintFailUnknown; | |||
| 185 | if (normalfind) { | |||
| 186 | double tan0[3], tan1[3]; | |||
| 187 | /* looking for a surface normal */ | |||
| 188 | if (tdir0) { /* have something to start with */ | |||
| 189 | ell_3v_perp_d(tan0, tdir0); | |||
| 190 | ELL_3V_CROSS(tan1, tan0, tdir0)((tan1)[0] = (tan0)[1]*(tdir0)[2] - (tan0)[2]*(tdir0)[1], (tan1 )[1] = (tan0)[2]*(tdir0)[0] - (tan0)[0]*(tdir0)[2], (tan1)[2] = (tan0)[0]*(tdir0)[1] - (tan0)[1]*(tdir0)[0]); | |||
| 191 | SLAP(tlen, tan1)if (_pullConstrTanSlap(pctx, point, (tlen), (tan1), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tan1)[0], (tan1)[1], (tan1)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 192 | SLAP(tlen, tan0)if (_pullConstrTanSlap(pctx, point, (tlen), (tan0), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tan0)[0], (tan0)[1], (tan0)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 193 | ELL_3V_CROSS(dir, tan1, tan0)((dir)[0] = (tan1)[1]*(tan0)[2] - (tan1)[2]*(tan0)[1], (dir)[ 1] = (tan1)[2]*(tan0)[0] - (tan1)[0]*(tan0)[2], (dir)[2] = (tan1 )[0]*(tan0)[1] - (tan1)[1]*(tan0)[0]); | |||
| 194 | } else { /* have to start from scratch */ | |||
| 195 | double tns[9] = {1,0,0, 0,1,0, 0,0,1}; | |||
| 196 | SLAP(tlen, tns+0)if (_pullConstrTanSlap(pctx, point, (tlen), (tns+0), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tns+0)[0], (tns+0)[1], (tns+0)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 197 | SLAP(tlen, tns+3)if (_pullConstrTanSlap(pctx, point, (tlen), (tns+3), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tns+3)[0], (tns+3)[1], (tns+3)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 198 | SLAP(tlen, tns+6)if (_pullConstrTanSlap(pctx, point, (tlen), (tns+6), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tns+6)[0], (tns+6)[1], (tns+6)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 199 | ell_3m_1d_nullspace_d(dir, tns); | |||
| 200 | } | |||
| 201 | } else { | |||
| 202 | /* looking for a curve tangent */ | |||
| 203 | if (tdir0) { /* have something to start with */ | |||
| 204 | ELL_3V_COPY(dir, tdir0)((dir)[0] = (tdir0)[0], (dir)[1] = (tdir0)[1], (dir)[2] = (tdir0 )[2]); | |||
| 205 | SLAP(tlen, dir)if (_pullConstrTanSlap(pctx, point, (tlen), (dir), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (dir)[0], (dir)[1], (dir)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 206 | /* SLAP(tlen, dir); (didn't have much effect, in one test) */ | |||
| 207 | } else { /* have to start from scratch */ | |||
| 208 | double tX[3] = {1,0,0}, tY[3] = {0,1,0}, tZ[3] = {0,0,1}; | |||
| 209 | SLAP(tlen, tX)if (_pullConstrTanSlap(pctx, point, (tlen), (tX), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tX)[0], (tX)[1], (tX)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 210 | SLAP(tlen, tY)if (_pullConstrTanSlap(pctx, point, (tlen), (tY), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tY)[0], (tY)[1], (tY)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 211 | if (ELL_3V_DOT(tX, tY)((tX)[0]*(tY)[0] + (tX)[1]*(tY)[1] + (tX)[2]*(tY)[2]) < 0) { ELL_3V_SCALE(tY, -1, tY)((tY)[0] = (-1)*(tY)[0], (tY)[1] = (-1)*(tY)[1], (tY)[2] = (- 1)*(tY)[2]); } | |||
| 212 | if (pctx->flag.zeroZ) { | |||
| 213 | ELL_3V_SET(tZ, 0, 0, 0)((tZ)[0] = (0), (tZ)[1] = (0), (tZ)[2] = (0)); | |||
| 214 | } else { | |||
| 215 | SLAP(tlen, tZ)if (_pullConstrTanSlap(pctx, point, (tlen), (tZ), constrFailP )) { biffAddf(pullBiffKey, "%s: looking for tangent, starting with (%g,%g,%g)" , me, (tZ)[0], (tZ)[1], (tZ)[2]); return 1; } if (*constrFailP ) { return 0; }; | |||
| 216 | if (ELL_3V_DOT(tX, tZ)((tX)[0]*(tZ)[0] + (tX)[1]*(tZ)[1] + (tX)[2]*(tZ)[2]) < 0) { ELL_3V_SCALE(tZ, -1, tZ)((tZ)[0] = (-1)*(tZ)[0], (tZ)[1] = (-1)*(tZ)[1], (tZ)[2] = (- 1)*(tZ)[2]); } | |||
| 217 | if (ELL_3V_DOT(tY, tZ)((tY)[0]*(tZ)[0] + (tY)[1]*(tZ)[1] + (tY)[2]*(tZ)[2]) < 0) { ELL_3V_SCALE(tY, -1, tZ)((tY)[0] = (-1)*(tZ)[0], (tY)[1] = (-1)*(tZ)[1], (tY)[2] = (- 1)*(tZ)[2]); } | |||
| 218 | } | |||
| 219 | ELL_3V_ADD3(dir, tX, tY, tZ)((dir)[0] = (tX)[0] + (tY)[0] + (tZ)[0], (dir)[1] = (tX)[1] + (tY)[1] + (tZ)[1], (dir)[2] = (tX)[2] + (tY)[2] + (tZ)[2]); | |||
| 220 | } | |||
| 221 | } | |||
| 222 | ELL_3V_NORM(dir, dir, tt)(tt = (sqrt((((dir))[0]*((dir))[0] + ((dir))[1]*((dir))[1] + ( (dir))[2]*((dir))[2]))), ((dir)[0] = (1.0/tt)*(dir)[0], (dir) [1] = (1.0/tt)*(dir)[1], (dir)[2] = (1.0/tt)*(dir)[2])); | |||
| 223 | if (!(tt > 0)) { | |||
| 224 | biffAddf(PULLpullBiffKey, "%s: computed direction is zero (%g)?", me, tt); | |||
| 225 | return 1; | |||
| 226 | } | |||
| 227 | return 0; | |||
| 228 | } | |||
| 229 | ||||
| 230 | /* | |||
| 231 | ******** pullTraceSet | |||
| 232 | ** | |||
| 233 | ** computes a single trace, according to the given parameters, | |||
| 234 | ** and store it in the pullTrace | |||
| 235 | int recordStrength: should strength be recorded along trace | |||
| 236 | double scaleDelta: discrete step along scale | |||
| 237 | double halfScaleWin: how far, along scale, trace should go in each direction | |||
| 238 | unsigned int arrIncr: increment for storing position (maybe strength) | |||
| 239 | const double _seedPos[4]: starting position | |||
| 240 | */ | |||
| 241 | int | |||
| 242 | pullTraceSet(pullContext *pctx, pullTrace *pts, | |||
| 243 | int recordStrength, | |||
| 244 | double scaleDelta, double halfScaleWin, | |||
| 245 | double orientTestLen, | |||
| 246 | unsigned int arrIncr, | |||
| 247 | const double _seedPos[4]) { | |||
| 248 | static const char me[]="pullTraceSet"; | |||
| 249 | pullPoint *point; | |||
| 250 | airArray *mop, *trceArr[2], *hstrnArr[2], *horinArr[2]; | |||
| 251 | double *trce[2], ssrange[2], *vert, *hstrn[2], *horin[2], *strn, *orin, *stab, | |||
| 252 | seedPos[4], polen, porin[3]; | |||
| 253 | int constrFail; | |||
| 254 | unsigned int dirIdx, lentmp, tidx, oidx, vertNum; | |||
| 255 | ||||
| 256 | if (!( pctx && pts && _seedPos )) { | |||
| 257 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 258 | return 1; | |||
| 259 | } | |||
| 260 | if (!( AIR_EXISTS(scaleDelta)(((int)(!((scaleDelta) - (scaleDelta))))) && scaleDelta > 0.0 )) { | |||
| 261 | biffAddf(PULLpullBiffKey, "%s: need existing scaleDelta > 0 (not %g)", | |||
| 262 | me, scaleDelta); | |||
| 263 | return 1; | |||
| 264 | } | |||
| 265 | if (!( halfScaleWin > 0 )) { | |||
| 266 | biffAddf(PULLpullBiffKey, "%s: need halfScaleWin > 0", me); | |||
| 267 | return 1; | |||
| 268 | } | |||
| 269 | if (!(pctx->constraint)) { | |||
| 270 | biffAddf(PULLpullBiffKey, "%s: given context doesn't have constraint set", me); | |||
| 271 | return 1; | |||
| 272 | } | |||
| 273 | pts->fdim = _pullConstraintDim(pctx); | |||
| 274 | if (0 > pts->fdim) { | |||
| 275 | biffAddf(PULLpullBiffKey, "%s: couldn't learn dimension of feature", me); | |||
| 276 | return 1; | |||
| 277 | } | |||
| 278 | if (pts->fdim == 2 && pctx->flag.zeroZ) { | |||
| 279 | biffAddf(PULLpullBiffKey, "%s: can't have feature dim 2 with zeroZ", me); | |||
| 280 | return 1; | |||
| 281 | } | |||
| 282 | if (!( AIR_EXISTS(orientTestLen)(((int)(!((orientTestLen) - (orientTestLen))))) && orientTestLen >= 0 )) { | |||
| 283 | biffAddf(PULLpullBiffKey, "%s: need non-negative orientTestLen (not %g)\n", | |||
| 284 | me, orientTestLen); | |||
| 285 | return 1; | |||
| 286 | } | |||
| 287 | if (pts->fdim && !orientTestLen) { | |||
| 288 | /* not really an error */ | |||
| 289 | /* | |||
| 290 | fprintf(stderr, "\n\n%s: WARNING: have a %d-D feature, but not " | |||
| 291 | "measuring its orientation\n\n\n", me, pts->fdim); | |||
| 292 | */ | |||
| 293 | } | |||
| 294 | /* The test for "should I measure orientation" | |||
| 295 | is "if (pts->fdim && orientTestLen)" */ | |||
| 296 | if (recordStrength && !pctx->ispec[pullInfoStrength]) { | |||
| 297 | biffAddf(PULLpullBiffKey, "%s: want to record strength but %s not set in context", | |||
| 298 | me, airEnumStr(pullInfo, pullInfoStrength)); | |||
| 299 | return 1; | |||
| 300 | } | |||
| 301 | if (pullConstraintScaleRange(pctx, ssrange)) { | |||
| 302 | biffAddf(PULLpullBiffKey, "%s: trouble getting scale range", me); | |||
| 303 | return 1; | |||
| 304 | } | |||
| 305 | ||||
| 306 | /* re-initialize termination descriptions (in case of trace re-use) */ | |||
| 307 | pts->whyStop[0] = pts->whyStop[1] = pullTraceStopUnknown; | |||
| 308 | pts->whyStopCFail[0] = pts->whyStopCFail[1] = pullConstraintFailUnknown; | |||
| 309 | pts->whyNowhere = pullTraceStopUnknown; | |||
| 310 | pts->whyNowhereCFail = pullConstraintFailUnknown; | |||
| 311 | ||||
| 312 | /* enforce zeroZ */ | |||
| 313 | ELL_4V_COPY(seedPos, _seedPos)((seedPos)[0] = (_seedPos)[0], (seedPos)[1] = (_seedPos)[1], ( seedPos)[2] = (_seedPos)[2], (seedPos)[3] = (_seedPos)[3]); | |||
| 314 | if (pctx->flag.zeroZ) { | |||
| 315 | seedPos[2] = 0.0; | |||
| 316 | } | |||
| 317 | ||||
| 318 | /* save seedPos in any case */ | |||
| 319 | ELL_4V_COPY(pts->seedPos, seedPos)((pts->seedPos)[0] = (seedPos)[0], (pts->seedPos)[1] = ( seedPos)[1], (pts->seedPos)[2] = (seedPos)[2], (pts->seedPos )[3] = (seedPos)[3]); | |||
| 320 | ||||
| 321 | mop = airMopNew(); | |||
| 322 | point = pullPointNew(pctx); /* we'll want to decrement idtagNext later */ | |||
| 323 | airMopAdd(mop, point, (airMopper)pullPointNix, airMopAlways); | |||
| 324 | ||||
| 325 | ELL_4V_COPY(point->pos, seedPos)((point->pos)[0] = (seedPos)[0], (point->pos)[1] = (seedPos )[1], (point->pos)[2] = (seedPos)[2], (point->pos)[3] = (seedPos)[3]); | |||
| 326 | /* | |||
| 327 | if (pctx->verbose) { | |||
| 328 | fprintf(stderr, "%s: trying at seed=(%g,%g,%g,%g)\n", me, | |||
| 329 | seedPos[0], seedPos[1], seedPos[2], seedPos[3]); | |||
| 330 | } | |||
| 331 | */ | |||
| 332 | /* The termination of the trace due to low stablity is handled here | |||
| 333 | (or could be), not by _pullConstraintSatisfy, so we set 0 for the | |||
| 334 | travelMax arg of _pullConstraintSatisfy (no travel limit) */ | |||
| 335 | if (_pullConstraintSatisfy(pctx->task[0], point, | |||
| 336 | 0 /* travmax */, &constrFail)) { | |||
| 337 | biffAddf(PULLpullBiffKey, "%s: constraint sat on seed point", me); | |||
| 338 | airMopError(mop); | |||
| 339 | return 1; | |||
| 340 | } | |||
| 341 | /* | |||
| 342 | if (pctx->verbose) { | |||
| 343 | fprintf(stderr, "%s: seed=(%g,%g,%g,%g) -> %s (%g,%g,%g,%g)\n", me, | |||
| 344 | seedPos[0], seedPos[1], seedPos[2], seedPos[3], | |||
| 345 | constrFail ? "!NO!" : "(yes)", | |||
| 346 | point->pos[0] - seedPos[0], point->pos[1] - seedPos[1], | |||
| 347 | point->pos[2] - seedPos[2], point->pos[3] - seedPos[3]); | |||
| 348 | } | |||
| 349 | */ | |||
| 350 | if (constrFail) { | |||
| 351 | pts->whyNowhere = pullTraceStopConstrFail; | |||
| 352 | airMopOkay(mop); | |||
| 353 | /* pctx->idtagNext -= 1; / * HACK * / */ | |||
| 354 | return 0; | |||
| 355 | } | |||
| 356 | if (point->status & PULL_STATUS_EDGE_BIT(1<< 4)) { | |||
| 357 | pts->whyNowhere = pullTraceStopVolumeEdge; | |||
| 358 | airMopOkay(mop); | |||
| 359 | /* pctx->idtagNext -= 1; / * HACK * / */ | |||
| 360 | return 0; | |||
| 361 | } | |||
| 362 | if (pctx->flag.zeroZ && point->pos[2] != 0) { | |||
| 363 | biffAddf(PULLpullBiffKey, "%s: zeroZ violated (a)", me); | |||
| 364 | airMopError(mop); | |||
| 365 | return 1; | |||
| 366 | } | |||
| 367 | ||||
| 368 | /* else constraint sat worked at seed point; we have work to do */ | |||
| 369 | if (pts->fdim && orientTestLen) { | |||
| 370 | /* learn orientation at seed point */ | |||
| 371 | polen = (orientTestLen | |||
| 372 | *pctx->voxelSizeSpace | |||
| 373 | /* if used, the effect of this this last (unprincipled) factor | |||
| 374 | is gradually increase the test distance with scale | |||
| 375 | *(1 + gageTauOfSig(_pullSigma(pctx, point->pos))) */ ); | |||
| 376 | double pos0[4], dp[4]; | |||
| 377 | int cf; | |||
| 378 | ELL_4V_COPY(pos0, point->pos)((pos0)[0] = (point->pos)[0], (pos0)[1] = (point->pos)[ 1], (pos0)[2] = (point->pos)[2], (pos0)[3] = (point->pos )[3]); | |||
| 379 | ||||
| 380 | if (_pullConstrOrientFind(pctx, point, pts->fdim == 2, | |||
| 381 | polen, NULL((void*)0) /* no initial guess */, porin, &cf)) { | |||
| 382 | biffAddf(PULLpullBiffKey, "%s: trying to find orientation at seed", me); | |||
| 383 | airMopError(mop); | |||
| 384 | return 1; | |||
| 385 | } | |||
| 386 | ELL_4V_SUB(dp, pos0, point->pos)((dp)[0] = (pos0)[0] - (point->pos)[0], (dp)[1] = (pos0)[1 ] - (point->pos)[1], (dp)[2] = (pos0)[2] - (point->pos) [2], (dp)[3] = (pos0)[3] - (point->pos)[3]); | |||
| 387 | /* | |||
| 388 | fprintf(stderr, "!%s: cf = %d (%s)\n", me, cf, airEnumStr(pullConstraintFail, cf)); | |||
| 389 | fprintf(stderr, "!%s: (fdim=%u) pos=[%g,%g,%g,%g] polen=%g porin=[%g,%g,%g] |%g|\n", | |||
| 390 | me, pts->fdim, | |||
| 391 | point->pos[0], point->pos[1], point->pos[2], point->pos[3], | |||
| 392 | polen, porin[0], porin[1], porin[2], ELL_4V_LEN(dp)); | |||
| 393 | */ | |||
| 394 | if (cf) { | |||
| 395 | pts->whyNowhere = pullTraceStopConstrFail; | |||
| 396 | pts->whyNowhereCFail = cf; | |||
| 397 | airMopOkay(mop); | |||
| 398 | /* pctx->idtagNext -= 1; / * HACK * / */ | |||
| 399 | return 0; | |||
| 400 | } | |||
| 401 | } else { | |||
| 402 | /* either feature is 0D points, or don't care about orientation */ | |||
| 403 | polen = AIR_NAN(airFloatQNaN.f); | |||
| 404 | ELL_3V_SET(porin, AIR_NAN, AIR_NAN, AIR_NAN)((porin)[0] = ((airFloatQNaN.f)), (porin)[1] = ((airFloatQNaN .f)), (porin)[2] = ((airFloatQNaN.f))); | |||
| 405 | } | |||
| 406 | ||||
| 407 | for (dirIdx=0; dirIdx<2; dirIdx++) { | |||
| 408 | trceArr[dirIdx] = airArrayNew((void**)(trce + dirIdx), NULL((void*)0), | |||
| 409 | 4*sizeof(double), arrIncr); | |||
| 410 | airMopAdd(mop, trceArr[dirIdx], (airMopper)airArrayNuke, airMopAlways); | |||
| 411 | if (recordStrength) { | |||
| 412 | hstrnArr[dirIdx] = airArrayNew((void**)(hstrn + dirIdx), NULL((void*)0), | |||
| 413 | sizeof(double), arrIncr); | |||
| 414 | airMopAdd(mop, hstrnArr[dirIdx], (airMopper)airArrayNuke, airMopAlways); | |||
| 415 | } else { | |||
| 416 | hstrnArr[dirIdx] = NULL((void*)0); | |||
| 417 | hstrn[dirIdx] = NULL((void*)0); | |||
| 418 | } | |||
| 419 | if (pts->fdim && orientTestLen) { | |||
| 420 | horinArr[dirIdx] = airArrayNew((void**)(horin + dirIdx), NULL((void*)0), | |||
| 421 | 3*sizeof(double), arrIncr); | |||
| 422 | airMopAdd(mop, horinArr[dirIdx], (airMopper)airArrayNuke, airMopAlways); | |||
| 423 | } else { | |||
| 424 | horinArr[dirIdx] = NULL((void*)0); | |||
| 425 | horin[dirIdx] = NULL((void*)0); | |||
| 426 | } | |||
| 427 | } | |||
| 428 | for (dirIdx=0; dirIdx<2; dirIdx++) { | |||
| 429 | unsigned int step; | |||
| 430 | double dscl; | |||
| 431 | dscl = (!dirIdx ? -1 : +1)*scaleDelta; | |||
| 432 | step = 0; | |||
| 433 | while (1) { | |||
| 434 | if (!step) { | |||
| 435 | /* first step in both directions requires special tricks */ | |||
| 436 | if (0 == dirIdx) { | |||
| 437 | /* this is done once, at the very start */ | |||
| 438 | /* save constraint sat of seed point */ | |||
| 439 | tidx = airArrayLenIncr(trceArr[0], 1); | |||
| 440 | ELL_4V_COPY(trce[0] + 4*tidx, point->pos)((trce[0] + 4*tidx)[0] = (point->pos)[0], (trce[0] + 4*tidx )[1] = (point->pos)[1], (trce[0] + 4*tidx)[2] = (point-> pos)[2], (trce[0] + 4*tidx)[3] = (point->pos)[3]); | |||
| 441 | if (recordStrength) { | |||
| 442 | airArrayLenIncr(hstrnArr[0], 1); | |||
| 443 | hstrn[0][0] = pullPointScalar(pctx, point, pullInfoStrength, | |||
| 444 | NULL((void*)0), NULL((void*)0)); | |||
| 445 | } | |||
| 446 | if (pts->fdim && orientTestLen) { | |||
| 447 | airArrayLenIncr(horinArr[0], 1); | |||
| 448 | ELL_3V_COPY(horin[0] + 3*0, porin)((horin[0] + 3*0)[0] = (porin)[0], (horin[0] + 3*0)[1] = (porin )[1], (horin[0] + 3*0)[2] = (porin)[2]); | |||
| 449 | } | |||
| 450 | } else { | |||
| 451 | /* re-set position from constraint sat of seed pos */ | |||
| 452 | ELL_4V_COPY(point->pos, trce[0] + 4*0)((point->pos)[0] = (trce[0] + 4*0)[0], (point->pos)[1] = (trce[0] + 4*0)[1], (point->pos)[2] = (trce[0] + 4*0)[2], (point->pos)[3] = (trce[0] + 4*0)[3]); | |||
| 453 | if (pts->fdim && orientTestLen) { | |||
| 454 | ELL_3V_COPY(porin, horin[0] + 3*0)((porin)[0] = (horin[0] + 3*0)[0], (porin)[1] = (horin[0] + 3 *0)[1], (porin)[2] = (horin[0] + 3*0)[2]); | |||
| 455 | } | |||
| 456 | } | |||
| 457 | } | |||
| 458 | /* nudge position along scale */ | |||
| 459 | point->pos[3] += dscl; | |||
| 460 | if (!AIR_IN_OP(ssrange[0], point->pos[3], ssrange[1])((ssrange[0]) < (point->pos[3]) && (point->pos [3]) < (ssrange[1]))) { | |||
| 461 | /* if we've stepped outside the range of scale for the volume | |||
| 462 | containing the constraint manifold, we're done */ | |||
| 463 | pts->whyStop[dirIdx] = pullTraceStopBounds; | |||
| 464 | break; | |||
| 465 | } | |||
| 466 | if (AIR_ABS(point->pos[3] - seedPos[3])((point->pos[3] - seedPos[3]) > 0.0f ? (point->pos[3 ] - seedPos[3]) : -(point->pos[3] - seedPos[3])) > halfScaleWin) { | |||
| 467 | /* we've moved along scale as far as allowed */ | |||
| 468 | pts->whyStop[dirIdx] = pullTraceStopLength; | |||
| 469 | break; | |||
| 470 | } | |||
| 471 | /* re-assert constraint */ | |||
| 472 | /* | |||
| 473 | fprintf(stderr, "%s(%u): pos = %g %g %g %g.... \n", me, | |||
| 474 | point->idtag, point->pos[0], point->pos[1], | |||
| 475 | point->pos[2], point->pos[3]); | |||
| 476 | */ | |||
| 477 | if (_pullConstraintSatisfy(pctx->task[0], point, | |||
| 478 | 0 /* travmax */, &constrFail)) { | |||
| 479 | biffAddf(PULLpullBiffKey, "%s: dir %u, step %u", me, dirIdx, step); | |||
| 480 | airMopError(mop); | |||
| 481 | return 1; | |||
| 482 | } | |||
| 483 | /* | |||
| 484 | if (pctx->verbose) { | |||
| 485 | fprintf(stderr, "%s(%u): ... %s(%d); pos = %g %g %g %g\n", me, | |||
| 486 | point->idtag, | |||
| 487 | constrFail ? "FAIL" : "(ok)", | |||
| 488 | constrFail, point->pos[0], point->pos[1], | |||
| 489 | point->pos[2], point->pos[3]); | |||
| 490 | } | |||
| 491 | */ | |||
| 492 | if (point->status & PULL_STATUS_EDGE_BIT(1<< 4)) { | |||
| 493 | pts->whyStop[dirIdx] = pullTraceStopVolumeEdge; | |||
| 494 | break; | |||
| 495 | } | |||
| 496 | if (constrFail) { | |||
| 497 | /* constraint sat failed; no error, we're just done | |||
| 498 | with stepping for this direction */ | |||
| 499 | pts->whyStop[dirIdx] = pullTraceStopConstrFail; | |||
| 500 | pts->whyStopCFail[dirIdx] = constrFail; | |||
| 501 | break; | |||
| 502 | } | |||
| 503 | if (pctx->flag.zeroZ && point->pos[2] != 0) { | |||
| 504 | biffAddf(PULLpullBiffKey, "%s: zeroZ violated (b)", me); | |||
| 505 | airMopError(mop); | |||
| 506 | return 1; | |||
| 507 | } | |||
| 508 | if (pts->fdim && orientTestLen) { | |||
| 509 | if (_pullConstrOrientFind(pctx, point, pts->fdim == 2, | |||
| 510 | polen, porin, porin, &constrFail)) { | |||
| 511 | biffAddf(PULLpullBiffKey, "%s: at dir %u, step %u", me, dirIdx, step); | |||
| 512 | airMopError(mop); | |||
| 513 | return 1; | |||
| 514 | } | |||
| 515 | } | |||
| 516 | if (trceArr[dirIdx]->len >= 2) { | |||
| 517 | /* see if we're moving too fast, by comparing with previous point */ | |||
| 518 | /* actually, screw that */ | |||
| 519 | } | |||
| 520 | /* else save new point on trace */ | |||
| 521 | tidx = airArrayLenIncr(trceArr[dirIdx], 1); | |||
| 522 | ELL_4V_COPY(trce[dirIdx] + 4*tidx, point->pos)((trce[dirIdx] + 4*tidx)[0] = (point->pos)[0], (trce[dirIdx ] + 4*tidx)[1] = (point->pos)[1], (trce[dirIdx] + 4*tidx)[ 2] = (point->pos)[2], (trce[dirIdx] + 4*tidx)[3] = (point-> pos)[3]); | |||
| 523 | if (recordStrength) { | |||
| 524 | tidx = airArrayLenIncr(hstrnArr[dirIdx], 1); | |||
| 525 | hstrn[dirIdx][tidx] = pullPointScalar(pctx, point, pullInfoStrength, | |||
| 526 | NULL((void*)0), NULL((void*)0)); | |||
| 527 | } | |||
| 528 | if (pts->fdim && orientTestLen) { | |||
| 529 | tidx = airArrayLenIncr(horinArr[dirIdx], 1); | |||
| 530 | ELL_3V_COPY(horin[dirIdx] + 3*tidx, porin)((horin[dirIdx] + 3*tidx)[0] = (porin)[0], (horin[dirIdx] + 3 *tidx)[1] = (porin)[1], (horin[dirIdx] + 3*tidx)[2] = (porin) [2]); | |||
| 531 | } | |||
| 532 | step++; | |||
| 533 | } | |||
| 534 | } | |||
| 535 | ||||
| 536 | /* transfer trace halves to pts->nvert */ | |||
| 537 | vertNum = trceArr[0]->len + trceArr[1]->len; | |||
| 538 | if (0 == vertNum || 1 == vertNum || 2 == vertNum) { | |||
| 539 | pts->whyNowhere = pullTraceStopStub; | |||
| 540 | airMopOkay(mop); | |||
| 541 | /* pctx->idtagNext -= 1; / * HACK * / */ | |||
| 542 | return 0; | |||
| 543 | } | |||
| 544 | ||||
| 545 | if (nrrdMaybeAlloc_va(pts->nvert, nrrdTypeDouble, 2, | |||
| 546 | AIR_CAST(size_t, 4)((size_t)(4)), | |||
| 547 | AIR_CAST(size_t, vertNum)((size_t)(vertNum))) | |||
| 548 | || nrrdMaybeAlloc_va(pts->nstab, nrrdTypeDouble, 2, | |||
| 549 | AIR_CAST(size_t, 2)((size_t)(2)), | |||
| 550 | AIR_CAST(size_t, vertNum)((size_t)(vertNum)))) { | |||
| 551 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: allocating output", me); | |||
| 552 | airMopError(mop); | |||
| 553 | return 1; | |||
| 554 | } | |||
| 555 | if (recordStrength) { | |||
| 556 | /* doing slicing is a simple form of allocation */ | |||
| 557 | if (nrrdSlice(pts->nstrn, pts->nvert, 0 /* axis */, 0 /* pos */)) { | |||
| 558 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: allocating strength output", me); | |||
| 559 | airMopError(mop); | |||
| 560 | return 1; | |||
| 561 | } | |||
| 562 | } | |||
| 563 | if (pts->fdim && orientTestLen) { | |||
| 564 | /* cropping just to allocate */ | |||
| 565 | size_t cmin[2] = {0, 0}, cmax[2] = {2, pts->nvert->axis[1].size-1}; | |||
| 566 | if (nrrdCrop(pts->norin, pts->nvert, cmin, cmax)) { | |||
| 567 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: allocating orientation output", me); | |||
| 568 | airMopError(mop); | |||
| 569 | return 1; | |||
| 570 | } | |||
| 571 | } | |||
| 572 | vert = AIR_CAST(double *, pts->nvert->data)((double *)(pts->nvert->data)); | |||
| 573 | if (recordStrength) { | |||
| 574 | strn = AIR_CAST(double *, pts->nstrn->data)((double *)(pts->nstrn->data)); | |||
| 575 | } else { | |||
| 576 | strn = NULL((void*)0); | |||
| 577 | } | |||
| 578 | if (pts->fdim && orientTestLen) { | |||
| 579 | orin = AIR_CAST(double *, pts->norin->data)((double *)(pts->norin->data)); | |||
| 580 | } else { | |||
| 581 | orin = NULL((void*)0); | |||
| 582 | } | |||
| 583 | stab = AIR_CAST(double *, pts->nstab->data)((double *)(pts->nstab->data)); | |||
| 584 | lentmp = trceArr[0]->len; | |||
| 585 | oidx = 0; | |||
| 586 | for (tidx=0; tidx<lentmp; tidx++) { | |||
| 587 | ELL_4V_COPY(vert + 4*oidx, trce[0] + 4*(lentmp - 1 - tidx))((vert + 4*oidx)[0] = (trce[0] + 4*(lentmp - 1 - tidx))[0], ( vert + 4*oidx)[1] = (trce[0] + 4*(lentmp - 1 - tidx))[1], (vert + 4*oidx)[2] = (trce[0] + 4*(lentmp - 1 - tidx))[2], (vert + 4*oidx)[3] = (trce[0] + 4*(lentmp - 1 - tidx))[3]); | |||
| 588 | if (strn) { | |||
| 589 | strn[oidx] = hstrn[0][lentmp - 1 - tidx]; | |||
| 590 | } | |||
| 591 | if (orin) { | |||
| 592 | ELL_3V_COPY(orin + 3*oidx, horin[0] + 3*(lentmp - 1 - tidx))((orin + 3*oidx)[0] = (horin[0] + 3*(lentmp - 1 - tidx))[0], ( orin + 3*oidx)[1] = (horin[0] + 3*(lentmp - 1 - tidx))[1], (orin + 3*oidx)[2] = (horin[0] + 3*(lentmp - 1 - tidx))[2]); | |||
| 593 | } | |||
| 594 | oidx++; | |||
| 595 | } | |||
| 596 | /* the last index written to (before oidx++) was the seed index */ | |||
| 597 | pts->seedIdx = oidx-1; | |||
| 598 | lentmp = trceArr[1]->len; | |||
| 599 | for (tidx=0; tidx<lentmp; tidx++) { | |||
| 600 | ELL_4V_COPY(vert + 4*oidx, trce[1] + 4*tidx)((vert + 4*oidx)[0] = (trce[1] + 4*tidx)[0], (vert + 4*oidx)[ 1] = (trce[1] + 4*tidx)[1], (vert + 4*oidx)[2] = (trce[1] + 4 *tidx)[2], (vert + 4*oidx)[3] = (trce[1] + 4*tidx)[3]); | |||
| 601 | if (strn) { | |||
| 602 | strn[oidx] = hstrn[1][tidx]; | |||
| 603 | } | |||
| 604 | if (orin) { | |||
| 605 | ELL_3V_COPY(orin + 3*oidx, horin[1] + 3*tidx)((orin + 3*oidx)[0] = (horin[1] + 3*tidx)[0], (orin + 3*oidx) [1] = (horin[1] + 3*tidx)[1], (orin + 3*oidx)[2] = (horin[1] + 3*tidx)[2]); | |||
| 606 | } | |||
| 607 | oidx++; | |||
| 608 | } | |||
| 609 | lentmp = pts->nstab->axis[1].size; | |||
| 610 | if (1 == lentmp) { | |||
| 611 | stab[0 + 2*0] = 0.0; | |||
| 612 | stab[1 + 2*0] = 0.0; | |||
| 613 | } else { | |||
| 614 | for (tidx=0; tidx<lentmp; tidx++) { | |||
| 615 | double *pA, *pB, *p0, *p1, *p2, *rA, *rB, *r0=NULL((void*)0), *r1=NULL((void*)0), *r2=NULL((void*)0); | |||
| 616 | p0 = vert + 4*(tidx-1); | |||
| 617 | p1 = vert + 4*tidx; | |||
| 618 | p2 = vert + 4*(tidx+1); | |||
| 619 | if (orin) { | |||
| 620 | r0 = orin + 3*(tidx-1); | |||
| 621 | r1 = orin + 3*tidx; | |||
| 622 | r2 = orin + 3*(tidx+1); | |||
| 623 | } | |||
| 624 | if (!tidx) { | |||
| 625 | /* first */ | |||
| 626 | pA = p1; rA = r1; | |||
| 627 | pB = p2; rB = r2; | |||
| 628 | } else if (tidx < lentmp-1) { | |||
| 629 | /* middle */ | |||
| 630 | pA = p0; rA = r0; | |||
| 631 | pB = p2; rB = r2; | |||
| 632 | } else { | |||
| 633 | /* last */ | |||
| 634 | pA = p0; rA = r0; | |||
| 635 | pB = p1; rB = r1; | |||
| 636 | } | |||
| 637 | pullTraceStability(stab + 0 + 2*tidx, stab + 1 + 2*tidx, | |||
| 638 | pA, pB, rA, rB, 0.5 /* sigma0 */, pctx); | |||
| 639 | } | |||
| 640 | } | |||
| 641 | ||||
| 642 | airMopOkay(mop); | |||
| 643 | /* pctx->idtagNext -= 1; / * HACK * / */ | |||
| 644 | return 0; | |||
| 645 | } | |||
| 646 | ||||
| 647 | typedef union { | |||
| 648 | pullTrace ***trace; | |||
| 649 | void **v; | |||
| 650 | } blahblahUnion; | |||
| 651 | ||||
| 652 | pullTraceMulti * | |||
| 653 | pullTraceMultiNew(void) { | |||
| 654 | /* static const char me[]="pullTraceMultiNew"; */ | |||
| 655 | pullTraceMulti *ret; | |||
| 656 | blahblahUnion bbu; | |||
| 657 | ||||
| 658 | ret = AIR_CALLOC(1, pullTraceMulti)(pullTraceMulti*)(calloc((1), sizeof(pullTraceMulti))); | |||
| 659 | if (ret) { | |||
| 660 | ret->trace = NULL((void*)0); | |||
| 661 | ret->traceNum = 0; | |||
| 662 | ret->traceArr = airArrayNew((bbu.trace = &(ret->trace), bbu.v), | |||
| 663 | &(ret->traceNum), sizeof(pullTrace*), | |||
| 664 | _PULL_TRACE_MULTI_INCR1024); | |||
| 665 | airArrayPointerCB(ret->traceArr, | |||
| 666 | NULL((void*)0), /* because we get handed pullTrace structs | |||
| 667 | that have already been allocated | |||
| 668 | (and then we own them) */ | |||
| 669 | (void *(*)(void *))pullTraceNix); | |||
| 670 | } | |||
| 671 | return ret; | |||
| 672 | } | |||
| 673 | ||||
| 674 | int | |||
| 675 | pullTraceMultiAdd(pullTraceMulti *mtrc, pullTrace *trc, int *addedP) { | |||
| 676 | static const char me[]="pullTraceMultiAdd"; | |||
| 677 | unsigned int indx; | |||
| 678 | ||||
| 679 | if (!(mtrc && trc && addedP)) { | |||
| 680 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 681 | return 1; | |||
| 682 | } | |||
| 683 | if (!(trc->nvert->data && trc->nvert->axis[1].size >= 3)) { | |||
| 684 | /* for now getting a stub trace is not an error | |||
| 685 | biffAddf(PULL, "%s: got stub trace", me); | |||
| 686 | return 1; */ | |||
| 687 | *addedP = AIR_FALSE0; | |||
| 688 | return 0; | |||
| 689 | } | |||
| 690 | if (!(trc->nstab->data | |||
| 691 | && trc->nstab->axis[1].size == trc->nvert->axis[1].size)) { | |||
| 692 | biffAddf(PULLpullBiffKey, "%s: stab data inconsistent", me); | |||
| 693 | return 1; | |||
| 694 | } | |||
| 695 | *addedP = AIR_TRUE1; | |||
| 696 | indx = airArrayLenIncr(mtrc->traceArr, 1); | |||
| 697 | if (!mtrc->trace) { | |||
| 698 | biffAddf(PULLpullBiffKey, "%s: alloc error", me); | |||
| 699 | return 1; | |||
| 700 | } | |||
| 701 | mtrc->trace[indx] = trc; | |||
| 702 | return 0; | |||
| 703 | } | |||
| 704 | ||||
| 705 | int | |||
| 706 | pullTraceMultiPlotAdd(Nrrd *nplot, const pullTraceMulti *mtrc, | |||
| 707 | const Nrrd *nfilt, int strengthUse, | |||
| 708 | int smooth, int flatWght, | |||
| 709 | unsigned int trcIdxMin, unsigned int trcNum, | |||
| 710 | Nrrd *nmaskedpos, | |||
| 711 | const Nrrd *nmask) { | |||
| 712 | static const char me[]="pullTraceMultiPlotAdd"; | |||
| 713 | double ssRange[2], vRange[2], *plot; | |||
| 714 | unsigned int sizeS, sizeT, trcIdx, trcIdxMax; | |||
| 715 | int *filt; | |||
| 716 | airArray *mop; | |||
| 717 | Nrrd *nsmst; /* smoothed stability */ | |||
| 718 | airArray *mposArr; | |||
| 719 | double *mpos; | |||
| 720 | unsigned char *mask; | |||
| 721 | ||||
| 722 | if (!(nplot && mtrc)) { | |||
| 723 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 724 | return 1; | |||
| 725 | } | |||
| 726 | if (nrrdCheck(nplot)) { | |||
| 727 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: trouble with nplot", me); | |||
| 728 | return 1; | |||
| 729 | } | |||
| 730 | if (nfilt) { | |||
| 731 | if (nrrdCheck(nfilt)) { | |||
| 732 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: trouble with nfilt", me); | |||
| 733 | return 1; | |||
| 734 | } | |||
| 735 | if (!(1 == nfilt->dim && nrrdTypeInt == nfilt->type)) { | |||
| 736 | biffAddf(PULLpullBiffKey, "%s: didn't get 1-D array of %s (got %u-D of %s)", me, | |||
| 737 | airEnumStr(nrrdType, nrrdTypeInt), nfilt->dim, | |||
| 738 | airEnumStr(nrrdType, nfilt->type)); | |||
| 739 | return 1; | |||
| 740 | } | |||
| 741 | } | |||
| 742 | if (!(2 == nplot->dim && nrrdTypeDouble == nplot->type)) { | |||
| 743 | biffAddf(PULLpullBiffKey, "%s: didn't get 2-D array of %s (got %u-D of %s)", me, | |||
| 744 | airEnumStr(nrrdType, nrrdTypeDouble), nplot->dim, | |||
| 745 | airEnumStr(nrrdType, nplot->type)); | |||
| 746 | return 1; | |||
| 747 | } | |||
| 748 | if (!(trcIdxMin < mtrc->traceNum)) { | |||
| 749 | biffAddf(PULLpullBiffKey, "%s: trcIdxMin %u not < traceNum %u", me, | |||
| 750 | trcIdxMin, mtrc->traceNum); | |||
| 751 | return 1; | |||
| 752 | } | |||
| 753 | if (trcNum) { | |||
| 754 | trcIdxMax = trcIdxMin + trcNum-1; | |||
| 755 | if (!(trcIdxMax < mtrc->traceNum)) { | |||
| 756 | biffAddf(PULLpullBiffKey, "%s: trcIdxMax %u = %u+%u-1 not < traceNum %u", me, | |||
| 757 | trcIdxMax, trcIdxMin, trcNum, mtrc->traceNum); | |||
| 758 | return 1; | |||
| 759 | } | |||
| 760 | } else { | |||
| 761 | trcIdxMax = mtrc->traceNum-1; | |||
| 762 | } | |||
| 763 | if (nmaskedpos || nmask) { | |||
| 764 | if (!( nmaskedpos && nmask )) { | |||
| 765 | biffAddf(PULLpullBiffKey, "%s: need either both or neither of nmaskedpos (%p)" | |||
| 766 | "and nmask (%p)", me, nmaskedpos, nmask); | |||
| 767 | return 1; | |||
| 768 | } | |||
| 769 | if (!( 2 == nmask->dim | |||
| 770 | && nrrdTypeUChar == nmask->type | |||
| 771 | && nplot->axis[0].size == nmask->axis[0].size | |||
| 772 | && nplot->axis[1].size == nmask->axis[1].size )) { | |||
| 773 | biffAddf(PULLpullBiffKey, "%s: got trace mask but wanted " | |||
| 774 | "2-D %s %u-by-%u (not %u-D %s %u-by-%u)\n", me, | |||
| 775 | airEnumStr(nrrdType, nrrdTypeUChar), | |||
| 776 | AIR_CAST(unsigned int, nplot->axis[0].size)((unsigned int)(nplot->axis[0].size)), | |||
| 777 | AIR_CAST(unsigned int, nplot->axis[1].size)((unsigned int)(nplot->axis[1].size)), | |||
| 778 | nmask->dim, | |||
| 779 | airEnumStr(nrrdType, nmask->type), | |||
| 780 | AIR_CAST(unsigned int, nmask->axis[0].size)((unsigned int)(nmask->axis[0].size)), | |||
| 781 | AIR_CAST(unsigned int, nmask->axis[1].size)((unsigned int)(nmask->axis[1].size))); | |||
| 782 | return 1; | |||
| 783 | } | |||
| 784 | mask = AIR_CAST(unsigned char *, nmask->data)((unsigned char *)(nmask->data)); | |||
| 785 | } else { | |||
| 786 | mask = NULL((void*)0); | |||
| 787 | } | |||
| 788 | ||||
| 789 | ssRange[0] = nplot->axis[0].min; | |||
| 790 | ssRange[1] = nplot->axis[0].max; | |||
| 791 | vRange[0] = nplot->axis[1].min; | |||
| 792 | vRange[1] = nplot->axis[1].max; | |||
| 793 | if (!( AIR_EXISTS(ssRange[0])(((int)(!((ssRange[0]) - (ssRange[0]))))) && AIR_EXISTS(ssRange[1])(((int)(!((ssRange[1]) - (ssRange[1]))))) && | |||
| 794 | AIR_EXISTS(vRange[0])(((int)(!((vRange[0]) - (vRange[0]))))) && AIR_EXISTS(vRange[1])(((int)(!((vRange[1]) - (vRange[1]))))) )) { | |||
| 795 | biffAddf(PULLpullBiffKey, "%s: need both axis 0 (%g,%g) and 1 (%g,%g) min,max", me, | |||
| 796 | ssRange[0], ssRange[1], vRange[0], vRange[1]); | |||
| 797 | return 1; | |||
| 798 | } | |||
| 799 | if (1 != vRange[0]) { | |||
| 800 | biffAddf(PULLpullBiffKey, "%s: expected vRange[0] == 1 not %g", me, vRange[0]); | |||
| 801 | return 1; | |||
| 802 | } | |||
| 803 | mop = airMopNew(); | |||
| 804 | ||||
| 805 | mpos = NULL((void*)0); | |||
| 806 | if (nmaskedpos && nmask) { | |||
| 807 | nrrdEmpty(nmaskedpos); | |||
| 808 | mposArr = airArrayNew((void**)(&mpos), NULL((void*)0), | |||
| 809 | 4*sizeof(double), 512 /* HEY */); | |||
| 810 | } else { | |||
| 811 | mposArr = NULL((void*)0); | |||
| 812 | } | |||
| 813 | ||||
| 814 | nsmst = nrrdNew(); | |||
| 815 | airMopAdd(mop, nsmst, (airMopper)nrrdNuke, airMopAlways); | |||
| 816 | plot = AIR_CAST(double *, nplot->data)((double *)(nplot->data)); | |||
| 817 | filt = (nfilt | |||
| 818 | ? AIR_CAST(int *, nfilt->data)((int *)(nfilt->data)) | |||
| 819 | : NULL((void*)0)); | |||
| 820 | sizeS = AIR_CAST(unsigned int, nplot->axis[0].size)((unsigned int)(nplot->axis[0].size)); | |||
| 821 | sizeT = AIR_CAST(unsigned int, nplot->axis[1].size)((unsigned int)(nplot->axis[1].size)); | |||
| 822 | for (trcIdx=trcIdxMin; trcIdx<=trcIdxMax; trcIdx++) { | |||
| 823 | int pntIdx, pntNum; | |||
| 824 | const pullTrace *trc; | |||
| 825 | const double *vert, *stab, *strn; | |||
| 826 | unsigned int maskInCount; | |||
| 827 | double maskInPos[4]; | |||
| 828 | ||||
| 829 | if (filt && !filt[trcIdx]) { | |||
| 830 | continue; | |||
| 831 | } | |||
| 832 | trc = mtrc->trace[trcIdx]; | |||
| 833 | if (pullTraceStopStub == trc->whyNowhere) { | |||
| 834 | continue; | |||
| 835 | } | |||
| 836 | if (strengthUse && !(trc->nstrn && trc->nstrn->data)) { | |||
| 837 | biffAddf(PULLpullBiffKey, "%s: requesting strength-based weighting, but don't have " | |||
| 838 | "strength info in trace %u", me, trcIdx); | |||
| 839 | airMopError(mop); return 1; | |||
| 840 | } | |||
| 841 | pntNum = AIR_CAST(int, trc->nvert->axis[1].size)((int)(trc->nvert->axis[1].size)); | |||
| 842 | vert = AIR_CAST(double *, trc->nvert->data)((double *)(trc->nvert->data)); | |||
| 843 | stab = AIR_CAST(double *, trc->nstab->data)((double *)(trc->nstab->data)); | |||
| 844 | if (smooth > 0) { | |||
| 845 | double *smst; | |||
| 846 | if (nrrdCopy(nsmst, trc->nstab)) { | |||
| 847 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: trouble w/ trace %u", me, trcIdx); | |||
| 848 | airMopError(mop); return 1; | |||
| 849 | } | |||
| 850 | smst = AIR_CAST(double *, nsmst->data)((double *)(nsmst->data)); | |||
| 851 | for (pntIdx=0; pntIdx<pntNum; pntIdx++) { | |||
| 852 | int ii, jj; | |||
| 853 | double ss, ww, ws; | |||
| 854 | ss = ws = 0; | |||
| 855 | for (jj=-smooth; jj<=smooth; jj++) { | |||
| 856 | ii = pntIdx+jj; | |||
| 857 | ii = AIR_CLAMP(0, ii, pntNum-1)((ii) < (0) ? (0) : ((ii) > (pntNum-1) ? (pntNum-1) : ( ii))); | |||
| 858 | ww = nrrdKernelBSpline3->eval1_d(AIR_AFFINE(-smooth-1, jj,( ((double)(2)-(-2))*((double)(jj)-(-smooth-1)) / ((double)(smooth +1)-(-smooth-1)) + (-2)) | |||
| 859 | smooth+1,-2,2)( ((double)(2)-(-2))*((double)(jj)-(-smooth-1)) / ((double)(smooth +1)-(-smooth-1)) + (-2)), NULL((void*)0)); | |||
| 860 | ws += ww; | |||
| 861 | ss += ww*stab[0 + 2*ii]*stab[1 + 2*ii]; | |||
| 862 | } | |||
| 863 | smst[pntIdx] = ss/ws; | |||
| 864 | } | |||
| 865 | /* now redirect stab */ | |||
| 866 | stab = smst; | |||
| 867 | } | |||
| 868 | strn = AIR_CAST(double *, (strengthUse && trc->nstrn((double *)((strengthUse && trc->nstrn ? trc->nstrn ->data : ((void*)0)))) | |||
| 869 | ? trc->nstrn->data : NULL))((double *)((strengthUse && trc->nstrn ? trc->nstrn ->data : ((void*)0)))); | |||
| 870 | /* would be nice to get some graphical indication of this */ | |||
| 871 | fprintf(stderr__stderrp, "!%s: trace %u in [%u,%u]: %u points; stops = %s(%s) | %s(%s)\n", | |||
| 872 | me, trcIdx, trcIdxMin, trcIdxMax, pntNum, | |||
| 873 | airEnumStr(pullTraceStop, trc->whyStop[0]), | |||
| 874 | (pullTraceStopConstrFail == trc->whyStop[0] | |||
| 875 | ? airEnumStr(pullConstraintFail, trc->whyStopCFail[0]) | |||
| 876 | : ""), | |||
| 877 | airEnumStr(pullTraceStop, trc->whyStop[1]), | |||
| 878 | (pullTraceStopConstrFail == trc->whyStop[1] | |||
| 879 | ? airEnumStr(pullConstraintFail, trc->whyStopCFail[1]) | |||
| 880 | : "")); | |||
| 881 | /* */ | |||
| 882 | ||||
| 883 | if (mask) { | |||
| 884 | maskInCount = 0; | |||
| 885 | ELL_4V_SET(maskInPos, 0, 0, 0, 0)((maskInPos)[0] = (0), (maskInPos)[1] = (0), (maskInPos)[2] = (0), (maskInPos)[3] = (0)); | |||
| 886 | } | |||
| 887 | for (pntIdx=0; pntIdx<pntNum; pntIdx++) { | |||
| 888 | const double *pp; | |||
| 889 | double add, ww; | |||
| 890 | unsigned int sidx, vidx; | |||
| 891 | pp = vert + 4*pntIdx; | |||
| 892 | if (!(AIR_IN_OP(ssRange[0], pp[3], ssRange[1])((ssRange[0]) < (pp[3]) && (pp[3]) < (ssRange[1 ])))) { | |||
| 893 | continue; | |||
| 894 | } | |||
| 895 | if (flatWght > 0) { | |||
| 896 | if (!pntIdx || pntIdx == pntNum-1) { | |||
| 897 | continue; | |||
| 898 | } | |||
| 899 | } else if (flatWght < 0) { | |||
| 900 | /* HACK: only show the seed point */ | |||
| 901 | if (AIR_CAST(unsigned int, pntIdx)((unsigned int)(pntIdx)) != trc->seedIdx) { | |||
| 902 | continue; | |||
| 903 | } | |||
| 904 | } | |||
| 905 | sidx = airIndex(ssRange[0], pp[3], ssRange[1], sizeS); | |||
| 906 | vidx = airIndexClamp(1, stab[0 + 2*pntIdx]*stab[1 + 2*pntIdx], 0, sizeT); | |||
| 907 | add = strn ? strn[pntIdx] : 1; | |||
| 908 | if (flatWght > 0) { | |||
| 909 | double dx = ( ((vert + 4*(pntIdx+1))[3] - (vert + 4*(pntIdx-1))[3]) | |||
| 910 | / (ssRange[1] - ssRange[0]) ); | |||
| 911 | /* | |||
| 912 | double dx = ( ((vert + 4*(pntIdx+1))[3] - pp[3]) | |||
| 913 | / (ssRange[1] - ssRange[0]) ); | |||
| 914 | */ | |||
| 915 | double dy = (stab[0 + 2*(pntIdx+1)]*stab[1 + 2*(pntIdx+1)] | |||
| 916 | - stab[0 + 2*(pntIdx-1)]*stab[1 + 2*(pntIdx-1)]); | |||
| 917 | ww = dx/sqrt(dx*dx + dy*dy); | |||
| 918 | } else { | |||
| 919 | ww = 1; | |||
| 920 | } | |||
| 921 | plot[sidx + sizeS*vidx] += AIR_MAX(0, ww*add)((0) > (ww*add) ? (0) : (ww*add)); | |||
| 922 | if (mask && mask[sidx + sizeS*vidx] > 200) { | |||
| 923 | ELL_4V_ADD2(maskInPos, maskInPos, pp)((maskInPos)[0] = (maskInPos)[0] + (pp)[0], (maskInPos)[1] = ( maskInPos)[1] + (pp)[1], (maskInPos)[2] = (maskInPos)[2] + (pp )[2], (maskInPos)[3] = (maskInPos)[3] + (pp)[3]); | |||
| 924 | maskInCount ++; | |||
| 925 | } | |||
| 926 | } | |||
| 927 | if (mask && maskInCount) { | |||
| 928 | unsigned int mpi = airArrayLenIncr(mposArr, 1); | |||
| 929 | ELL_4V_SCALE(mpos + 4*mpi, 1.0/maskInCount, maskInPos)((mpos + 4*mpi)[0] = (maskInPos)[0]*1.0/maskInCount, (mpos + 4 *mpi)[1] = (maskInPos)[1]*1.0/maskInCount, (mpos + 4*mpi)[2] = (maskInPos)[2]*1.0/maskInCount, (mpos + 4*mpi)[3] = (maskInPos )[3]*1.0/maskInCount); | |||
| 930 | } | |||
| 931 | } | |||
| 932 | if (mask && mposArr->len) { | |||
| 933 | if (nrrdMaybeAlloc_va(nmaskedpos, nrrdTypeDouble, 2, | |||
| 934 | AIR_CAST(size_t, 4)((size_t)(4)), | |||
| 935 | AIR_CAST(size_t, mposArr->len)((size_t)(mposArr->len)))) { | |||
| 936 | biffAddf(PULLpullBiffKey, "%s: couldn't allocate masked pos", me); | |||
| 937 | airMopError(mop); return 1; | |||
| 938 | } | |||
| 939 | memcpy(nmaskedpos->data, mposArr->data,__builtin___memcpy_chk (nmaskedpos->data, mposArr->data , 4*(mposArr->len)*sizeof(double), __builtin_object_size ( nmaskedpos->data, 0)) | |||
| 940 | 4*(mposArr->len)*sizeof(double))__builtin___memcpy_chk (nmaskedpos->data, mposArr->data , 4*(mposArr->len)*sizeof(double), __builtin_object_size ( nmaskedpos->data, 0)); | |||
| 941 | } | |||
| 942 | airMopOkay(mop); | |||
| 943 | return 0; | |||
| 944 | } | |||
| 945 | ||||
| 946 | static size_t | |||
| 947 | nsizeof(const Nrrd *nrrd) { | |||
| 948 | return (nrrd | |||
| 949 | ? nrrdElementSize(nrrd)*nrrdElementNumber(nrrd) | |||
| 950 | : 0); | |||
| 951 | } | |||
| 952 | ||||
| 953 | size_t | |||
| 954 | pullTraceMultiSizeof(const pullTraceMulti *mtrc) { | |||
| 955 | size_t ret; | |||
| 956 | unsigned int ti; | |||
| 957 | ||||
| 958 | if (!mtrc) { | |||
| 959 | return 0; | |||
| 960 | } | |||
| 961 | ret = 0; | |||
| 962 | for (ti=0; ti<mtrc->traceNum; ti++) { | |||
| 963 | ret += sizeof(pullTrace); | |||
| 964 | ret += nsizeof(mtrc->trace[ti]->nvert); | |||
| 965 | ret += nsizeof(mtrc->trace[ti]->nstrn); | |||
| 966 | ret += nsizeof(mtrc->trace[ti]->nstab); | |||
| 967 | } | |||
| 968 | ret += sizeof(pullTrace*)*(mtrc->traceArr->size); | |||
| 969 | return ret; | |||
| 970 | } | |||
| 971 | ||||
| 972 | pullTraceMulti * | |||
| 973 | pullTraceMultiNix(pullTraceMulti *mtrc) { | |||
| 974 | ||||
| 975 | if (mtrc) { | |||
| 976 | airArrayNuke(mtrc->traceArr); | |||
| 977 | free(mtrc); | |||
| 978 | } | |||
| 979 | return NULL((void*)0); | |||
| 980 | } | |||
| 981 | ||||
| 982 | ||||
| 983 | #define PULL_MTRC_MAGIC"PULLMTRC0001" "PULLMTRC0001" | |||
| 984 | #define DEMARK_STR"======" "======" | |||
| 985 | ||||
| 986 | static int | |||
| 987 | tracewrite(FILE *file, const pullTrace *trc, unsigned int ti) { | |||
| 988 | static const char me[]="tracewrite"; | |||
| 989 | ||||
| 990 | /* | |||
| 991 | this was used to get ascii coordinates for a trace, | |||
| 992 | to help isolate (via emacs) one trace from a saved multi-trace | |||
| 993 | NrrdIoState *nio = nrrdIoStateNew(); | |||
| 994 | nio->encoding = nrrdEncodingAscii; | |||
| 995 | */ | |||
| 996 | ||||
| 997 | fprintf(file, "%s %u\n", DEMARK_STR"======", ti); | |||
| 998 | ell_4v_print_d(file, trc->seedPos); | |||
| 999 | #define WRITE(FF) \ | |||
| 1000 | if (trc->FF && trc->FF->data) { \ | |||
| 1001 | if (nrrdWrite(file, trc->FF, NULL((void*)0) /* nio */ )) { \ | |||
| 1002 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: trouble with " #FF , me); \ | |||
| 1003 | return 1; \ | |||
| 1004 | } \ | |||
| 1005 | } else { \ | |||
| 1006 | fprintf(file, "NULL"); \ | |||
| 1007 | } \ | |||
| 1008 | fprintf(file, "\n") | |||
| 1009 | ||||
| 1010 | fprintf(file, "nrrds: vert strn stab = %d %d %d\n", | |||
| 1011 | trc->nvert && trc->nvert->data, | |||
| 1012 | trc->nstrn && trc->nstrn->data, | |||
| 1013 | trc->nstab && trc->nstab->data); | |||
| 1014 | WRITE(nvert); | |||
| 1015 | WRITE(nstrn); | |||
| 1016 | WRITE(nstab); | |||
| 1017 | fprintf(file, "%u\n", trc->seedIdx); | |||
| 1018 | fprintf(file, "%s %s %s\n", | |||
| 1019 | airEnumStr(pullTraceStop, trc->whyStop[0]), | |||
| 1020 | airEnumStr(pullTraceStop, trc->whyStop[1]), | |||
| 1021 | airEnumStr(pullTraceStop, trc->whyNowhere)); | |||
| 1022 | #undef WRITE | |||
| 1023 | return 0; | |||
| 1024 | } | |||
| 1025 | ||||
| 1026 | int | |||
| 1027 | pullTraceMultiWrite(FILE *file, const pullTraceMulti *mtrc) { | |||
| 1028 | static const char me[]="pullTraceMultiWrite"; | |||
| 1029 | unsigned int ti; | |||
| 1030 | ||||
| 1031 | if (!(file && mtrc)) { | |||
| 1032 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 1033 | return 1; | |||
| 1034 | } | |||
| 1035 | fprintf(file, "%s\n", PULL_MTRC_MAGIC"PULLMTRC0001"); | |||
| 1036 | fprintf(file, "%u traces\n", mtrc->traceNum); | |||
| 1037 | ||||
| 1038 | for (ti=0; ti<mtrc->traceNum; ti++) { | |||
| 1039 | if (tracewrite(file, mtrc->trace[ti], ti)) { | |||
| 1040 | biffAddf(PULLpullBiffKey, "%s: trace %u/%u", me, ti, mtrc->traceNum); | |||
| 1041 | return 1; | |||
| 1042 | } | |||
| 1043 | } | |||
| 1044 | return 0; | |||
| 1045 | } | |||
| 1046 | ||||
| 1047 | static int | |||
| 1048 | traceread(pullTrace *trc, FILE *file, unsigned int _ti) { | |||
| 1049 | static const char me[]="traceread"; | |||
| 1050 | char line[AIR_STRLEN_MED(256+1)], name[AIR_STRLEN_MED(256+1)]; | |||
| 1051 | unsigned int ti, lineLen; | |||
| 1052 | int stops[3], hackhack, vertHN, strnHN, stabHN; /* HN == have nrrd */ | |||
| 1053 | ||||
| 1054 | sprintf(name, "separator")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "separator"); | |||
| 1055 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1056 | if (!lineLen) { | |||
| 1057 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1058 | return 1; | |||
| 1059 | } | |||
| 1060 | if (1 != sscanf(line, DEMARK_STR"======" " %u", &ti)) { | |||
| 1061 | biffAddf(PULLpullBiffKey, "%s: \"%s\" doesn't look like %s line", me, line, name); | |||
| 1062 | return 1; | |||
| 1063 | } | |||
| 1064 | if (ti != _ti) { | |||
| 1065 | biffAddf(PULLpullBiffKey, "%s: read trace index %u but wanted %u", me, ti, _ti); | |||
| 1066 | return 1; | |||
| 1067 | } | |||
| 1068 | sprintf(name, "seed pos")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "seed pos"); | |||
| 1069 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1070 | if (!lineLen) { | |||
| 1071 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1072 | return 1; | |||
| 1073 | } | |||
| 1074 | if (4 != sscanf(line, "%lg %lg %lg %lg", trc->seedPos + 0, | |||
| 1075 | trc->seedPos + 1, trc->seedPos + 2, trc->seedPos + 3)) { | |||
| 1076 | biffAddf(PULLpullBiffKey, "%s: couldn't parse %s line \"%s\" as 4 doubles", | |||
| 1077 | me, name, line); | |||
| 1078 | return 1; | |||
| 1079 | } | |||
| 1080 | sprintf(name, "have nrrds")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "have nrrds"); | |||
| 1081 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1082 | if (!lineLen) { | |||
| 1083 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1084 | return 1; | |||
| 1085 | } | |||
| 1086 | if (3 != sscanf(line, "nrrds: vert strn stab = %d %d %d", | |||
| 1087 | &vertHN, &strnHN, &stabHN)) { | |||
| 1088 | biffAddf(PULLpullBiffKey, "%s: couldn't parse %s line", me, name); | |||
| 1089 | return 1; | |||
| 1090 | } | |||
| 1091 | #define READ(FF)if (FFHN) { if (nrrdRead(trc->nFF, file, ((void*)0))) { biffMovef (pullBiffKey, nrrdBiffKey, "%s: trouble with " "FF" , me); return 1; } fgetc(file); } else { airOneLine(file, line, (256+1)); } \ | |||
| 1092 | if (FF##HN) { \ | |||
| 1093 | if (nrrdRead(trc->n##FF, file, NULL((void*)0))) { \ | |||
| 1094 | biffMovef(PULLpullBiffKey, NRRDnrrdBiffKey, "%s: trouble with " #FF , me); \ | |||
| 1095 | return 1; \ | |||
| 1096 | } \ | |||
| 1097 | fgetc(file); \ | |||
| 1098 | } else { \ | |||
| 1099 | airOneLine(file, line, AIR_STRLEN_MED(256+1)); \ | |||
| 1100 | } | |||
| 1101 | hackhack = nrrdStateVerboseIO; /* should be fixed in Teem v2 */ | |||
| 1102 | nrrdStateVerboseIO = 0; | |||
| 1103 | READ(vert)if (vertHN) { if (nrrdRead(trc->nvert, file, ((void*)0))) { biffMovef(pullBiffKey, nrrdBiffKey, "%s: trouble with " "vert" , me); return 1; } fgetc(file); } else { airOneLine(file, line , (256+1)); }; | |||
| 1104 | READ(strn)if (strnHN) { if (nrrdRead(trc->nstrn, file, ((void*)0))) { biffMovef(pullBiffKey, nrrdBiffKey, "%s: trouble with " "strn" , me); return 1; } fgetc(file); } else { airOneLine(file, line , (256+1)); }; | |||
| 1105 | READ(stab)if (stabHN) { if (nrrdRead(trc->nstab, file, ((void*)0))) { biffMovef(pullBiffKey, nrrdBiffKey, "%s: trouble with " "stab" , me); return 1; } fgetc(file); } else { airOneLine(file, line , (256+1)); }; | |||
| 1106 | nrrdStateVerboseIO = hackhack; | |||
| 1107 | ||||
| 1108 | sprintf(name, "seed idx")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "seed idx"); | |||
| 1109 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1110 | if (!lineLen) { | |||
| 1111 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1112 | return 1; | |||
| 1113 | } | |||
| 1114 | if (1 != sscanf(line, "%u", &(trc->seedIdx))) { | |||
| 1115 | biffAddf(PULLpullBiffKey, "%s: didn't parse uint from %s line \"%s\"", | |||
| 1116 | me, name, line); | |||
| 1117 | return 1; | |||
| 1118 | } | |||
| 1119 | sprintf(name, "stops")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "stops"); | |||
| 1120 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1121 | if (!lineLen) { | |||
| 1122 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1123 | return 1; | |||
| 1124 | } | |||
| 1125 | if (3 != airParseStrE(stops, line, " ", 3, pullTraceStop)) { | |||
| 1126 | biffAddf(PULLpullBiffKey, "%s: didn't see 3 %s on %s line \"%s\"", me, | |||
| 1127 | pullTraceStop->name, name, line); | |||
| 1128 | return 1; | |||
| 1129 | } | |||
| 1130 | ||||
| 1131 | return 0; | |||
| 1132 | } | |||
| 1133 | int | |||
| 1134 | pullTraceMultiRead(pullTraceMulti *mtrc, FILE *file) { | |||
| 1135 | static const char me[]="pullTraceMultiRead"; | |||
| 1136 | char line[AIR_STRLEN_MED(256+1)], name[AIR_STRLEN_MED(256+1)]; | |||
| 1137 | unsigned int lineLen, ti, tnum; | |||
| 1138 | pullTrace *trc; | |||
| 1139 | ||||
| 1140 | if (!(mtrc && file)) { | |||
| ||||
| 1141 | biffAddf(PULLpullBiffKey, "%s: got NULL pointer", me); | |||
| 1142 | return 1; | |||
| 1143 | } | |||
| 1144 | airArrayLenSet(mtrc->traceArr, 0); | |||
| 1145 | sprintf(name, "magic")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "magic"); | |||
| 1146 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1147 | if (!lineLen) { | |||
| 1148 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1149 | return 1; | |||
| 1150 | } | |||
| 1151 | if (strcmp(line, PULL_MTRC_MAGIC"PULLMTRC0001")) { | |||
| 1152 | biffAddf(PULLpullBiffKey, "%s: %s line \"%s\" not expected \"%s\"", | |||
| 1153 | me, name, line, PULL_MTRC_MAGIC"PULLMTRC0001"); | |||
| 1154 | return 1; | |||
| 1155 | } | |||
| 1156 | ||||
| 1157 | sprintf(name, "# of traces")__builtin___sprintf_chk (name, 0, __builtin_object_size (name , 2 > 1 ? 1 : 0), "# of traces"); | |||
| 1158 | lineLen = airOneLine(file, line, AIR_STRLEN_MED(256+1)); | |||
| 1159 | if (!lineLen) { | |||
| 1160 | biffAddf(PULLpullBiffKey, "%s: didn't get %s line", me, name); | |||
| 1161 | return 1; | |||
| 1162 | } | |||
| 1163 | if (1 != sscanf(line, "%u traces", &tnum)) { | |||
| 1164 | biffAddf(PULLpullBiffKey, "%s: \"%s\" doesn't look like %s line", me, line, name); | |||
| 1165 | return 1; | |||
| 1166 | } | |||
| 1167 | for (ti=0; ti<tnum; ti++) { | |||
| 1168 | int added; | |||
| 1169 | trc = pullTraceNew(); | |||
| 1170 | if (traceread(trc, file, ti)) { | |||
| 1171 | biffAddf(PULLpullBiffKey, "%s: on trace %u/%u", me, ti, tnum); | |||
| ||||
| 1172 | return 1; | |||
| 1173 | } | |||
| 1174 | if (pullTraceMultiAdd(mtrc, trc, &added)) { | |||
| 1175 | biffAddf(PULLpullBiffKey, "%s: adding trace %u/%u", me, ti, tnum); | |||
| 1176 | return 1; | |||
| 1177 | } | |||
| 1178 | if (!added) { | |||
| 1179 | trc = pullTraceNix(trc); | |||
| 1180 | } | |||
| 1181 | } | |||
| 1182 | ||||
| 1183 | return 0; | |||
| 1184 | } |