#include "clitkXdrImageIO.h"
#include "clitkCommon.h"
-#include <sys/stat.h>\r
+#include <sys/stat.h>
//From mbfield.h
-#ifndef unix\r
-//#define _read readfix\r
-#endif\r
-#define AVSINT ptrdiff_t\r
-#define AVS_ERROR\r
-#define AVS_OK\r
-\r
+#ifndef unix
+//#define _read readfix
+#endif
+#define AVSINT ptrdiff_t
+#define AVS_ERROR
+#define AVS_OK
+
//From portdefs.h
-#ifndef __LARGE__\r
-# if defined(__GNUC__) || defined(unix)\r
- typedef long long Q_INT64;\r
- typedef unsigned long long Q_UINT64;\r
-# define Q_INT64_CONST(x) (x##ll)\r
-# define Q_UINT64_CONST(x) (x##llu) /* gcc also allows ull */\r
- /* When using MINGW with MS(V)CRT DLL, use MS format modifier. */\r
-# ifdef __MSVCRT__\r
-# define Q_INT64_FORMAT "I64"\r
-# else\r
-# define Q_INT64_FORMAT "L"\r
-# endif\r
-# elif defined(__BORLANDC__) || defined(__WATCOMC__) || defined(_MSC_VER)\r
- typedef __int64 Q_INT64;\r
- typedef unsigned __int64 Q_UINT64;\r
-# define Q_INT64_CONST(x) (x##i64)\r
-# define Q_UINT64_CONST(x) (x##ui64) /* i64u is not allowed! */\r
-# ifdef _MSC_VER\r
-# define Q_INT64_FORMAT "I64"\r
-# else\r
-# define Q_INT64_FORMAT "L"\r
-# endif\r
-# else\r
-# error No 64 bit integers known for this compiler, edit portdefs.h.\r
-# endif\r
-#endif\r
-\r
+#ifdef unix
+#define O_BINARY 0
+#define setmode(a,b) 0
+#endif
+
+#ifndef __LARGE__
+# if defined(__GNUC__) || defined(unix)
+ typedef long long Q_INT64;
+ typedef unsigned long long Q_UINT64;
+# define Q_INT64_CONST(x) (x##ll)
+# define Q_UINT64_CONST(x) (x##llu) /* gcc also allows ull */
+ /* When using MINGW with MS(V)CRT DLL, use MS format modifier. */
+# ifdef __MSVCRT__
+# define Q_INT64_FORMAT "I64"
+# else
+# define Q_INT64_FORMAT "L"
+# endif
+# elif defined(__BORLANDC__) || defined(__WATCOMC__) || defined(_MSC_VER)
+ typedef __int64 Q_INT64;
+ typedef unsigned __int64 Q_UINT64;
+# define Q_INT64_CONST(x) (x##i64)
+# define Q_UINT64_CONST(x) (x##ui64) /* i64u is not allowed! */
+# ifdef _MSC_VER
+# define Q_INT64_FORMAT "I64"
+# else
+# define Q_INT64_FORMAT "L"
+# endif
+# else
+# error No 64 bit integers known for this compiler, edit portdefs.h.
+# endif
+#endif
+
bool clitk::XdrImageIO::CanWriteFile(const char* FileNameToWrite)
{ std::string filename(FileNameToWrite);
std::string filenameext = GetExtension(filename);
}
void clitk::XdrImageIO::Write(const void* buffer)
-{ char *s = "";
+{ char *s = const_cast<char*>("");
WriteImage( m_FileName.c_str(), s, s, 0, -1, 0, 2, 0, 0, 0, 0, buffer);
}
-\r
-/************************************************************************/\r
-/* */\r
-/* file : AVS_WXDR.C */\r
-/* */\r
-/* purpose : AVS module for writing XDR and RAW files */\r
-/* */\r
-/* author : Lambert Zijp (based on a true story by */\r
-/* Marcel van Herk) */\r
-/* */\r
-/* date : 19970716 */\r
-/* */\r
-/* portability: AVS requires sizeof(void *)==sizeof(int) */\r
-/* This module assumes sizeof(int)>=4 */\r
-/* */\r
-/* notes : This module has been integrated in QUIRT */\r
-/* */\r
-/************************************************************************/\r
-/* Updates:\r
-When Who What\r
-19980212 ljz Creation (i.e. Removed from 'mbfield4.c')\r
-19980304 ljz Added QUIRT_NEXT_PARAMETER_FILE\r
-19980310 ljz Fix: XDRs were written in text-mode when\r
- quirt_init() is not called\r
-19980319 ljz Undone last change: _fmode is set in DllEntryPoint\r
-19980408 tp Moved M$-specific includes;\r
- removed AVS portability anyway\r
-19980611 mvh Added offset to write_raw (write into file)\r
-20000214 lsp Check file handle returned by open before use\r
-20000313 nd Two AVSerror messages added.\r
-20000322 lsp Matched compute func declaration with desc func\r
-20000630 ljz Added WRITE_RAW_LE, to write little-endian pixel-data\r
-20000806 lsp Compute func now 'int', conformal to ANSI C++\r
-20000821 ljz Added NkiCompression parameter to WRITE_XDR. If used,\r
- mode 2 is recommended.\r
-20000504 mvh Added write coordinates in header option to WRITE_XDR\r
-20010720 mvh Added option to write to stdout: pass "" as filename\r
-20010723 mvh Made it working by redefining write; setmode does not work\r
-20010725 bb Changed include order of windows.h and mbavs2q.h\r
- for MS6 compiler.\r
-20020124 mvh+kg The datatype in the header is now e.g. "xdr_integer"\r
- The datatypes as "integer" were intended as having native\r
- byte order, while we wrote bigendian (high byte first). Our\r
- older files read therefore OK on HP and SUN but not on linux.\r
-20020124 mvh+kg On AVS5 xdr_byte is illegal. Replace only that one by 'byte'\r
-20030311 bb Added check on write errors. Disc full errors during a\r
- write have nasty side-effects for other programs' open files.\r
- Added check on file handle to prevent closing of stdout.\r
-20030430 mvh Added append mode for WRITE_XDR\r
-20030717 ljz Added support for NkiCompressionModes 3 and 4 (4 is recommended)\r
-20040426 mvh ELEKTA NKI-XVI0.1 RELEASE\r
-20040910 mvh Write uncompressed and warn if the compress malloc fails, layout\r
-20040920 mvh Fixed the above option (failed in append mode)\r
-20040924 mvh Fixed warn in option\r
-20041117 mvh ELEKTA NKI-XVI0.1g RELEASE\r
-20050302 ljz Merant tracker ID #1867: Check on bad compression-ratio causing\r
- access violation.\r
-20050308 ljz+mvh ELEKTA NKI-XVI0.1j RELEASE\r
-20050411 mvh Fails on I178_s1_4H.3d mode 1: compression fails and written file corrupted\r
-20060903 mvh Added WRITE_MEM_XDR and QUERY_MEM_XDR\r
-20070330 mvh WRITE_XDR failed on G:\20605104_hypo lung\20605104_incl corr+doseab3.PACK\r
-20071015 mvh ELEKTA NKI-XVI3.08 RELEASE\r
-20080411 lsp+mw NULL->0 in WriteImage() ; __sun__ doesn't know <io.h>\r
-20080825 mgw Corrected slash in sys/stat.h include\r
-20081031 lsp+mvh ELEKTA NKI-XVI4.15 RELEASE\r
-20081119 lsp+sr Removed tell(f) call for Linux\r
-20081203 lsp __sun__ -> unix\r
-20090114 mvh ELEKTA NKI-XVI4.22 RELEASE\r
-20090529 lsp+sr Work around the 64 MB limitation of write() in Windows\r
-20090802 mvh ELEKTA NKI-XVI4.29 RELEASE\r
-20091209 lsp Added WRITE_MEM_RAW\r
-20091214 lsp 64 bits adaptations: pass high address of buffer as well,\r
- replaced int by AVSINT when requested, process buffer in chunks\r
- to be able to interpret return value of write() (signed int) correctly,\r
- replaced out-of-range constants 0xc0 and 0x80 by their signed equivalents\r
-20091216 lsp+ljz Prepared compression for more than 4294967295 (UINT_MAX) shorts\r
- by using bigger NKI_MODE2_64BITS struct (backwards compatible)\r
-20091216 lsp Disabled #define write for clarity: checked_write() is used throughout\r
- Use more space for pCompressed to be able to call WRITE_XDR with compression \r
- on small fields (<10 pixels for mode 2 or 4, <2 pixels for mode 1 and 3)\r
-*/\r
-\r
-/************************************************************************/\r
-/* MODULE DOCUMENTATION */\r
-/************************************************************************/\r
-/*\r
-AVS Modules Lambert Zijp XDR writer & RAW writer\r
-\r
-NAME\r
- XDR writer - Module for writing XDR and RAW files\r
-\r
-SUMMARY\r
- Name XDR writer\r
-\r
- Availability : specify in which module libraries\r
-\r
- Source AVS_WXDR (C++ interface)\r
-\r
- Type Render / Output\r
-\r
- QUIRT name WRITE_XDR & WRITE_RAW\r
-\r
- Inputs Input field = field\r
-\r
- Outputs none\r
-\r
- Parameters Name Type Default Min Max\r
- File name string\r
- Header info string (WRITE_XDR only!)\r
- Header file string (WRITE_XDR only!)\r
- File Offset integer (WRITE_RAW only!)\r
- NKI compression integer (WRITE_XDR only!)\r
- Coords in header boolean (WRITE_XDR only!)\r
- Append boolean (WRITE_XDR only!)\r
-\r
-DESCRIPTION\r
- WRITE_RAW: No header, no coordinates.\r
- Data only are written (high byte first); optionally into an\r
- existing file (if you specify an offset)\r
-\r
- WRITE_RAW_LE: No header, no coordinates.\r
- Data only are written (low byte first);\r
- optionally into an existing file (if you specify an offset)\r
-\r
- WRITE_XDR: Successively is written to file:\r
- - The string '#AVS wants ...'.\r
- - The optional Header info\r
- - The contents of the optional Header file\r
- - An ascii description of the Input field\r
- - Optionally the coordinates (coord%axis%[%pixel%]=%coord%)\r
- (%axis% is 1 based, %pixel% 0 based, %coord% is float)\r
- - Two bytes containing ascii character 0x0c\r
- - The Data in binary (high byte first).\r
- Or, if NKI_Compression is greater than zero, compressed data.\r
- - The Coordinates in binary IEEE float (high byte first)\r
-\r
-INPUTS\r
- Input field (Required; field)\r
- Describe input here....\r
-\r
-PARAMETERS\r
- File name\r
- A string described here....\r
-\r
- Header info\r
- A string described here....\r
-\r
- Header file\r
- A string described here....\r
-\r
- File offset\r
- An integer (default -1). If set, WRITE_RAW writes the data\r
- into an existing file.\r
-\r
- NKI compression\r
- An integer described here....\r
-\r
- Write coordinates in header\r
- A boolean described here....\r
-\r
- Append to file\r
- A boolean described here....\r
-\r
-OUTPUTS\r
- none\r
-\r
-PORTABILITY\r
- QUIRT, C++\r
-\r
-LIBRARIES\r
- (optional): in which AVS libraries available\r
-\r
-TYPE\r
- Render / Output\r
- (optional): further specify type of Render / Output module\r
-\r
-FILE FORMATS\r
- (optional): which type of files are read|written\r
-\r
-EXAMPLE\r
- The following network shows how .....\r
-\r
- READ IMA\r
- |\r
- |\r
- XDR WRITER\r
- | |\r
- | |\r
-\r
-LIMITATIONS\r
- (optional): describe limitations here\r
-\r
-RELATED MODULES\r
- Modules that can provide input:\r
- .....\r
- Modules that could be used in place of XDR writer:\r
- .....\r
- Modules that can take output:\r
- .....\r
-\r
-RELATED FILES\r
- (optional): The following files\r
- {are needed for|are output from|give examples of|further document}\r
- XDR writer:\r
- .....\r
-\r
-\r
-QUIRT MANUAL\r
- WRITE_XDR AVS module XDR writer\r
- numerical_expression Input field = field\r
- string_expression name of xdr file to create\r
- string_expression extra header info (default "")\r
- string_expression text file with more header info (default none)\r
- numerical_expression NKI compression (default 0)\r
- numerical_expression Coordinates in header (boolean, default 0)\r
- numerical_expression Append to file (boolean, default 0)\r
-\r
- WRITE_RAW AVS module Raw writer\r
- numerical_expression Input field = field\r
- string_expression name of file to write\r
- numerical_expression Offset; if not -1 (default) writes into file at offset\r
-\r
- WRITE_RAW_LE AVS module Raw writer little endian\r
- numerical_expression Input field = field\r
- string_expression name of file to write\r
- numerical_expression Offset; if not -1 (default) writes into file at offset\r
-\r
- WRITE_MEM_XDR\r
- numerical_expression Input field = field\r
- numerical_expression point to buffer to write xdr data into (use QUERY_MEM_XDR to find length)\r
- string_expression extra header info (default "")\r
- string_expression text file with more header info (default none)\r
- numerical_expression NKI compression (default 0)\r
- numerical_expression Coordinates in header (boolean, default 0)\r
-\r
- QUERY_MEM_XDR\r
- numerical_expression Input field = field\r
- name recieves required length for data\r
- string_expression extra header info (default "")\r
- string_expression text file with more header info (default none)\r
- numerical_expression NKI compression (default 0)\r
- numerical_expression Coordinates in header (boolean, default 0)\r
-\r
-Release 1.0 19980212 Lambert Zijp XDR writer & RAW writer\r
-*/\r
-/************************************************************************/\r
-/* INCLUDE FILES */\r
-/************************************************************************/\r
-\r
-#include <string.h>\r
-#include <stdio.h>\r
-#include <math.h>\r
-#include <stdlib.h>\r
-#include <limits.h>\r
-#ifndef unix\r
-#include <io.h>\r
-#endif\r
-#include <fcntl.h>\r
-#include <errno.h>\r
-\r
-#include <algorithm>\r
-\r
-#ifdef WIN32\r
-// don't use min() and max() macros indirectly defined by windows.h, \r
-// but use portable std::min() and std:max() instead\r
-#ifndef NOMINMAX\r
-#define NOMINMAX\r
-#endif\r
-#include <windows.h>\r
-#endif\r
-\r
-/************************************************************************/\r
-/* DEFINES, ENUMERATED TYPES AND CONSTANTS */\r
-/************************************************************************/\r
-\r
-#undef fileno
-#define fileno _fileno
-
-#undef close
-#define close _close
-
-#undef open
-#define open _open
-
-#undef lseek
-#define lseek _lseek
-
-#undef creat
-#define creat _creat
-
-#pragma pack (1)\r
-\r
-// Fields with data size>8GB (having UINT_MAX short pixels) cannot be compressed using\r
-// NKI_MODE2 struct because iOrgSize has type "unsigned int". In that case use NKI_MODE2_64BITS.\r
-// The type of structure is indicated as follows:\r
-//\r
-// iOrgSize==0: NKI_MODE2_64BITS\r
-// otherwise : NKI_MODE2\r
-//\r
-// Compression modes 1 and 3 (without CRCs) only use the first 2 members (iOrgSize and iMode).\r
-\r
-typedef struct\r
-{\r
- unsigned int iOrgSize; /* in pixels (i.e. shorts) */\r
- unsigned int iMode; /* 1, 2, 3 or 4 */\r
- unsigned int iCompressedSize; /* in bytes, excluding header */\r
- unsigned int iOrgCRC; /* CRC of the data (no coords etc) */\r
- unsigned int iCompressedCRC; /* CRC of the compressed data, excluding this header */\r
-} NKI_MODE2;\r
-\r
-typedef struct\r
-{\r
- unsigned int iOrgSize; /* in pixels (i.e. shorts) */\r
- unsigned int iMode; /* 1, 2, 3 or 4 */\r
- unsigned int iCompressedSize; /* in bytes, excluding header */\r
- unsigned int iOrgCRC; /* CRC of the data (no coords etc) */\r
- unsigned int iCompressedCRC; /* CRC of the compressed data, excluding this header */\r
- unsigned int iPad; /* unused */\r
- Q_UINT64 i64OrgSize; /* used for more than UINT_MAX pixels, indicated by iOrgSize==0 (0-vector not compressed) */ \r
- Q_UINT64 i64CompressedSize; /* value in BYTES, used for more than UINT_MAX PIXELS, indicated by iCompressedSize==0 */ \r
- Q_UINT64 i64Future1;\r
- Q_UINT64 i64Future2;\r
-} NKI_MODE2_64BITS;\r
-\r
-#pragma pack ()\r
-\r
-// Changed next to static function in stead of macro so it can\r
-// have a return value to check in the calling function.\r
-// It could be made inline as well, but there is no real time\r
-// punishment from the extra layer of function calls.\r
-\r
-// note: some compilers do not like comments ending in a backslash.\r
-// so use macro functions to exclude.\r
-\r
-\r
-/************************************************************************/\r
-/* GLOBAL VARIABLES */\r
-/************************************************************************/\r
-\r
-static const unsigned long CRC32_table[256] = {\r
- 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,\r
- 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,\r
- 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,\r
- 0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,\r
- 0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,\r
- 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,\r
- 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,\r
- 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,\r
- 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,\r
- 0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,\r
- 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,\r
- 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,\r
- 0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,\r
- 0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,\r
- 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,\r
- 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,\r
- 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,\r
- 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,\r
- 0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,\r
- 0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,\r
- 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,\r
- 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,\r
- 0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,\r
- 0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,\r
- 0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,\r
- 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,\r
- 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,\r
- 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,\r
- 0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,\r
- 0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,\r
- 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,\r
- 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,\r
- 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,\r
- 0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,\r
- 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,\r
- 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,\r
- 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,\r
- 0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,\r
- 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,\r
- 0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,\r
- 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,\r
- 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,\r
- 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d\r
-};\r
-\r
-/************************************************************************/\r
-/* MODULE FUNCTIONS */\r
-/************************************************************************/\r
-\r
-#ifdef __WATCOMC__\r
-_WCRTLINK\r
-#endif\r
-int writefix(int file, const void *buf, unsigned int count)\r
-{ int i, j, k, total=0;\r
-\r
- for (i=0; i<count; i+=16384)\r
- { j = count - i;\r
- if (j>16384) j=16384;\r
-\r
- k=_write(file, (char *)buf+i, j);\r
- if (k < 0) return k;\r
-\r
- total += k;\r
-\r
- if (k != j) break;\r
- }\r
-\r
- return total;\r
-}\r
-\r
-\r
-/*\r
- Version of write() that takes special action in case of\r
- standard output. Based on commented out macro above.\r
- This function overloads the <cstdio> (or stdio.h for old style C++)\r
- write() function.\r
-*/\r
-\r
-// Like the original macro, we do /not/ want writefix from mbfield.c.\r
-#ifdef write\r
-#undef write\r
-#endif\r
-static int wxdr_write(int handle, const void * buf, unsigned len)\r
-{\r
- // if (handle == 1) // stdout\r
- if (handle == fileno(stdout))\r
- {\r
-#ifdef WIN32\r
- // Behave as C standard library write(): return number of bytes\r
- // written or -1 and errno set on error.\r
- fflush(stdout);\r
- DWORD dwBytesWritten;\r
- if (!WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), buf, len,\r
- &dwBytesWritten, NULL))\r
- {\r
- // There is no simple 1-to-1 mapping between GetLastError()\r
- // values that WriteFile() can return (quite a lot) and the two\r
- // errno values that write() can return. So return EACCES in\r
- // almost all cases.\r
- switch (GetLastError())\r
- { case ERROR_INVALID_HANDLE:\r
- errno = EBADF ; break;\r
- default:\r
- errno = EACCES; break;\r
- }\r
- return -1;\r
- }\r
- else\r
- return (int)dwBytesWritten; // May still be < len!\r
- // And... write() may write a maximum of UINT_MAX-1 bytes, whereas\r
- // WriteFile() may write UINT_MAX bytes at once. But since\r
- // int(UINT_MAX) == -1 this will pose an actual problem in the\r
- // (far?) future.\r
-#else // !WIN32\r
- const int oldmode = setmode(handle, O_BINARY);\r
- const int iBytesWritten = write(handle, buf, len);\r
- const int saveerrno = errno; // setmode() may change errno.\r
- if (oldmode != -1) setmode(handle, oldmode);\r
- errno = saveerrno;\r
- return iBytesWritten;\r
-#endif // !WIN32\r
- }\r
- else\r
- return _write(handle, buf, len);\r
-}\r
-\r
-/*\r
- Checked write().\r
- Behaves like win32 WriteFile() and returns a Boolean to indicate\r
- success or failure, where failure almost invariably means disc full.\r
-\r
- !!! SIDE EFFECT !!!\r
-\r
- In case of failure, this function issues an AVS error message\r
- and closes the file (if handle != 1). It is up to the calling\r
- function to return the AVS_ERROR state and before that do things\r
- like close other files, free memory etc. This way, there is less\r
- chance of erroneously duplicated code, like in:\r
- written = write(f, temp, strlen(temp));\r
- if (written == -1 || written != strlen(temp))\r
- { AVSerror(...);\r
- if (f != fileno(stdout)) close(f);\r
+
+/************************************************************************/
+/* */
+/* file : AVS_WXDR.C */
+/* */
+/* purpose : AVS module for writing XDR and RAW files */
+/* */
+/* author : Lambert Zijp (based on a true story by */
+/* Marcel van Herk) */
+/* */
+/* date : 19970716 */
+/* */
+/* portability: AVS requires sizeof(void *)==sizeof(int) */
+/* This module assumes sizeof(int)>=4 */
+/* */
+/* notes : This module has been integrated in QUIRT */
+/* */
+/************************************************************************/
+/* Updates:
+When Who What
+19980212 ljz Creation (i.e. Removed from 'mbfield4.c')
+19980304 ljz Added QUIRT_NEXT_PARAMETER_FILE
+19980310 ljz Fix: XDRs were written in text-mode when
+ quirt_init() is not called
+19980319 ljz Undone last change: _fmode is set in DllEntryPoint
+19980408 tp Moved M$-specific includes;
+ removed AVS portability anyway
+19980611 mvh Added offset to write_raw (write into file)
+20000214 lsp Check file handle returned by open before use
+20000313 nd Two AVSerror messages added.
+20000322 lsp Matched compute func declaration with desc func
+20000630 ljz Added WRITE_RAW_LE, to write little-endian pixel-data
+20000806 lsp Compute func now 'int', conformal to ANSI C++
+20000821 ljz Added NkiCompression parameter to WRITE_XDR. If used,
+ mode 2 is recommended.
+20000504 mvh Added write coordinates in header option to WRITE_XDR
+20010720 mvh Added option to write to stdout: pass "" as filename
+20010723 mvh Made it working by redefining write; setmode does not work
+20010725 bb Changed include order of windows.h and mbavs2q.h
+ for MS6 compiler.
+20020124 mvh+kg The datatype in the header is now e.g. "xdr_integer"
+ The datatypes as "integer" were intended as having native
+ byte order, while we wrote bigendian (high byte first). Our
+ older files read therefore OK on HP and SUN but not on linux.
+20020124 mvh+kg On AVS5 xdr_byte is illegal. Replace only that one by 'byte'
+20030311 bb Added check on write errors. Disc full errors during a
+ write have nasty side-effects for other programs' open files.
+ Added check on file handle to prevent closing of stdout.
+20030430 mvh Added append mode for WRITE_XDR
+20030717 ljz Added support for NkiCompressionModes 3 and 4 (4 is recommended)
+20040426 mvh ELEKTA NKI-XVI0.1 RELEASE
+20040910 mvh Write uncompressed and warn if the compress malloc fails, layout
+20040920 mvh Fixed the above option (failed in append mode)
+20040924 mvh Fixed warn in option
+20041117 mvh ELEKTA NKI-XVI0.1g RELEASE
+20050302 ljz Merant tracker ID #1867: Check on bad compression-ratio causing
+ access violation.
+20050308 ljz+mvh ELEKTA NKI-XVI0.1j RELEASE
+20050411 mvh Fails on I178_s1_4H.3d mode 1: compression fails and written file corrupted
+20060903 mvh Added WRITE_MEM_XDR and QUERY_MEM_XDR
+20070330 mvh WRITE_XDR failed on G:\20605104_hypo lung\20605104_incl corr+doseab3.PACK
+20071015 mvh ELEKTA NKI-XVI3.08 RELEASE
+20080411 lsp+mw NULL->0 in WriteImage() ; __sun__ doesn't know <io.h>
+20080825 mgw Corrected slash in sys/stat.h include
+20081031 lsp+mvh ELEKTA NKI-XVI4.15 RELEASE
+20081119 lsp+sr Removed tell(f) call for Linux
+20081203 lsp __sun__ -> unix
+20090114 mvh ELEKTA NKI-XVI4.22 RELEASE
+20090529 lsp+sr Work around the 64 MB limitation of write() in Windows
+20090802 mvh ELEKTA NKI-XVI4.29 RELEASE
+20091209 lsp Added WRITE_MEM_RAW
+20091214 lsp 64 bits adaptations: pass high address of buffer as well,
+ replaced int by AVSINT when requested, process buffer in chunks
+ to be able to interpret return value of write() (signed int) correctly,
+ replaced out-of-range constants 0xc0 and 0x80 by their signed equivalents
+20091216 lsp+ljz Prepared compression for more than 4294967295 (UINT_MAX) shorts
+ by using bigger NKI_MODE2_64BITS struct (backwards compatible)
+20091216 lsp Disabled #define write for clarity: checked_write() is used throughout
+ Use more space for pCompressed to be able to call WRITE_XDR with compression
+ on small fields (<10 pixels for mode 2 or 4, <2 pixels for mode 1 and 3)
+*/
+
+/************************************************************************/
+/* MODULE DOCUMENTATION */
+/************************************************************************/
+/*
+AVS Modules Lambert Zijp XDR writer & RAW writer
+
+NAME
+ XDR writer - Module for writing XDR and RAW files
+
+SUMMARY
+ Name XDR writer
+
+ Availability : specify in which module libraries
+
+ Source AVS_WXDR (C++ interface)
+
+ Type Render / Output
+
+ QUIRT name WRITE_XDR & WRITE_RAW
+
+ Inputs Input field = field
+
+ Outputs none
+
+ Parameters Name Type Default Min Max
+ File name string
+ Header info string (WRITE_XDR only!)
+ Header file string (WRITE_XDR only!)
+ File Offset integer (WRITE_RAW only!)
+ NKI compression integer (WRITE_XDR only!)
+ Coords in header boolean (WRITE_XDR only!)
+ Append boolean (WRITE_XDR only!)
+
+DESCRIPTION
+ WRITE_RAW: No header, no coordinates.
+ Data only are written (high byte first); optionally into an
+ existing file (if you specify an offset)
+
+ WRITE_RAW_LE: No header, no coordinates.
+ Data only are written (low byte first);
+ optionally into an existing file (if you specify an offset)
+
+ WRITE_XDR: Successively is written to file:
+ - The string '#AVS wants ...'.
+ - The optional Header info
+ - The contents of the optional Header file
+ - An ascii description of the Input field
+ - Optionally the coordinates (coord%axis%[%pixel%]=%coord%)
+ (%axis% is 1 based, %pixel% 0 based, %coord% is float)
+ - Two bytes containing ascii character 0x0c
+ - The Data in binary (high byte first).
+ Or, if NKI_Compression is greater than zero, compressed data.
+ - The Coordinates in binary IEEE float (high byte first)
+
+INPUTS
+ Input field (Required; field)
+ Describe input here....
+
+PARAMETERS
+ File name
+ A string described here....
+
+ Header info
+ A string described here....
+
+ Header file
+ A string described here....
+
+ File offset
+ An integer (default -1). If set, WRITE_RAW writes the data
+ into an existing file.
+
+ NKI compression
+ An integer described here....
+
+ Write coordinates in header
+ A boolean described here....
+
+ Append to file
+ A boolean described here....
+
+OUTPUTS
+ none
+
+PORTABILITY
+ QUIRT, C++
+
+LIBRARIES
+ (optional): in which AVS libraries available
+
+TYPE
+ Render / Output
+ (optional): further specify type of Render / Output module
+
+FILE FORMATS
+ (optional): which type of files are read|written
+
+EXAMPLE
+ The following network shows how .....
+
+ READ IMA
+ |
+ |
+ XDR WRITER
+ | |
+ | |
+
+LIMITATIONS
+ (optional): describe limitations here
+
+RELATED MODULES
+ Modules that can provide input:
+ .....
+ Modules that could be used in place of XDR writer:
+ .....
+ Modules that can take output:
+ .....
+
+RELATED FILES
+ (optional): The following files
+ {are needed for|are output from|give examples of|further document}
+ XDR writer:
+ .....
+
+
+QUIRT MANUAL
+ WRITE_XDR AVS module XDR writer
+ numerical_expression Input field = field
+ string_expression name of xdr file to create
+ string_expression extra header info (default "")
+ string_expression text file with more header info (default none)
+ numerical_expression NKI compression (default 0)
+ numerical_expression Coordinates in header (boolean, default 0)
+ numerical_expression Append to file (boolean, default 0)
+
+ WRITE_RAW AVS module Raw writer
+ numerical_expression Input field = field
+ string_expression name of file to write
+ numerical_expression Offset; if not -1 (default) writes into file at offset
+
+ WRITE_RAW_LE AVS module Raw writer little endian
+ numerical_expression Input field = field
+ string_expression name of file to write
+ numerical_expression Offset; if not -1 (default) writes into file at offset
+
+ WRITE_MEM_XDR
+ numerical_expression Input field = field
+ numerical_expression point to buffer to write xdr data into (use QUERY_MEM_XDR to find length)
+ string_expression extra header info (default "")
+ string_expression text file with more header info (default none)
+ numerical_expression NKI compression (default 0)
+ numerical_expression Coordinates in header (boolean, default 0)
+
+ QUERY_MEM_XDR
+ numerical_expression Input field = field
+ name recieves required length for data
+ string_expression extra header info (default "")
+ string_expression text file with more header info (default none)
+ numerical_expression NKI compression (default 0)
+ numerical_expression Coordinates in header (boolean, default 0)
+
+Release 1.0 19980212 Lambert Zijp XDR writer & RAW writer
+*/
+/************************************************************************/
+/* INCLUDE FILES */
+/************************************************************************/
+
+#include <string.h>
+#include <stdio.h>
+#include <math.h>
+#include <stdlib.h>
+#include <limits.h>
+#ifndef unix
+#include <io.h>
+#endif
+#include <fcntl.h>
+#include <errno.h>
+
+#include <algorithm>
+
+#ifdef WIN32
+// don't use min() and max() macros indirectly defined by windows.h,
+// but use portable std::min() and std:max() instead
+#ifndef NOMINMAX
+#define NOMINMAX
+#endif
+#include <windows.h>
+#endif
+
+/************************************************************************/
+/* DEFINES, ENUMERATED TYPES AND CONSTANTS */
+/************************************************************************/
+
+//#undef fileno
+//#define fileno _fileno
+//
+//#undef close
+//#define close _close
+//
+//#undef open
+//#define open _open
+//
+//#undef lseek
+//#define lseek _lseek
+//
+//#undef creat
+//#define creat _creat
+//
+#pragma pack (1)
+
+// Fields with data size>8GB (having UINT_MAX short pixels) cannot be compressed using
+// NKI_MODE2 struct because iOrgSize has type "unsigned int". In that case use NKI_MODE2_64BITS.
+// The type of structure is indicated as follows:
+//
+// iOrgSize==0: NKI_MODE2_64BITS
+// otherwise : NKI_MODE2
+//
+// Compression modes 1 and 3 (without CRCs) only use the first 2 members (iOrgSize and iMode).
+
+typedef struct
+{
+ unsigned int iOrgSize; /* in pixels (i.e. shorts) */
+ unsigned int iMode; /* 1, 2, 3 or 4 */
+ unsigned int iCompressedSize; /* in bytes, excluding header */
+ unsigned int iOrgCRC; /* CRC of the data (no coords etc) */
+ unsigned int iCompressedCRC; /* CRC of the compressed data, excluding this header */
+} NKI_MODE2;
+
+typedef struct
+{
+ unsigned int iOrgSize; /* in pixels (i.e. shorts) */
+ unsigned int iMode; /* 1, 2, 3 or 4 */
+ unsigned int iCompressedSize; /* in bytes, excluding header */
+ unsigned int iOrgCRC; /* CRC of the data (no coords etc) */
+ unsigned int iCompressedCRC; /* CRC of the compressed data, excluding this header */
+ unsigned int iPad; /* unused */
+ Q_UINT64 i64OrgSize; /* used for more than UINT_MAX pixels, indicated by iOrgSize==0 (0-vector not compressed) */
+ Q_UINT64 i64CompressedSize; /* value in BYTES, used for more than UINT_MAX PIXELS, indicated by iCompressedSize==0 */
+ Q_UINT64 i64Future1;
+ Q_UINT64 i64Future2;
+} NKI_MODE2_64BITS;
+
+#pragma pack ()
+
+// Changed next to static function in stead of macro so it can
+// have a return value to check in the calling function.
+// It could be made inline as well, but there is no real time
+// punishment from the extra layer of function calls.
+
+// note: some compilers do not like comments ending in a backslash.
+// so use macro functions to exclude.
+
+
+/************************************************************************/
+/* GLOBAL VARIABLES */
+/************************************************************************/
+
+static const unsigned long CRC32_table[256] = {
+ 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,
+ 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
+ 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
+ 0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
+ 0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
+ 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
+ 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,
+ 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
+ 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
+ 0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
+ 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,
+ 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
+ 0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,
+ 0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
+ 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
+ 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
+ 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,
+ 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
+ 0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,
+ 0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
+ 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
+ 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
+ 0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,
+ 0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
+ 0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
+ 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
+ 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
+ 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
+ 0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,
+ 0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
+ 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,
+ 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
+ 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
+ 0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
+ 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
+ 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
+ 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,
+ 0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
+ 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
+ 0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
+ 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,
+ 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
+ 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d
+};
+
+/************************************************************************/
+/* MODULE FUNCTIONS */
+/************************************************************************/
+
+#ifdef __WATCOMC__
+_WCRTLINK
+#endif
+int writefix(int file, const void *buf, unsigned int count)
+{ int j, k, total=0;
+
+ for (unsigned i=0; i<count; i+=16384)
+ { j = count - i;
+ if (j>16384) j=16384;
+
+ k=write(file, (char *)buf+i, j);
+ if (k < 0) return k;
+
+ total += k;
+
+ if (k != j) break;
+ }
+
+ return total;
+}
+
+
+/*
+ Version of write() that takes special action in case of
+ standard output. Based on commented out macro above.
+ This function overloads the <cstdio> (or stdio.h for old style C++)
+ write() function.
+*/
+
+// Like the original macro, we do /not/ want writefix from mbfield.c.
+#ifdef write
+#undef write
+#endif
+static int wxdr_write(int handle, const void * buf, unsigned len)
+{
+ // if (handle == 1) // stdout
+ if (handle == fileno(stdout))
+ {
+#ifdef WIN32
+ // Behave as C standard library write(): return number of bytes
+ // written or -1 and errno set on error.
+ fflush(stdout);
+ DWORD dwBytesWritten;
+ if (!WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), buf, len,
+ &dwBytesWritten, NULL))
+ {
+ // There is no simple 1-to-1 mapping between GetLastError()
+ // values that WriteFile() can return (quite a lot) and the two
+ // errno values that write() can return. So return EACCES in
+ // almost all cases.
+ switch (GetLastError())
+ { case ERROR_INVALID_HANDLE:
+ errno = EBADF ; break;
+ default:
+ errno = EACCES; break;
+ }
+ return -1;
+ }
+ else
+ return (int)dwBytesWritten; // May still be < len!
+ // And... write() may write a maximum of UINT_MAX-1 bytes, whereas
+ // WriteFile() may write UINT_MAX bytes at once. But since
+ // int(UINT_MAX) == -1 this will pose an actual problem in the
+ // (far?) future.
+#else // !WIN32
+ //const int oldmode = setmode(handle, O_BINARY);//commented out by joel
+ const int iBytesWritten = write(handle, buf, len);
+ const int saveerrno = errno; // setmode() may change errno.
+ //if (oldmode != -1) setmode(handle, oldmode); //commented out by joel
+ errno = saveerrno;
+ return iBytesWritten;
+#endif // !WIN32
+ }
+ else
+ return write(handle, buf, len);
+}
+
+/*
+ Checked write().
+ Behaves like win32 WriteFile() and returns a Boolean to indicate
+ success or failure, where failure almost invariably means disc full.
+
+ !!! SIDE EFFECT !!!
+
+ In case of failure, this function issues an AVS error message
+ and closes the file (if handle != 1). It is up to the calling
+ function to return the AVS_ERROR state and before that do things
+ like close other files, free memory etc. This way, there is less
+ chance of erroneously duplicated code, like in:
+ written = write(f, temp, strlen(temp));
+ if (written == -1 || written != strlen(temp))
+ { AVSerror(...);
+ if (f != fileno(stdout)) close(f);
return AVS_ERROR;
- }\r
- written = write(f, buf, buflength)\r
- if (written == -1 || written != strlen(temp)) {\r
- // oops, wrong length copy'n'pasted\r
-\r
- If more elaborate error handling is needed then the calling\r
- functuon should use the (overloaded) write() and act on its return\r
- value (and the value of errno) accordingly.\r
-\r
- It does /not/ close stdout.\r
-\r
- !!! SIDE EFFECT !!!\r
-\r
- Note that checked_write() takes a size_t for len, whereas write() takes\r
- an unsigned int of 4 bytes. On a 64 bits OS a size_t will be an 8 byte integer, \r
- enabling more than UINT_MAX bytes to write at once.\r
-*/\r
-static BOOL checked_write(int handle, const void * buf, size_t len, char **buffer)\r
-{ if (buffer && !handle)\r
- { memcpy(*buffer, buf, len);\r
- (*buffer) += len;\r
- return TRUE;\r
- }\r
- if (buffer && handle)\r
- { (*buffer) += len;\r
- return TRUE;\r
- }\r
- else\r
- { for(int i=0; i<2; i++)\r
- { int byteswritten;\r
- size_t remaining;\r
- int chunksize;\r
-\r
- //If write fails, test if not related to big buffer problem\r
- //Bug report http://support.microsoft.com/kb/899149 entitled\r
- //"You cannot call the fwrite function to write to a buffer\r
- // that is larger than 64 MB in Visual C++ 2005,\r
- // in Visual C++ .NET 2003, or in Visual C++ .NET 2002"\r
- // NB: same thing for write function in binary mode\r
- if (i==0)\r
- { remaining = len;\r
- // call wxdr_write (for handle!=fileno(stdout) a wrapper for write) several times \r
- // to interpret the signed 32-bit return value correctly\r
- while (remaining>0)\r
- { chunksize = (int)std::min(remaining, (size_t)INT_MAX);\r
- byteswritten = wxdr_write(handle, buf, chunksize);\r
- if (byteswritten == chunksize)\r
- remaining -= chunksize;\r
- else\r
- break; // try writefix in the next round\r
- }\r
- if (remaining == 0)\r
- return TRUE;\r
- }\r
- else\r
- { remaining = len; \r
- // call writefix (in mbfield.c) several times to interpret the signed 32-bit \r
- // return value correctly. writefix uses chunks of 16384 bytes\r
- while (remaining>0)\r
- { chunksize = (int)std::min(remaining, (size_t)INT_MAX);\r
- byteswritten = writefix(handle, buf, chunksize);\r
- if (byteswritten == chunksize)\r
- remaining -= chunksize;\r
- else\r
- break; // even writefix failed: return error \r
- }\r
- if (remaining == 0)\r
- return TRUE;\r
- }\r
- }\r
- // Note: file is open in binary mode, no need to compensate\r
- // for a value of byteswritten > len due to \n -> \r\n conversions.\r
- // (write() on a text stream is implementation dependent.)\r
- if (handle != fileno(stdout)) close(handle);\r
- AVSerror("Avs_wxdr: write failed, disk full?");\r
- return FALSE;\r
- }\r
-}\r
-\r
-/* coder for NKI private compressed pixel data\r
- arguments: dest = (in) points to area where compressed destination data is written (byte)\r
- src = (in) points to uncompressed source data (short)\r
- npixels = (in) number of pixels to compress\r
-\r
- The return value is the number of bytes in the compressed data (maximal 3*npixels+10, typical 0.52*npixels)\r
-\r
- if iMode == 1 then\r
- - The first 4 bytes contain the number of short-int pixels\r
- - The following 4 bytes contain iMode=1\r
- - The rest is the compressed image\r
-\r
- if iMode == 2 then\r
- - The first 4 bytes contain the number of short-int pixels\r
- - The following 4 bytes contain iMode=2\r
- - The following 4 bytes contain the size of the compressed image (in bytes)\r
- - The following 4 bytes contain the CRC of the original image\r
- - The following 4 bytes contain the CRC of the compressed image\r
- - The rest is the compressed image\r
- - The compressed size will be even (padded by a zero if necessary).\r
-\r
- if iMode == 3 then\r
- - The first 4 bytes contain the number of short-int pixels\r
- - The following 4 bytes contain iMode=3\r
- - The rest is the compressed image, including 4 bit differences\r
-\r
- if iMode == 4 then\r
- - The first 4 bytes contain the number of short-int pixels\r
- - The following 4 bytes contain iMode=4\r
- - The following 4 bytes contain the size of the compressed image (in bytes)\r
- - The following 4 bytes contain the CRC of the original image\r
- - The following 4 bytes contain 0\r
- - The rest is the compressed image, including 4 bit differences\r
- - The compressed size will be even (padded by a zero if necessary).\r
-\r
- iMode 1 and iMode 2 are identical, except for the CRC data that is included for iMode 2\r
- iMode 3 and iMode 4 are identical, except for the CRC data that is included for iMode 4\r
-*/\r
-\r
-// optimized settings for the 4 bit run compressor (mode 3 and 4)\r
-\r
-#define MINZEROS 5 // shortest RLE (2 byte overhead, but breaks 4bit run)\r
-#define MIN4BIT 6 // shortest 4 bit run (6 bytes compressed to 5 bytes)\r
-\r
-// This internal routine converts an 8 bit difference string into a 4 bit one\r
-static signed char *recompress4bit(int n, signed char *dest)\r
-{ signed char *p, *q;\r
- int val;\r
-\r
- n = n & 0xfe;\r
- dest -= n;\r
- p = dest;\r
- val = (((int)p[0])<<4) | (p[1]&15);\r
- p += 2;\r
- *dest++ = -0x40; // 192 (0xc0) does not fit between -128..127: maps to -64 (0x40) in 2's complement\r
- *dest++ = (signed char)n;\r
- q = dest++;\r
- n -= 2;\r
- while(n>0)\r
- { *dest++ = (signed char)((((int)p[0])<<4) | (p[1]&15));\r
- p += 2;\r
- n -= 2;\r
- }\r
- q[0] = (signed char)val;\r
-\r
- return dest;\r
-}\r
-\r
-\r
-static size_t nki_private_compress(signed char *dest, short int *src, size_t npixels, int iMode)\r
-{ unsigned long iCRC;\r
- unsigned long iCRC2;\r
- unsigned int iHeaderSize=8; // value for iMode==1 and iMode==3\r
- register int val;\r
- size_t i,j;\r
- NKI_MODE2* pHeader = (NKI_MODE2*)dest;\r
- NKI_MODE2_64BITS* pHeader_64bits = (NKI_MODE2_64BITS*)dest;\r
- size_t iBufferSize;\r
-\r
- iBufferSize = (npixels / 2) * 3; // Buffer is sizeof(NKI_MODE2_64BITS) + 10 bytes larger\r
-\r
- /* Up till now only Mode=1 .. 4 are supported */\r
- if ((iMode < 1) || (iMode > 4))\r
- return 0;\r
-\r
- /* Create the header */\r
- pHeader->iMode = iMode;\r
-\r
- if (sizeof(int*)>sizeof(int) && npixels>UINT_MAX) // On a 64 bits OS we want to store files>4GB \r
- { pHeader_64bits->iOrgSize = 0; // This indicates>4GB file (0-vector is not compressed)\r
- pHeader_64bits->i64OrgSize = npixels; \r
- iHeaderSize = sizeof(NKI_MODE2_64BITS);\r
- dest += sizeof(NKI_MODE2_64BITS);\r
- }\r
- else\r
- { pHeader->iOrgSize = (unsigned int)(npixels & UINT_MAX); // store 32 bit number as first member\r
-\r
- if (iMode==2 || iMode==4)\r
- iHeaderSize = sizeof(NKI_MODE2);\r
- dest += iHeaderSize;\r
- }\r
-\r
- /* Create the compressed image */\r
-\r
- if (iMode == 1)\r
- { *(short int *)dest = *src;\r
- dest+=2;\r
-\r
- npixels--;\r
-\r
- do\r
- { val = src[1] - src[0];\r
- src++;\r
-\r
- if (val == 0) /* run length-encode zero differences */\r
- { for (i=2;; i++)\r
- { if (i>=npixels || src[i-1]!=src[-1] || i==256)\r
- { if (i==2)\r
- *dest++=0;\r
- else\r
- { *dest++ = -128; // hexadecimal 0x80\r
- *dest++ = (signed char)(i-1);\r
- npixels -= (i-2);\r
- src += (i-2);\r
- }\r
- break;\r
- }\r
- }\r
- }\r
- else if (val >= -64 && val <= 63) /* small difference coded as one byte */\r
- { *dest = (signed char)val;\r
- dest++;\r
- }\r
- else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */\r
- { dest[0] = (signed char)((val>>8) ^ 0x40);\r
- dest[1] = (signed char)val;\r
- dest+=2;\r
- }\r
- else /* if very large differences code abs val as three bytes */\r
- { *dest++ = 0x7F;\r
- *dest++ = (signed char)(src[0]>>8);\r
- *dest++ = (signed char)(src[0]);\r
- }\r
- /* Are we beyond the allocated memory? */\r
- if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)\r
- return 0;\r
- }\r
- while (--npixels);\r
- }\r
-\r
- else if (iMode == 2)\r
- { iCRC = 0;\r
- iCRC2 = 0;\r
-\r
- *(short int *)dest = val = *src;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char) val ] ^ ((iCRC2 >> 8));\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)(val>>8)] ^ ((iCRC2 >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) val ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(val>>8)] ^ ((iCRC >> 8));\r
- dest+=2;\r
- npixels--;\r
-\r
- do\r
- { val = src[1] - src[0];\r
- src++;\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));\r
-\r
- if (val == 0) /* run length-encode zero differences */\r
- { for (i=2;; i++)\r
- { if (i>=npixels || src[i-1]!=src[-1] || i==256)\r
- { if (i==2)\r
- { *dest++=0;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0 ] ^ ((iCRC2 >> 8));\r
- }\r
- else\r
- { *dest++ = -128; // hexadecimal 0x80\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0x80 ] ^ ((iCRC2 >> 8));\r
- *dest++ = (signed char)(i-1);\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (i-1)] ^ ((iCRC2 >> 8));\r
- npixels -= (i-2);\r
-\r
- for (j=0; j<i-2; j++)\r
- { src++;\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));\r
- }\r
- }\r
- break;\r
- }\r
- }\r
- }\r
- else if (val >= -64 && val <= 63) /* small difference coded as one byte */\r
- { *dest = (signed char)val;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));\r
- dest++;\r
- }\r
- else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */\r
- { dest[0] = (signed char)((val>>8) ^ 0x40);\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)dest[0] ] ^ ((iCRC2 >> 8));\r
- dest[1] = (signed char)val;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));\r
- dest+=2;\r
- }\r
- else /* if very large differences code abs val as three bytes */\r
- { dest[0] = 0x7F;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0x7f ] ^ ((iCRC2 >> 8));\r
- val = src[0];\r
- dest[1] = (signed char)(val>>8);\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)(val>>8)] ^ ((iCRC2 >> 8));\r
- dest[2] = (signed char)val;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));\r
- dest+=3;\r
- }\r
- /* Are we beyond the allocated memory? */\r
- if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)\r
- return 0;\r
- }\r
- while (--npixels);\r
- \r
- if ((dest - (signed char*)pHeader - iHeaderSize)<UINT_MAX) // store 32 bit number as third member\r
- pHeader->iCompressedSize = \r
- (unsigned int)(dest - (signed char*)pHeader - iHeaderSize);\r
- else // store 64 bit number in extended structure\r
- pHeader_64bits->i64CompressedSize = dest - (signed char*)pHeader -iHeaderSize;\r
-\r
- /* Pad it to get an even length */\r
- if (pHeader->iCompressedSize & 1)\r
- { *dest++ = 0;\r
- iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0] ^ ((iCRC2 >> 8));\r
- pHeader->iCompressedSize++;\r
- }\r
-\r
- pHeader->iOrgCRC = iCRC;\r
- pHeader->iCompressedCRC = iCRC2;\r
- }\r
-\r
- /* Create the compressed image - compressor with added 4 bit run */\r
-\r
- else if (iMode == 3)\r
- { int n4bit=0;\r
- *(short int *)dest = *src;\r
- dest+=2;\r
- npixels--;\r
-\r
- do\r
- { val = src[1] - src[0];\r
- src++;\r
-\r
- if (val == 0) /* run length-encode zero differences */\r
- { for (i=2;; i++)\r
- { if (i>=npixels || src[i-1]!=src[-1] || i==256)\r
- { if (i<=MINZEROS) /* too short run -> write zeros */\r
- { for (j=0; j<i-1; j++)\r
- { *dest++=0;\r
- n4bit++;\r
-\r
- if(n4bit>=254) /* maximum length 4 bit run */\r
- { dest = recompress4bit(n4bit, dest);\r
- n4bit = 0;\r
- }\r
- }\r
- }\r
- else\r
- { if (n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- *dest++ = -128; // hexadecimal 0x80\r
- *dest++ = (signed char)(i-1);\r
- }\r
-\r
- npixels -= (i-2);\r
- src += (i-2);\r
- break;\r
- }\r
- }\r
- }\r
- else if (val >= -63 && val <= 63) /* small difference coded as one byte */\r
- { if (val >= -8 && val <= 7)\r
- { *dest++ = (signed char)val;\r
- n4bit++;\r
-\r
- if(n4bit>=254) /* maximum length 4 bit run */\r
- { dest = recompress4bit(n4bit, dest);\r
- n4bit=0;\r
- }\r
- }\r
- else if(n4bit>=MIN4BIT) /* end and write 4 bit run */\r
- { j = val;\r
- dest = recompress4bit(n4bit, dest);\r
- n4bit=0;\r
- *dest++ = (signed char)j;\r
- }\r
- else\r
- { *dest++ = (signed char)val; /* end 4 bit run */\r
- n4bit = 0;\r
- }\r
- }\r
- else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */\r
- { j = val;\r
-\r
- if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- dest[0] = (signed char)((j>>8) ^ 0x40);\r
- dest[1] = (signed char)j;\r
- dest+=2;\r
- }\r
- else /* if very large differences code abs val as three bytes */\r
- { j = src[0];\r
-\r
- if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- *dest++ = 0x7F;\r
- *dest++ = (signed char)(j>>8);\r
- *dest++ = (signed char)j;\r
- }\r
- /* Are we beyond the allocated memory? */\r
- if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)\r
- return 0;\r
- }\r
- while (--npixels);\r
- }\r
-\r
- /* Create the compressed image - compressor with added 4 bit run and CRC */\r
-\r
- else if (iMode == 4)\r
- { int n4bit=0;\r
- iCRC = 0;\r
-\r
- *(short int *)dest = val = *src;\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) val ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(val>>8)] ^ ((iCRC >> 8));\r
- dest+=2;\r
- npixels--;\r
-\r
- do\r
- { val = src[1] - src[0];\r
- src++;\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));\r
-\r
- if (val == 0) /* run length-encode zero differences */\r
- { for (i=2;; i++)\r
- { if (i>=npixels || src[i-1]!=src[-1] || i==256)\r
- { if (i<=MINZEROS) /* too short run -> write zeros */\r
- { for (j=0; j<i-1; j++)\r
- { *dest++=0;\r
- n4bit++;\r
-\r
- if(n4bit>=254) /* maximum length 4 bit run */\r
- { dest = recompress4bit(n4bit, dest);\r
- n4bit = 0;\r
- }\r
- }\r
- }\r
- else\r
- { if (n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- *dest++ = -128; // hexadecimal 0x80\r
- *dest++ = (signed char)(i-1);\r
- }\r
-\r
- npixels -= (i-2);\r
- for (j=0; j<i-2; j++)\r
- { src++;\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));\r
- iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));\r
- }\r
- break;\r
- }\r
- }\r
- }\r
- else if (val >= -63 && val <= 63) /* small difference coded as one byte */\r
- { if (val >= -8 && val <= 7)\r
- { *dest++ = (signed char)val;\r
- n4bit++;\r
-\r
- if(n4bit>=254) /* maximum length 4 bit run */\r
- { dest = recompress4bit(n4bit, dest);\r
- n4bit=0;\r
- }\r
- }\r
- else if(n4bit>=MIN4BIT) /* end and write 4 bit run */\r
- { j = val;\r
- dest = recompress4bit(n4bit, dest);\r
- n4bit=0;\r
- *dest++ = (signed char)j;\r
- }\r
- else\r
- { *dest++ = (signed char)val; /* end 4 bit run */\r
- n4bit = 0;\r
- }\r
- }\r
- else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */\r
- { j = val;\r
-\r
- if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- dest[0] = (signed char)((j>>8) ^ 0x40);\r
- dest[1] = (signed char)j;\r
- dest+=2;\r
- }\r
- else /* if very large differences code abs val as three bytes */\r
- { j = src[0];\r
-\r
- if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */\r
- dest = recompress4bit(n4bit, dest);\r
-\r
- n4bit=0;\r
- *dest++ = 0x7F;\r
- *dest++ = (signed char)(j>>8);\r
- *dest++ = (signed char)j;\r
- }\r
- /* Are we beyond the allocated memory? */\r
- if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)\r
- return 0;\r
- }\r
- while (--npixels);\r
-\r
- if ((dest - (signed char*)pHeader - iHeaderSize)<UINT_MAX) // store 32 bit number as third member\r
- pHeader->iCompressedSize = \r
- (unsigned int)(dest - (signed char*)pHeader - iHeaderSize);\r
- else // store 64 bit number in extended structure\r
- { pHeader_64bits->iCompressedSize = 0;\r
- pHeader_64bits->i64CompressedSize = dest - (signed char*)pHeader -iHeaderSize;\r
- }\r
-\r
- /* Pad it to get an even length */\r
- if (pHeader->iCompressedSize & 1)\r
- { *dest++ = 0;\r
- pHeader->iCompressedSize++;\r
- }\r
-\r
- pHeader->iOrgCRC = iCRC;\r
- pHeader->iCompressedCRC = 0;\r
- }\r
-\r
- return dest - (signed char*)pHeader;\r
-}\r
-\r
-\r
-void clitk::XdrImageIO::WriteImage(const char* file, char* headerinfo, char* headerfile, int raw,\r
- int offset, char bLittleEndian, int iNkiCompression,\r
- int wcoords, int append, int getsize, char *tobuffer, const void* data)\r
-{ AVSINT total=1;\r
- int i;\r
- AVSINT coords=0;\r
- int f=0;\r
- char temp[256];\r
- char *c;\r
- char cSwap;\r
- FILE *fp;\r
- long swap_test = 0x1000000;\r
- signed char* pCompressed = NULL;\r
- size_t FilePos=0;\r
- char **buffer = NULL;\r
- int len=0;\r
- char *buf2;\r
- size_t slen;\r
-\r
- if (bLittleEndian)\r
- swap_test = 0x00000001;\r
-\r
- if (getsize)\r
- { swap_test = 0xffffffff; // never swap to save time\r
- buffer = (char **) &len;\r
- f = 1;\r
- }\r
-\r
- if (tobuffer)\r
- { buf2 = (char *)tobuffer;\r
- buffer = &buf2;\r
- f = 0;\r
- }\r
-\r
- for (i=0; i<GetNumberOfDimensions(); i++)\r
- { total *= GetDimensions(i);\r
- coords += GetDimensions(i);\r
- }\r
-\r
- /* Try allocate the compressed fielddata - compression disabled if alloc fails */\r
- if ((iNkiCompression > 0) &&\r
- (GetComponentType() == itk::ImageIOBase::SHORT) &&\r
- (GetPixelType() == itk::ImageIOBase::SCALAR))\r
- { pCompressed = (signed char *)malloc((total/2) * 3 + sizeof(NKI_MODE2_64BITS) + 10);\r
- if (pCompressed==NULL)\r
- { iNkiCompression = 0;\r
- AVSwarning("Avs_wxdr: not enough memory to compress data, saving uncompressed");\r
- }\r
- }\r
-\r
- if (!(tobuffer || getsize))\r
- { if (offset != -1)\r
- { f = open(file, O_RDWR, 0);\r
- if (f < 0)\r
- {\r
- AVSerror("Avs_wxdr: Opening " << file << "failed.\n" << strerror(errno));\r
- free(pCompressed);\r
+ }
+ written = write(f, buf, buflength)
+ if (written == -1 || written != strlen(temp)) {
+ // oops, wrong length copy'n'pasted
+
+ If more elaborate error handling is needed then the calling
+ functuon should use the (overloaded) write() and act on its return
+ value (and the value of errno) accordingly.
+
+ It does /not/ close stdout.
+
+ !!! SIDE EFFECT !!!
+
+ Note that checked_write() takes a size_t for len, whereas write() takes
+ an unsigned int of 4 bytes. On a 64 bits OS a size_t will be an 8 byte integer,
+ enabling more than UINT_MAX bytes to write at once.
+*/
+static bool checked_write(int handle, const void * buf, size_t len, char **buffer)
+{ if (buffer && !handle)
+ { memcpy(*buffer, buf, len);
+ (*buffer) += len;
+ return true;
+ }
+ if (buffer && handle)
+ { (*buffer) += len;
+ return true;
+ }
+ else
+ { for(int i=0; i<2; i++)
+ { int byteswritten;
+ size_t remaining;
+ int chunksize;
+
+ //If write fails, test if not related to big buffer problem
+ //Bug report http://support.microsoft.com/kb/899149 entitled
+ //"You cannot call the fwrite function to write to a buffer
+ // that is larger than 64 MB in Visual C++ 2005,
+ // in Visual C++ .NET 2003, or in Visual C++ .NET 2002"
+ // NB: same thing for write function in binary mode
+ if (i==0)
+ { remaining = len;
+ // call wxdr_write (for handle!=fileno(stdout) a wrapper for write) several times
+ // to interpret the signed 32-bit return value correctly
+ while (remaining>0)
+ { chunksize = (int)std::min(remaining, (size_t)INT_MAX);
+ byteswritten = wxdr_write(handle, buf, chunksize);
+ if (byteswritten == chunksize)
+ remaining -= chunksize;
+ else
+ break; // try writefix in the next round
+ }
+ if (remaining == 0)
+ return true;
+ }
+ else
+ { remaining = len;
+ // call writefix (in mbfield.c) several times to interpret the signed 32-bit
+ // return value correctly. writefix uses chunks of 16384 bytes
+ while (remaining>0)
+ { chunksize = (int)std::min(remaining, (size_t)INT_MAX);
+ byteswritten = writefix(handle, buf, chunksize);
+ if (byteswritten == chunksize)
+ remaining -= chunksize;
+ else
+ break; // even writefix failed: return error
+ }
+ if (remaining == 0)
+ return true;
+ }
+ }
+ // Note: file is open in binary mode, no need to compensate
+ // for a value of byteswritten > len due to \n -> \r\n conversions.
+ // (write() on a text stream is implementation dependent.)
+ if (handle != fileno(stdout)) close(handle);
+ AVSerror("Avs_wxdr: write failed, disk full?");
+ return false;
+ }
+}
+
+/* coder for NKI private compressed pixel data
+ arguments: dest = (in) points to area where compressed destination data is written (byte)
+ src = (in) points to uncompressed source data (short)
+ npixels = (in) number of pixels to compress
+
+ The return value is the number of bytes in the compressed data (maximal 3*npixels+10, typical 0.52*npixels)
+
+ if iMode == 1 then
+ - The first 4 bytes contain the number of short-int pixels
+ - The following 4 bytes contain iMode=1
+ - The rest is the compressed image
+
+ if iMode == 2 then
+ - The first 4 bytes contain the number of short-int pixels
+ - The following 4 bytes contain iMode=2
+ - The following 4 bytes contain the size of the compressed image (in bytes)
+ - The following 4 bytes contain the CRC of the original image
+ - The following 4 bytes contain the CRC of the compressed image
+ - The rest is the compressed image
+ - The compressed size will be even (padded by a zero if necessary).
+
+ if iMode == 3 then
+ - The first 4 bytes contain the number of short-int pixels
+ - The following 4 bytes contain iMode=3
+ - The rest is the compressed image, including 4 bit differences
+
+ if iMode == 4 then
+ - The first 4 bytes contain the number of short-int pixels
+ - The following 4 bytes contain iMode=4
+ - The following 4 bytes contain the size of the compressed image (in bytes)
+ - The following 4 bytes contain the CRC of the original image
+ - The following 4 bytes contain 0
+ - The rest is the compressed image, including 4 bit differences
+ - The compressed size will be even (padded by a zero if necessary).
+
+ iMode 1 and iMode 2 are identical, except for the CRC data that is included for iMode 2
+ iMode 3 and iMode 4 are identical, except for the CRC data that is included for iMode 4
+*/
+
+// optimized settings for the 4 bit run compressor (mode 3 and 4)
+
+#define MINZEROS 5 // shortest RLE (2 byte overhead, but breaks 4bit run)
+#define MIN4BIT 6 // shortest 4 bit run (6 bytes compressed to 5 bytes)
+
+// This internal routine converts an 8 bit difference string into a 4 bit one
+static signed char *recompress4bit(int n, signed char *dest)
+{ signed char *p, *q;
+ int val;
+
+ n = n & 0xfe;
+ dest -= n;
+ p = dest;
+ val = (((int)p[0])<<4) | (p[1]&15);
+ p += 2;
+ *dest++ = -0x40; // 192 (0xc0) does not fit between -128..127: maps to -64 (0x40) in 2's complement
+ *dest++ = (signed char)n;
+ q = dest++;
+ n -= 2;
+ while(n>0)
+ { *dest++ = (signed char)((((int)p[0])<<4) | (p[1]&15));
+ p += 2;
+ n -= 2;
+ }
+ q[0] = (signed char)val;
+
+ return dest;
+}
+
+
+static size_t nki_private_compress(signed char *dest, short int *src, size_t npixels, int iMode)
+{ unsigned long iCRC;
+ unsigned long iCRC2;
+ unsigned int iHeaderSize=8; // value for iMode==1 and iMode==3
+ register int val;
+ size_t i,j;
+ NKI_MODE2* pHeader = (NKI_MODE2*)dest;
+ NKI_MODE2_64BITS* pHeader_64bits = (NKI_MODE2_64BITS*)dest;
+ size_t iBufferSize;
+
+ iBufferSize = (npixels / 2) * 3; // Buffer is sizeof(NKI_MODE2_64BITS) + 10 bytes larger
+
+ /* Up till now only Mode=1 .. 4 are supported */
+ if ((iMode < 1) || (iMode > 4))
+ return 0;
+
+ /* Create the header */
+ pHeader->iMode = iMode;
+
+ if (sizeof(int*)>sizeof(int) && npixels>UINT_MAX) // On a 64 bits OS we want to store files>4GB
+ { pHeader_64bits->iOrgSize = 0; // This indicates>4GB file (0-vector is not compressed)
+ pHeader_64bits->i64OrgSize = npixels;
+ iHeaderSize = sizeof(NKI_MODE2_64BITS);
+ dest += sizeof(NKI_MODE2_64BITS);
+ }
+ else
+ { pHeader->iOrgSize = (unsigned int)(npixels & UINT_MAX); // store 32 bit number as first member
+
+ if (iMode==2 || iMode==4)
+ iHeaderSize = sizeof(NKI_MODE2);
+ dest += iHeaderSize;
+ }
+
+ /* Create the compressed image */
+
+ if (iMode == 1)
+ { *(short int *)dest = *src;
+ dest+=2;
+
+ npixels--;
+
+ do
+ { val = src[1] - src[0];
+ src++;
+
+ if (val == 0) /* run length-encode zero differences */
+ { for (i=2;; i++)
+ { if (i>=npixels || src[i-1]!=src[-1] || i==256)
+ { if (i==2)
+ *dest++=0;
+ else
+ { *dest++ = -128; // hexadecimal 0x80
+ *dest++ = (signed char)(i-1);
+ npixels -= (i-2);
+ src += (i-2);
+ }
+ break;
+ }
+ }
+ }
+ else if (val >= -64 && val <= 63) /* small difference coded as one byte */
+ { *dest = (signed char)val;
+ dest++;
+ }
+ else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */
+ { dest[0] = (signed char)((val>>8) ^ 0x40);
+ dest[1] = (signed char)val;
+ dest+=2;
+ }
+ else /* if very large differences code abs val as three bytes */
+ { *dest++ = 0x7F;
+ *dest++ = (signed char)(src[0]>>8);
+ *dest++ = (signed char)(src[0]);
+ }
+ /* Are we beyond the allocated memory? */
+ if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)
+ return 0;
+ }
+ while (--npixels);
+ }
+
+ else if (iMode == 2)
+ { iCRC = 0;
+ iCRC2 = 0;
+
+ *(short int *)dest = val = *src;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char) val ] ^ ((iCRC2 >> 8));
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)(val>>8)] ^ ((iCRC2 >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) val ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(val>>8)] ^ ((iCRC >> 8));
+ dest+=2;
+ npixels--;
+
+ do
+ { val = src[1] - src[0];
+ src++;
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));
+
+ if (val == 0) /* run length-encode zero differences */
+ { for (i=2;; i++)
+ { if (i>=npixels || src[i-1]!=src[-1] || i==256)
+ { if (i==2)
+ { *dest++=0;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0 ] ^ ((iCRC2 >> 8));
+ }
+ else
+ { *dest++ = -128; // hexadecimal 0x80
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0x80 ] ^ ((iCRC2 >> 8));
+ *dest++ = (signed char)(i-1);
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (i-1)] ^ ((iCRC2 >> 8));
+ npixels -= (i-2);
+
+ for (j=0; j<i-2; j++)
+ { src++;
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));
+ }
+ }
+ break;
+ }
+ }
+ }
+ else if (val >= -64 && val <= 63) /* small difference coded as one byte */
+ { *dest = (signed char)val;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));
+ dest++;
+ }
+ else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */
+ { dest[0] = (signed char)((val>>8) ^ 0x40);
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)dest[0] ] ^ ((iCRC2 >> 8));
+ dest[1] = (signed char)val;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));
+ dest+=2;
+ }
+ else /* if very large differences code abs val as three bytes */
+ { dest[0] = 0x7F;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0x7f ] ^ ((iCRC2 >> 8));
+ val = src[0];
+ dest[1] = (signed char)(val>>8);
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)(val>>8)] ^ ((iCRC2 >> 8));
+ dest[2] = (signed char)val;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ (unsigned char)val ] ^ ((iCRC2 >> 8));
+ dest+=3;
+ }
+ /* Are we beyond the allocated memory? */
+ if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)
+ return 0;
+ }
+ while (--npixels);
+
+ if ((dest - (signed char*)pHeader - iHeaderSize)<UINT_MAX) // store 32 bit number as third member
+ pHeader->iCompressedSize =
+ (unsigned int)(dest - (signed char*)pHeader - iHeaderSize);
+ else // store 64 bit number in extended structure
+ pHeader_64bits->i64CompressedSize = dest - (signed char*)pHeader -iHeaderSize;
+
+ /* Pad it to get an even length */
+ if (pHeader->iCompressedSize & 1)
+ { *dest++ = 0;
+ iCRC2 = CRC32_table[(unsigned char)iCRC2 ^ 0] ^ ((iCRC2 >> 8));
+ pHeader->iCompressedSize++;
+ }
+
+ pHeader->iOrgCRC = iCRC;
+ pHeader->iCompressedCRC = iCRC2;
+ }
+
+ /* Create the compressed image - compressor with added 4 bit run */
+
+ else if (iMode == 3)
+ { int n4bit=0;
+ *(short int *)dest = *src;
+ dest+=2;
+ npixels--;
+
+ do
+ { val = src[1] - src[0];
+ src++;
+
+ if (val == 0) /* run length-encode zero differences */
+ { for (i=2;; i++)
+ { if (i>=npixels || src[i-1]!=src[-1] || i==256)
+ { if (i<=MINZEROS) /* too short run -> write zeros */
+ { for (j=0; j<i-1; j++)
+ { *dest++=0;
+ n4bit++;
+
+ if(n4bit>=254) /* maximum length 4 bit run */
+ { dest = recompress4bit(n4bit, dest);
+ n4bit = 0;
+ }
+ }
+ }
+ else
+ { if (n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ *dest++ = -128; // hexadecimal 0x80
+ *dest++ = (signed char)(i-1);
+ }
+
+ npixels -= (i-2);
+ src += (i-2);
+ break;
+ }
+ }
+ }
+ else if (val >= -63 && val <= 63) /* small difference coded as one byte */
+ { if (val >= -8 && val <= 7)
+ { *dest++ = (signed char)val;
+ n4bit++;
+
+ if(n4bit>=254) /* maximum length 4 bit run */
+ { dest = recompress4bit(n4bit, dest);
+ n4bit=0;
+ }
+ }
+ else if(n4bit>=MIN4BIT) /* end and write 4 bit run */
+ { j = val;
+ dest = recompress4bit(n4bit, dest);
+ n4bit=0;
+ *dest++ = (signed char)j;
+ }
+ else
+ { *dest++ = (signed char)val; /* end 4 bit run */
+ n4bit = 0;
+ }
+ }
+ else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */
+ { j = val;
+
+ if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ dest[0] = (signed char)((j>>8) ^ 0x40);
+ dest[1] = (signed char)j;
+ dest+=2;
+ }
+ else /* if very large differences code abs val as three bytes */
+ { j = src[0];
+
+ if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ *dest++ = 0x7F;
+ *dest++ = (signed char)(j>>8);
+ *dest++ = (signed char)j;
+ }
+ /* Are we beyond the allocated memory? */
+ if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)
+ return 0;
+ }
+ while (--npixels);
+ }
+
+ /* Create the compressed image - compressor with added 4 bit run and CRC */
+
+ else if (iMode == 4)
+ { int n4bit=0;
+ iCRC = 0;
+
+ *(short int *)dest = val = *src;
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) val ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(val>>8)] ^ ((iCRC >> 8));
+ dest+=2;
+ npixels--;
+
+ do
+ { val = src[1] - src[0];
+ src++;
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));
+
+ if (val == 0) /* run length-encode zero differences */
+ { for (i=2;; i++)
+ { if (i>=npixels || src[i-1]!=src[-1] || i==256)
+ { if (i<=MINZEROS) /* too short run -> write zeros */
+ { for (j=0; j<i-1; j++)
+ { *dest++=0;
+ n4bit++;
+
+ if(n4bit>=254) /* maximum length 4 bit run */
+ { dest = recompress4bit(n4bit, dest);
+ n4bit = 0;
+ }
+ }
+ }
+ else
+ { if (n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ *dest++ = -128; // hexadecimal 0x80
+ *dest++ = (signed char)(i-1);
+ }
+
+ npixels -= (i-2);
+ for (j=0; j<i-2; j++)
+ { src++;
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char) src[0] ] ^ ((iCRC >> 8));
+ iCRC = CRC32_table[(unsigned char)iCRC ^ (unsigned char)(src[0]>>8)] ^ ((iCRC >> 8));
+ }
+ break;
+ }
+ }
+ }
+ else if (val >= -63 && val <= 63) /* small difference coded as one byte */
+ { if (val >= -8 && val <= 7)
+ { *dest++ = (signed char)val;
+ n4bit++;
+
+ if(n4bit>=254) /* maximum length 4 bit run */
+ { dest = recompress4bit(n4bit, dest);
+ n4bit=0;
+ }
+ }
+ else if(n4bit>=MIN4BIT) /* end and write 4 bit run */
+ { j = val;
+ dest = recompress4bit(n4bit, dest);
+ n4bit=0;
+ *dest++ = (signed char)j;
+ }
+ else
+ { *dest++ = (signed char)val; /* end 4 bit run */
+ n4bit = 0;
+ }
+ }
+ else if (val >= -0x3F00 && val <= 0x3EFF) /* large differences coded as two bytes */
+ { j = val;
+
+ if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ dest[0] = (signed char)((j>>8) ^ 0x40);
+ dest[1] = (signed char)j;
+ dest+=2;
+ }
+ else /* if very large differences code abs val as three bytes */
+ { j = src[0];
+
+ if(n4bit>=MIN4BIT) /* end (and write) 4 bit run */
+ dest = recompress4bit(n4bit, dest);
+
+ n4bit=0;
+ *dest++ = 0x7F;
+ *dest++ = (signed char)(j>>8);
+ *dest++ = (signed char)j;
+ }
+ /* Are we beyond the allocated memory? */
+ if ((size_t)(dest - (signed char*)pHeader) > iBufferSize)
+ return 0;
+ }
+ while (--npixels);
+
+ if ((dest - (signed char*)pHeader - iHeaderSize)<UINT_MAX) // store 32 bit number as third member
+ pHeader->iCompressedSize =
+ (unsigned int)(dest - (signed char*)pHeader - iHeaderSize);
+ else // store 64 bit number in extended structure
+ { pHeader_64bits->iCompressedSize = 0;
+ pHeader_64bits->i64CompressedSize = dest - (signed char*)pHeader -iHeaderSize;
+ }
+
+ /* Pad it to get an even length */
+ if (pHeader->iCompressedSize & 1)
+ { *dest++ = 0;
+ pHeader->iCompressedSize++;
+ }
+
+ pHeader->iOrgCRC = iCRC;
+ pHeader->iCompressedCRC = 0;
+ }
+
+ return dest - (signed char*)pHeader;
+}
+
+
+void clitk::XdrImageIO::WriteImage(const char* file, char* headerinfo, char* headerfile, int raw,
+ int offset, char bLittleEndian, int iNkiCompression,
+ int wcoords, int append, int getsize, char *tobuffer, const void* data)
+{ AVSINT total=1;
+ unsigned int i;
+ AVSINT coords=0;
+ int f=0;
+ char temp[256];
+ char *c;
+ char cSwap;
+ FILE *fp;
+ long swap_test = 0x1000000;
+ signed char* pCompressed = NULL;
+ size_t FilePos=0;
+ char **buffer = NULL;
+ int len=0;
+ char *buf2;
+ size_t slen;
+
+ if (bLittleEndian)
+ swap_test = 0x00000001;
+
+ if (getsize)
+ { swap_test = 0xffffffff; // never swap to save time
+ buffer = (char **) &len;
+ f = 1;
+ }
+
+ if (tobuffer)
+ { buf2 = (char *)tobuffer;
+ buffer = &buf2;
+ f = 0;
+ }
+
+ for (i=0; i<GetNumberOfDimensions(); i++)
+ { total *= GetDimensions(i);
+ coords += GetDimensions(i);
+ }
+
+ /* Try allocate the compressed fielddata - compression disabled if alloc fails */
+ if ((iNkiCompression > 0) &&
+ (GetComponentType() == itk::ImageIOBase::SHORT) &&
+ (GetPixelType() == itk::ImageIOBase::SCALAR))
+ { pCompressed = (signed char *)malloc((total/2) * 3 + sizeof(NKI_MODE2_64BITS) + 10);
+ if (pCompressed==NULL)
+ { iNkiCompression = 0;
+ AVSwarning("Avs_wxdr: not enough memory to compress data, saving uncompressed");
+ }
+ }
+
+ if (!(tobuffer || getsize))
+ { if (offset != -1)
+ { f = open(file, O_RDWR, 0);
+ if (f < 0)
+ {
+ AVSerror("Avs_wxdr: Opening " << file << "failed.\n" << strerror(errno));
+ free(pCompressed);
return AVS_ERROR;
- }\r
- lseek(f, offset, SEEK_SET);\r
- }\r
- else\r
- { if (strlen(file)==0)\r
- f = fileno(stdout);\r
- else\r
- { if (append)\r
- f = open(file, O_RDWR | O_APPEND, 0);\r
- else\r
- f = creat(file, S_IWRITE | S_IREAD);\r
- }\r
-\r
- if (f < 0)\r
- { AVSerror("Avs_wxdr: Creating " << file << " failed.\n" << strerror(errno));\r
- free(pCompressed);\r
+ }
+ lseek(f, offset, SEEK_SET);
+ }
+ else
+ { if (strlen(file)==0)
+ f = fileno(stdout);
+ else
+ { if (append)
+ f = open(file, O_RDWR | O_APPEND, 0);
+ else
+ f = creat(file, S_IWRITE | S_IREAD);
+ }
+
+ if (f < 0)
+ { AVSerror("Avs_wxdr: Creating " << file << " failed.\n" << strerror(errno));
+ free(pCompressed);
return AVS_ERROR;
- }\r
- }\r
- }\r
-\r
- if (!raw)\r
- { sprintf(temp, "# AVS wants to have the first line starting with its name\n");\r
- slen = strlen(temp);\r
-\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ }
+ }
+
+ if (!raw)
+ { sprintf(temp, "# AVS wants to have the first line starting with its name\n");
+ slen = strlen(temp);
+
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
-\r
- slen = strlen(headerinfo);\r
- if (slen && !checked_write(f, headerinfo, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+
+ slen = strlen(headerinfo);
+ if (slen && !checked_write(f, headerinfo, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
-\r
- if (!checked_write(f, "\n", 1, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+
+ if (!checked_write(f, "\n", 1, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos++;\r
-\r
- if (strlen(headerfile))\r
- { fp = fopen(headerfile, "rt");\r
- if (fp)\r
- { for (;;)\r
- { if (fgets(temp, 255, fp) == NULL) break;\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { fclose(fp);\r
- free(pCompressed);\r
+ }
+ FilePos++;
+
+ if (strlen(headerfile))
+ { fp = fopen(headerfile, "rt");
+ if (fp)
+ { for (;;)
+ { if (fgets(temp, 255, fp) == NULL) break;
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { fclose(fp);
+ free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
- }\r
- fclose(fp);\r
- if (!checked_write(f, "\n", 1, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+ }
+ fclose(fp);
+ if (!checked_write(f, "\n", 1, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos++;\r
- }\r
- }\r
-\r
- sprintf(temp, "ndim=%d\n", GetNumberOfDimensions());\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos++;
+ }
+ }
+
+ sprintf(temp, "ndim=%d\n", GetNumberOfDimensions());
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
- }\r
-\r
- for (i=0; i<GetNumberOfDimensions(); i++)\r
- { if (!raw)\r
- { sprintf(temp, "dim%d=%d\n", i+1, GetDimensions(i));\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+ }
+
+ for (i=0; i<GetNumberOfDimensions(); i++)
+ { if (!raw)
+ { sprintf(temp, "dim%d=%d\n", i+1, GetDimensions(i));
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
- }\r
- }\r
-\r
- if (!raw)\r
- { sprintf(temp, "nspace=%d\n", GetNumberOfDimensions());\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+ }
+ }
+
+ if (!raw)
+ { sprintf(temp, "nspace=%d\n", GetNumberOfDimensions());
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
-\r
- sprintf(temp, "veclen=%d\n", GetNumberOfComponents());\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+
+ sprintf(temp, "veclen=%d\n", GetNumberOfComponents());
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
-\r
- switch(GetComponentType())\r
- { case itk::ImageIOBase::CHAR : strcpy(temp, "data=byte\n"); break;\r
- case itk::ImageIOBase::SHORT : strcpy(temp, "data=xdr_short\n"); break;\r
- case itk::ImageIOBase::INT : strcpy(temp, "data=xdr_integer\n"); break;\r
- case itk::ImageIOBase::FLOAT : strcpy(temp, "data=xdr_real\n"); break;\r
- case itk::ImageIOBase::DOUBLE : strcpy(temp, "data=xdr_double\n"); break;\r
- default : if (f != fileno(stdout)) close(f);\r
- free(pCompressed);\r
+ }
+ FilePos += slen;
+
+ switch(GetComponentType())
+ { case itk::ImageIOBase::CHAR : strcpy(temp, "data=byte\n"); break;
+ case itk::ImageIOBase::SHORT : strcpy(temp, "data=xdr_short\n"); break;
+ case itk::ImageIOBase::INT : strcpy(temp, "data=xdr_integer\n"); break;
+ case itk::ImageIOBase::FLOAT : strcpy(temp, "data=xdr_real\n"); break;
+ case itk::ImageIOBase::DOUBLE : strcpy(temp, "data=xdr_double\n"); break;
+ default : if (f != fileno(stdout)) close(f);
+ free(pCompressed);
return AVS_ERROR;
- }\r
- slen = strlen(temp);\r
- if (!checked_write(f, temp, slen, buffer))\r
- { free(pCompressed);\r
+ }
+ slen = strlen(temp);
+ if (!checked_write(f, temp, slen, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- FilePos += slen;\r
- }\r
-\r
-\r
- //FilePos = tell(f);\r
-ONCE_AGAIN:\r
-\r
-\r
- //switch(input->uniform)\r
- //{ case UNIFORM : \r
- strcpy(temp, "field=uniform\n");\r
- coords = GetNumberOfDimensions() * 2;\r
- // break;\r
- // case RECTILINEAR : strcpy(temp, "field=rectilinear\n");\r
- // break;\r
- // case IRREGULAR : strcpy(temp, "field=irregular\n");\r
- // coords = total * input->nspace;\r
- // break;\r
- // default : if (f != fileno(stdout)) close(f);\r
- // free(pCompressed);\r
- // return;\r
- //}\r
-\r
- if (!raw)\r
- { if (!checked_write(f, temp, strlen(temp), buffer))\r
- { free(pCompressed);\r
+ }
+ FilePos += slen;
+ }
+
+
+ //FilePos = tell(f);
+ONCE_AGAIN:
+
+
+ //switch(input->uniform)
+ //{ case UNIFORM :
+ strcpy(temp, "field=uniform\n");
+ coords = GetNumberOfDimensions() * 2;
+ // break;
+ // case RECTILINEAR : strcpy(temp, "field=rectilinear\n");
+ // break;
+ // case IRREGULAR : strcpy(temp, "field=irregular\n");
+ // coords = total * input->nspace;
+ // break;
+ // default : if (f != fileno(stdout)) close(f);
+ // free(pCompressed);
+ // return;
+ //}
+
+ if (!raw)
+ { if (!checked_write(f, temp, strlen(temp), buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
-\r
- if ((iNkiCompression > 0) &&\r
- (GetComponentType() == itk::ImageIOBase::SHORT) &&\r
- (GetPixelType() == itk::ImageIOBase::SCALAR))\r
- { sprintf(temp, "nki_compression=%d", iNkiCompression);\r
- if (!checked_write(f, temp, strlen(temp), buffer))\r
- { free(pCompressed);\r
- return AVS_ERROR;\r
- }\r
- }\r
-\r
- temp[0] = temp[1] = 12;\r
- if (!checked_write(f, temp, 2, buffer))\r
- { free(pCompressed);\r
+ }
+
+ if ((iNkiCompression > 0) &&
+ (GetComponentType() == itk::ImageIOBase::SHORT) &&
+ (GetPixelType() == itk::ImageIOBase::SCALAR))
+ { sprintf(temp, "nki_compression=%d", iNkiCompression);
+ if (!checked_write(f, temp, strlen(temp), buffer))
+ { free(pCompressed);
+ return AVS_ERROR;
+ }
+ }
+
+ temp[0] = temp[1] = 12;
+ if (!checked_write(f, temp, 2, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- }\r
-\r
- total *= GetPixelSize();\r
-\r
- if ((!raw) && (iNkiCompression > 0) &&\r
- (GetComponentType() == itk::ImageIOBase::SHORT) &&\r
- (GetPixelType() == itk::ImageIOBase::SCALAR))\r
- { size_t iCompressedLength;\r
-\r
- iCompressedLength = nki_private_compress(pCompressed,\r
- (short int *)(data), total/2, iNkiCompression);\r
-\r
- if (iCompressedLength > 0)\r
- { if (!checked_write(f, pCompressed, iCompressedLength, buffer))\r
- { free(pCompressed);\r
+ }
+ }
+
+ total *= GetPixelSize();
+
+ if ((!raw) && (iNkiCompression > 0) &&
+ (GetComponentType() == itk::ImageIOBase::SHORT) &&
+ (GetPixelType() == itk::ImageIOBase::SCALAR))
+ { size_t iCompressedLength;
+
+ iCompressedLength = nki_private_compress(pCompressed,
+ (short int *)(data), total/2, iNkiCompression);
+
+ if (iCompressedLength > 0)
+ { if (!checked_write(f, pCompressed, iCompressedLength, buffer))
+ { free(pCompressed);
return AVS_ERROR;
- }\r
- free(pCompressed);\r
- goto WRITE_COORDS;\r
- }\r
-\r
- /* Compressionratio was poor: let's write uncompressed */\r
- iNkiCompression = 0;\r
- total /= 2;\r
- free(pCompressed);\r
- pCompressed = NULL;\r
- _lseeki64(f, (unsigned int)FilePos, SEEK_SET); // use _lseeki64 just in case header size > UINT_MAX bytes\r
- goto ONCE_AGAIN;\r
- }\r
-\r
- /* swap data if required (xdr is low-endian) */\r
-\r
- if (!(*(char *)(&swap_test)))\r
- { if (GetComponentSize()==2)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=2)\r
- { cSwap = c[i]; c[i] = c[i+1]; c[i+1] = cSwap;\r
- }\r
- }\r
- else if (GetComponentSize()==4)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=4)\r
- { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;\r
- }\r
- }\r
- else if (GetComponentSize()==8)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=8)\r
- { cSwap = c[i]; c[i] = c[i+7]; c[i+7] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+6]; c[i+6] = cSwap;\r
- cSwap = c[i+2]; c[i+2] = c[i+5]; c[i+5] = cSwap;\r
- cSwap = c[i+3]; c[i+3] = c[i+4]; c[i+4] = cSwap;\r
- }\r
- }\r
- }\r
-\r
- if (total)\r
- { if (!checked_write(f, data, total, buffer))\r
- return AVS_ERROR;\r
- }\r
-\r
- /* swap data back if was swapped before writing */\r
-\r
- if (!(*(char *)(&swap_test)))\r
- { if (GetComponentSize()==2)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=2)\r
- { cSwap = c[i]; c[i] = c[i+1]; c[i+1] = cSwap;\r
- }\r
- }\r
- else if (GetComponentSize()==4)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=4)\r
- { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;\r
- }\r
- }\r
- else if (GetComponentSize()==8)\r
- { c = (char *)data;\r
- for (i=0; i<total; i+=8)\r
- { cSwap = c[i]; c[i] = c[i+7]; c[i+7] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+6]; c[i+6] = cSwap;\r
- cSwap = c[i+2]; c[i+2] = c[i+5]; c[i+5] = cSwap;\r
- cSwap = c[i+3]; c[i+3] = c[i+4]; c[i+4] = cSwap;\r
- }\r
- }\r
- }\r
-\r
-WRITE_COORDS:\r
- float *points;\r
- points = (float *)malloc(sizeof(float)*GetNumberOfDimensions()*2);\r
+ }
+ free(pCompressed);
+ goto WRITE_COORDS;
+ }
+
+ /* Compressionratio was poor: let's write uncompressed */
+ iNkiCompression = 0;
+ total /= 2;
+ free(pCompressed);
+ pCompressed = NULL;
+ lseek(f, (unsigned int)FilePos, SEEK_SET); // use _lseeki64 just in case header size > UINT_MAX bytes
+ goto ONCE_AGAIN;
+ }
+
+ /* swap data if required (xdr is low-endian) */
+
+ if (!(*(char *)(&swap_test)))
+ { if (GetComponentSize()==2)
+ { c = (char *)data;
+ for (i=0; i<total; i+=2)
+ { cSwap = c[i]; c[i] = c[i+1]; c[i+1] = cSwap;
+ }
+ }
+ else if (GetComponentSize()==4)
+ { c = (char *)data;
+ for (i=0; i<total; i+=4)
+ { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;
+ }
+ }
+ else if (GetComponentSize()==8)
+ { c = (char *)data;
+ for (i=0; i<total; i+=8)
+ { cSwap = c[i]; c[i] = c[i+7]; c[i+7] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+6]; c[i+6] = cSwap;
+ cSwap = c[i+2]; c[i+2] = c[i+5]; c[i+5] = cSwap;
+ cSwap = c[i+3]; c[i+3] = c[i+4]; c[i+4] = cSwap;
+ }
+ }
+ }
+
+ if (total)
+ { if (!checked_write(f, data, total, buffer))
+ return AVS_ERROR;
+ }
+
+ /* swap data back if was swapped before writing */
+
+ if (!(*(char *)(&swap_test)))
+ { if (GetComponentSize()==2)
+ { c = (char *)data;
+ for (i=0; i<total; i+=2)
+ { cSwap = c[i]; c[i] = c[i+1]; c[i+1] = cSwap;
+ }
+ }
+ else if (GetComponentSize()==4)
+ { c = (char *)data;
+ for (i=0; i<total; i+=4)
+ { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;
+ }
+ }
+ else if (GetComponentSize()==8)
+ { c = (char *)data;
+ for (i=0; i<total; i+=8)
+ { cSwap = c[i]; c[i] = c[i+7]; c[i+7] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+6]; c[i+6] = cSwap;
+ cSwap = c[i+2]; c[i+2] = c[i+5]; c[i+5] = cSwap;
+ cSwap = c[i+3]; c[i+3] = c[i+4]; c[i+4] = cSwap;
+ }
+ }
+ }
+
+WRITE_COORDS:
+ float *points;
+ points = (float *)malloc(sizeof(float)*GetNumberOfDimensions()*2);
for (i=0; i<GetNumberOfDimensions(); i++)
{
points[i*2 ] = 0.1 * GetOrigin(i);
points[i*2+1] = 0.1 * (GetOrigin(i) + GetSpacing(i)*(GetDimensions(i)-1));
}
-\r
- if (coords && !raw) /* write AVS coordinates ? */\r
- { coords *= sizeof(float);\r
- if (!(*(char *)(&swap_test)))\r
- { c = (char *)(points); /* swap bytes */\r
- for (i=0; i<coords; i+=4)\r
- { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;\r
- }\r
- }\r
-\r
- if (!checked_write(f, points, coords, buffer))\r
- return AVS_ERROR;\r
-\r
- if (!(*(char *)(&swap_test)))\r
- { c = (char *)(points); /* swap bytes back */\r
- for (i=0; i<coords; i+=4)\r
- { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;\r
- cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;\r
- }\r
- }\r
- }\r
-\r
- if (!(tobuffer || getsize))\r
- if (f != fileno(stdout)) close(f);\r
-\r
- if (getsize) return;\r
- return AVS_OK;\r
-}\r
+
+ if (coords && !raw) /* write AVS coordinates ? */
+ { coords *= sizeof(float);
+ if (!(*(char *)(&swap_test)))
+ { c = (char *)(points); /* swap bytes */
+ for (i=0; i<coords; i+=4)
+ { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;
+ }
+ }
+
+ if (!checked_write(f, points, coords, buffer))
+ return AVS_ERROR;
+
+ if (!(*(char *)(&swap_test)))
+ { c = (char *)(points); /* swap bytes back */
+ for (i=0; i<coords; i+=4)
+ { cSwap = c[i]; c[i] = c[i+3]; c[i+3] = cSwap;
+ cSwap = c[i+1]; c[i+1] = c[i+2]; c[i+2] = cSwap;
+ }
+ }
+ }
+
+ if (!(tobuffer || getsize))
+ if (f != fileno(stdout)) close(f);
+
+ if (getsize) return;
+ return AVS_OK;
+}