GFitsImageByte.i
5.55 KB
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
120
121
122
/***************************************************************************
* GFitsImageByte.i - FITS Byte image class *
* ----------------------------------------------------------------------- *
* copyright (C) 2010-2017 by Juergen Knoedlseder *
* ----------------------------------------------------------------------- *
* *
* This program is free software: you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation, either version 3 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
* *
***************************************************************************/
/**
* @file GFitsImageByte.i
* @brief FITS Byte image class definition
* @author Juergen Knoedlseder
*/
%{
/* Put headers and other declarations here that are needed for compilation */
#include "GFitsImageByte.hpp"
#include "GException.hpp"
%}
/***********************************************************************//**
* @class GFitsImageByte
*
* @brief FITS Byte image class
***************************************************************************/
class GFitsImageByte : public GFitsImage {
public:
// Constructors and destructors
GFitsImageByte(void);
GFitsImageByte(const int& nx, const unsigned char* pixels = NULL);
GFitsImageByte(const int& nx, const int& ny, const unsigned char* pixels = NULL);
GFitsImageByte(const int& nx, const int& ny, const int& nz, const unsigned char* pixels = NULL);
GFitsImageByte(const int& nx, const int& ny, const int& nz, const int& nt, const unsigned char* pixels = NULL);
GFitsImageByte(const std::vector<int>& naxes, const unsigned char* pixels = NULL);
GFitsImageByte(const GFitsImage& image);
GFitsImageByte(const GFitsImageByte& image);
virtual ~GFitsImageByte(void);
// Methods
void clear(void);
GFitsImageByte* clone(void) const;
std::string classname(void) const;
double pixel(const int& ix) const;
double pixel(const int& ix, const int& iy) const;
double pixel(const int& ix, const int& iy, const int& iz) const;
double pixel(const int& ix, const int& iy, const int& iz, const int& it) const;
void* pixels(void);
int type(void) const;
};
/***********************************************************************//**
* @brief GFitsImageByte class extension
***************************************************************************/
%extend GFitsImageByte {
unsigned char __getitem__(int GFitsImageInx[]) {
// Check image dimensions
for (int i = 0; i < GFitsImageInx[0]; ++i) {
if (GFitsImageInx[i+1] < 0 || GFitsImageInx[i+1] >= self->naxes(i)) {
throw GException::out_of_range("__getitem__(int)",
"FITS image axis "+
gammalib::str(i)+" index",
GFitsImageInx[i+1],
self->naxes(i));
}
}
// Return pixel value
if (GFitsImageInx[0] == 1) {
return (*self)(GFitsImageInx[1]);
}
else if (GFitsImageInx[0] == 2) {
return (*self)(GFitsImageInx[1], GFitsImageInx[2]);
}
else if (GFitsImageInx[0] == 3) {
return (*self)(GFitsImageInx[1], GFitsImageInx[2], GFitsImageInx[3]);
}
else if (GFitsImageInx[0] == 4) {
return (*self)(GFitsImageInx[1], GFitsImageInx[2], GFitsImageInx[3],
GFitsImageInx[4]);
}
else {
throw GException::fits_wrong_image_operator("__getitem__(int)",
self->naxis(), GFitsImageInx[0]);
}
}
void __setitem__(int GFitsImageInx[], unsigned char value) {
if (GFitsImageInx[0] == 1) {
self->at(GFitsImageInx[1]) = value;
}
else if (GFitsImageInx[0] == 2) {
self->at(GFitsImageInx[1], GFitsImageInx[2]) = value;
}
else if (GFitsImageInx[0] == 3) {
self->at(GFitsImageInx[1], GFitsImageInx[2], GFitsImageInx[3]) =
value;
}
else if (GFitsImageInx[0] == 4) {
self->at(GFitsImageInx[1], GFitsImageInx[2], GFitsImageInx[3],
GFitsImageInx[4]) = value;
}
else {
throw GException::fits_wrong_image_operator("__setitem__(int)",
self->naxis(),
GFitsImageInx[0]);
}
}
GFitsImageByte copy() {
return (*self);
}
};