--- /dev/null
+/**
+ * @file clitkXdrImageIO.cxx
+ * @author Simon Rit <simon.rit@gmail.com>
+ * @date Sun Jun 1 22:12:20 2008
+ *
+ * @brief
+ *
+ *
+ */
+
+#include "clitkXdrImageIO.h"
+#include "clitkCommon.h"
+
+#include <sys/stat.h>\r
+
+//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
+//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
+bool clitk::XdrImageIO::CanWriteFile(const char* FileNameToWrite)
+{ std::string filename(FileNameToWrite);
+ std::string filenameext = GetExtension(filename);
+ if (filenameext != std::string("xdr")) return false;
+ return true;
+}
+
+void clitk::XdrImageIO::Write(const void* buffer)
+{ char *s = "";
+ 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
+ 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
+ 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
+ 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
+ return AVS_ERROR;
+ }\r
+ FilePos += slen;\r
+\r
+ slen = strlen(headerinfo);\r
+ if (slen && !checked_write(f, headerinfo, slen, buffer))\r
+ { free(pCompressed);\r
+ return AVS_ERROR;
+ }\r
+ FilePos += slen;\r
+\r
+ if (!checked_write(f, "\n", 1, buffer))\r
+ { free(pCompressed);\r
+ 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
+ return AVS_ERROR;
+ }\r
+ FilePos += slen;\r
+ }\r
+ fclose(fp);\r
+ if (!checked_write(f, "\n", 1, buffer))\r
+ { free(pCompressed);\r
+ 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
+ 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
+ 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
+ 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
+ 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
+ return AVS_ERROR;
+ }\r
+ slen = strlen(temp);\r
+ if (!checked_write(f, temp, slen, buffer))\r
+ { free(pCompressed);\r
+ 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
+ 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
+ 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
+ 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
+ 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
git://git.creatis.insa-lyon.fr / clitk.git / blobdiff