-#ifndef NKITKXDRIMAGEIO_CXX
-#define NKITKXDRIMAGEIO_CXX
-
/**
- * @file nkitkXDRImageIO.cxx
+ * @file clitkXdrImageIO.cxx
* @author Simon Rit <simon.rit@gmail.com>
* @date Sun Jun 1 22:12:20 2008
*
*
*/
-#include "nkitkXDRImageIO.h"
+#include "clitkXdrImageIO.h"
// std include
#include <iostream>
//defines
#define MAXDIM 5
-#define AVSerror(v) std::cerr << "Error in nkitk::XDRImageIO. Message:" << v << std::endl;
#ifdef _WIN32
#ifdef memicmp
#undef memicmp
20080414 lsp+mgw __sun__ doesn't know <io.h>
*/
-/************************************************************************/
-/* MODULE DOCUMENTATION */
-/************************************************************************/
-/*
- READ_XDR Read XDR file (may be compressed) into field
- name output field handle
- file_expression name of file
- numerical_expression start XDR data in file offset (default 0)
- NOTE: compressed XDR data may not be part of a larger file
-
- READ_XDR_HEADER Get entry from xdr header
- %name Output = string block
- file_expression file name default extension ''
- %string_expression name of element to load
-
- READ_XDR_PREVIEW Read and downsize XDR file (for bitmap)
- name output field handle
- file_expression name of file
- numerical_expression start XDR data in file offset (default 0)
- NOTE: this command is not supported for compressed XDR files
-
- ENUM_XDR_HEADER
- %name Output = string block
- file_expression file name default extension ''
- numerical_expression number of element to find name of
-*/
-/************************************************************************/
-/* INCLUDE FILES */
-/************************************************************************/
-
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <ctype.h>
-#if 0
-#ifndef __sun__
-#include <io.h>
-#endif
-#include "mbavs2q.h"
-
-#ifndef QUIRT
-#include <memory.h>
-#include <avs/avs.h>
-#include <avs/field.h>
-#else
-#include "mbfield.h"
-#endif
-#endif
/************************************************************************/
/* DEFINES, ENUMERATED TYPES AND CONSTANTS */
"XDR file reading error",
"Out of memory",
"Decompression failed",
- "Format not handled by nkitkXDRImageIO (RECTILINEAR or IRREGULAR field)"
+ "Format not handled by clitkXdrImageIO (RECTILINEAR or IRREGULAR field)"
};
};
-/************************************************************************/
-/* PROTOTYPES */
-/************************************************************************/
-#if 0
-int Rxdr_compute(AVSfield** ppOut, char* pszFileName, int iOffset);
-
-int RxdrPreview_compute(AVSfield** ppOut, char* pszFileName, int iOffset);
-
-int RxdrHeader_compute(char *pszOut, char *pszFileName, char *pszEntry);
-
-int RxdrEnum_compute(char *pszOut, char *pszFileName, int iEntry);
-
-/******************************************************************************/
-/* AVS/QUIRT INTERFACE */
-/******************************************************************************/
-
-void Rxdr_desc(void)
-{
- int param;
-
- /* Set the module name and type */
- AVSset_module_name("XDR reader", MODULE_DATA);
-
- /* Create output ports for the resulting fields */
- AVScreate_output_port("Output", "field");
-
- /* declare widgets */
- QUIRT_NEXT_PARAMETER_FILE("");
- param = AVSadd_parameter("FileName", "string", "/data/AVS/", NULL, "");
- AVSconnect_widget(param, "browser");
- AVSadd_parameter_prop(param, "height", "integer", 8);
-
- param = AVSadd_parameter("Offset", "integer", 0, 0, INT_UNBOUND);
- AVSconnect_widget(param, "typein_integer");
-
- AVSset_compute_proc(CF Rxdr_compute);
-}
-AVS_TO_QUIRT(READ_XDR, Rxdr_desc);
-
-void RxdrPreview_desc(void)
-{
- int param;
-
- /* Set the module name and type */
- AVSset_module_name("XDR preview reader", MODULE_DATA);
-
- /* Create output ports for the resulting fields */
- AVScreate_output_port("Output", "field");
-
- /* declare widgets */
- QUIRT_NEXT_PARAMETER_FILE("");
- param = AVSadd_parameter("FileName", "string", "/data/AVS/", NULL, "");
- AVSconnect_widget(param, "browser");
- AVSadd_parameter_prop(param, "height", "integer", 8);
-
- param = AVSadd_parameter("Offset", "integer", 0, 0, INT_UNBOUND);
- AVSconnect_widget(param, "typein_integer");
-
- AVSset_compute_proc(CF RxdrPreview_compute);
-}
-AVS_TO_QUIRT(READ_XDR_PREVIEW, RxdrPreview_desc);
-
-void RxdrHeader_desc(void)
-{ /* Set the module name and type */
- AVSset_module_name("Read XDR header", MODULE_DATA);
-
- /* Create output ports for the resulting fields */
- AVSadd_parameter("Output", "string_block", "", NULL, "");
-
- QUIRT_NEXT_PARAMETER_FILE("");
- AVSadd_parameter("FileName", "string", "", NULL, "");
-
- AVSadd_parameter("Entry", "string", "", NULL, "");
-
- AVSset_compute_proc(CF RxdrHeader_compute);
-}
-AVS_TO_QUIRT(READ_XDR_HEADER, RxdrHeader_desc);
-
-void RxdrEnum_desc(void)
-{ /* Set the module name and type */
- AVSset_module_name("Enumerate XDR header", MODULE_DATA);
-
- /* Create output ports for the resulting fields */
- AVSadd_parameter("Output", "string_block", "", NULL, "");
-
- QUIRT_NEXT_PARAMETER_FILE("");
- AVSadd_parameter("FileName", "string", "", NULL, "");
-
- AVSadd_parameter("Entry", "integer", 0, INT_UNBOUND, INT_UNBOUND);
-
- AVSset_compute_proc(CF RxdrEnum_compute);
-}
-AVS_TO_QUIRT(ENUM_XDR_HEADER, RxdrEnum_desc);
-
-#ifndef QUIRT
-AVSinit_modules(void)
-{
- AVSmodule_from_desc( (int_desc_func)Rxdr_desc );
- AVSmodule_from_desc( (int_desc_func)RxdrHeader_desc );
- AVSmodule_from_desc( (int_desc_func)RxdrEnum_desc );
-}
-#endif
-#endif
/************************************************************************/
/* MODULE FUNCTIONS */
/************************************************************************/
return NULL;
}
-/* help routine, enumerates XDR file for names of keyword entrys, returns NULL
- if not found, else returns pointer to start of line upto '='
-*/
-// //Commented out because it does not seem to be needed for vv
-//static char *enum_header(char *file, int iEntry, int offset)
-//{
-// int i, count;
-// static char temp[512];
-// FILE *f;
-// char *p;
-//
-// if ((f = fopen(file, "rt")) == NULL) return NULL;
-// if (offset) fseek(f, offset, SEEK_SET);
-//
-// count = 0;
-//
-// for (i=0; i<200; )
-// {
-// if (fgets(temp, 500, f) == NULL ) break; /* end of file */
-//
-// if (temp[0] == 12 ) break; /* ^L end of header */
-// if (temp[0] != '#') i++; /* The first 200 non comment lines must be read before data is opened. */
-// if ((p = strchr(temp+1, '=')) == NULL) continue; /* no '=' */
-// if (count++ != iEntry) continue; /* no match */
-//
-// *p=0; /* match, end at = */
-//
-// fclose (f);
-// return temp;
-// }
-//
-// fclose(f);
-// return NULL;
-//}
-
-
static int get_nki_compressed_size(FILE *f)
{
NKI_MODE2 Header;
//====================================================================
// Read image information (copied from XDRreader)
-int nkitk::XDRImageIO::ReadImageInformationWithError()
+int clitk::XdrImageIO::ReadImageInformationWithError()
{
int offset=0;
itk::Vector<int,MAXDIM> dim;
- int veclen=1; //, data=AVS_TYPE_BYTE, field=UNIFORM;
-#if 0
- int iNkiCompression = 0;
-#endif
- int total=1/*, datasize=0, iNumRead, HeaderSize*/;
+ int veclen=1;
+ int total=1;
unsigned int coords=0,i,j,ndim,nspace;
char temp[512];
FILE *fstream;
char *c;
-#if 0
- char *buff;
- AVSfield FieldTemplate;
-#endif
+
long swap_test = 0x1000000; /* first byte is 1 when low-endian */
forcenoswap=0;
char *file = const_cast<char *>(m_FileName.c_str());
//====================================================================
// Read image information (copied from XDRreader)
-void nkitk::XDRImageIO::ReadImageInformation() {
+void clitk::XdrImageIO::ReadImageInformation() {
int result = ReadImageInformationWithError();
- if (result) ITKError("nkitk::XDRImageIO::ReadImageInformation",result);
+ if (result) ITKError("clitk::XdrImageIO::ReadImageInformation",result);
}
//====================================================================
-// Read Image Content (copied from XDRreader)
-int nkitk::XDRImageIO::ReadWithError(void * buffer)
+// Read Image Content (copied from Xdr reader)
+int clitk::XdrImageIO::ReadWithError(void * buffer)
{ //AVSINT dim[5];
int /*ndim,*/ nspace/*, veclen=1, data=AVS_TYPE_BYTE, field=UNIFORM*/;
int iNkiCompression = 0;
if (c) HeaderSize = 32768;
else HeaderSize = 2048;
-#if 0
- FIELDdefault(&FieldTemplate);
- FieldTemplate.ndim = ndim;
- FieldTemplate.nspace = nspace;
- FieldTemplate.veclen = veclen;
- FieldTemplate.type = data;
- FieldTemplate.uniform = field;
- FieldTemplate.size = datasize;
-#endif
-
fstream = fopen(file, "rb");
if (fstream == NULL)
return ER_XDR_OPEN;
if (i==iNumRead) return ER_XDR_NOCTRLL;
-#if 0
- if (*output) AVSfield_free(*output);
- *output = (AVSfield *) AVSfield_alloc(&FieldTemplate, dim);
- if (*output == NULL)
- {
- fclose(fstream);
- return ER_OUTOFMEMORY;
- }
-
- total *= datasize * veclen;
- coords *= sizeof(float);
-#endif
total = GetImageSizeInBytes();
//We add casts because the resulting quantity can be negative.
if (!pCompressed)
{
fclose(fstream);
-#if 0
- if (*output) AVSfield_free(*output);
- *output = NULL;
-#endif
return ER_OUTOFMEMORY;
}
if (fread( (void *)pCompressed, 1, iSize, fstream ) != iSize)
{
fclose(fstream);
-#if 0
- if (*output) AVSfield_free(*output);
- *output = NULL;
-#endif
return ER_XDR_READ;
}
if (!nki_private_decompress((short*)buffer, pCompressed, iSize))
{
fclose(fstream);
-#if 0
- if (*output) AVSfield_free(*output);
- *output = NULL;
-#endif
return ER_DECOMPRESSION;
}
if (fread( (void *)buffer, 1, total, fstream ) != total)
{
fclose(fstream);
-#if 0
- if (*output) AVSfield_free(*output);
- *output = NULL;
-#endif
return ER_XDR_READ;
}
}
}
READ_COORDS:
-#if 0
- if (coords) /* expect AVS coordinates ? */
- {
- if (fread( (*output)->points, 1, coords, fstream ) == coords)
- { /* swap data if read-ok and required (xdr is low-endian) */
-
- if (!(*(char *)(&swap_test)) && !forcenoswap)
- {
- c = (char *)(*output)->points;
- for (i=0; i<coords; i+=4)
- {
- j = c[i];
- c[i] = c[i+3];
- c[i+3] = j;
- j = c[i+1];
- c[i+1] = c[i+2];
- c[i+2] = j;
- }
- }
- }
- }
-#endif
fclose(fstream);
return OK;
}
//====================================================================
-// Read image information (copied from XDRreader)
-void nkitk::XDRImageIO::Read(void * buffer) {
+// Read image information (copied from Xdr reader)
+void clitk::XdrImageIO::Read(void * buffer) {
int result = ReadWithError(buffer);
- if (result) ITKError("nkitk::XDRImageIO::Read",result);
-}
-
-#if 0
-static int XDRreader_preview(AVSfield **output, char *file, int offset)
-{
- AVSINT dim[MAXDIM], dim2[MAXDIM], ds[MAXDIM];
- int ndim, nspace, veclen=1, data=AVS_TYPE_BYTE, field=UNIFORM;
- int iNkiCompression = 0;
- int total=1, i, j, k, l, m, coords=0, datasize=0, iNumRead, len, start, sliceax;
- char temp[512];
- FILE *fstream;
- char *c;
- char *buff;
- AVSfield FieldTemplate;
- long swap_test = 0x1000000; /* first byte is 1 when low-endian */
- int forcenoswap=0;
-
- fstream = fopen(file, "rt");
- if (fstream == NULL) return ER_XDR_OPEN;
- fgets(temp, 500, fstream);
- fclose(fstream);
-
- if (memcmp(temp, "# AVS field file (produced by avs_nfwrite.c)", 44)==0) forcenoswap=1;
-
- c = scan_header(file, "ndim", offset, 1);
- if (!c) return ER_XDR_NDIM;
- ndim = atoi(c);
- if (ndim<1 || ndim>MAXDIM) return ER_XDR_NDIM;
- nspace = ndim;
-
- /* defaults for dimensions and downsize */
-
- for (i=0; i<MAXDIM; i++)
- dim[i] = dim2[i] = ds[i] = 1;
-
- for (i=0; i<ndim; i++)
- {
- sprintf(temp, "dim%d", i+1);
- c = scan_header(file, temp, offset, 1);
- if (!c) return ER_XDR_DIM;
- dim[i]=atoi(c);
- if (dim[i]<1 || dim[i]>200000L) return ER_XDR_DIM;
-
- total *= dim[i];
- coords += dim[i];
- }
-
- c = scan_header(file, "nspace", offset, 1);
- if (c) nspace = atoi(c);
- if (nspace<1 || ndim > MAXDIM) return ER_XDR_NSPACE;
-
- c = scan_header(file, "veclen", offset, 1);
- if (c) veclen = atoi(c);
- if (veclen<0 || veclen>100) return ER_XDR_VECLEN;
-
- c = scan_header(file, "data", offset, 1);
-
- if (c)
- {
- if (memicmp(c, "byte", 4) == 0) data=AVS_TYPE_BYTE, datasize=1;
- else if (memicmp(c, "short", 5) == 0) data=AVS_TYPE_SHORT, datasize=2;
- else if (memicmp(c, "int" , 3) == 0) data=AVS_TYPE_INTEGER,datasize=4;
- else if (memicmp(c, "real", 4) == 0) data=AVS_TYPE_REAL, datasize=4;
- else if (memicmp(c, "float", 5) == 0) data=AVS_TYPE_REAL, datasize=4;
- else if (memicmp(c, "double",6) == 0) data=AVS_TYPE_DOUBLE, datasize=8;
-
- else if (memicmp(c, "xdr_byte", 8) == 0) data=AVS_TYPE_BYTE, datasize=1, forcenoswap=0;
- else if (memicmp(c, "xdr_short", 9) == 0) data=AVS_TYPE_SHORT, datasize=2, forcenoswap=0;
- else if (memicmp(c, "xdr_int" , 7) == 0) data=AVS_TYPE_INTEGER,datasize=4, forcenoswap=0;
- else if (memicmp(c, "xdr_real", 8) == 0) data=AVS_TYPE_REAL, datasize=4, forcenoswap=0;
- else if (memicmp(c, "xdr_float", 9) == 0) data=AVS_TYPE_REAL, datasize=4, forcenoswap=0;
- else if (memicmp(c, "xdr_double",10)== 0) data=AVS_TYPE_DOUBLE, datasize=8, forcenoswap=0;
- else return ER_XDR_DATA;
- }
-
- c = scan_header(file, "field", offset, 1);
- if (c)
- {
- if (memicmp(c, "unifo", 5) == 0) field=UNIFORM, coords=nspace*2;
- else if (memicmp(c, "recti", 5) == 0) field=RECTILINEAR;
- else if (memicmp(c, "irreg", 5) == 0) field=IRREGULAR, coords=total*nspace;
- else return ER_XDR_FIELD;
- }
- else
- coords=0;
-
- c = scan_header(file, "nki_compression", offset, 1);
- if (c) iNkiCompression = atoi(c);
- if (iNkiCompression)
- {
- fclose(fstream);
- return ER_ILLCOMMFUNCT;
- }
-
- FIELDdefault(&FieldTemplate);
- FieldTemplate.ndim = ndim;
- FieldTemplate.nspace = nspace;
- FieldTemplate.veclen = veclen;
- FieldTemplate.type = data;
- FieldTemplate.uniform = field;
- FieldTemplate.size = datasize;
-
- fstream = fopen(file, "rb");
-
- if (fstream == NULL)
- return ER_XDR_OPEN;
-
- buff = (char *)malloc(8192);
- if (buff == NULL)
- {
- return ER_OUTOFMEMORY;
- }
- memset(buff, 0, 8192);
- fseek(fstream, offset, SEEK_SET);
-
- while (1)
- {
- if (fgets(temp, 500, fstream) == NULL )
- return ER_XDR_NOCTRLL; /* end of file */
-
- if (temp[0] == 10) continue;
-
- if (temp[0] == 12)
- {
- fseek(fstream, -2, SEEK_CUR);
- break;
- } /* ^L end of header */
-
- if (temp[0] != '#') break;
- }
- start = ftell(fstream);
-
- iNumRead = fread(buff, 1, 8192, fstream);
- if (iNumRead < 1)
- {
- free(buff);
- fclose(fstream);
- return ER_XDR_READ;
- }
-
- for (i=0; i<iNumRead; i++) {
- if (buff[i] == 12) break;
- }
-
- free(buff);
-
- if (i==iNumRead) return ER_XDR_NOCTRLL;
-
- start += i+2;
-
- /* determine slice axis and downsize slice axis to 3, others to 32 */
-
- sliceax = 2;
-
- if (ndim>=3)
- {
- if (dim[0]==dim[1]) sliceax = 2;
- else sliceax = 1;
- }
-
- total = 1;
-
- for (i=0; i<ndim; i++)
- {
- if (i==sliceax) ds[i] = dim[i] / 3;
- else ds[i] = dim[i] / 32;
- if (ds[i]==0) ds[i]=1;
-
- dim2[i] = dim[i]/ds[i];
- if (dim2[i]==0) dim2[i] = 1;
-
- total *= dim2[i];
- }
-
- if (*output) AVSfield_free(*output);
- *output = (AVSfield *) AVSfield_alloc(&FieldTemplate, dim2);
- if (*output == NULL)
- {
- fclose(fstream);
- return ER_OUTOFMEMORY;
- }
-
- total *= datasize * veclen;
- coords *= sizeof(float);
-
- /* Read and downsize the data */
-
- if (total)
- {
- c = (char *)((*output)->field_data);
-
- buff = (char *)malloc(dim[0] * datasize * veclen);
-
- for (i=0; i<dim2[4]; i++)
- for (j=0; j<dim2[3]; j++)
- for (k=0; k<dim2[2]; k++)
- for (l=0; l<dim2[1]; l++)
- {
- len = i*ds[4];
- len = len * dim[3] + j*ds[3];
- len = len * dim[2] + k*ds[2];
- len = len * dim[1] + l*ds[1];
- len = len * dim[0] * datasize * veclen;
-
- // This is the time-critical statement
- fseek(fstream, start + len, SEEK_SET);
-
- fread(buff, 1, dim[0]*datasize*veclen, fstream);
- for (m=0; m<dim2[0]; m++)
- {
- memcpy(c, buff+m*datasize*veclen*ds[0], datasize*veclen);
- c += datasize*veclen;
- };
- };
-
- free (buff);
- }
-
- /* swap data if required (xdr is low-endian) */
-
- if (!(*(char *)(&swap_test)) && !forcenoswap)
- {
- if (datasize==2)
- {
- c = (char *)(*output)->field_data;
- for (i=0; i<total; i+=2)
- {
- j = c[i];
- c[i] = c[i+1];
- c[i+1] = j;
- }
- }
- else if (datasize==4)
- {
- c = (char *)(*output)->field_data;
- for (i=0; i<total; i+=4)
- {
- j = c[i];
- c[i] = c[i+3];
- c[i+3] = j;
- j = c[i+1];
- c[i+1] = c[i+2];
- c[i+2] = j;
- }
- }
- else if (datasize==8)
- {
- c = (char *)(*output)->field_data;
- for (i=0; i<total; i+=8)
- {
- j = c[i];
- c[i] = c[i+7];
- c[i+7] = j;
- j = c[i+1];
- c[i+1] = c[i+6];
- c[i+6] = j;
- j = c[i+2];
- c[i+2] = c[i+5];
- c[i+5] = j;
- j = c[i+3];
- c[i+3] = c[i+4];
- c[i+4] = j;
- }
- }
- }
-
-
- /*
- if (coords)
- { if (read( fHandle,(*output)->points, coords ) == coords)
- { if (!(*(char *)(&swap_test)) && !forcenoswap)
- { c = (char *)(*output)->points;
- for (i=0; i<coords; i+=4)
- { j = c[i]; c[i] = c[i+3]; c[i+3] = j;
- j = c[i+1]; c[i+1] = c[i+2]; c[i+2] = j;
- }
- }
- }
- }
- */
-
- fclose(fstream);
- return OK;
+ if (result) ITKError("clitk::XdrImageIO::Read",result);
}
-int Rxdr_compute(AVSfield** ppOut, char* pszFileName, int iOffset)
-{
- int rc;
- char szMsg[64];
-
- rc = XDRreader(ppOut, pszFileName, iOffset);
- if (rc == OK)
- rc = AVS_OK;
- else
- {
- strcpy(szMsg, "Avs_rxdr: ");
- strcat(szMsg, gl_ErrorMsg[rc]);
- AVSerror(szMsg);
- rc = AVS_ERROR;
- }
- return rc;
-}
-
-int RxdrPreview_compute(AVSfield** ppOut, char* pszFileName, int iOffset)
-{
- int rc;
- char szMsg[64];
-
- rc = XDRreader_preview(ppOut, pszFileName, iOffset);
- if (rc == OK)
- rc = AVS_OK;
- else
- {
- strcpy(szMsg, "Avs_rxdr: ");
- strcat(szMsg, gl_ErrorMsg[rc]);
- AVSerror(szMsg);
- rc = AVS_ERROR;
- }
- return rc;
-}
-
-int RxdrHeader_compute(char *pszOut, char *pszFileName, char *pszEntry)
-{
- char *cc;
-
- cc = scan_header(pszFileName, pszEntry, 0, 0);
- if (cc) strcpy(pszOut, cc);
- else *pszOut = 0;
-
- return AVS_OK;
-}
-
-int RxdrEnum_compute(char *pszOut, char *pszFileName, int iEntry)
-{
- char *cc;
-
- cc = enum_header(pszFileName, iEntry, 0);
- if (cc) strcpy(pszOut, cc);
- else *pszOut = 0;
-
- return AVS_OK;
-}
-#endif
-
//====================================================================
// Read Image Information
-bool nkitk::XDRImageIO::CanReadFile(const char* FileNameToRead)
+bool clitk::XdrImageIO::CanReadFile(const char* FileNameToRead)
{
char temp[512];
FILE *fstream;
fstream = fopen(FileNameToRead, "rt");
if (fstream == NULL)
+ {
+ AVSerror("Couldn't open file " << FileNameToRead);
return false;
+ }
fgets(temp, 500, fstream);
fclose(fstream);
return false;
} ////
-void nkitk::XDRImageIO::ITKError(std::string funcName, int msgID) {
+void clitk::XdrImageIO::ITKError(std::string funcName, int msgID) {
itkExceptionMacro(<< "Error in " << funcName << ". Message: " << gl_ErrorMsg[msgID]);
}
-#endif /* end #define NKITKXDRIMAGEIO_CXX */
-
--- /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 / commitdiff