1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// Copyright (C) 1999-2017
// 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;
// Note: currently, hcompress will not work with float or double data with
// BLANK defined, due to the fact that the decompress code takes a int() and
// not an unsigned int()
#include "hcompress.h"
#include "util.h"
extern "C" {
int fits_hdecompress(unsigned char *input, int smooth, int *a, int *ny, int *nx, int *scale, int *status);
int fits_hdecompress64(unsigned char *input, int smooth, long long *a, int *ny, int *nx, int *scale, int *status);
}
template<class T> FitsHcompressm<T>::FitsHcompressm(FitsFile* fits)
: FitsCompressm<T>(fits)
{
// hcompress parameters
smooth_ = 0;
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,"SMOOTH",4))
smooth_ = fits->getInteger(val,4);
}
}
FitsCompressm<T>::uncompress(fits);
}
template <class T> int FitsHcompressm<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 nx,ny,scale;
int status=0;
int ll=0;
switch (FitsCompressm<T>::bitpix_) {
case 8:
case 16:
{
int* obuf = new int[ocnt];
if (fits_hdecompress(ibuf, smooth_, obuf, &nx, &ny, &scale, &status)) {
internalError("Fitsy++ hcompress 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++)
dest[kk*FitsCompressm<T>::width_*FitsCompressm<T>::height_ + jj*FitsCompressm<T>::width_ + ii] = FitsCompressm<T>::getValue(obuf+ll,zs,zz,blank);
if (obuf)
delete [] obuf;
}
break;
case 32:
case -32:
case -64:
{
long long* obuf = new long long[ocnt];
if (fits_hdecompress64(ibuf, smooth_, obuf, &nx, &ny, &scale, &status)) {
internalError("Fitsy++ hcompress 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++)
dest[kk*FitsCompressm<T>::width_*FitsCompressm<T>::height_ + jj*FitsCompressm<T>::width_ + ii] = FitsCompressm<T>::getValue((int*)obuf+ll,zs,zz,blank);
if (obuf)
delete [] obuf;
}
break;
}
return 1;
}
template class FitsHcompressm<unsigned char>;
template class FitsHcompressm<short>;
template class FitsHcompressm<unsigned short>;
template class FitsHcompressm<int>;
template class FitsHcompressm<long long>;
template class FitsHcompressm<float>;
template class FitsHcompressm<double>;
|
