source: Sophya/trunk/SophyaExt/FitsIOServer/fitsspherehealpix.cc@ 2906

Last change on this file since 2906 was 2898, checked in by ansari, 20 years ago

Ajout methode FitsManager::ScanFile() et corrections diverses - Reza 13/01/2006

File size: 8.5 KB
Line 
1#include "sopnamsp.h"
2#include "pexceptions.h"
3#include "fitsspherehealpix.h"
4#include "tarray.h"
5#include "fitstarray.h"
6
7
8
9///////////////////////////////////////////////////////////////////////
10// ----objets delegues pour la gestion de persistance I/O format fits--
11// SphereHealpix
12////////////////////////////////////////////////////////////////////
13
14/*!
15 \class SOPHYA::FITS_SphereHEALPix
16 \ingroup FitsIOServer
17 FITS format I/O handler for SOPHYA::SphereHEALPix objects.
18*/
19
20
21template <class T>
22FITS_SphereHEALPix<T>::FITS_SphereHEALPix()
23{
24 dobj_= new SphereHEALPix<T>;
25 ownobj_= true;
26}
27
28template <class T>
29FITS_SphereHEALPix<T>::FITS_SphereHEALPix(char inputfile[],int hdunum)
30{
31 dobj_= new SphereHEALPix<T>;
32 ownobj_= true;
33 Read(inputfile,hdunum);
34}
35
36
37template <class T>
38FITS_SphereHEALPix<T>::FITS_SphereHEALPix(const SphereHEALPix<T>& obj)
39{
40 dobj_= new SphereHEALPix<T>(obj);
41 ownobj_= true;
42}
43
44template <class T>
45FITS_SphereHEALPix<T>::FITS_SphereHEALPix(SphereHEALPix<T>* obj)
46{
47 dobj_= obj;
48 ownobj_= false;
49}
50
51template <class T>
52FITS_SphereHEALPix<T>::~FITS_SphereHEALPix()
53{
54 if (ownobj_ && dobj_) delete dobj_;
55}
56
57template <class T>
58AnyDataObj* FITS_SphereHEALPix<T>::DataObj()
59{
60 return(dobj_);
61}
62
63template <class T>
64void FITS_SphereHEALPix<T>::SetDataObj(AnyDataObj & o)
65{
66 SphereHEALPix<T> * po = dynamic_cast< SphereHEALPix<T> * >(&o);
67 if (po == NULL) return;
68 if (ownobj_ && dobj_) delete dobj_;
69 dobj_ = po;
70 ownobj_ = false;
71}
72
73template <class T>
74int FITS_SphereHEALPix<T>::CheckHandling(AnyDataObj & o)
75{
76 if ( typeid(o) == typeid( SphereHEALPix< T > ) ) return 2;
77 SphereHEALPix<T> * po = dynamic_cast< SphereHEALPix<T> * >(&o);
78 if (po == NULL) return 0;
79 else return 1;
80}
81
82template <class T>
83int FITS_SphereHEALPix<T>::CheckReadability(FitsInOutFile& is)
84{
85 if (is.CurrentHDUType() != BINARY_TBL ) return 0;
86 string key;
87 key = "PIXTYPE";
88 if ( is.KeyValue(key) != "HEALPIX") return 0;
89
90 bool nosk = false;
91 key = "ORDERING";
92 is.KeyValue(key, nosk);
93 bool nosk2 = false;
94 key = "NSIDE";
95 is.KeyValue(key, nosk2);
96 if (nosk || nosk2) return 0;
97
98 vector<string> colnames;
99 vector<int> coltypes;
100 vector<long> repcnt;
101 vector<long> width;
102 long ncols = is.GetColInfo(colnames, coltypes, repcnt, width);
103 if (ncols < 1) return 0;
104 T x = 0;
105 if (coltypes[0] == FitsTypes::DataType(x)) return 2 ;
106 else return 1;
107}
108
109template <class T>
110FitsHandlerInterface* FITS_SphereHEALPix<T>::Clone()
111{
112 return new FITS_SphereHEALPix<T>() ;
113}
114
115
116template <class T>
117void FITS_SphereHEALPix<T>::WriteToFits(FitsOutFile& os)
118{
119 if(dobj_ == NULL)
120 {
121 cout << " WriteToFits:: dobj_= null " << endl;
122 return;
123 }
124
125 DVList dvl( dobj_->Info() );
126
127 SphereCoordSys* cs= dobj_->GetCoordSys();
128 string description= cs->description();
129 dvl["COORDSYS"]= description;
130
131 dvl["PIXTYPE"] = "HEALPIX";
132 dvl.SetComment("PIXTYPE", "HEALPIX Pixelization");
133 dvl["ORDERING"]= dobj_->TypeOfMap();
134 dvl.SetComment("ORDERING", "Pixel ordering scheme, either RING or NESTED");
135
136 int nSide= dobj_->SizeIndex();
137 dvl["NSIDE"]= (int_4) nSide;
138 dvl.SetComment("NSIDE","Resolution parameter for HEALPIX" );
139
140 int nPix= dobj_->NbPixels();
141 dvl["FIRSTPIX"]= (int_4) 0;
142 dvl.SetComment("FIRSTPIX", "First pixel # (0 based)");
143 dvl["LASTPIX"]= (int_4) nPix-1;
144 dvl.SetComment("LASTPIX", "Last pixel # (0 based)");
145 dvl["Content"]= "SphereHEALPix";
146 dvl.SetComment("Content", "name of SOPHYA object");
147 dvl["SOPCLSNM"]= "SOPHYA::SphereHEALPix<T>";
148 dvl.SetComment("SOPCLSNM", "SOPHYA class name");
149
150 // On ecrit les dataBlocks
151 vector<string> Noms;
152 Noms.push_back(dvl.GetS("Content"));
153 // string extname("SIMULATION");
154 string extname = os.NextExtensionName();
155
156 string Type;
157 if (typeid(T) == typeid(r_8) ) Type+='D';
158 else if (typeid(T) == typeid(r_4) ) Type+='E';
159 else if (typeid(T) == typeid(int_4) ) Type+='J';
160 else
161 {
162 cout << "FITS_SphereHEALPix<T>::WriteToFits()- Unsupported data type ("
163 << typeid(T).name() << ") throwing IOExc " << endl;
164 throw IOExc("FITS_SphereHEALPix<T>::WriteToFits() unsupported data type");
165 }
166 vector<int> dummy;
167 os.makeHeaderBntblOnFits(Type, Noms, nPix, 1, &dvl, extname, dummy);
168 os.PutColToFits(0, nPix, dobj_->pixels_.Data());
169}
170
171template <class T>
172void FITS_SphereHEALPix<T>::ReadFromFits(FitsInFile& is)
173{
174 if(dobj_ == NULL)
175 {
176 dobj_= new SphereHEALPix<T>;
177 ownobj_= true;
178 }
179 int nbcols, nbentries;
180 //
181
182 nbcols = is.NbColsFromFits();
183 if (nbcols != 1)
184 {
185 throw IOExc("le fichier fits n'est pas une sphere Healpix");
186 }
187 DVList dvl=is.DVListFromFits();
188 nbentries = is.NentriesFromFits(0);
189 int lastpix=dvl.GetI("LASTPIX");
190 if (lastpix>0)
191 {
192 if (nbentries!=lastpix+1)
193 {
194 nbentries=lastpix+1;
195 }
196 }
197 // Let's Read the SphereCoordSys object
198 int id= SphereCoordSys_NEUTRAL;
199 string description= "NEUTRAL SphereCoordSystem";
200 string coordsys= dvl.GetS("COORDSYS");
201 // $CHECK$ - Reza 2/05/2000 - compare(size_t, size_t,...) passe pas sur g++
202 // if(coordsys.compare(0,7,"ROTATION",0,7) == 0) $CHECK$
203 if(coordsys.substr(0,8) == "ROTATION")
204 {
205 id= SphereCoordSys_ROTATION;
206 description= "ROTATION SphereCoordSystem";
207 }
208 // else if(coordsys.compare(0,4,"OTHER",0,4) == 0) $CHECK$ Reza 2/05/2000
209 else if(coordsys.substr(0,5) == "OTHER" )
210 {
211 id= SphereCoordSys_OTHER;
212 description= "OTHER SphereCoordSystem";
213 }
214 dobj_->SetCoordSys(new SphereCoordSys(id,description));
215
216 string ordering= dvl.GetS("ORDERING");
217 // if(ordering.compare(0,3,"RING",0,3) != 0) $CHECK$ Reza 2/05/2000
218 if(ordering.substr(0,4) != "RING" )
219 {
220 cerr << "FITS_SphereHEALPix/Error Not supported ORDERING= "
221 << ordering << " substr(0,4)=[" << ordering.substr(0,4) << "]" << endl;
222 throw IOExc(" FITS_SphereHEALPix:: numerotation non RING");
223 }
224
225 int nside= dvl.GetI("NSIDE");
226 int nPix = 0;
227 if(nside <= 0)
228 {
229 // throw IOExc("FITS_SphereHEALPix:: No resol parameter nside");
230 if (lastpix>0)
231 {
232 nside = sqrt((double)(nbentries/12));
233 }
234 else
235 {
236 // dvl.Print();
237 // int naxis2 = dvl.GetI("NAXIS2");
238 // if (naxis2 > 0 )
239 // {
240 nside = sqrt((double)(nbentries/12));
241 // }
242 // else
243 // throw IOExc("FITS_SphereHEALPix:: No NSIDE nor LASTPIX nor NAXIS2 on file");
244 }
245 }
246 nPix = 12*nside*nside;
247 if (nPix != nbentries)
248 {
249 cout << "WARNING: le nombre de pixels relu " << nbentries << " est incoherent avec la pixelisation Healpix qui donne nPix= " << nPix << endl;
250 }
251 double Omega= 4.0*Pi/nPix;
252 dobj_->setParameters(nside,nPix,Omega);
253 // On lit les DataBlocks;
254 //
255 dobj_->pixels_.ReSize(nPix);
256 is.GetSingleColumn(dobj_->pixels_.Data(),nPix);
257 //
258
259 // on effectue le decoupage en tranches
260 dobj_->SetThetaSlices();
261 //
262 dobj_->Info() = is.DVListFromFits();
263}
264
265template <class T>
266void FITS_SphereHEALPix<T>::Mollweide_picture_projection(char filename[])
267{
268 int ni = 300;
269 int nj = 600;
270 TMatrix<float> map(ni, nj);
271
272 int i,j;
273 for(i=0; i < ni; i++)
274 {
275 double yd = (i+0.5)/ni-0.5;
276 double facteur=2.*Pi/sin(acos(yd*2));
277 double theta = (0.5-yd)*Pi;
278 for(j=0; j < nj; j++)
279 {
280 double xa = (j+0.5)/nj-0.5;
281 double phi = xa*facteur+Pi;
282 float th=float(theta);
283 float ff=float(phi);
284 if (phi<2*Pi && phi>= 0)
285 {
286 map(i,j) = (float)dobj_->PixValSph(th, ff);
287 }
288 }
289 }
290 FitsOutFile fout(filename);
291 fout.firstImageOnPrimaryHeader(true);
292 FITS_TArray<r_4> fta(map);
293 fta.Write(fout);
294
295}
296template <class T>
297void FITS_SphereHEALPix<T>::sinus_picture_projection(char filename[])
298{
299 int ni = 300;
300 int nj = 600;
301 TMatrix<float> map(ni, nj);
302
303 int i,j;
304 for(i=0; i < ni; i++)
305 {
306 double yd = (i+0.5)/ni-0.5;
307 double theta = (0.5-yd)*Pi;
308 double facteur=1./sin(theta);
309 for(j=0; j < nj; j++)
310 {
311 double xa = (j+0.5)/nj-0.5;
312 double phi = 2.*Pi*xa*facteur+Pi;
313 float th=float(theta);
314 float ff=float(phi);
315 if (phi<2*Pi && phi>= 0)
316 {
317 map(i,j) = (float)dobj_->PixValSph(th, ff);
318 }
319 }
320 }
321 FitsOutFile fout(filename);
322 fout.firstImageOnPrimaryHeader(true);
323 FITS_TArray<r_4> fta(map);
324 fta.Write(fout);
325
326
327}
328
329
330#ifdef __CXX_PRAGMA_TEMPLATES__
331#pragma define_template FITS_SphereHEALPix<r_8>
332#pragma define_template FITS_SphereHEALPix<r_4>
333#pragma define_template FITS_SphereHEALPix<int_4>
334#endif
335#if defined(ANSI_TEMPLATES) || defined(GNU_TEMPLATES)
336namespace SOPHYA {
337template class FITS_SphereHEALPix<r_8>;
338template class FITS_SphereHEALPix<r_4>;
339template class FITS_SphereHEALPix<int_4>;
340}
341#endif
Note: See TracBrowser for help on using the repository browser.