[7ee2b42] | 1 | /* extend.c |
---|
[d1b1380] | 2 | * Produce an extended elevation |
---|
[6a6cb97] | 3 | * Copyright (C) 1995-2002,2005,2010 Olly Betts |
---|
[dc2a4bf] | 4 | * Copyright (C) 2004,2005 John Pybus |
---|
[846746e] | 5 | * |
---|
[89231c4] | 6 | * This program is free software; you can redistribute it and/or modify |
---|
| 7 | * it under the terms of the GNU General Public License as published by |
---|
| 8 | * the Free Software Foundation; either version 2 of the License, or |
---|
| 9 | * (at your option) any later version. |
---|
[846746e] | 10 | * |
---|
| 11 | * This program is distributed in the hope that it will be useful, |
---|
| 12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
[89231c4] | 13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 14 | * GNU General Public License for more details. |
---|
[846746e] | 15 | * |
---|
[89231c4] | 16 | * You should have received a copy of the GNU General Public License |
---|
| 17 | * along with this program; if not, write to the Free Software |
---|
[6a6cb97] | 18 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
---|
[d1b1380] | 19 | */ |
---|
| 20 | |
---|
[a420b49] | 21 | #ifdef HAVE_CONFIG_H |
---|
| 22 | # include <config.h> |
---|
| 23 | #endif |
---|
[d1b1380] | 24 | |
---|
[e55ee28] | 25 | #include <float.h> |
---|
[d1b1380] | 26 | #include <stdio.h> |
---|
| 27 | #include <stdlib.h> |
---|
| 28 | #include <string.h> |
---|
| 29 | |
---|
[acc20b1] | 30 | #include "cmdline.h" |
---|
[693388e] | 31 | #include "debug.h" |
---|
[d1b1380] | 32 | #include "filelist.h" |
---|
[693388e] | 33 | #include "filename.h" |
---|
| 34 | #include "hash.h" |
---|
[d1b1380] | 35 | #include "img.h" |
---|
[693388e] | 36 | #include "message.h" |
---|
| 37 | #include "useful.h" |
---|
[d1b1380] | 38 | |
---|
[693388e] | 39 | /* To save memory we should probably use the prefix hash for the prefix on |
---|
| 40 | * point labels (FIXME) */ |
---|
[ea1c342] | 41 | |
---|
| 42 | typedef struct stn { |
---|
| 43 | const char *label; |
---|
| 44 | int flags; |
---|
[ee11c6a7] | 45 | const struct stn *next; |
---|
[ea1c342] | 46 | } stn; |
---|
| 47 | |
---|
[d1b1380] | 48 | typedef struct POINT { |
---|
[0ed0e16] | 49 | img_point p; |
---|
[ea1c342] | 50 | const stn *stns; |
---|
[647407d] | 51 | unsigned int order; |
---|
[f48f66f] | 52 | char dir; |
---|
| 53 | char fDone; |
---|
| 54 | char fBroken; |
---|
[d1b1380] | 55 | struct POINT *next; |
---|
| 56 | } point; |
---|
| 57 | |
---|
| 58 | typedef struct LEG { |
---|
| 59 | point *fr, *to; |
---|
[693388e] | 60 | const char *prefix; |
---|
[f48f66f] | 61 | char dir; |
---|
| 62 | char fDone; |
---|
| 63 | char broken; |
---|
[95c3272] | 64 | int flags; |
---|
| 65 | struct LEG *next; |
---|
[d1b1380] | 66 | } leg; |
---|
| 67 | |
---|
[f48f66f] | 68 | /* Values for leg.broken: */ |
---|
| 69 | #define BREAK_FR 0x01 |
---|
| 70 | #define BREAK_TO 0x02 |
---|
[863401f] | 71 | |
---|
[f48f66f] | 72 | /* Values for point.dir and leg.dir: */ |
---|
| 73 | #define ELEFT 0x01 |
---|
| 74 | #define ERIGHT 0x02 |
---|
| 75 | #define ESWAP 0x04 |
---|
[863401f] | 76 | |
---|
[f48f66f] | 77 | static point headpoint = {{0, 0, 0}, NULL, 0, 0, 0, 0, NULL}; |
---|
[d1b1380] | 78 | |
---|
[f48f66f] | 79 | static leg headleg = {NULL, NULL, NULL, 0, 0, 0, 0, NULL}; |
---|
[d1b1380] | 80 | |
---|
[168df28] | 81 | static img *pimg_out; |
---|
[d1b1380] | 82 | |
---|
[dc2a4bf] | 83 | static void do_stn(point *, double, const char *, int, int); |
---|
[d1b1380] | 84 | |
---|
[693388e] | 85 | typedef struct pfx { |
---|
| 86 | const char *label; |
---|
| 87 | struct pfx *next; |
---|
| 88 | } pfx; |
---|
| 89 | |
---|
| 90 | static pfx **htab; |
---|
| 91 | |
---|
[a03f5d1] | 92 | #define HTAB_SIZE 0x2000 |
---|
| 93 | |
---|
[693388e] | 94 | static const char * |
---|
| 95 | find_prefix(const char *prefix) |
---|
| 96 | { |
---|
| 97 | pfx *p; |
---|
| 98 | int hash; |
---|
| 99 | |
---|
[4c07c51] | 100 | SVX_ASSERT(prefix); |
---|
[421b7d2] | 101 | |
---|
[a03f5d1] | 102 | hash = hash_string(prefix) & (HTAB_SIZE - 1); |
---|
[693388e] | 103 | for (p = htab[hash]; p; p = p->next) { |
---|
| 104 | if (strcmp(prefix, p->label) == 0) return p->label; |
---|
| 105 | } |
---|
| 106 | |
---|
| 107 | p = osnew(pfx); |
---|
| 108 | p->label = osstrdup(prefix); |
---|
| 109 | p->next = htab[hash]; |
---|
| 110 | htab[hash] = p; |
---|
| 111 | |
---|
| 112 | return p->label; |
---|
| 113 | } |
---|
| 114 | |
---|
[a420b49] | 115 | static point * |
---|
[0ed0e16] | 116 | find_point(const img_point *pt) |
---|
[a420b49] | 117 | { |
---|
[d1b1380] | 118 | point *p; |
---|
[647407d] | 119 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
[0ed0e16] | 120 | if (pt->x == p->p.x && pt->y == p->p.y && pt->z == p->p.z) { |
---|
[647407d] | 121 | return p; |
---|
| 122 | } |
---|
| 123 | } |
---|
[a420b49] | 124 | |
---|
| 125 | p = osmalloc(ossizeof(point)); |
---|
[0ed0e16] | 126 | p->p = *pt; |
---|
[ea1c342] | 127 | p->stns = NULL; |
---|
[7ee2b42] | 128 | p->order = 0; |
---|
[f48f66f] | 129 | p->dir = 0; |
---|
[863401f] | 130 | p->fDone = 0; |
---|
[168df28] | 131 | p->fBroken = 0; |
---|
[a420b49] | 132 | p->next = headpoint.next; |
---|
| 133 | headpoint.next = p; |
---|
[d1b1380] | 134 | return p; |
---|
| 135 | } |
---|
| 136 | |
---|
[a420b49] | 137 | static void |
---|
[693388e] | 138 | add_leg(point *fr, point *to, const char *prefix, int flags) |
---|
[a420b49] | 139 | { |
---|
[95c3272] | 140 | leg *l; |
---|
[7ee2b42] | 141 | fr->order++; |
---|
| 142 | to->order++; |
---|
[95c3272] | 143 | l = osmalloc(ossizeof(leg)); |
---|
| 144 | l->fr = fr; |
---|
| 145 | l->to = to; |
---|
[693388e] | 146 | if (prefix) |
---|
| 147 | l->prefix = find_prefix(prefix); |
---|
| 148 | else |
---|
| 149 | l->prefix = NULL; |
---|
[95c3272] | 150 | l->next = headleg.next; |
---|
[f48f66f] | 151 | l->dir = 0; |
---|
[95c3272] | 152 | l->fDone = 0; |
---|
[168df28] | 153 | l->broken = 0; |
---|
[95c3272] | 154 | l->flags = flags; |
---|
| 155 | headleg.next = l; |
---|
[d1b1380] | 156 | } |
---|
| 157 | |
---|
[a420b49] | 158 | static void |
---|
[95c3272] | 159 | add_label(point *p, const char *label, int flags) |
---|
[a420b49] | 160 | { |
---|
[ea1c342] | 161 | stn *s = osnew(stn); |
---|
| 162 | s->label = osstrdup(label); |
---|
| 163 | s->flags = flags; |
---|
| 164 | s->next = p->stns; |
---|
| 165 | p->stns = s; |
---|
[d1b1380] | 166 | } |
---|
| 167 | |
---|
[863401f] | 168 | /* Read in config file */ |
---|
| 169 | |
---|
| 170 | |
---|
| 171 | /* lifted from img.c Should be put somewhere common? JPNP*/ |
---|
| 172 | static char * |
---|
| 173 | getline_alloc(FILE *fh, size_t ilen) |
---|
| 174 | { |
---|
| 175 | int ch; |
---|
| 176 | size_t i = 0; |
---|
| 177 | size_t len = ilen; |
---|
| 178 | char *buf = xosmalloc(len); |
---|
| 179 | if (!buf) return NULL; |
---|
| 180 | |
---|
| 181 | ch = getc(fh); |
---|
| 182 | while (ch != '\n' && ch != '\r' && ch != EOF) { |
---|
| 183 | buf[i++] = ch; |
---|
| 184 | if (i == len - 1) { |
---|
| 185 | char *p; |
---|
| 186 | len += len; |
---|
| 187 | p = xosrealloc(buf, len); |
---|
| 188 | if (!p) { |
---|
| 189 | osfree(buf); |
---|
| 190 | return NULL; |
---|
| 191 | } |
---|
| 192 | buf = p; |
---|
| 193 | } |
---|
| 194 | ch = getc(fh); |
---|
| 195 | } |
---|
| 196 | if (ch == '\n' || ch == '\r') { |
---|
| 197 | int otherone = ch ^ ('\n' ^ '\r'); |
---|
| 198 | ch = getc(fh); |
---|
| 199 | /* if it's not the other eol character, put it back */ |
---|
| 200 | if (ch != otherone) ungetc(ch, fh); |
---|
| 201 | } |
---|
| 202 | buf[i++] = '\0'; |
---|
| 203 | return buf; |
---|
| 204 | } |
---|
| 205 | |
---|
[c0b1b1a] | 206 | static int lineno = 0; |
---|
| 207 | static point *start = NULL; |
---|
[863401f] | 208 | |
---|
| 209 | static char* |
---|
| 210 | delimword(char *ln, char** lr) |
---|
| 211 | { |
---|
| 212 | char *le; |
---|
| 213 | |
---|
| 214 | while (*ln == ' ' || *ln == '\t' || *ln == '\n' || *ln == '\r') |
---|
| 215 | ln++; |
---|
| 216 | |
---|
| 217 | le = ln; |
---|
| 218 | while (*le != ' ' && *le != '\t' && *le != '\n' && *le != '\r' && *le != ';' && *le != '\0') |
---|
| 219 | le++; |
---|
| 220 | |
---|
| 221 | if (*le == '\0' || *le == ';') { |
---|
| 222 | *lr = le; |
---|
| 223 | } else { |
---|
| 224 | *lr = le + 1; |
---|
| 225 | } |
---|
[168df28] | 226 | |
---|
[863401f] | 227 | *le = '\0'; |
---|
| 228 | return ln; |
---|
| 229 | } |
---|
| 230 | |
---|
| 231 | static void |
---|
[cbb0e89] | 232 | parseconfigline(const char *fnm, char *ln) |
---|
[863401f] | 233 | { |
---|
| 234 | point *p; |
---|
| 235 | const stn *s; |
---|
| 236 | const stn *t; |
---|
| 237 | leg *l; |
---|
| 238 | char *lc = NULL; |
---|
| 239 | |
---|
| 240 | ln = delimword(ln, &lc); |
---|
| 241 | |
---|
| 242 | if (*ln == '\0') return; |
---|
| 243 | |
---|
| 244 | if (strcmp(ln, "*start")==0) { |
---|
| 245 | ln = delimword(lc, &lc); |
---|
[cbb0e89] | 246 | if (*ln == 0) |
---|
| 247 | fatalerror_in_file(fnm, lineno, /*Expecting station name*/28); |
---|
[863401f] | 248 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 249 | for (s = p->stns; s; s = s->next) { |
---|
| 250 | if (strcmp(s->label, ln)==0) { |
---|
| 251 | start = p; |
---|
[cbb0e89] | 252 | printf(msg(/*Starting from station %s*/512),ln); |
---|
[863401f] | 253 | putnl(); |
---|
| 254 | goto loopend; |
---|
| 255 | } |
---|
| 256 | } |
---|
| 257 | } |
---|
[cbb0e89] | 258 | warning_in_file(fnm, lineno, /*Failed to find station %s*/510, ln); |
---|
[863401f] | 259 | } else if (strcmp(ln, "*eleft")==0) { |
---|
| 260 | char *ll = delimword(lc, &lc); |
---|
[cbb0e89] | 261 | if (*ll == 0) |
---|
| 262 | fatalerror_in_file(fnm, lineno, /*Expecting station name*/28); |
---|
[863401f] | 263 | ln = delimword(lc, &lc); |
---|
| 264 | if (*ln == 0) { /* One argument, look for point to switch at. */ |
---|
| 265 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 266 | for (s = p->stns; s; s = s->next) { |
---|
| 267 | if (strcmp(s->label, ll)==0) { |
---|
[cbb0e89] | 268 | printf(msg(/*Extending to the left from station %s*/513), ll); |
---|
[863401f] | 269 | putnl(); |
---|
[f48f66f] | 270 | p->dir = ELEFT; |
---|
[863401f] | 271 | goto loopend; |
---|
| 272 | } |
---|
| 273 | } |
---|
| 274 | } |
---|
[cbb0e89] | 275 | warning_in_file(fnm, lineno, /*Failed to find station %s*/510, ll); |
---|
[863401f] | 276 | } else { /* Two arguments look for a specific leg */ |
---|
| 277 | for (l = headleg.next; l; l=l->next) { |
---|
| 278 | point * fr = l->fr; |
---|
| 279 | point * to = l->to; |
---|
| 280 | if (fr && to) { |
---|
| 281 | for (s=fr->stns; s; s=s->next) { |
---|
| 282 | int b = 0; |
---|
| 283 | if (strcmp(s->label,ll)==0 || (strcmp(s->label, ln)==0 && (b = 1)) ) { |
---|
| 284 | char * lr = (b ? ll : ln); |
---|
| 285 | for (t=to->stns; t; t=t->next) { |
---|
| 286 | if (strcmp(t->label,lr)==0) { |
---|
[cbb0e89] | 287 | printf(msg(/*Extending to the left from leg %s -> %s*/515), s->label, t->label); |
---|
[863401f] | 288 | putnl(); |
---|
[f48f66f] | 289 | l->dir = ELEFT; |
---|
[863401f] | 290 | goto loopend; |
---|
| 291 | } |
---|
| 292 | } |
---|
| 293 | } |
---|
| 294 | } |
---|
| 295 | } |
---|
| 296 | } |
---|
[cbb0e89] | 297 | warning_in_file(fnm, lineno, /*Failed to find leg %s -> %s*/511, ll, ln); |
---|
[863401f] | 298 | } |
---|
| 299 | } else if (strcmp(ln, "*eright")==0) { |
---|
| 300 | char *ll = delimword(lc, &lc); |
---|
[cbb0e89] | 301 | if (*ll == 0) |
---|
| 302 | fatalerror_in_file(fnm, lineno, /*Expecting station name*/28); |
---|
[863401f] | 303 | ln = delimword(lc, &lc); |
---|
| 304 | if (*ln == 0) { /* One argument, look for point to switch at. */ |
---|
| 305 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 306 | for (s = p->stns; s; s = s->next) { |
---|
| 307 | if (strcmp(s->label, ll)==0) { |
---|
[cbb0e89] | 308 | printf(msg(/*Extending to the right from station %s*/514), ll); |
---|
[863401f] | 309 | putnl(); |
---|
[f48f66f] | 310 | p->dir = ERIGHT; |
---|
[863401f] | 311 | goto loopend; |
---|
| 312 | } |
---|
| 313 | } |
---|
| 314 | } |
---|
[cbb0e89] | 315 | warning_in_file(fnm, lineno, /*Failed to find station %s*/510, ll); |
---|
[863401f] | 316 | } else { /* Two arguments look for a specific leg */ |
---|
| 317 | for (l = headleg.next; l; l=l->next) { |
---|
| 318 | point * fr = l->fr; |
---|
| 319 | point * to = l->to; |
---|
| 320 | if (fr && to) { |
---|
| 321 | for (s=fr->stns; s; s=s->next) { |
---|
| 322 | int b = 0; |
---|
| 323 | if (strcmp(s->label,ll)==0 || (strcmp(s->label, ln)==0 && (b = 1)) ) { |
---|
| 324 | char * lr = (b ? ll : ln); |
---|
| 325 | for (t=to->stns; t; t=t->next) { |
---|
| 326 | if (strcmp(t->label,lr)==0) { |
---|
[cbb0e89] | 327 | printf(msg(/*Extending to the right from leg %s -> %s*/516), s->label, t->label); |
---|
| 328 | putnl(); |
---|
[f48f66f] | 329 | l->dir=ERIGHT; |
---|
[863401f] | 330 | goto loopend; |
---|
| 331 | } |
---|
| 332 | } |
---|
| 333 | } |
---|
| 334 | } |
---|
| 335 | } |
---|
| 336 | } |
---|
[cbb0e89] | 337 | warning_in_file(fnm, lineno, /*Failed to find leg %s -> %s*/511, ll, ln); |
---|
[863401f] | 338 | } |
---|
| 339 | } else if (strcmp(ln, "*eswap")==0) { |
---|
| 340 | char *ll = delimword(lc, &lc); |
---|
[cbb0e89] | 341 | if (*ll == 0) |
---|
| 342 | fatalerror_in_file(fnm, lineno, /*Expecting station name*/28); |
---|
[863401f] | 343 | ln = delimword(lc, &lc); |
---|
| 344 | if (*ln == 0) { /* One argument, look for point to switch at. */ |
---|
| 345 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 346 | for (s = p->stns; s; s = s->next) { |
---|
| 347 | if (strcmp(s->label, ll)==0) { |
---|
[cbb0e89] | 348 | printf(msg(/*Swapping extend direction from station %s*/519),ll); |
---|
[863401f] | 349 | putnl(); |
---|
[f48f66f] | 350 | p->dir = ESWAP; |
---|
[863401f] | 351 | goto loopend; |
---|
| 352 | } |
---|
| 353 | } |
---|
| 354 | } |
---|
[cbb0e89] | 355 | warning_in_file(fnm, lineno, /*Failed to find station %s*/510, ll); |
---|
[863401f] | 356 | } else { /* Two arguments look for a specific leg */ |
---|
| 357 | for (l = headleg.next; l; l=l->next) { |
---|
| 358 | point * fr = l->fr; |
---|
| 359 | point * to = l->to; |
---|
| 360 | if (fr && to) { |
---|
| 361 | for (s=fr->stns; s; s=s->next) { |
---|
| 362 | int b = 0; |
---|
| 363 | if (strcmp(s->label,ll)==0 || (strcmp(s->label, ln)==0 && (b = 1)) ) { |
---|
| 364 | char * lr = (b ? ll : ln); |
---|
| 365 | for (t=to->stns; t; t=t->next) { |
---|
| 366 | if (strcmp(t->label,lr)==0) { |
---|
[cbb0e89] | 367 | printf(msg(/*Swapping extend direction from leg %s -> %s*/520), s->label, t->label); |
---|
| 368 | putnl(); |
---|
[f48f66f] | 369 | l->dir = ESWAP; |
---|
[863401f] | 370 | goto loopend; |
---|
| 371 | } |
---|
| 372 | } |
---|
| 373 | } |
---|
| 374 | } |
---|
| 375 | } |
---|
| 376 | } |
---|
[cbb0e89] | 377 | warning_in_file(fnm, lineno, /*Failed to find leg %s -> %s*/511, ll, ln); |
---|
[863401f] | 378 | } |
---|
| 379 | } else if (strcmp(ln, "*break")==0) { |
---|
| 380 | char *ll = delimword(lc, &lc); |
---|
[cbb0e89] | 381 | if (*ll == 0) |
---|
| 382 | fatalerror_in_file(fnm, lineno, /*Expecting station name*/28); |
---|
[863401f] | 383 | ln = delimword(lc, &lc); |
---|
| 384 | if (*ln == 0) { /* One argument, look for point to break at. */ |
---|
| 385 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 386 | for (s = p->stns; s; s = s->next) { |
---|
| 387 | if (strcmp(s->label, ll)==0) { |
---|
[cbb0e89] | 388 | printf(msg(/*Breaking survey loop at station %s*/517), ll); |
---|
[863401f] | 389 | putnl(); |
---|
[f48f66f] | 390 | p->fBroken = 1; |
---|
[863401f] | 391 | goto loopend; |
---|
| 392 | } |
---|
| 393 | } |
---|
| 394 | } |
---|
[cbb0e89] | 395 | warning_in_file(fnm, lineno, /*Failed to find station %s*/510, ll); |
---|
[863401f] | 396 | } else { /* Two arguments look for a specific leg */ |
---|
| 397 | for (l = headleg.next; l; l=l->next) { |
---|
| 398 | point * fr = l->fr; |
---|
| 399 | point * to = l->to; |
---|
[f48f66f] | 400 | if (fr && to) { |
---|
[863401f] | 401 | for (s=fr->stns; s; s=s->next) { |
---|
| 402 | int b = 0; |
---|
| 403 | if (strcmp(s->label,ll)==0 || (strcmp(s->label, ln)==0 && (b = 1)) ) { |
---|
| 404 | char * lr = (b ? ll : ln); |
---|
| 405 | for (t=to->stns; t; t=t->next) { |
---|
| 406 | if (strcmp(t->label,lr)==0) { |
---|
[cbb0e89] | 407 | printf(msg(/*Breaking survey loop at leg %s -> %s*/518), s->label, t->label); |
---|
[863401f] | 408 | putnl(); |
---|
[f48f66f] | 409 | l->broken = (b ? BREAK_TO : BREAK_FR); |
---|
[863401f] | 410 | goto loopend; |
---|
| 411 | } |
---|
| 412 | } |
---|
| 413 | } |
---|
| 414 | } |
---|
| 415 | } |
---|
| 416 | } |
---|
[cbb0e89] | 417 | warning_in_file(fnm, lineno, /*Failed to find leg %s -> %s*/511, ll, ln); |
---|
[863401f] | 418 | } |
---|
| 419 | } else { |
---|
[cbb0e89] | 420 | fatalerror_in_file(fnm, lineno, /*Unknown command `%s'*/12, ln); |
---|
[863401f] | 421 | } |
---|
| 422 | loopend: |
---|
| 423 | ln = delimword(lc, &lc); |
---|
| 424 | if (*ln != 0) { |
---|
[cbb0e89] | 425 | fatalerror_in_file(fnm, lineno, /*End of line not blank*/15); |
---|
| 426 | /* FIXME: give ln as context? */ |
---|
[863401f] | 427 | } |
---|
| 428 | } |
---|
| 429 | |
---|
[acc20b1] | 430 | static const struct option long_opts[] = { |
---|
| 431 | /* const char *name; int has_arg (0 no_argument, 1 required_*, 2 optional_*); int *flag; int val; */ |
---|
[76bbb7c9] | 432 | {"survey", required_argument, 0, 's'}, |
---|
[863401f] | 433 | {"specfile", required_argument, 0, 'p'}, |
---|
[acc20b1] | 434 | {"help", no_argument, 0, HLP_HELP}, |
---|
| 435 | {"version", no_argument, 0, HLP_VERSION}, |
---|
| 436 | {0, 0, 0, 0} |
---|
| 437 | }; |
---|
| 438 | |
---|
[863401f] | 439 | #define short_opts "s:p:" |
---|
[acc20b1] | 440 | |
---|
| 441 | static struct help_msg help[] = { |
---|
| 442 | /* <-- */ |
---|
[e4138b5] | 443 | {HLP_ENCODELONG(0), "only load the sub-survey with this prefix"}, |
---|
[863401f] | 444 | {HLP_ENCODELONG(1), "apply specifications from the named file"}, |
---|
[acc20b1] | 445 | {0, 0} |
---|
| 446 | }; |
---|
| 447 | |
---|
[a420b49] | 448 | int |
---|
| 449 | main(int argc, char **argv) |
---|
| 450 | { |
---|
[6f185e5] | 451 | const char *fnm_in, *fnm_out; |
---|
[a2ad284] | 452 | char *desc; |
---|
[0ed0e16] | 453 | img_point pt; |
---|
[d1b1380] | 454 | int result; |
---|
[647407d] | 455 | point *fr = NULL, *to; |
---|
[badeec8] | 456 | double zMax = -DBL_MAX; |
---|
[647407d] | 457 | point *p; |
---|
[76bbb7c9] | 458 | const char *survey = NULL; |
---|
[863401f] | 459 | const char *specfile = NULL; |
---|
[168df28] | 460 | img *pimg; |
---|
[d1b1380] | 461 | |
---|
[bdfe97f] | 462 | msg_init(argv); |
---|
[d1b1380] | 463 | |
---|
[acc20b1] | 464 | cmdline_set_syntax_message("INPUT_3D_FILE [OUTPUT_3D_FILE]", NULL); |
---|
| 465 | cmdline_init(argc, argv, short_opts, long_opts, NULL, help, 1, 2); |
---|
[76bbb7c9] | 466 | while (1) { |
---|
| 467 | int opt = cmdline_getopt(); |
---|
| 468 | if (opt == EOF) break; |
---|
| 469 | if (opt == 's') survey = optarg; |
---|
[863401f] | 470 | if (opt == 'p') specfile = optarg; |
---|
[acc20b1] | 471 | } |
---|
[6f185e5] | 472 | fnm_in = argv[optind++]; |
---|
[acc20b1] | 473 | if (argv[optind]) { |
---|
[6f185e5] | 474 | fnm_out = argv[optind]; |
---|
[acc20b1] | 475 | } else { |
---|
[168df28] | 476 | char * base_in = base_from_fnm(fnm_in); |
---|
| 477 | char * base_out = osmalloc(strlen(base_in) + 8); |
---|
| 478 | strcpy(base_out, base_in); |
---|
| 479 | strcat(base_out, "_extend"); |
---|
| 480 | fnm_out = add_ext(base_out, EXT_SVX_3D); |
---|
| 481 | osfree(base_in); |
---|
| 482 | osfree(base_out); |
---|
[d1b1380] | 483 | } |
---|
| 484 | |
---|
| 485 | /* try to open image file, and check it has correct header */ |
---|
[a2ad284] | 486 | pimg = img_open_survey(fnm_in, survey); |
---|
[6f185e5] | 487 | if (pimg == NULL) fatalerror(img_error(), fnm_in); |
---|
[d1b1380] | 488 | |
---|
| 489 | putnl(); |
---|
[a420b49] | 490 | puts(msg(/*Reading in data - please wait...*/105)); |
---|
[d1b1380] | 491 | |
---|
[a03f5d1] | 492 | htab = osmalloc(ossizeof(pfx*) * HTAB_SIZE); |
---|
| 493 | { |
---|
| 494 | int i; |
---|
| 495 | for (i = 0; i < HTAB_SIZE; ++i) htab[i] = NULL; |
---|
| 496 | } |
---|
[421b7d2] | 497 | |
---|
[d1b1380] | 498 | do { |
---|
[23f7ea7] | 499 | result = img_read_item(pimg, &pt); |
---|
[d1b1380] | 500 | switch (result) { |
---|
[a420b49] | 501 | case img_MOVE: |
---|
[421b7d2] | 502 | fr = find_point(&pt); |
---|
| 503 | break; |
---|
[a420b49] | 504 | case img_LINE: |
---|
[421b7d2] | 505 | if (!fr) { |
---|
[76bbb7c9] | 506 | result = img_BAD; |
---|
| 507 | break; |
---|
[421b7d2] | 508 | } |
---|
| 509 | to = find_point(&pt); |
---|
| 510 | if (!(pimg->flags & (img_FLAG_SURFACE|img_FLAG_SPLAY))) |
---|
[693388e] | 511 | add_leg(fr, to, pimg->label, pimg->flags); |
---|
[421b7d2] | 512 | fr = to; |
---|
| 513 | break; |
---|
[a420b49] | 514 | case img_LABEL: |
---|
[388ad50] | 515 | to = find_point(&pt); |
---|
| 516 | add_label(to, pimg->label, pimg->flags); |
---|
[421b7d2] | 517 | break; |
---|
[ed0f5b6] | 518 | case img_BAD: |
---|
[6f185e5] | 519 | (void)img_close(pimg); |
---|
| 520 | fatalerror(img_error(), fnm_in); |
---|
[d1b1380] | 521 | } |
---|
[ed0f5b6] | 522 | } while (result != img_STOP); |
---|
[d1b1380] | 523 | |
---|
[5b2b04f] | 524 | desc = osmalloc(strlen(pimg->title) + 11 + 1); |
---|
[a2ad284] | 525 | strcpy(desc, pimg->title); |
---|
| 526 | strcat(desc, " (extended)"); |
---|
| 527 | |
---|
[6f185e5] | 528 | (void)img_close(pimg); |
---|
[d1b1380] | 529 | |
---|
[863401f] | 530 | if (specfile) { |
---|
| 531 | FILE *fs = NULL; |
---|
[cbb0e89] | 532 | char *fnm_used; |
---|
| 533 | printf(msg(/*Applying specfile: `%s'*/521), specfile); |
---|
[863401f] | 534 | putnl(); |
---|
[cbb0e89] | 535 | fs = fopenWithPthAndExt("", specfile, NULL, "r", &fnm_used); |
---|
[863401f] | 536 | if (fs == NULL) fatalerror(/*Unable to open file*/93, specfile); |
---|
| 537 | while (!feof(fs)) { |
---|
[cbb0e89] | 538 | char *lbuf = getline_alloc(fs, 32); |
---|
[863401f] | 539 | lineno++; |
---|
[cbb0e89] | 540 | if (!lbuf) |
---|
| 541 | fatalerror_in_file(fnm_used, lineno, /*Error reading file*/18); |
---|
| 542 | parseconfigline(fnm_used, lbuf); |
---|
[863401f] | 543 | osfree(lbuf); |
---|
[647407d] | 544 | } |
---|
[cbb0e89] | 545 | osfree(fnm_used); |
---|
[647407d] | 546 | } |
---|
[863401f] | 547 | |
---|
| 548 | if (start == NULL) { /* i.e. start wasn't specified in specfile */ |
---|
[168df28] | 549 | |
---|
[863401f] | 550 | /* start at the highest entrance with some legs attached */ |
---|
[647407d] | 551 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
[863401f] | 552 | if (p->order > 0 && p->p.z > zMax) { |
---|
| 553 | const stn *s; |
---|
| 554 | for (s = p->stns; s; s = s->next) { |
---|
| 555 | if (s->flags & img_SFLAG_ENTRANCE) { |
---|
| 556 | start = p; |
---|
| 557 | zMax = p->p.z; |
---|
| 558 | break; |
---|
| 559 | } |
---|
| 560 | } |
---|
[647407d] | 561 | } |
---|
| 562 | } |
---|
| 563 | if (start == NULL) { |
---|
[863401f] | 564 | /* if no entrances with legs, start at the highest 1-node */ |
---|
[7ee2b42] | 565 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
[863401f] | 566 | if (p->order == 1 && p->p.z > zMax) { |
---|
[7ee2b42] | 567 | start = p; |
---|
| 568 | zMax = p->p.z; |
---|
| 569 | } |
---|
| 570 | } |
---|
[863401f] | 571 | /* of course we may have no 1-nodes... */ |
---|
[d216f5d] | 572 | if (start == NULL) { |
---|
| 573 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
[863401f] | 574 | if (p->order != 0 && p->p.z > zMax) { |
---|
[d216f5d] | 575 | start = p; |
---|
| 576 | zMax = p->p.z; |
---|
| 577 | } |
---|
| 578 | } |
---|
[863401f] | 579 | if (start == NULL) { |
---|
| 580 | /* There are no legs - just pick the highest station... */ |
---|
| 581 | for (p = headpoint.next; p != NULL; p = p->next) { |
---|
| 582 | if (p->p.z > zMax) { |
---|
| 583 | start = p; |
---|
| 584 | zMax = p->p.z; |
---|
| 585 | } |
---|
| 586 | } |
---|
| 587 | if (!start) fatalerror(/*No survey data*/43); |
---|
| 588 | } |
---|
[d216f5d] | 589 | } |
---|
[647407d] | 590 | } |
---|
| 591 | } |
---|
[863401f] | 592 | |
---|
[cbb0e89] | 593 | printf(msg(/*Writing %s...*/522), fnm_out); |
---|
[863401f] | 594 | putnl(); |
---|
[6a6cb97] | 595 | img_output_version = 3; |
---|
[168df28] | 596 | pimg_out = img_open_write(fnm_out, desc, fTrue); |
---|
[d1b1380] | 597 | |
---|
[dc2a4bf] | 598 | /* Only does single connected component currently. */ |
---|
| 599 | do_stn(start, 0.0, NULL, ERIGHT, 0); |
---|
[168df28] | 600 | if (!img_close(pimg_out)) { |
---|
[6f185e5] | 601 | (void)remove(fnm_out); |
---|
| 602 | fatalerror(img_error(), fnm_out); |
---|
| 603 | } |
---|
[cb3d1e2] | 604 | |
---|
[d1b1380] | 605 | return EXIT_SUCCESS; |
---|
| 606 | } |
---|
| 607 | |
---|
[f48f66f] | 608 | static int adjust_direction(int dir, int by) { |
---|
| 609 | if (by == ESWAP) |
---|
| 610 | return dir ^ (ELEFT|ERIGHT); |
---|
| 611 | if (by) |
---|
| 612 | return by; |
---|
| 613 | return dir; |
---|
| 614 | } |
---|
| 615 | |
---|
[a420b49] | 616 | static void |
---|
[dc2a4bf] | 617 | do_stn(point *p, double X, const char *prefix, int dir, int labOnly) |
---|
[a420b49] | 618 | { |
---|
| 619 | leg *l, *lp; |
---|
[badeec8] | 620 | double dX; |
---|
[ee11c6a7] | 621 | const stn *s; |
---|
[dc2a4bf] | 622 | int odir = dir; |
---|
[ea1c342] | 623 | |
---|
| 624 | for (s = p->stns; s; s = s->next) { |
---|
[168df28] | 625 | img_write_item(pimg_out, img_LABEL, s->flags, s->label, X, 0, p->p.z); |
---|
[ea1c342] | 626 | } |
---|
[f48f66f] | 627 | if (labOnly || p->fBroken) { |
---|
| 628 | return; |
---|
[863401f] | 629 | } |
---|
| 630 | |
---|
[7d9b3a9] | 631 | lp = &headleg; |
---|
[b39c5f6] | 632 | for (l = lp->next; l; lp = l, l = lp->next) { |
---|
[dc2a4bf] | 633 | dir = odir; |
---|
[f48f66f] | 634 | if (l->fDone) { |
---|
[863401f] | 635 | /* this case happens if a recursive call causes the next leg to be |
---|
[b39c5f6] | 636 | * removed, leaving our next pointing to a leg which has been dealt |
---|
| 637 | * with... */ |
---|
[f48f66f] | 638 | continue; |
---|
| 639 | } |
---|
| 640 | if (l->prefix != prefix) { |
---|
| 641 | continue; |
---|
| 642 | } |
---|
| 643 | if (l->to == p) { |
---|
| 644 | if (l->broken & BREAK_TO) continue; |
---|
| 645 | lp->next = l->next; |
---|
| 646 | /* adjust direction of extension if necessary */ |
---|
| 647 | dir = adjust_direction(dir, l->to->dir); |
---|
| 648 | dir = adjust_direction(dir, l->dir); |
---|
| 649 | |
---|
| 650 | dX = hypot(l->fr->p.x - l->to->p.x, l->fr->p.y - l->to->p.y); |
---|
| 651 | if (dir == ELEFT) dX = -dX; |
---|
[168df28] | 652 | img_write_item(pimg_out, img_MOVE, 0, NULL, X + dX, 0, l->fr->p.z); |
---|
| 653 | img_write_item(pimg_out, img_LINE, l->flags, l->prefix, |
---|
[f48f66f] | 654 | X, 0, l->to->p.z); |
---|
| 655 | l->fDone = 1; |
---|
| 656 | do_stn(l->fr, X + dX, l->prefix, dir, (l->broken & BREAK_FR)); |
---|
| 657 | l = lp; |
---|
| 658 | } else if (l->fr == p) { |
---|
| 659 | if (l->broken & BREAK_FR) continue; |
---|
| 660 | lp->next = l->next; |
---|
| 661 | /* adjust direction of extension if necessary */ |
---|
| 662 | dir = adjust_direction(dir, l->fr->dir); |
---|
| 663 | dir = adjust_direction(dir, l->dir); |
---|
| 664 | |
---|
| 665 | dX = hypot(l->fr->p.x - l->to->p.x, l->fr->p.y - l->to->p.y); |
---|
| 666 | if (dir == ELEFT) dX = -dX; |
---|
[168df28] | 667 | img_write_item(pimg_out, img_MOVE, 0, NULL, X, 0, l->fr->p.z); |
---|
| 668 | img_write_item(pimg_out, img_LINE, l->flags, l->prefix, |
---|
[f48f66f] | 669 | X + dX, 0, l->to->p.z); |
---|
| 670 | l->fDone = 1; |
---|
| 671 | do_stn(l->to, X + dX, l->prefix, dir, (l->broken & BREAK_TO)); |
---|
| 672 | l = lp; |
---|
[b39c5f6] | 673 | } |
---|
| 674 | } |
---|
| 675 | lp = &headleg; |
---|
[7d9b3a9] | 676 | for (l = lp->next; l; lp = l, l = lp->next) { |
---|
[dc2a4bf] | 677 | dir = odir; |
---|
[f48f66f] | 678 | if (l->fDone) { |
---|
[7d9b3a9] | 679 | /* this case happens iff a recursive call causes the next leg to be |
---|
| 680 | * removed, leaving our next pointing to a leg which has been dealt |
---|
| 681 | * with... */ |
---|
[f48f66f] | 682 | continue; |
---|
| 683 | } |
---|
| 684 | if (l->to == p) { |
---|
| 685 | if (l->broken & BREAK_TO) continue; |
---|
| 686 | lp->next = l->next; |
---|
| 687 | /* adjust direction of extension if necessary */ |
---|
| 688 | dir = adjust_direction(dir, l->to->dir); |
---|
| 689 | dir = adjust_direction(dir, l->dir); |
---|
| 690 | |
---|
| 691 | dX = hypot(l->fr->p.x - l->to->p.x, l->fr->p.y - l->to->p.y); |
---|
| 692 | if (dir == ELEFT) dX = -dX; |
---|
[168df28] | 693 | img_write_item(pimg_out, img_MOVE, 0, NULL, X + dX, 0, l->fr->p.z); |
---|
| 694 | img_write_item(pimg_out, img_LINE, l->flags, l->prefix, |
---|
[f48f66f] | 695 | X, 0, l->to->p.z); |
---|
| 696 | l->fDone = 1; |
---|
| 697 | do_stn(l->fr, X + dX, l->prefix, dir, (l->broken & BREAK_FR)); |
---|
| 698 | l = lp; |
---|
| 699 | } else if (l->fr == p) { |
---|
| 700 | if (l->broken & BREAK_FR) continue; |
---|
| 701 | lp->next = l->next; |
---|
| 702 | /* adjust direction of extension if necessary */ |
---|
| 703 | dir = adjust_direction(dir, l->fr->dir); |
---|
| 704 | dir = adjust_direction(dir, l->dir); |
---|
| 705 | |
---|
| 706 | dX = hypot(l->fr->p.x - l->to->p.x, l->fr->p.y - l->to->p.y); |
---|
| 707 | if (dir == ELEFT) dX = -dX; |
---|
[168df28] | 708 | img_write_item(pimg_out, img_MOVE, 0, NULL, X, 0, l->fr->p.z); |
---|
| 709 | img_write_item(pimg_out, img_LINE, l->flags, l->prefix, |
---|
[f48f66f] | 710 | X + dX, 0, l->to->p.z); |
---|
| 711 | l->fDone = 1; |
---|
| 712 | do_stn(l->to, X + dX, l->prefix, dir, (l->broken & BREAK_TO)); |
---|
| 713 | l = lp; |
---|
[d1b1380] | 714 | } |
---|
| 715 | } |
---|
| 716 | } |
---|