1 : /******************************************************************************
2 : * $Id: gridlib.c 17435 2009-07-23 22:36:32Z rouault $
3 : *
4 : * Project: Arc/Info Binary Grid Translator
5 : * Purpose: Grid file reading code.
6 : * Author: Frank Warmerdam, warmerdam@pobox.com
7 : *
8 : ******************************************************************************
9 : * Copyright (c) 1999, Frank Warmerdam
10 : *
11 : * Permission is hereby granted, free of charge, to any person obtaining a
12 : * copy of this software and associated documentation files (the "Software"),
13 : * to deal in the Software without restriction, including without limitation
14 : * the rights to use, copy, modify, merge, publish, distribute, sublicense,
15 : * and/or sell copies of the Software, and to permit persons to whom the
16 : * Software is furnished to do so, subject to the following conditions:
17 : *
18 : * The above copyright notice and this permission notice shall be included
19 : * in all copies or substantial portions of the Software.
20 : *
21 : * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
22 : * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23 : * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
24 : * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25 : * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
26 : * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
27 : * DEALINGS IN THE SOFTWARE.
28 : ****************************************************************************/
29 :
30 : #include "aigrid.h"
31 :
32 : CPL_CVSID("$Id: gridlib.c 17435 2009-07-23 22:36:32Z rouault $");
33 :
34 : /************************************************************************/
35 : /* AIGProcessRaw32bitFloatBlock() */
36 : /* */
37 : /* Process a block using ``00'' (32 bit) raw format. */
38 : /************************************************************************/
39 :
40 : static
41 0 : CPLErr AIGProcessRaw32BitFloatBlock( GByte *pabyCur, int nDataSize, int nMin,
42 : int nBlockXSize, int nBlockYSize,
43 : float * pafData )
44 :
45 : {
46 : int i;
47 :
48 : (void) nMin;
49 0 : if( nDataSize < nBlockXSize*nBlockYSize*4 )
50 : {
51 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
52 0 : return CE_Failure;
53 : }
54 :
55 : /* -------------------------------------------------------------------- */
56 : /* Collect raw data. */
57 : /* -------------------------------------------------------------------- */
58 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
59 : {
60 : float fWork;
61 :
62 : #ifdef CPL_LSB
63 0 : ((GByte *) &fWork)[3] = *(pabyCur++);
64 0 : ((GByte *) &fWork)[2] = *(pabyCur++);
65 0 : ((GByte *) &fWork)[1] = *(pabyCur++);
66 0 : ((GByte *) &fWork)[0] = *(pabyCur++);
67 : #else
68 : ((GByte *) &fWork)[0] = *(pabyCur++);
69 : ((GByte *) &fWork)[1] = *(pabyCur++);
70 : ((GByte *) &fWork)[2] = *(pabyCur++);
71 : ((GByte *) &fWork)[3] = *(pabyCur++);
72 : #endif
73 :
74 0 : pafData[i] = fWork;
75 : }
76 :
77 0 : return( CE_None );
78 : }
79 :
80 : /************************************************************************/
81 : /* AIGProcessIntConstBlock() */
82 : /* */
83 : /* Process a block using ``00'' constant 32bit integer format. */
84 : /************************************************************************/
85 :
86 : static
87 4034 : CPLErr AIGProcessIntConstBlock( GByte *pabyCur, int nDataSize, int nMin,
88 : int nBlockXSize, int nBlockYSize,
89 : GInt32 * panData )
90 :
91 : {
92 : int i;
93 :
94 : (void) pabyCur;
95 : (void) nDataSize;
96 :
97 : /* -------------------------------------------------------------------- */
98 : /* Apply constant min value. */
99 : /* -------------------------------------------------------------------- */
100 4134850 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
101 4130816 : panData[i] = nMin;
102 :
103 4034 : return( CE_None );
104 : }
105 :
106 : /************************************************************************/
107 : /* AIGProcess32bitRawBlock() */
108 : /* */
109 : /* Process a block using ``20'' (thirtytwo bit) raw format. */
110 : /************************************************************************/
111 :
112 : static
113 0 : CPLErr AIGProcessRaw32BitBlock( GByte *pabyCur, int nDataSize, int nMin,
114 : int nBlockXSize, int nBlockYSize,
115 : GInt32 * panData )
116 :
117 : {
118 : int i;
119 :
120 0 : if( nDataSize < nBlockXSize*nBlockYSize*4 )
121 : {
122 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
123 0 : return CE_Failure;
124 : }
125 :
126 : /* -------------------------------------------------------------------- */
127 : /* Collect raw data. */
128 : /* -------------------------------------------------------------------- */
129 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
130 : {
131 0 : panData[i] = pabyCur[0] * 256 * 256 * 256
132 0 : + pabyCur[1] * 256 * 256
133 0 : + pabyCur[2] * 256
134 0 : + pabyCur[3] + nMin;
135 0 : pabyCur += 4;
136 : }
137 :
138 0 : return( CE_None );
139 : }
140 :
141 : /************************************************************************/
142 : /* AIGProcess16bitRawBlock() */
143 : /* */
144 : /* Process a block using ``10'' (sixteen bit) raw format. */
145 : /************************************************************************/
146 :
147 : static
148 345 : CPLErr AIGProcessRaw16BitBlock( GByte *pabyCur, int nDataSize, int nMin,
149 : int nBlockXSize, int nBlockYSize,
150 : GInt32 * panData )
151 :
152 : {
153 : int i;
154 :
155 345 : if( nDataSize < nBlockXSize*nBlockYSize*2 )
156 : {
157 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
158 0 : return CE_Failure;
159 : }
160 :
161 : /* -------------------------------------------------------------------- */
162 : /* Collect raw data. */
163 : /* -------------------------------------------------------------------- */
164 353625 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
165 : {
166 353280 : panData[i] = pabyCur[0] * 256 + pabyCur[1] + nMin;
167 353280 : pabyCur += 2;
168 : }
169 :
170 345 : return( CE_None );
171 : }
172 :
173 : /************************************************************************/
174 : /* AIGProcess4BitRawBlock() */
175 : /* */
176 : /* Process a block using ``08'' raw format. */
177 : /************************************************************************/
178 :
179 : static
180 0 : CPLErr AIGProcessRaw4BitBlock( GByte *pabyCur, int nDataSize, int nMin,
181 : int nBlockXSize, int nBlockYSize,
182 : GInt32 * panData )
183 :
184 : {
185 : int i;
186 :
187 0 : if( nDataSize < (nBlockXSize*nBlockYSize+1)/2 )
188 : {
189 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
190 0 : return CE_Failure;
191 : }
192 :
193 : /* -------------------------------------------------------------------- */
194 : /* Collect raw data. */
195 : /* -------------------------------------------------------------------- */
196 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
197 : {
198 0 : if( i % 2 == 0 )
199 0 : panData[i] = ((*(pabyCur) & 0xf0) >> 4) + nMin;
200 : else
201 0 : panData[i] = (*(pabyCur++) & 0xf) + nMin;
202 : }
203 :
204 0 : return( CE_None );
205 : }
206 :
207 : /************************************************************************/
208 : /* AIGProcess1BitRawBlock() */
209 : /* */
210 : /* Process a block using ``0x01'' raw format. */
211 : /************************************************************************/
212 :
213 : static
214 0 : CPLErr AIGProcessRaw1BitBlock( GByte *pabyCur, int nDataSize, int nMin,
215 : int nBlockXSize, int nBlockYSize,
216 : GInt32 * panData )
217 :
218 : {
219 : int i;
220 :
221 0 : if( nDataSize < (nBlockXSize*nBlockYSize+7)/8 )
222 : {
223 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
224 0 : return CE_Failure;
225 : }
226 :
227 : /* -------------------------------------------------------------------- */
228 : /* Collect raw data. */
229 : /* -------------------------------------------------------------------- */
230 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
231 : {
232 0 : if( pabyCur[i>>3] & (0x80 >> (i&0x7)) )
233 0 : panData[i] = 1 + nMin;
234 : else
235 0 : panData[i] = 0 + nMin;
236 : }
237 :
238 0 : return( CE_None );
239 : }
240 :
241 : /************************************************************************/
242 : /* AIGProcessRawBlock() */
243 : /* */
244 : /* Process a block using ``08'' raw format. */
245 : /************************************************************************/
246 :
247 : static
248 0 : CPLErr AIGProcessRawBlock( GByte *pabyCur, int nDataSize, int nMin,
249 : int nBlockXSize, int nBlockYSize, GInt32 * panData )
250 :
251 : {
252 : int i;
253 :
254 0 : if( nDataSize < nBlockXSize*nBlockYSize )
255 : {
256 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
257 0 : return CE_Failure;
258 : }
259 :
260 : /* -------------------------------------------------------------------- */
261 : /* Collect raw data. */
262 : /* -------------------------------------------------------------------- */
263 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
264 : {
265 0 : panData[i] = *(pabyCur++) + nMin;
266 : }
267 :
268 0 : return( CE_None );
269 : }
270 :
271 : /************************************************************************/
272 : /* AIGProcessFFBlock() */
273 : /* */
274 : /* Process a type 0xFF (CCITT RLE) compressed block. */
275 : /************************************************************************/
276 :
277 : static
278 0 : CPLErr AIGProcessFFBlock( GByte *pabyCur, int nDataSize, int nMin,
279 : int nBlockXSize, int nBlockYSize,
280 : GInt32 * panData )
281 :
282 : {
283 : /* -------------------------------------------------------------------- */
284 : /* Convert CCITT compress bitstream into 1bit raw data. */
285 : /* -------------------------------------------------------------------- */
286 : CPLErr eErr;
287 0 : int i, nDstBytes = (nBlockXSize * nBlockYSize + 7) / 8;
288 : unsigned char *pabyIntermediate;
289 :
290 0 : pabyIntermediate = (unsigned char *) VSIMalloc(nDstBytes);
291 0 : if (pabyIntermediate == NULL)
292 : {
293 0 : CPLError(CE_Failure, CPLE_OutOfMemory,
294 : "Cannot allocate %d bytes", nDstBytes);
295 0 : return CE_Failure;
296 : }
297 :
298 0 : eErr = DecompressCCITTRLETile( pabyCur, nDataSize,
299 : pabyIntermediate, nDstBytes,
300 : nBlockXSize, nBlockYSize );
301 0 : if( eErr != CE_None )
302 : {
303 0 : CPLFree(pabyIntermediate);
304 0 : return eErr;
305 : }
306 :
307 : /* -------------------------------------------------------------------- */
308 : /* Convert the bit buffer into 32bit integers and account for */
309 : /* nMin. */
310 : /* -------------------------------------------------------------------- */
311 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
312 : {
313 0 : if( pabyIntermediate[i>>3] & (0x80 >> (i&0x7)) )
314 0 : panData[i] = nMin+1;
315 : else
316 0 : panData[i] = nMin;
317 : }
318 :
319 0 : CPLFree( pabyIntermediate );
320 :
321 0 : return( CE_None );
322 : }
323 :
324 :
325 :
326 : /************************************************************************/
327 : /* AIGProcessBlock() */
328 : /* */
329 : /* Process a block using ``D7'', ``E0'' or ``DF'' compression. */
330 : /************************************************************************/
331 :
332 : static
333 1519 : CPLErr AIGProcessBlock( GByte *pabyCur, int nDataSize, int nMin, int nMagic,
334 : int nBlockXSize, int nBlockYSize, GInt32 * panData )
335 :
336 : {
337 : int nTotPixels, nPixels;
338 : int i;
339 :
340 : /* ==================================================================== */
341 : /* Process runs till we are done. */
342 : /* ==================================================================== */
343 1519 : nTotPixels = nBlockXSize * nBlockYSize;
344 1519 : nPixels = 0;
345 :
346 29509 : while( nPixels < nTotPixels && nDataSize > 0 )
347 : {
348 26471 : int nMarker = *(pabyCur++);
349 :
350 26471 : nDataSize--;
351 :
352 : /* -------------------------------------------------------------------- */
353 : /* Repeat data - four byte data block (0xE0) */
354 : /* -------------------------------------------------------------------- */
355 26471 : if( nMagic == 0xE0 )
356 : {
357 : GInt32 nValue;
358 :
359 0 : if( nMarker + nPixels > nTotPixels )
360 : {
361 0 : CPLError( CE_Failure, CPLE_AppDefined,
362 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
363 : nTotPixels - nPixels, nMarker );
364 0 : return CE_Failure;
365 : }
366 :
367 0 : if( nDataSize < 4 )
368 : {
369 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
370 0 : return CE_Failure;
371 : }
372 :
373 0 : nValue = 0;
374 0 : memcpy( &nValue, pabyCur, 4 );
375 0 : pabyCur += 4;
376 0 : nDataSize -= 4;
377 :
378 0 : nValue = CPL_MSBWORD32( nValue );
379 :
380 0 : nValue += nMin;
381 0 : for( i = 0; i < nMarker; i++ )
382 0 : panData[nPixels++] = nValue;
383 : }
384 :
385 : /* -------------------------------------------------------------------- */
386 : /* Repeat data - two byte data block (0xF0) */
387 : /* -------------------------------------------------------------------- */
388 26471 : else if( nMagic == 0xF0 )
389 : {
390 : GInt32 nValue;
391 :
392 0 : if( nMarker + nPixels > nTotPixels )
393 : {
394 0 : CPLError( CE_Failure, CPLE_AppDefined,
395 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
396 : nTotPixels - nPixels, nMarker );
397 0 : return CE_Failure;
398 : }
399 :
400 0 : if( nDataSize < 2 )
401 : {
402 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
403 0 : return CE_Failure;
404 : }
405 :
406 0 : nValue = (pabyCur[0] * 256 + pabyCur[1]) + nMin;
407 0 : pabyCur += 2;
408 0 : nDataSize -= 2;
409 :
410 0 : for( i = 0; i < nMarker; i++ )
411 0 : panData[nPixels++] = nValue;
412 : }
413 :
414 : /* -------------------------------------------------------------------- */
415 : /* Repeat data - one byte data block (0xFC) */
416 : /* -------------------------------------------------------------------- */
417 26471 : else if( nMagic == 0xFC || nMagic == 0xF8 )
418 : {
419 : GInt32 nValue;
420 :
421 0 : if( nMarker + nPixels > nTotPixels )
422 : {
423 0 : CPLError( CE_Failure, CPLE_AppDefined,
424 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
425 : nTotPixels - nPixels, nMarker );
426 0 : return CE_Failure;
427 : }
428 :
429 0 : if( nDataSize < 1 )
430 : {
431 0 : CPLError(CE_Failure, CPLE_AppDefined, "Block too small");
432 0 : return CE_Failure;
433 : }
434 :
435 0 : nValue = *(pabyCur++) + nMin;
436 0 : nDataSize--;
437 :
438 0 : for( i = 0; i < nMarker; i++ )
439 0 : panData[nPixels++] = nValue;
440 : }
441 :
442 : /* -------------------------------------------------------------------- */
443 : /* Repeat data - no actual data, just assign minimum (0xDF) */
444 : /* -------------------------------------------------------------------- */
445 26481 : else if( nMagic == 0xDF && nMarker < 128 )
446 : {
447 10 : if( nMarker + nPixels > nTotPixels )
448 : {
449 0 : CPLError( CE_Failure, CPLE_AppDefined,
450 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
451 : nTotPixels - nPixels, nMarker );
452 0 : return CE_Failure;
453 : }
454 :
455 20 : for( i = 0; i < nMarker; i++ )
456 10 : panData[nPixels++] = nMin;
457 : }
458 :
459 : /* -------------------------------------------------------------------- */
460 : /* Literal data (0xD7): 8bit values. */
461 : /* -------------------------------------------------------------------- */
462 28322 : else if( nMagic == 0xD7 && nMarker < 128 )
463 : {
464 1861 : if( nMarker + nPixels > nTotPixels )
465 : {
466 0 : CPLError( CE_Failure, CPLE_AppDefined,
467 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
468 : nTotPixels - nPixels, nMarker );
469 0 : return CE_Failure;
470 : }
471 :
472 31220 : while( nMarker > 0 && nDataSize > 0 )
473 : {
474 27498 : panData[nPixels++] = *(pabyCur++) + nMin;
475 27498 : nMarker--;
476 27498 : nDataSize--;
477 : }
478 : }
479 :
480 : /* -------------------------------------------------------------------- */
481 : /* Literal data (0xCF): 16 bit values. */
482 : /* -------------------------------------------------------------------- */
483 35874 : else if( nMagic == 0xCF && nMarker < 128 )
484 : {
485 : GInt32 nValue;
486 :
487 11274 : if( nMarker + nPixels > nTotPixels )
488 : {
489 0 : CPLError( CE_Failure, CPLE_AppDefined,
490 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
491 : nTotPixels - nPixels, nMarker );
492 0 : return CE_Failure;
493 : }
494 :
495 711741 : while( nMarker > 0 && nDataSize >= 2 )
496 : {
497 689193 : nValue = pabyCur[0] * 256 + pabyCur[1] + nMin;
498 689193 : panData[nPixels++] = nValue;
499 689193 : pabyCur += 2;
500 :
501 689193 : nMarker--;
502 689193 : nDataSize -= 2;
503 : }
504 : }
505 :
506 : /* -------------------------------------------------------------------- */
507 : /* Nodata repeat */
508 : /* -------------------------------------------------------------------- */
509 13326 : else if( nMarker > 128 )
510 : {
511 13326 : nMarker = 256 - nMarker;
512 :
513 13326 : if( nMarker + nPixels > nTotPixels )
514 : {
515 0 : CPLError( CE_Failure, CPLE_AppDefined,
516 : "Run too long in AIGProcessBlock, needed %d values, got %d.",
517 : nTotPixels - nPixels, nMarker );
518 0 : return CE_Failure;
519 : }
520 :
521 865407 : while( nMarker > 0 )
522 : {
523 838755 : panData[nPixels++] = ESRI_GRID_NO_DATA;
524 838755 : nMarker--;
525 : }
526 : }
527 :
528 : else
529 : {
530 0 : return CE_Failure;
531 : }
532 :
533 : }
534 :
535 1519 : if( nPixels < nTotPixels || nDataSize < 0 )
536 : {
537 0 : CPLError( CE_Failure, CPLE_AppDefined,
538 : "Ran out of data processing block with nMagic=%d.",
539 : nMagic );
540 0 : return CE_Failure;
541 : }
542 :
543 1519 : return CE_None;
544 : }
545 :
546 : /************************************************************************/
547 : /* AIGReadBlock() */
548 : /* */
549 : /* Read a single block of integer grid data. */
550 : /************************************************************************/
551 :
552 5898 : CPLErr AIGReadBlock( FILE * fp, GUInt32 nBlockOffset, int nBlockSize,
553 : int nBlockXSize, int nBlockYSize,
554 : GInt32 *panData, int nCellType )
555 :
556 : {
557 : GByte *pabyRaw, *pabyCur;
558 : CPLErr eErr;
559 5898 : int i, nMagic, nMinSize=0, nDataSize;
560 5898 : GInt32 nMin = 0;
561 :
562 : /* -------------------------------------------------------------------- */
563 : /* If the block has zero size it is all dummies. */
564 : /* -------------------------------------------------------------------- */
565 5898 : if( nBlockSize == 0 )
566 : {
567 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
568 0 : panData[i] = ESRI_GRID_NO_DATA;
569 :
570 0 : return( CE_None );
571 : }
572 :
573 : /* -------------------------------------------------------------------- */
574 : /* Read the block into memory. */
575 : /* -------------------------------------------------------------------- */
576 5898 : if (nBlockSize <= 0 || nBlockSize > 65535 * 2)
577 : {
578 0 : CPLError(CE_Failure, CPLE_AppDefined, "Invalid block size : %d", nBlockSize);
579 0 : return CE_Failure;
580 : }
581 :
582 5898 : pabyRaw = (GByte *) VSIMalloc(nBlockSize+2);
583 5898 : if (pabyRaw == NULL)
584 : {
585 0 : CPLError(CE_Failure, CPLE_AppDefined, "Cannot allocate memory for block");
586 0 : return CE_Failure;
587 : }
588 :
589 11796 : if( VSIFSeekL( fp, nBlockOffset, SEEK_SET ) != 0
590 11796 : || VSIFReadL( pabyRaw, nBlockSize+2, 1, fp ) != 1 )
591 : {
592 0 : memset( panData, 0, nBlockXSize*nBlockYSize*4 );
593 0 : CPLError( CE_Failure, CPLE_AppDefined,
594 : "Read of %d bytes from offset %d for grid block failed.",
595 : nBlockSize+2, nBlockOffset );
596 0 : CPLFree( pabyRaw );
597 0 : return CE_Failure;
598 : }
599 :
600 : /* -------------------------------------------------------------------- */
601 : /* Verify the block size. */
602 : /* -------------------------------------------------------------------- */
603 5898 : if( nBlockSize != (pabyRaw[0]*256 + pabyRaw[1])*2 )
604 : {
605 0 : memset( panData, 0, nBlockXSize*nBlockYSize*4 );
606 0 : CPLError( CE_Failure, CPLE_AppDefined,
607 : "Block is corrupt, block size was %d, but expected to be %d.",
608 0 : (pabyRaw[0]*256 + pabyRaw[1])*2, nBlockSize );
609 0 : CPLFree( pabyRaw );
610 0 : return CE_Failure;
611 : }
612 :
613 5898 : nDataSize = nBlockSize;
614 :
615 : /* -------------------------------------------------------------------- */
616 : /* Handle float files directly. */
617 : /* -------------------------------------------------------------------- */
618 5898 : if( nCellType == AIG_CELLTYPE_FLOAT )
619 : {
620 0 : AIGProcessRaw32BitFloatBlock( pabyRaw + 2, nDataSize, 0,
621 : nBlockXSize, nBlockYSize,
622 : (float *) panData );
623 0 : CPLFree( pabyRaw );
624 :
625 0 : return CE_None;
626 : }
627 :
628 : /* -------------------------------------------------------------------- */
629 : /* Collect minimum value. */
630 : /* -------------------------------------------------------------------- */
631 :
632 : /* The first 2 bytes that give the block size are not included in nDataSize */
633 : /* and have already been safely read */
634 5898 : pabyCur = pabyRaw + 2;
635 :
636 : /* Need at least 2 byte to read the nMinSize and the nMagic */
637 5898 : if (nDataSize < 2)
638 : {
639 0 : CPLError( CE_Failure, CPLE_AppDefined,
640 : "Corrupt block. Need 2 bytes to read nMagic and nMinSize, only %d available",
641 : nDataSize);
642 0 : CPLFree( pabyRaw );
643 0 : return CE_Failure;
644 : }
645 5898 : nMagic = pabyCur[0];
646 5898 : nMinSize = pabyCur[1];
647 5898 : pabyCur += 2;
648 5898 : nDataSize -= 2;
649 :
650 : /* Need at least nMinSize bytes to read the nMin value */
651 5898 : if (nDataSize < nMinSize)
652 : {
653 0 : CPLError( CE_Failure, CPLE_AppDefined,
654 : "Corrupt block. Need %d bytes to read nMin. Only %d available",
655 : nMinSize, nDataSize);
656 0 : CPLFree( pabyRaw );
657 0 : return CE_Failure;
658 : }
659 :
660 5898 : if( nMinSize > 4 )
661 : {
662 0 : memset( panData, 0, nBlockXSize*nBlockYSize*4 );
663 0 : CPLError( CE_Failure, CPLE_AppDefined,
664 : "Corrupt 'minsize' of %d in block header. Read aborted.",
665 : nMinSize );
666 0 : CPLFree( pabyRaw );
667 0 : return CE_Failure;
668 : }
669 :
670 5898 : if( nMinSize == 4 )
671 : {
672 4034 : memcpy( &nMin, pabyCur, 4 );
673 4034 : nMin = CPL_MSBWORD32( nMin );
674 4034 : pabyCur += 4;
675 : }
676 : else
677 : {
678 1864 : nMin = 0;
679 3470 : for( i = 0; i < nMinSize; i++ )
680 : {
681 1606 : nMin = nMin * 256 + *pabyCur;
682 1606 : pabyCur++;
683 : }
684 :
685 : /* If nMinSize = 0, then we might have only 4 bytes in pabyRaw */
686 : /* don't try to read the 5th one then */
687 1864 : if( nMinSize != 0 && pabyRaw[4] > 127 )
688 : {
689 0 : if( nMinSize == 2 )
690 0 : nMin = nMin - 65536;
691 0 : else if( nMinSize == 1 )
692 0 : nMin = nMin - 256;
693 0 : else if( nMinSize == 3 )
694 0 : nMin = nMin - 256*256*256;
695 : }
696 : }
697 :
698 5898 : nDataSize -= nMinSize;
699 :
700 : /* -------------------------------------------------------------------- */
701 : /* Call an apppropriate handler depending on magic code. */
702 : /* -------------------------------------------------------------------- */
703 :
704 5898 : if( nMagic == 0x08 )
705 : {
706 0 : AIGProcessRawBlock( pabyCur, nDataSize, nMin,
707 : nBlockXSize, nBlockYSize,
708 : panData );
709 : }
710 5898 : else if( nMagic == 0x04 )
711 : {
712 0 : AIGProcessRaw4BitBlock( pabyCur, nDataSize, nMin,
713 : nBlockXSize, nBlockYSize,
714 : panData );
715 : }
716 5898 : else if( nMagic == 0x01 )
717 : {
718 0 : AIGProcessRaw1BitBlock( pabyCur, nDataSize, nMin,
719 : nBlockXSize, nBlockYSize,
720 : panData );
721 : }
722 5898 : else if( nMagic == 0x00 )
723 : {
724 4034 : AIGProcessIntConstBlock( pabyCur, nDataSize, nMin,
725 : nBlockXSize, nBlockYSize, panData );
726 : }
727 1864 : else if( nMagic == 0x10 )
728 : {
729 345 : AIGProcessRaw16BitBlock( pabyCur, nDataSize, nMin,
730 : nBlockXSize, nBlockYSize,
731 : panData );
732 : }
733 1519 : else if( nMagic == 0x20 )
734 : {
735 0 : AIGProcessRaw32BitBlock( pabyCur, nDataSize, nMin,
736 : nBlockXSize, nBlockYSize,
737 : panData );
738 : }
739 1519 : else if( nMagic == 0xFF )
740 : {
741 0 : AIGProcessFFBlock( pabyCur, nDataSize, nMin,
742 : nBlockXSize, nBlockYSize,
743 : panData );
744 : }
745 : else
746 : {
747 1519 : eErr = AIGProcessBlock( pabyCur, nDataSize, nMin, nMagic,
748 : nBlockXSize, nBlockYSize, panData );
749 :
750 1519 : if( eErr == CE_Failure )
751 : {
752 : static int bHasWarned = FALSE;
753 :
754 0 : for( i = 0; i < nBlockXSize * nBlockYSize; i++ )
755 0 : panData[i] = ESRI_GRID_NO_DATA;
756 :
757 0 : if( !bHasWarned )
758 : {
759 0 : CPLError( CE_Warning, CPLE_AppDefined,
760 : "Unsupported Arc/Info Binary Grid tile of type 0x%X"
761 : " encountered.\n"
762 : "This and subsequent unsupported tile types set to"
763 : " no data value.\n",
764 : nMagic );
765 0 : bHasWarned = TRUE;
766 : }
767 : }
768 : }
769 :
770 5898 : CPLFree( pabyRaw );
771 :
772 5898 : return CE_None;
773 : }
774 :
775 : /************************************************************************/
776 : /* AIGReadHeader() */
777 : /* */
778 : /* Read the hdr.adf file, and populate the given info structure */
779 : /* appropriately. */
780 : /************************************************************************/
781 :
782 7 : CPLErr AIGReadHeader( const char * pszCoverName, AIGInfo_t * psInfo )
783 :
784 : {
785 : char *pszHDRFilename;
786 : FILE *fp;
787 : GByte abyData[308];
788 :
789 : /* -------------------------------------------------------------------- */
790 : /* Open the file hdr.adf file. */
791 : /* -------------------------------------------------------------------- */
792 7 : pszHDRFilename = (char *) CPLMalloc(strlen(pszCoverName)+30);
793 7 : sprintf( pszHDRFilename, "%s/hdr.adf", pszCoverName );
794 :
795 7 : fp = AIGLLOpen( pszHDRFilename, "rb" );
796 :
797 7 : if( fp == NULL )
798 : {
799 0 : CPLError( CE_Failure, CPLE_OpenFailed,
800 : "Failed to open grid header file:\n%s\n", pszHDRFilename );
801 :
802 0 : CPLFree( pszHDRFilename );
803 0 : return( CE_Failure );
804 : }
805 :
806 7 : CPLFree( pszHDRFilename );
807 :
808 : /* -------------------------------------------------------------------- */
809 : /* Read the whole file (we expect it to always be 308 bytes */
810 : /* long. */
811 : /* -------------------------------------------------------------------- */
812 :
813 7 : VSIFReadL( abyData, 1, 308, fp );
814 :
815 7 : VSIFCloseL( fp );
816 :
817 : /* -------------------------------------------------------------------- */
818 : /* Read the block size information. */
819 : /* -------------------------------------------------------------------- */
820 7 : memcpy( &(psInfo->nCellType), abyData+16, 4 );
821 7 : memcpy( &(psInfo->nBlocksPerRow), abyData+288, 4 );
822 7 : memcpy( &(psInfo->nBlocksPerColumn), abyData+292, 4 );
823 7 : memcpy( &(psInfo->nBlockXSize), abyData+296, 4 );
824 7 : memcpy( &(psInfo->nBlockYSize), abyData+304, 4 );
825 7 : memcpy( &(psInfo->dfCellSizeX), abyData+256, 8 );
826 7 : memcpy( &(psInfo->dfCellSizeY), abyData+264, 8 );
827 :
828 : #ifdef CPL_LSB
829 7 : psInfo->nCellType = CPL_SWAP32( psInfo->nCellType );
830 7 : psInfo->nBlocksPerRow = CPL_SWAP32( psInfo->nBlocksPerRow );
831 7 : psInfo->nBlocksPerColumn = CPL_SWAP32( psInfo->nBlocksPerColumn );
832 7 : psInfo->nBlockXSize = CPL_SWAP32( psInfo->nBlockXSize );
833 7 : psInfo->nBlockYSize = CPL_SWAP32( psInfo->nBlockYSize );
834 7 : CPL_SWAPDOUBLE( &(psInfo->dfCellSizeX) );
835 7 : CPL_SWAPDOUBLE( &(psInfo->dfCellSizeY) );
836 : #endif
837 :
838 7 : return( CE_None );
839 : }
840 :
841 : /************************************************************************/
842 : /* AIGReadBlockIndex() */
843 : /* */
844 : /* Read the w001001x.adf file, and populate the given info */
845 : /* structure with the block offsets, and sizes. */
846 : /************************************************************************/
847 :
848 3 : CPLErr AIGReadBlockIndex( AIGInfo_t * psInfo, AIGTileInfo *psTInfo,
849 : const char *pszBasename )
850 :
851 : {
852 : char *pszHDRFilename;
853 : FILE *fp;
854 : int nLength, i;
855 : GInt32 nValue;
856 : GUInt32 *panIndex;
857 : GByte abyHeader[8];
858 :
859 : /* -------------------------------------------------------------------- */
860 : /* Open the file hdr.adf file. */
861 : /* -------------------------------------------------------------------- */
862 3 : pszHDRFilename = (char *) CPLMalloc(strlen(psInfo->pszCoverName)+40);
863 3 : sprintf( pszHDRFilename, "%s/%sx.adf", psInfo->pszCoverName, pszBasename );
864 :
865 3 : fp = AIGLLOpen( pszHDRFilename, "rb" );
866 :
867 3 : if( fp == NULL )
868 : {
869 0 : CPLError( CE_Failure, CPLE_OpenFailed,
870 : "Failed to open grid block index file:\n%s\n",
871 : pszHDRFilename );
872 :
873 0 : CPLFree( pszHDRFilename );
874 0 : return( CE_Failure );
875 : }
876 :
877 3 : CPLFree( pszHDRFilename );
878 :
879 : /* -------------------------------------------------------------------- */
880 : /* Verify the magic number. This is often corrupted by CR/LF */
881 : /* translation. */
882 : /* -------------------------------------------------------------------- */
883 3 : VSIFReadL( abyHeader, 1, 8, fp );
884 3 : if( abyHeader[3] == 0x0D && abyHeader[4] == 0x0A )
885 : {
886 0 : CPLError( CE_Failure, CPLE_AppDefined,
887 : "w001001x.adf file header has been corrupted by unix to dos text conversion." );
888 0 : VSIFCloseL( fp );
889 0 : return CE_Failure;
890 : }
891 :
892 18 : if( abyHeader[0] != 0x00
893 3 : || abyHeader[1] != 0x00
894 3 : || abyHeader[2] != 0x27
895 3 : || abyHeader[3] != 0x0A
896 3 : || abyHeader[4] != 0xFF
897 3 : || abyHeader[5] != 0xFF )
898 : {
899 0 : CPLError( CE_Failure, CPLE_AppDefined,
900 : "w001001x.adf file header magic number is corrupt." );
901 0 : VSIFCloseL( fp );
902 0 : return CE_Failure;
903 : }
904 :
905 : /* -------------------------------------------------------------------- */
906 : /* Get the file length (in 2 byte shorts) */
907 : /* -------------------------------------------------------------------- */
908 3 : VSIFSeekL( fp, 24, SEEK_SET );
909 3 : VSIFReadL( &nValue, 1, 4, fp );
910 :
911 : // FIXME? : risk of overflow in multiplication
912 3 : nLength = CPL_MSBWORD32(nValue) * 2;
913 :
914 : /* -------------------------------------------------------------------- */
915 : /* Allocate buffer, and read the file (from beyond the header) */
916 : /* into the buffer. */
917 : /* -------------------------------------------------------------------- */
918 3 : psTInfo->nBlocks = (nLength-100) / 8;
919 3 : panIndex = (GUInt32 *) VSIMalloc2(psTInfo->nBlocks, 8);
920 3 : if (panIndex == NULL)
921 : {
922 0 : CPLError(CE_Failure, CPLE_OutOfMemory,
923 : "AIGReadBlockIndex: Out of memory. Probably due to corrupted w001001x.adf file");
924 0 : VSIFCloseL( fp );
925 0 : return CE_Failure;
926 : }
927 3 : VSIFSeekL( fp, 100, SEEK_SET );
928 3 : if ((int)VSIFReadL( panIndex, 8, psTInfo->nBlocks, fp ) != psTInfo->nBlocks)
929 : {
930 0 : CPLError(CE_Failure, CPLE_AppDefined,
931 : "AIGReadBlockIndex: Cannot read block info");
932 0 : VSIFCloseL( fp );
933 0 : CPLFree( panIndex );
934 0 : return CE_Failure;
935 : }
936 :
937 3 : VSIFCloseL( fp );
938 :
939 : /* -------------------------------------------------------------------- */
940 : /* Allocate AIGInfo block info arrays. */
941 : /* -------------------------------------------------------------------- */
942 3 : psTInfo->panBlockOffset = (GUInt32 *) VSIMalloc2(4, psTInfo->nBlocks);
943 3 : psTInfo->panBlockSize = (int *) VSIMalloc2(4, psTInfo->nBlocks);
944 6 : if (psTInfo->panBlockOffset == NULL ||
945 3 : psTInfo->panBlockSize == NULL)
946 : {
947 0 : CPLError(CE_Failure, CPLE_OutOfMemory,
948 : "AIGReadBlockIndex: Out of memory. Probably due to corrupted w001001x.adf file");
949 0 : CPLFree( psTInfo->panBlockOffset );
950 0 : CPLFree( psTInfo->panBlockSize );
951 0 : CPLFree( panIndex );
952 0 : return CE_Failure;
953 : }
954 :
955 : /* -------------------------------------------------------------------- */
956 : /* Populate the block information. */
957 : /* -------------------------------------------------------------------- */
958 5901 : for( i = 0; i < psTInfo->nBlocks; i++ )
959 : {
960 5898 : psTInfo->panBlockOffset[i] = CPL_MSBWORD32(panIndex[i*2]) * 2;
961 5898 : psTInfo->panBlockSize[i] = CPL_MSBWORD32(panIndex[i*2+1]) * 2;
962 : }
963 :
964 3 : CPLFree( panIndex );
965 :
966 3 : return( CE_None );
967 : }
968 :
969 : /************************************************************************/
970 : /* AIGReadBounds() */
971 : /* */
972 : /* Read the dblbnd.adf file for the georeferenced bounds. */
973 : /************************************************************************/
974 :
975 7 : CPLErr AIGReadBounds( const char * pszCoverName, AIGInfo_t * psInfo )
976 :
977 : {
978 : char *pszHDRFilename;
979 : FILE *fp;
980 : double adfBound[4];
981 :
982 : /* -------------------------------------------------------------------- */
983 : /* Open the file dblbnd.adf file. */
984 : /* -------------------------------------------------------------------- */
985 7 : pszHDRFilename = (char *) CPLMalloc(strlen(pszCoverName)+40);
986 7 : sprintf( pszHDRFilename, "%s/dblbnd.adf", pszCoverName );
987 :
988 7 : fp = AIGLLOpen( pszHDRFilename, "rb" );
989 :
990 7 : if( fp == NULL )
991 : {
992 0 : CPLError( CE_Failure, CPLE_OpenFailed,
993 : "Failed to open grid bounds file:\n%s\n",
994 : pszHDRFilename );
995 :
996 0 : CPLFree( pszHDRFilename );
997 0 : return( CE_Failure );
998 : }
999 :
1000 7 : CPLFree( pszHDRFilename );
1001 :
1002 : /* -------------------------------------------------------------------- */
1003 : /* Get the contents - four doubles. */
1004 : /* -------------------------------------------------------------------- */
1005 7 : VSIFReadL( adfBound, 1, 32, fp );
1006 :
1007 7 : VSIFCloseL( fp );
1008 :
1009 : #ifdef CPL_LSB
1010 7 : CPL_SWAPDOUBLE(adfBound+0);
1011 7 : CPL_SWAPDOUBLE(adfBound+1);
1012 7 : CPL_SWAPDOUBLE(adfBound+2);
1013 7 : CPL_SWAPDOUBLE(adfBound+3);
1014 : #endif
1015 :
1016 7 : psInfo->dfLLX = adfBound[0];
1017 7 : psInfo->dfLLY = adfBound[1];
1018 7 : psInfo->dfURX = adfBound[2];
1019 7 : psInfo->dfURY = adfBound[3];
1020 :
1021 7 : return( CE_None );
1022 : }
1023 :
1024 : /************************************************************************/
1025 : /* AIGReadStatistics() */
1026 : /* */
1027 : /* Read the sta.adf file for the layer statistics. */
1028 : /************************************************************************/
1029 :
1030 7 : CPLErr AIGReadStatistics( const char * pszCoverName, AIGInfo_t * psInfo )
1031 :
1032 : {
1033 : char *pszHDRFilename;
1034 : FILE *fp;
1035 : double adfStats[4];
1036 :
1037 7 : psInfo->dfMin = 0.0;
1038 7 : psInfo->dfMax = 0.0;
1039 7 : psInfo->dfMean = 0.0;
1040 7 : psInfo->dfStdDev = 0.0;
1041 :
1042 : /* -------------------------------------------------------------------- */
1043 : /* Open the file sta.adf file. */
1044 : /* -------------------------------------------------------------------- */
1045 7 : pszHDRFilename = (char *) CPLMalloc(strlen(pszCoverName)+40);
1046 7 : sprintf( pszHDRFilename, "%s/sta.adf", pszCoverName );
1047 :
1048 7 : fp = AIGLLOpen( pszHDRFilename, "rb" );
1049 :
1050 7 : if( fp == NULL )
1051 : {
1052 0 : CPLError( CE_Failure, CPLE_OpenFailed,
1053 : "Failed to open grid statistics file:\n%s\n",
1054 : pszHDRFilename );
1055 :
1056 0 : CPLFree( pszHDRFilename );
1057 0 : return( CE_Failure );
1058 : }
1059 :
1060 7 : CPLFree( pszHDRFilename );
1061 :
1062 : /* -------------------------------------------------------------------- */
1063 : /* Get the contents - four doubles. */
1064 : /* -------------------------------------------------------------------- */
1065 7 : VSIFReadL( adfStats, 1, 32, fp );
1066 :
1067 7 : VSIFCloseL( fp );
1068 :
1069 : #ifdef CPL_LSB
1070 7 : CPL_SWAPDOUBLE(adfStats+0);
1071 7 : CPL_SWAPDOUBLE(adfStats+1);
1072 7 : CPL_SWAPDOUBLE(adfStats+2);
1073 7 : CPL_SWAPDOUBLE(adfStats+3);
1074 : #endif
1075 :
1076 7 : psInfo->dfMin = adfStats[0];
1077 7 : psInfo->dfMax = adfStats[1];
1078 7 : psInfo->dfMean = adfStats[2];
1079 7 : psInfo->dfStdDev = adfStats[3];
1080 :
1081 7 : return( CE_None );
1082 : }
1083 :
|
