diff options
Diffstat (limited to 'fitsy++/rice.C')
| -rw-r--r-- | fitsy++/rice.C | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/fitsy++/rice.C b/fitsy++/rice.C new file mode 100644 index 0000000..6ee13dc --- /dev/null +++ b/fitsy++/rice.C @@ -0,0 +1,157 @@ +// Copyright (C) 1999-2018 +// Smithsonian Astrophysical Observatory, Cambridge, MA, USA +// For conditions of distribution and use, see copyright notice in "copyright" + +#include <iostream> +#include <sstream> +#include <iomanip> +using namespace std; + +#include "rice.h" +#include "util.h" + +extern "C" { + int fits_rdecomp (unsigned char *c, + int clen, + unsigned int array[], + int nx, + int nblock); + int fits_rdecomp_short (unsigned char *c, + int clen, + unsigned short array[], + int nx, + int nblock); + int fits_rdecomp_byte (unsigned char *c, + int clen, + unsigned char array[], + int nx, + int nblock); +} + +template<class T> FitsRicem<T>::FitsRicem(FitsFile* fits) + : FitsCompressm<T>(fits) +{ + // rice parameters + block_ = 32; + bytepix_ = 4; + noisebit_ = 4; + char name[] = "ZNAME "; + char val[] = "ZVAL "; + for (int ii=0; ii<9; ii++) { + name[5] = '0'+ii; + val[4] = '0'+ii; + if (fits->find(name)) { + char* str = fits->getString(name); + if (!strncmp(str,"BLOCK",4)) + block_ = fits->getInteger(val,32); + else if (!strncmp(str,"BYTEPIX",4)) + bytepix_ = fits->getInteger(val,4); + else if (!strncmp(str,"NOISEBIT",4)) + noisebit_ = fits->getInteger(val,4); + } + } + + FitsCompressm<T>::uncompress(fits); +} + +template <class T> int FitsRicem<T>::compressed(T* dest, char* sptr, + char* heap, + int kkstart, int kkstop, + int jjstart, int jjstop, + int iistart, int iistop) +{ + double zs = FitsCompressm<T>::bscale_; + if (FitsCompressm<T>::zscale_) + zs = FitsCompressm<T>::zscale_->value(sptr,0); + + double zz = FitsCompressm<T>::bzero_; + if (FitsCompressm<T>::zzero_) + zz = FitsCompressm<T>::zzero_->value(sptr,0); + + int blank = FitsCompressm<T>::blank_; + if (FitsCompressm<T>::zblank_) + blank = (int)FitsCompressm<T>::zblank_->value(sptr,0); + + int icnt=0; + unsigned char* ibuf = (unsigned char*)((FitsBinColumnArray*)FitsCompressm<T>::compress_)->get(heap, sptr, &icnt); + + // ibuf can be NULL + if (!ibuf || !icnt) + return 0; + + int ocnt = FitsCompressm<T>::tilesize_; + int ll=0; + + switch (bytepix_) { + case 1: + { + char* obuf = new char[ocnt]; + if (fits_rdecomp_byte(ibuf, icnt, (unsigned char*)obuf, ocnt, block_)) { + internalError("Fitsy++ rice bad inflate result"); + return 0; + } + for (int kk=kkstart; kk<kkstop; kk++) + for (int jj=jjstart; jj<jjstop; jj++) + for (int ii=iistart; ii<iistop; ii++,ll++) { + // very carefull about type conversions + size_t id = kk*FitsCompressm<T>::width_*FitsCompressm<T>::height_ + jj*FitsCompressm<T>::width_ + ii; + T val = FitsCompressm<T>::getValue(obuf+ll,zs,zz,blank); + dest[id] = val; + } + + if (obuf) + delete [] obuf; + } + break; + case 2: + { + short* obuf = new short[ocnt]; + if (fits_rdecomp_short(ibuf, icnt, (unsigned short*)obuf, ocnt, block_)) { + internalError("Fitsy++ rice bad inflate result"); + return 0; + } + for (int kk=kkstart; kk<kkstop; kk++) + for (int jj=jjstart; jj<jjstop; jj++) + for (int ii=iistart; ii<iistop; ii++,ll++) { + // very carefull about type conversions + size_t id = kk*FitsCompressm<T>::width_*FitsCompressm<T>::height_ + jj*FitsCompressm<T>::width_ + ii; + T val = FitsCompressm<T>::getValue(obuf+ll,zs,zz,blank); + dest[id] = val; + } + + if (obuf) + delete [] obuf; + } + break; + case 4: + { + int* obuf = new int[ocnt]; + if (fits_rdecomp(ibuf, icnt, (unsigned int*)obuf, ocnt, block_)) { + internalError("Fitsy++ rice bad inflate result"); + return 0; + } + for (int kk=kkstart; kk<kkstop; kk++) + for (int jj=jjstart; jj<jjstop; jj++) + for (int ii=iistart; ii<iistop; ii++,ll++) { + // very carefull about type conversions + int id = kk*FitsCompressm<T>::width_*FitsCompressm<T>::height_ + jj*FitsCompressm<T>::width_ + ii; + T val = FitsCompressm<T>::getValue(obuf+ll,zs,zz,blank); + dest[id] = val; + } + + if (obuf) + delete [] obuf; + } + break; + } + + return 1; +} + +template class FitsRicem<unsigned char>; +template class FitsRicem<short>; +template class FitsRicem<unsigned short>; +template class FitsRicem<int>; +template class FitsRicem<long long>; +template class FitsRicem<float>; +template class FitsRicem<double>; |