source: Sophya/trunk/SophyaPI/PIext/piacmd.cc@ 208

Last change on this file since 208 was 208, checked in by ercodmgr, 27 years ago

Ajout savefits Reza 02/03/99

File size: 38.3 KB
Line 
1#include <stdio.h>
2#include <stdlib.h>
3#include <math.h>
4
5#include <iostream.h>
6#include <string>
7#include <vector>
8#include <map>
9#if defined(__KCC__)
10using std::string ;
11#include <vector.h>
12#include <map.h>
13#endif
14
15#include "ctimer.h"
16
17#include "nobjmgr.h"
18#include "piacmd.h"
19#include "pistdimgapp.h"
20
21#include "histos.h"
22#include "histos2.h"
23#include "hisprof.h"
24#include "ntuple.h"
25#include "generaldata.h"
26
27
28// Pour le decoupage des commandes en lignes
29typedef vector<string> cmdtok;
30
31// Fonction pour faire le Fit de PSF sur image
32static int FitPSFImg(cmdtok& tokens);
33static int Photom_Ouv(cmdtok& tokens);
34
35
36/* --Methode-- */
37PIACmd::PIACmd(NamedObjMgr* omg, PIStdImgApp* app)
38{
39if (omg) { mOmg= false; mObjMgr = omg; }
40else { mObjMgr = new NamedObjMgr; mOmg = true; }
41mImgApp = app;
42system("cp history.pic hisold.pic");
43hist.open("history.pic");
44trace = false; timing = false;
45gltimer = NULL;
46dynlink = NULL;
47}
48
49/* --Methode-- */
50PIACmd::~PIACmd()
51{
52if (mOmg) delete mObjMgr;
53hist.close();
54if (gltimer) { delete gltimer; gltimer = NULL; }
55}
56
57/* --Methode-- */
58int PIACmd::Exec(string& file)
59{
60char line_buff[512];
61FILE *fip;
62
63if ( (fip = fopen(file.c_str(),"r")) == NULL ) {
64 cerr << "PIACmd::Exec() Error opening file " << file << endl;
65 hist << "##! PIACmd::Exec() Error opening file " << file << endl;
66 return(0);
67 }
68
69hist << "### Executing commands from " << file << endl;
70if (trace) {
71 mImgApp->GetConsole()->AddStr("### Executing commands from ", PIVA_Magenta);
72 mImgApp->GetConsole()->AddStr(file.c_str(), PIVA_Magenta);
73 mImgApp->GetConsole()->AddStr("\n", PIVA_Magenta);
74 }
75
76while (fgets(line_buff,511,fip) != NULL)
77 {
78 if (trace) mImgApp->GetConsole()->AddStr(line_buff, PIVA_Magenta);
79 line_buff[strlen(line_buff)-1] = '\0'; /* LF/CR de la fin */
80 string line(line_buff);
81 Do(line);
82 }
83hist << "### End of Exec( " << file << " ) " << endl;
84if (trace) {
85 mImgApp->GetConsole()->AddStr("### End of Exec( ", PIVA_Magenta);
86 mImgApp->GetConsole()->AddStr(file.c_str(), PIVA_Magenta);
87 mImgApp->GetConsole()->AddStr(" ) \n", PIVA_Magenta);
88 }
89
90return(0);
91}
92
93/* --Methode-- */
94int PIACmd::Do(string& s)
95{
96
97cmdtok tokens;
98if (s.length() < 1) return(0);
99
100hist << s << endl; // On enregistre les commandes
101
102if (s[0] == '#') {
103 cout << "PIACmd::Do() Comment-Line:" << s << endl;
104 return(0);
105 }
106string toks,kw;
107size_t p = s.find_first_not_of(" ");
108s = s.substr(p);
109p = 0;
110size_t q = s.find_first_of(" ");
111size_t l = s.length();
112
113if (q < l)
114 { kw = s.substr(p,q-p); toks = s.substr(q, l-q); }
115else { kw = s.substr(p,l-p); toks = ""; }
116
117q = 0;
118while (q < l) {
119 p = toks.find_first_not_of(" ",q+1); // au debut d'un token
120 if (p>=l) break;
121 q = toks.find_first_of(" ",p); // la fin du token;
122 string token = toks.substr(p,q-p);
123 tokens.push_back(token);
124 }
125
126for(int k=0; k<tokens.size(); k++) { // On remplace les $varname par la valeur de la variable
127 if ((tokens[k])[0] != '$') continue;
128 CmdVarList::iterator it = mVars.find(tokens[k].substr(1));
129 if (it != mVars.end()) tokens[k] = (*it).second;
130}
131
132// cout << "PIACmd::Do() DBG KeyW= " << kw << " NbArgs= " << tokens.size() << endl;
133// for(int ii=0; ii<tokens.size(); ii++)
134// cout << "arg[ " << ii << " ] : " << tokens[ii] << endl;
135
136// >>>>>>>>>>> Commande d'interpreteur
137if (kw == "help") {
138 cout << "\n PIACmd::Do() ----- Help : Liste des commandes ------------ \n";
139 cout << "help shelp exec set listvars traceon traceoff \n";
140 cout << "timingon timingoff shell \n";
141 cout << "link call zone gratt stacknext \n" ;
142 cout << "openppf saveall listobjs openfits savefits \n";
143 cout << "print rename del delobjs \n";
144 cout << "newh1d newh2d newprof newgfd func func2d \n" ;
145 cout << "disp surf nt2d nt3d gfd2d gfd3d \n" ;
146 cout << "plot2d plot3d projh1d projh2d projprof \n";
147 cout << "fillvec fillnt fillgd1 fillgd2 fft fit fitpsf aper\n";
148 cout << "*** Commande sans args -> liste d'arguments\n";
149 cout << "*** dopt: DisplayOption = next/same/win/stack,color,... \n";
150 cout << "*** $varname remplace par la valeur de la variable varname\n" << endl;
151 }
152
153else if (kw == "shelp" ) {
154 cout << "\n PIACmd::Do() ------------- SHelp ------------ " << endl;
155 cout << "exec filename -- set varname string -- clr varname -- listvars \n";
156 cout << "traceon/off -- timingon/off -- shell cmdline\n";
157 cout << "link fnameso f1 [f2 f3] -- call userf [arg1 arg2 ...]\n";
158 cout << "zone nx ny -- gratt attributes_list -- stacknext \n";
159 cout << "openfits file -- openppf file -- saveall file -- listobjs --\n";
160 cout << "print nomobj -- rename nomobj nomnew -- del nomobj -- delobjs pattern (*,?) \n";
161 cout << "newh1d nomh xmin xmax nbin -- newh2d nomh xmin xmax nbinx ymin ymax nbiny\n";
162 cout << "newprof nomh xmin xmax nbin [ymin ymax] \n";
163 cout << "newgfd nomgfd nvar nalloc [errx(0/1)]\n";
164 cout << "func expfunc xmin xmax npt opt\n";
165 cout << "func2d expfunc xmin xmax nptx ymin ymax npty opt\n";
166 cout << "disp nomobj opt -- surf nomobj opt\n";
167 cout << "nt2d nomobj varx vary errx erry opt\n";
168 cout << "nt3d nomobj varx vary varz errx erry errz dopt\n";
169 cout << "gfd2d nomobj numvarx erreur=(x y xy) opt\n";
170 cout << "gfd3d nomobj numvarx numvary erreur=(x y z xy xz yz xyz) opt\n";
171 cout << "plot2d nomobj expx expy [experrx experry] expcut dopt\n";
172 cout << "plot3d nomobj expx expy expz expcut dopt\n";
173 cout << "projh1d nomobj expx expwt expcut nomh1 dopt\n";
174 cout << "projh2d nomobj expx expy expwt expcut nomh2 dopt\n";
175 cout << "projprof nomobj expx expy expwt expcut nomprof dopt\n";
176 cout << "fillnt nomobj expx expy expz expt expcut nomnt dopt\n";
177 cout << "fillvec nomobj expx expcut nomvec dopt\n";
178 cout << "fillgd1 nomobj expx expy experry expcut nomgfd\n";
179 cout << "fillgd2 nomobj expx expy expz experrz expcut nomgfd\n";
180 cout << "fft nomobj dopt\n";
181 cout << "fit nomobj func [p:p1,.,pn s:s1,.,sn m:m1,.,mn M:M1,.,Mn I:i1,i2 o:... o:...]\n";
182 cout << "fitpsf psf_typ [options_fit]\n";
183 cout << "aper fond_typ [options_calcul]\n";
184}
185
186else if (kw == "set") {
187 if (tokens.size() < 2) { cout << "PIACmd::Do() Usage: set varname string" << endl; return(0); }
188 mVars[tokens[0]] = tokens[1];
189 }
190else if (kw == "clr") {
191 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: clr varname" << endl; return(0); }
192 CmdVarList::iterator it = mVars.find(tokens[0]);
193 if (it != mVars.end()) mVars.erase(it);
194 else cerr << "PIACmd::Do() Pas de variable de nom " << tokens[0] << endl;
195 }
196else if (kw == "listvars") {
197 cout << "PIACmd::Do() Variable List , VarName = Value \n";
198 CmdVarList::iterator it;
199 for(it = mVars.begin(); it != mVars.end(); it++)
200 cout << (*it).first << " = " << (*it).second << "\n";
201 cout << endl;
202 }
203else if (kw == "traceon") { cout << "PIACmd::Do() -> Trace ON mode " << endl; trace = true; }
204else if (kw == "traceoff") { cout << "PIACmd::Do() -> Trace OFF mode " << endl; trace = false; }
205else if (kw == "timingon") {
206 cout << "PIACmd::Do() -> Timing ON mode " << endl;
207 if (gltimer) delete gltimer; gltimer = new Timer("PIA-CmdInterpreter "); timing = true; }
208else if (kw == "timingoff") {
209 cout << "PIACmd::Do() -> Timing OFF mode " << endl;
210 if (gltimer) delete gltimer; gltimer = NULL; timing = false; }
211else if (kw == "exec") {
212 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: exec filename" << endl; return(0); }
213 Exec(tokens[0]);
214 }
215else if (kw == "shell") {
216 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: shell cmdline" << endl; return(0); }
217 system(toks.c_str());
218 }
219
220// >>>>>>>>>>> Fenetre graphique , changement d'attributs graphiques
221else if (kw == "zone") {
222 if (tokens.size() < 2) { cout << "PIACmd::Do() Usage: zone nx ny" << endl; return(0); }
223 int nx, ny;
224 nx = ny = 1;
225 nx = atoi(tokens[0].c_str()); ny = atoi(tokens[1].c_str());
226 mObjMgr->SetGraphicWinZone(nx, ny);
227 }
228else if (kw == "gratt") {
229 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: gratt attributes_list (att=def->defaut)" << endl; return(0); }
230 mObjMgr->SetGraphicAttributes(tokens[0]);
231 }
232else if (kw == "stacknext") mImgApp->StackWinNext();
233
234// >>>>>>>>>>> Link dynamique de fonctions C++
235else if (kw == "link" ) {
236 if (tokens.size() < 2) { cout << "PIACmd::Do() Usage: link fnameso f1 [f2 f3]" << endl; return(0); }
237 string sph = "";
238 for(int gg=0; gg<5; gg++) tokens.push_back(sph);
239 int rc = LinkUserFuncs(tokens[0], tokens[1], tokens[2], tokens[3]);
240 if (rc == 0) cout << "PIAcmd::Do(): Link from " << tokens[0] << " OK " << endl;
241}
242else if (kw == "call" ) {
243 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: call userf [arg1 arg2 ...]" << endl; return(0); }
244 UsFmap::iterator it = usfmap.find(tokens[0]);
245 if (it == usfmap.end()) {
246 cerr << "PIAcmd::Do() No User Function " << tokens[0] << endl;
247 return(0);
248 }
249 cout << "PIAcmd::Do() Call " << tokens[0] << "( ... )" << endl;
250// on est oblige de faire un cast etant donne qu'on
251// utilise donc des DlFunction (Reza 20/08/98) voir commentaire ds .h (pb g++)
252 DlUserProcFunction fuf = (DlUserProcFunction)(*it).second;
253 TRY {
254 tokens.erase(tokens.begin());
255 fuf(tokens);
256 } CATCH(merr) {
257 fflush(stdout);
258 cout << endl;
259 cerr << endl;
260 string es = PeidaExc(merr);
261 cerr << "PIACmd::Do() Call UserFunc Exception :" << merr << es;
262 }
263}
264
265// >>>>>>>>>>> lecture/ecriture des objets, gestion des objets
266else if (kw == "openfits" ) {
267 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: openfits file " << endl; return(0); }
268 else mObjMgr->ReadFits(tokens[0]);
269}
270else if (kw == "savefits" ) {
271 if (tokens.size() < 2) { cout << "PIACmd::Do() Usage: savefits nomobj filename " << endl; return(0); }
272 else mObjMgr->SaveFits(tokens[0], tokens[1]);
273}
274else if (kw == "openppf" ) {
275 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: openppf file " << endl; return(0); }
276 mObjMgr->ReadAll(tokens[0]);
277}
278else if (kw == "saveall" ) {
279 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: saveall file " << endl; return(0); }
280 mObjMgr->SaveAll(tokens[0]);
281}
282else if (kw == "print" ) {
283 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: print nomobj " << endl; return(0); }
284 mObjMgr->PrintObj(tokens[0]);
285}
286else if (kw == "rename" ) {
287 if (tokens.size() < 2) { cout << "PIACmd::Do() Usage: rename nomobj nomnew" << endl; return(0); }
288 mObjMgr->RenameObj(tokens[0], tokens[1]);
289}
290else if (kw == "del" ) {
291 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: del nomobj " << endl; return(0); }
292 mObjMgr->DelObj(tokens[0]);
293}
294else if (kw == "delobjs" ) {
295 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: delobjs nomobjpattern (*,?) " << endl; return(0); }
296 mObjMgr->DelObjects(tokens[0]);
297}
298else if (kw == "listobjs") mObjMgr->ListObjs();
299
300// >>>>>>>>>>> Creation d'histos 1D-2D
301else if (kw == "newh1d") {
302 if (tokens.size() < 4) { cout << "PIACmd::Do() Usage: newh1d nom xmin xmax nbin" << endl; return(0); }
303 int nbx;
304 float xmin, xmax;
305 nbx = 100;
306 xmin = 0.; xmax = 1.;
307 nbx = atoi(tokens[3].c_str());
308 xmin = atof(tokens[1].c_str()); xmax = atof(tokens[2].c_str());
309 Histo* h = new Histo(xmin, xmax, nbx);
310 mObjMgr->AddObj(h, tokens[0]);
311 }
312else if (kw == "newh2d") {
313 if (tokens.size() < 7) {
314 cout << "PIACmd::Do() Usage:newh2d nomh xmin xmax nbinx ymin ymax nbiny" << endl;
315 return(0);
316 }
317 int nbx, nby;
318 float xmin, xmax;
319 float ymin, ymax;
320 nbx = nby = 50;
321 xmin = 0.; xmax = 1.;
322 ymin = 0.; ymax = 1.;
323 nbx = atoi(tokens[3].c_str());
324 nby = atoi(tokens[6].c_str());
325 xmin = atof(tokens[1].c_str()); xmax = atof(tokens[2].c_str());
326 ymin = atof(tokens[4].c_str()); ymax = atof(tokens[5].c_str());
327 Histo2D* h = new Histo2D(xmin, xmax, nbx, ymin, ymax, nby);
328 mObjMgr->AddObj(h, tokens[0]);
329 }
330else if (kw == "newprof") {
331 if (tokens.size() < 4)
332 { cout << "PIACmd::Do() Usage: newprof nom xmin xmax nbin [ymin ymax]" << endl; return(0); }
333 int nbx;
334 float xmin, xmax, ymin = 1., ymax = -1.;
335 if(tokens.size() > 5)
336 {ymin = atof(tokens[4].c_str()); ymax = atof(tokens[5].c_str());}
337 nbx = 100;
338 xmin = 0.; xmax = 1.;
339 nbx = atoi(tokens[3].c_str());
340 xmin = atof(tokens[1].c_str()); xmax = atof(tokens[2].c_str());
341 HProf* h = new HProf(xmin, xmax, nbx, ymin, ymax);
342 mObjMgr->AddObj(h, tokens[0]);
343 }
344else if (kw == "newgfd") {
345 if (tokens.size() < 3)
346 { cout << "PIACmd::Do() Usage: newgfd nvar nalloc [errx(0/1)]" << endl; return(0); }
347 int nvar, nalloc, errx=0;
348 if (tokens.size() > 3)
349 { errx = atoi(tokens[3].c_str()); if(errx>0) errx=1; else errx = 0;}
350 nvar = atoi(tokens[1].c_str()); nalloc = atoi(tokens[2].c_str());
351 if(nvar>0 && nalloc>0) {
352 GeneralFitData* gfd = new GeneralFitData(nvar,nalloc,errx);
353 mObjMgr->AddObj(gfd, tokens[0]);
354 }
355 }
356
357// >>>>>>>>>>> Affichage des objets
358else if ( (kw == "disp") || (kw == "surf") ) {
359 if (tokens.size() < 1) { cout << "PIACmd::Do() Usage: disp nomobj opt" << endl; return(0); }
360 string opt = "n";
361 if (tokens.size() > 1) opt = tokens[1];
362 if (kw == "disp") mObjMgr->DisplayObj(tokens[0], opt);
363 else if (kw == "surf") mObjMgr->DisplaySurf3D(tokens[0], opt);
364 }
365
366else if (kw == "nt2d") {
367 if (tokens.size() < 5) { cout << "PIACmd::Do() Usage: nt2d nomobj varx vary errx erry opt" << endl; return(0); }
368 string opt = "n";
369 if (tokens.size() > 5) opt = tokens[5];
370 string ph = "";
371 mObjMgr->DisplayNT(tokens[0],tokens[1],tokens[2], ph, tokens[3], tokens[4], ph, opt);
372 }
373else if (kw == "nt3d") {
374 if (tokens.size() < 7) { cout << "PIACmd::Do() Usage: nt3d nomobj varx vary varz errx erry errz opt" << endl; return(0); }
375 string opt = "n";
376 if (tokens.size() > 7) opt = tokens[7];
377 mObjMgr->DisplayNT(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], tokens[5], tokens[6], opt);
378 }
379
380else if (kw == "gfd2d") {
381 if(tokens.size()<2)
382 {cout<<"PIACmd::Do() Usage: gfd2d nomobj numvarx erreur=(x y xy) opt"<<endl;
383 return(0);}
384 string numvary = "";
385 string err = "";
386 string opt = "n";
387 if(tokens.size()>2) err = tokens[2];
388 if(tokens.size()>3) opt = tokens[3];
389 mObjMgr->DisplayGFD(tokens[0],tokens[1],numvary,err,opt);
390 }
391else if (kw == "gfd3d") {
392 if(tokens.size()<3)
393 {cout<<"PIACmd::Do() Usage: gfd3d nomobj numvarx numvary erreur=(x y z xy xz yz xyz) opt"<<endl;
394 return(0);}
395 string err = "";
396 string opt = "n";
397 if(tokens.size()>3) err = tokens[3];
398 if(tokens.size()>4) opt = tokens[4];
399 mObjMgr->DisplayGFD(tokens[0],tokens[1],tokens[2],err,opt);
400 }
401
402// >>>>>>>>>>> Trace de fonctions
403else if ( (kw == "func") ) {
404 if (tokens.size() < 4) { cout << "PIACmd::Do() Usage: func expr xmin xmax npt opt" << endl; return(0); }
405 string opt = "n";
406 if (tokens.size() > 4) opt = tokens[4];
407 int np;
408 float xmin, xmax;
409 np = 100;
410 xmin = 0.; xmax = 1.;
411 np = atoi(tokens[3].c_str());
412 xmin = atof(tokens[1].c_str()); xmax = atof(tokens[2].c_str());
413 mObjMgr->PlotFunc(tokens[0], xmin, xmax, np, opt);
414 }
415else if ( (kw == "func2d") ) {
416 if (tokens.size() < 7) {
417 cout << "PIACmd::Do() Usage: func2d expr xmin xmax nptx ymin ymax npty opt" << endl;
418 return(0);
419 }
420 int npx, npy;
421 float xmin, xmax;
422 float ymin, ymax;
423 npx = npy = 50;
424 xmin = 0.; xmax = 1.;
425 ymin = 0.; ymax = 1.;
426 npx = atoi(tokens[3].c_str());
427 npy = atoi(tokens[6].c_str());
428 xmin = atof(tokens[1].c_str()); xmax = atof(tokens[2].c_str());
429 ymin = atof(tokens[4].c_str()); ymax = atof(tokens[5].c_str());
430 string opt = "n";
431 if (tokens.size() > 7) opt = tokens[7];
432 mObjMgr->PlotFunc2D(tokens[0], xmin, xmax, ymin, ymax, npx, npy, opt);
433 }
434
435// >>>>>>>>>>> Trace d'expressions de N_Tuple, StarList, etc ...
436else if (kw == "plot2d" ) {
437 if (tokens.size() < 4) {
438 cout << "PIACmd::Do() Usage: plot2d nomobj expx expy [experrx experry] expcut opt" << endl;
439 return(0);
440 }
441 string errx = ""; string erry = ""; string ecut = "1";
442 string opt = "n";
443 if (tokens.size() < 6) { // Plot sans les erreurs
444 ecut = tokens[3];
445 if (tokens.size() > 4) opt = tokens[4];
446 }
447 else { // Plot avec les erreurs
448 errx = tokens[3]; erry = tokens[4]; ecut = tokens[5];
449 if (tokens.size() > 6) opt = tokens[6];
450 }
451 mObjMgr->DisplayPoints2D(tokens[0],tokens[1],tokens[2],errx,erry,ecut,opt);
452 }
453
454else if (kw == "plot3d" ) {
455 if (tokens.size() < 5) {
456 cout << "PIACmd::Do() Usage: plot3d nomobj expx expy expz expcut opt" << endl;
457 return(0);
458 }
459 string opt = "n";
460 if (tokens.size() > 5) opt = tokens[5];
461 mObjMgr->DisplayPoints3D(tokens[0],tokens[1],tokens[2],tokens[3], tokens[4], opt);
462 }
463
464else if (kw == "projh1d" ) {
465 if (tokens.size() < 5) {
466 cout << "PIACmd::Do() Usage: projh1d nomobj expx expwt expcut nomh1 opt" << endl;
467 return(0);
468 }
469 string opt = "n";
470 if (tokens.size() > 5) opt = tokens[5];
471 mObjMgr->ProjectH1(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], opt);
472 }
473
474else if (kw == "projh2d" ) {
475 if (tokens.size() < 6) {
476 cout << "PIACmd::Do() Usage: projh2 nomobj expx expy expwt expcut nomh2 opt" << endl;
477 return(0);
478 }
479 string opt = "n";
480 if (tokens.size() > 6) opt = tokens[6];
481
482 mObjMgr->ProjectH2(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], tokens[5], opt);
483 }
484
485else if (kw == "projprof" ) {
486 if (tokens.size() < 6) {
487 cout << "PIACmd::Do() Usage: projprof nomobj expx expy expwt expcut nomprof opt" << endl;
488 return(0);
489 }
490 string opt = "n";
491 if (tokens.size() > 6) opt = tokens[6];
492
493 mObjMgr->ProjectHProf(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], tokens[5], opt);
494 }
495
496else if (kw == "fillnt" ) {
497 if (tokens.size() < 7) {
498 cout << "PIACmd::Do() Usage: fillnt nomobj expx expy expz expt expcut nomnt" << endl;
499 return(0);
500 }
501 mObjMgr->FillNT(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], tokens[5], tokens[6] );
502 }
503
504else if (kw == "fillvec" ) {
505 if (tokens.size() < 4) {
506 cout << "PIACmd::Do() Usage: fillvec nomobj expx expcut nomvec opt" << endl;
507 return(0);
508 }
509 string opt = "n";
510 if (tokens.size() > 4) opt = tokens[4];
511 mObjMgr->FillVect(tokens[0],tokens[1],tokens[2], tokens[3], opt);
512 }
513
514else if (kw == "fillgd1" ) {
515 if (tokens.size() < 5) {
516 cout << "PIACmd::Do() Usage: fillgd1 nomobj expx expy experry expcut nomgfd" << endl;
517 return(0);
518 }
519 string nomgfd = "";
520 if (tokens.size() > 5) nomgfd = tokens[5];
521 string expy = "";
522 mObjMgr->FillGFD(tokens[0],tokens[1], expy, tokens[2], tokens[3], tokens[4],nomgfd);
523 }
524
525else if (kw == "fillgd2" ) {
526 if (tokens.size() < 6) {
527 cout << "PIACmd::Do() Usage: fillgd2 nomobj expx expy expz experrz expcut nomgfd" << endl;
528 return(0);
529 }
530 string nomgfd = "";
531 if (tokens.size() > 6) nomgfd = tokens[6];
532 mObjMgr->FillGFD(tokens[0],tokens[1],tokens[2], tokens[3], tokens[4], tokens[5],nomgfd);
533 }
534
535// Calcul TF
536else if (kw == "fft" ) {
537 if (tokens.size() < 1) {
538 cout << "PIACmd::Do() Usage: fft nomobj opt" << endl;
539 return(0);
540 }
541 string opt = "n";
542 if (tokens.size() > 1) opt = tokens[1];
543 mObjMgr->FFT(tokens[0], opt);
544 }
545
546// Fit 1D sur objets 1D. Egalement Fit 2D sur objets 2D.
547else if (kw == "fit") {
548 if (tokens.size() < 2) {
549 cout <<"PIACmd::Do() Usage:fit nomobj func \n"
550 <<" [p:p1,...,pn s:s1,...,sn m:m1,...,mn M:M1,...,Mn o:... o:...]\n";
551 return(0);
552 }
553 string p=""; string s=""; string m=""; string M=""; string O="";
554 if (tokens.size()>2)
555 for(int ip=2;ip<tokens.size();ip++) {
556 if(tokens[ip].length()<=2) continue;
557 const char *c = tokens[ip].c_str();
558 if(c[1]!=':') continue;
559 if(c[0]=='p') p=c+2;
560 else if(c[0]=='s') s=c+2;
561 else if(c[0]=='m') m=c+2;
562 else if(c[0]=='M') M=c+2;
563 else if(c[0]=='o') {O += ","; O += c+2;}
564 }
565 mObjMgr->Fit12D(tokens[0],tokens[1],p,s,m,M,O);
566}
567
568#ifndef SANS_StarReco
569// FITPSF sur image
570else if (kw == "fitpsf" ) {
571 if (tokens.size() < 1) {
572 cout << "PIACmd::Do() Usage: fitpsf psf_typ [options_fit]" << endl;
573 return(0);
574 }
575 FitPSFImg(tokens);
576 }
577
578// Photometrie d'ouvertue sur image
579else if (kw == "aper" ) {
580 if (tokens.size() < 1) {
581 cout << "PIACmd::Do() Usage: aper fond_typ [opt_calcul opt_disp]" << endl;
582 return(0);
583 }
584 Photom_Ouv(tokens);
585 }
586#endif
587
588else {
589 cerr << "PIACmd::Do() Erreur - Commande " << kw << " inconuue ! " << endl;
590 return(-1);
591 }
592if (timing) gltimer->Split();
593
594return(0);
595}
596
597
598
599/* --Methode-- */
600int PIACmd::LinkUserFuncs(string& fnameso, string& func1, string& func2, string& func3)
601// string& func4, string& func5)
602{
603string cmd;
604int rc;
605
606if (dynlink) delete dynlink; dynlink = NULL;
607usfmap.clear();
608
609dynlink = new PDynLinkMgr(fnameso, true);
610if (dynlink == NULL) {
611 string sn = fnameso;
612 cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur ouverture SO " << sn << endl;
613 return(2);
614 }
615
616int nok=0;
617// on utilise donc des DlFunction (Reza 20/08/98) voir commentaire ds .h (pb g++)
618// DlUserProcFunction f = NULL;
619DlFunction f = NULL;
620if ((func1.length() < 1) || (func1 == "-") || (func1 == ".") ) goto fin;
621// f = (DlUserProcFunction) dlsym(dlhandle, func1.c_str());
622f = dynlink->GetFunction(func1);
623if (f) { nok++; usfmap[func1] = f; }
624else cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur linking " << func1 << endl;
625
626if ((func2.length() < 1) || (func2 == "-") || (func2 == ".") ) goto fin;
627// f = (DlUserProcFunction) dlsym(dlhandle, func2.c_str());
628f = dynlink->GetFunction(func2);
629if (f) { nok++; usfmap[func2] = f; }
630else cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur linking " << func2 << endl;
631
632if ((func3.length() < 1) || (func3 == "-") || (func3 == ".") ) goto fin;
633// f = (DlUserProcFunction) dlsym(dlhandle, func3.c_str());
634f = dynlink->GetFunction(func3);
635if (f) { nok++; usfmap[func3] = f; }
636else cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur linking " << func3 << endl;
637
638/* Pb compile g++ 2.7.2
639if ((func4.length() < 1) || (func4 == "-") || (func4 == ".") ) goto fin;
640// f = (DlUserProcFunction) dlsym(dlhandle, func4.c_str());
641f = dynlink->GetFunction(func4);
642if (f) { nok++; usfmap[func4] = f; }
643else cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur linking " << func4 << endl;
644
645if ((func5.length() < 1) || (func5 == "-") || (func5 == ".") ) goto fin;
646// f = (DlUserProcFunction) dlsym(dlhandle, func5.c_str());
647f = dynlink->GetFunction(func5);
648if (f) { nok++; usfmap[func5] = f; }
649else cerr << "PIACmd/LinkUserFuncs_Erreur: Erreur linking " << func5 << endl;
650*/
651fin:
652if (nok < 1) { if (dynlink) delete dynlink; dynlink = NULL; return(3); }
653else return(0);
654}
655
656/////////////////////////////////////////////////////////////////////////
657/////////////////////////////////////////////////////////////////////////
658/////////////////////////////// Fonctions ///////////////////////////////
659/////////////////////////////////////////////////////////////////////////
660/////////////////////////////////////////////////////////////////////////
661#ifndef SANS_StarReco
662#include "psf.h"
663#include "dynccd.h"
664#include "pimgadapter.h"
665#include "nbmath.h"
666#include "nbtri.h"
667
668/* Nouvelle-Fonction rz + cmv 15/10/98 */
669int FitPSFImg(cmdtok& tokens)
670// Fit de PSF sur une Image.
671//| tokens[0] = Type de PSF ()
672//| : G = fit gaussienne+fond (volume)
673//| : Gi = fit gaussienne+fond integree (volume)
674//| : d = fit DL de gaussienne+fond (volume)
675//| : di = fit DL de gaussienne+fond integree (volume)
676//| : D = fit DL de gaussienne+fond avec coeff variable p6 (volume)
677//| : Di = fit DL de gaussienne+fond integree avec coeff variable p6 (volume)
678//| : M = fit Moffat+fond (expos=p6) (volume)
679//| : Mi = fit Moffat+fond integree (expos=p6) (volume)
680//| tokens[1] = options pour le fit ex: l1.0,G2.,R7.,m10.,M65000.,P3.,f,n,r,i,w10/450
681//| : r = ecriture de l'image des residus (def=non)
682//| : f = ecriture de l'image de de la PSF fitee (def=non)
683//| : i = ecriture du pave initial selectionne pour le fit (def=non)
684//| : n = ecriture de l'image du bruit (def=non)
685//| : Gaa.bbb = gain en e-/ADU (def=1)
686//| : Raa.bbb = readout noise en e- (def=0)
687//| : maa.bbb = limite basse dynamique en ADU (def=-9.e19)
688//| : Maa.bbb = limite haute dynamique en ADU (def=9.e19)
689//| : Paa.bbb = valeur du parametre supplementaire pour
690//| : la.a = print a=FitPSFImg, b=PSF.Fit PSF.IniFit
691//| : wa/b = "a" est le facteur de zoom de la fenetre de representation
692//| des resultats (def=6) et b sa taille maximale (def=500).
693//| Return: 0 si Ok, <0 sinon.
694{
695// Decodage des arguments
696if(tokens.size()<1) return(-1);
697int typ_psf=PSF::psfGaussienne;
698bool wres=false, wpsf=false, wpav=false, wnois=false;
699float gain=1., readout=0., pixmin=-9.e19, pixmax=9.e19;
700int zoom=6, wmax=500;
701float par6=-1.;
702int lp=1, lpg=0;
703string dummy,dum;
704
705if(tokens.size()>=2) {
706 dummy = "," + tokens[1]+ ",";
707 if(strstr(dummy.c_str(),",r,")) wres=true;
708 if(strstr(dummy.c_str(),",f,")) wpsf=true;
709 if(strstr(dummy.c_str(),",i")) wpav=true;
710 if(strstr(dummy.c_str(),",n")) wnois=true;
711 if(strstr(dummy.c_str(),",G")) {
712 size_t p = dummy.find(",G"); size_t q = dummy.find_first_of(',',p+1);
713 dum = dummy.substr(p,q-p);
714 if(dum.length()>2) sscanf(dum.c_str(),",G%f",&gain);
715 }
716 if(strstr(dummy.c_str(),",P")) {
717 size_t p = dummy.find(",P"); size_t q = dummy.find_first_of(',',p+1);
718 dum = dummy.substr(p,q-p);
719 if(dum.length()>2) sscanf(dum.c_str(),",P%f",&par6);
720 }
721 if(strstr(dummy.c_str(),",R")) {
722 size_t p = dummy.find(",R"); size_t q = dummy.find_first_of(',',p+1);
723 dum = dummy.substr(p,q-p);
724 if(dum.length()>2) sscanf(dum.c_str(),",R%f",&readout);
725 }
726 if(strstr(dummy.c_str(),",m")) {
727 size_t p = dummy.find(",m"); size_t q = dummy.find_first_of(',',p+1);
728 dum = dummy.substr(p,q-p);
729 if(dum.length()>2) sscanf(dum.c_str(),",m%f",&pixmin);
730 }
731 if(strstr(dummy.c_str(),",M")) {
732 size_t p = dummy.find(",M"); size_t q = dummy.find_first_of(',',p+1);
733 dum = dummy.substr(p,q-p);
734 if(dum.length()>2) sscanf(dum.c_str(),",M%f",&pixmax);
735 }
736 if(strstr(dummy.c_str(),",l")) { // niveau de print
737 size_t p = dummy.find(",l"); size_t q = dummy.find_first_of(',',p+1);
738 dum = dummy.substr(p,q-p);
739 float ab;
740 if(dum.length()>2) sscanf(dum.c_str(),",l%f",&ab);
741 if(ab<0) ab = 0.;
742 lp = (int) ab; lpg = (int) 10.*(ab-lp);
743 }
744 if(strstr(dummy.c_str(),",w")) { // niveau de print
745 size_t p = dummy.find(",w"); size_t q = dummy.find_first_of(',',p+1);
746 dum = dummy.substr(p,q-p);
747 if(dum.length()>2) sscanf(dum.c_str(),",w%d/%d",&zoom,&wmax);
748 }
749}
750
751dummy = "," + tokens[0]+ ",";
752if(strstr(dummy.c_str(),",G,")) typ_psf=PSF::psfGaussienne;
753else if(strstr(dummy.c_str(),",Gi,")) typ_psf=PSF::psfGaussInteg;
754else if(strstr(dummy.c_str(),",d,") ) typ_psf=PSF::psfDelGaussienne;
755else if(strstr(dummy.c_str(),",di,")) typ_psf=PSF::psfDelGaussinteg;
756else if(strstr(dummy.c_str(),",D,") ) {typ_psf=PSF::psfDel1Gaussienne; if(par6<=0.) par6=1.;}
757else if(strstr(dummy.c_str(),",Di,")) {typ_psf=PSF::psfDel1GaussInteg; if(par6<=0.) par6=1.;}
758else if(strstr(dummy.c_str(),",M,") ) {typ_psf=PSF::psfMoffat; if(par6<=1.) par6=2.5;}
759else if(strstr(dummy.c_str(),",Mi,")) {typ_psf=PSF::psfMoffInteg; if(par6<=1.) par6=2.5;}
760
761if(gain<=0.) gain=1.; if(readout<0.) readout=0.;
762if(zoom<1) zoom=1; if(zoom>10) zoom=10; // Il semble que le zoom soit limite a 10 !!!
763if(wmax>500) wmax=500;
764if(lp<0) lp=1; if(lpg<0) lpg=0;
765
766// Wigdet,piimage et image courants
767NamedObjMgr omg;
768PIStdImgApp* app = omg.GetImgApp();
769if(app == NULL)
770 {cout<<"Pas de PIStdImgApp*"<<endl; return(-2);}
771PIBaseWdg* curw= app->CurrentBaseWdg() ;
772if(curw == NULL)
773 {cout<<"Pas de PIBaseWdg*"<<endl; return(-3);}
774if(curw->kind() != PIImage::ClassId)
775 {cout<<"Current window not an Image"<<endl; return(-4);}
776PIImage* curpimg = (PIImage*)curw;
777if(curpimg == NULL)
778 {cout<<"Pas de PIImage*"<<endl; return(-5);}
779P2DArrayAdapter* img = curpimg->Image();
780if(img == NULL)
781 {cout<<"Pas de Image*"<<endl; return(-6);}
782string nom = "fitpsf_";
783
784// Definition du pave courant sur la piimage
785int dx = curpimg->XSzPave(), dy = curpimg->YSzPave();
786int x0 = curpimg->XPave()-dx/2, y0 = curpimg->YPave()-dy/2;
787if(x0<0) x0 = 0; if(y0<0) y0 = 0;
788int x1 = x0+dx, y1 = y0+dy;
789if(x1>img->XSize()) x1=img->XSize(); if(y1>img->YSize()) y1=img->YSize();
790dx = x1-x0; dy = y1-y0;
791int dxy = (dx>dy)?dx:dy;
792ImageR4* pav = new ImageR4(dx, dy);
793{for(int j=0;j<dy;j++) for(int i=0;i<dx;i++) (*pav)(i,j)=(*img)(i+x0,j+y0);}
794
795// Bruit sur le pave
796DynCCD mdynccd(kPhotonNoise,pixmin,pixmax,gain,readout);
797ImageR4 *nois = NoiseImage(pav, &mdynccd);
798
799// Print
800int w=2*zoom*dxy+25; if(w>wmax){w=wmax;zoom=(wmax-25)/(2*dxy);if(zoom<1)zoom=1;}
801if(lp>1)
802 {cout<<"FitPSFImg: fun="<<tokens[0]<<","<<typ_psf
803 <<" Write res="<<wres<<" psf="<<wpsf<<" pav="<<wpav<<" nois="<<wnois
804 <<" Par6="<<par6<<" lp="<<lp<<","<<lpg<<endl;
805 mdynccd.Print();
806 cout<<"... dx,y="<<dxy<<" zoom="<<zoom<<" w="<<w<<endl;}
807
808// PSF et Fit
809PSF mypsf(typ_psf,100.0);
810int npar = mypsf.NParm();
811PSFStar mypsfstar(&mypsf);
812if(npar>7) mypsfstar.Parm(6)=(double)par6; // Cas moffat et Dl gauss avec coeff var.
813int rc = mypsfstar.IniFit(*pav,*nois,lpg);
814if(rc != 0)
815 {cout<<"PIDemoApp::FitPSFImg(), Echec IniFit (rc="<<rc<<")"<<endl;
816 delete nois; delete pav; return(-10); }
817rc = mypsfstar.Fit(*pav,*nois,lpg);
818if(rc != 0)
819 {cout<<"PIDemoApp::FitPSFImg(), Echec Fit (rc="<<rc<<")"<<endl;
820 delete nois; delete pav; return(-11); }
821if(lp) mypsfstar.Print(2);
822float smax,axisrat,tiltdeg;
823rc = paramga((float) mypsfstar.SigX(),(float) mypsfstar.SigY()
824 ,(float) mypsfstar.Rho(),&smax,&axisrat,&tiltdeg);
825if(rc==0 && lp)
826 printf("... sa=%g sb=%g (%g) teta=%g\n",smax,smax*axisrat,axisrat,tiltdeg);
827else printf("Erreur paramga \n");
828
829// Creation des pave resultats
830ImageR4* res = new ImageR4(dx, dy);
831ImageR4* psf = new ImageR4(dx, dy);
832CopieImage(*res,*pav); *res -= mypsfstar;
833*psf = 0.; *psf += mypsfstar;
834
835// Si on demande d'ecrire au moins un des resultats (pav,nois,res,psf) => menage!
836bool delobj = (wpav||wnois||wres||wpsf)?true:false;
837
838// Display des resultats, fenetre 2x2 {{pav,nois},{res,psf}}
839app->CreateGraphWin(2,2,w,w); app->SetZoomAtt(zoom);
840dummy = nom+"pav"; dum = dummy+"*"; if(delobj) omg.DelObjects(dum);
841if(wpav) {
842 // Si on veut garder l'objet
843 omg.AddObj(pav,dummy);
844 omg.DisplayObj(omg.LastObjName());
845} else {
846 // "true" pour tuer (delete) de l'objet ImageR4 lors de la destruction de PIImage
847 app->DispImage(new ImageAdapter<r_4>(pav,true),dummy,Disp_Next);
848}
849dummy = nom+"nois"; dum = dummy+"*"; if(delobj) omg.DelObjects(dum);
850if(wnois) {
851 omg.AddObj(nois,dummy);
852 omg.DisplayObj(omg.LastObjName());
853} else {
854 app->DispImage(new ImageAdapter<r_4>(nois,true),dummy,Disp_Next);
855}
856dummy = nom+"res"; dum = dummy+"*"; if(delobj) omg.DelObjects(dum);
857if(wres) {
858 omg.AddObj(res,dummy);
859 omg.DisplayObj(omg.LastObjName());
860} else {
861 app->DispImage(new ImageAdapter<r_4>(res,true),dummy,Disp_Next);
862}
863dummy = nom+"psf"; dum = dummy+"*"; if(delobj) omg.DelObjects(dum);
864if(wpsf) {
865 omg.AddObj(psf,dummy);
866 omg.DisplayObj(omg.LastObjName());
867} else {
868 app->DispImage(new ImageAdapter<r_4>(psf,true),dummy,Disp_Next);
869}
870
871return(0);
872}
873
874/* Nouvelle-Fonction cmv 16/10/98 */
875int Photom_Ouv(cmdtok& tokens)
876// Photometrie d'ouverture sur une Image.
877//| tokens[0] = Type de calcul du fond et du flux.
878//| : b = fond calcule sur les pixels du bord du pave.
879//| : a = fond sur les pixels autres que ceux servant au calcul du flux.
880//| : faa.bbb = fond fixe a aa.bbb
881//| defaut : "b"
882//| tokens[1] = options pour le fit ex: l1
883//| : ra = "a" rayon minimal de scan (def=0)
884//| : ga = "a" taille de la fenetre glissante pour les extrapolations (def=4)
885//| : la = "a" niveau de print
886//| : c = la surface pour la photometrie est un carre (defaut=cercle)
887//| : C = recentrage du pave sur le pixel le + brillant (defaut=non)
888//| tokens[2] = options graphiques
889//| Return: 0 si Ok, <0 sinon.
890{
891// Decodage des arguments
892if(tokens.size()<1) return(-1);
893int typ_fond=1, rmin=0, rmax, npt_gliss=4, lp=0;
894float fondimpos=0., fondbord=0., fondautre;
895bool cercle=true, centre=false;
896string opt="";
897string dummy; string dum;
898
899if(tokens.size()>=3) opt = tokens[2];
900if(tokens.size()>=2) {
901 dummy = "," + tokens[1]+ ",";
902 if(strstr(dummy.c_str(),",r")) {
903 size_t p = dummy.find(",r"); size_t q = dummy.find_first_of(',',p+1);
904 dum = dummy.substr(p,q-p);
905 if(dum.length()>2) sscanf(dum.c_str(),",r%d",&rmin);
906 }
907 if(strstr(dummy.c_str(),",g")) {
908 size_t p = dummy.find(",g"); size_t q = dummy.find_first_of(',',p+1);
909 dum = dummy.substr(p,q-p);
910 if(dum.length()>2) sscanf(dum.c_str(),",g%d",&npt_gliss);
911 }
912 if(strstr(dummy.c_str(),",l")) {
913 size_t p = dummy.find(",l"); size_t q = dummy.find_first_of(',',p+1);
914 dum = dummy.substr(p,q-p);
915 if(dum.length()>2) sscanf(dum.c_str(),",l%d",&lp);
916 }
917 if(strstr(dummy.c_str(),",c")) cercle = false;
918 if(strstr(dummy.c_str(),",C")) centre = true;
919}
920
921dummy = "," + tokens[0]+ ",";
922if(strstr(dummy.c_str(),",b,")) typ_fond=1;
923else if(strstr(dummy.c_str(),",a,")) typ_fond=2;
924else if(strstr(dummy.c_str(),",f")) {
925 typ_fond=3;
926 size_t p = dummy.find(",f"); size_t q = dummy.find_first_of(',',p+1);
927 dum = dummy.substr(p,q-p);
928 if(dum.length()>2) sscanf(dum.c_str(),",f%f",&fondimpos);
929}
930
931if(rmin<0) rmin=0; if(npt_gliss<=0) npt_gliss=4; if(lp<0) lp=1;
932
933// Wigdet,piimage et image courants
934NamedObjMgr omg;
935PIStdImgApp* app = omg.GetImgApp();
936if(app == NULL)
937 {cout<<"Pas de PIStdImgApp*"<<endl; return(-2);}
938PIBaseWdg* curw= app->CurrentBaseWdg() ;
939if(curw == NULL)
940 {cout<<"Pas de PIBaseWdg*"<<endl; return(-3);}
941if(curw->kind() != PIImage::ClassId)
942 {cout<<"Current window not an Image"<<endl; return(-4);}
943PIImage* curpimg = (PIImage*)curw;
944if(curpimg == NULL)
945 {cout<<"Pas de PIImage*"<<endl; return(-5);}
946P2DArrayAdapter* img = curpimg->Image();
947if(img == NULL)
948 {cout<<"Pas de Image*"<<endl; return(-6);}
949string nom = "ouver_";
950
951// Definition du pave courant sur la piimage
952int xc = curpimg->XPave(), yc = curpimg->YPave();
953int dx = curpimg->XSzPave(), dy = curpimg->YSzPave();
954int dxym = (dx<dy)?dx:dy; if(dxym%2==0) dxym++; rmax = dxym/2;
955if(dxym<3||dxym>img->XSize()||dxym>img->YSize())
956 {cout<<"Pave trop petit/grand: "<<dxym<<endl; return(-10);}
957if(rmin>=rmax)
958 {cout<<"rayon mini="<<rmin<<">= rmax="<<rmax<<endl; return(-11);}
959if(centre) { // Recentrage autour du pixel le + brillant?
960 int xcnew=-1, ycnew=-1; float pxnew=-1.e20,pxold=-1.e20,px;
961 for(int j=yc-rmax;j<=yc+rmax;j++) {
962 if(j<0 || j>=img->YSize()) continue;
963 for(int i=xc-rmax;i<=xc+rmax;i++) {
964 if(i<0 || i>=img->XSize()) continue;
965 px = (*img)(i,j);
966 if(i==xc && j==yc) pxold = px;
967 if((xcnew==-1 && ycnew==-1) || px>=pxnew) {xcnew=i; ycnew=j; pxnew=px;}
968 // cout<<"xcnew,ycnew,pxnew="<<xcnew<<" "<<ycnew<<" "<<pxnew<<endl;
969 } }
970 if(lp>1) cout<<"Re-centrage(x,y,px): old=("<<xc<<","<<yc<<","<<pxold<<")"
971 <<" -> new=("<<xcnew<<","<<ycnew<<","<<pxnew<<")"<<endl;
972 if(xcnew!=-1 && ycnew!=-1 && pxnew>pxold) {xc=xcnew; yc=ycnew;}
973}
974int x0=xc-rmax, y0=yc-rmax, x1=xc+rmax, y1=yc+rmax;
975if(x0<0||y0<0||x1>=img->XSize()||y1>=img->YSize())
976 {cout<<"Pave hors image"<<endl; return(-12);}
977int ic = rmax, jc = rmax;
978ImageR4* pav = new ImageR4(dxym, dxym);
979{for(int j=-rmax;j<=rmax;j++) for(int i=-rmax;i<=rmax;i++)
980 (*pav)(ic+i,jc+j)=(*img)(xc+i,yc+j);}
981
982// Print
983if(lp>1)
984 cout<<"photom_ouv[cercle="<<cercle<<",typ="<<typ_fond<<"]: pav["
985 <<dx<<","<<dy<<","<<dxym
986 <<"] xc,yc=["<<xc<<","<<yc<<"] ic,jc="<<ic<<","<<jc<<endl
987 <<" (fond="<<fondimpos<<") pxmil="<<(*pav)(ic,ic)
988 <<" -> Rph de "<<rmin<<" a "<<rmax<<endl;
989
990// Ouverture des buffers et objets necessaires
991float *p = new float[dxym*dxym];
992r_4 xnt[4];
993NTuple *ntfl=NULL, *ntex=NULL;
994{char* name[4]={"npx","flux","fond","mag"}; ntfl = new NTuple(4,name);}
995{char* name[3]={"k","a","f0"}; ntex = new NTuple(3,name);}
996
997// Calcul du fond sur les pixels du bord du pave
998{
999int nf = 0;
1000for(int i=0;i<dxym;i++) {p[nf]=(*pav)(i,0); nf++; p[nf]=(*pav)(i,dxym-1); nf++;}
1001for(int j=1;j<dxym-1;j++) {p[nf]=(*pav)(0,j); nf++; p[nf]=(*pav)(dxym-1,j); nf++;}
1002qsort(p,(size_t) nf,sizeof(float),qSort_Float);
1003fondbord = (p[nf/2]+p[(nf-1)/2])/2.;
1004if(lp>0) cout<<"Fond_Bord: "<<fondbord<<" (avec 1 px au bord="<<nf<<")"<<endl;
1005}
1006
1007// Photometrie d'ouverture variable.
1008int rfin = -1;
1009for(int r=rmin;r<=rmax;r++) {
1010 double sum=0.; int ns=0, nf=0;
1011 for(int i=0;i<dxym;i++) for(int j=0;j<dxym;j++) {
1012 double ray2 = (i-ic)*(i-ic)+(j-jc)*(j-jc)-0.1;
1013 if( (cercle && ray2<=r*r) ||
1014 (!cercle && ic-r<=i && i<=ic+r && jc-r<=j && j<=jc+r) )
1015 {sum += (*pav)(i,j); ns++;}
1016 else {p[nf] = (*pav)(i,j); nf++;}
1017 }
1018 if(typ_fond==2) {
1019 if(nf<2) continue;
1020 qsort(p,(size_t) nf,sizeof(float),qSort_Float);
1021 fondautre = (p[nf/2]+p[(nf-1)/2])/2.;
1022 }
1023 double fond=-1.e19;
1024 if(typ_fond==3) fond=fondimpos; else if(typ_fond==2) fond=fondautre; else fond=fondbord;
1025 rfin = r;
1026 xnt[0] = ns;
1027 xnt[1] = sum-fond*ns;
1028 xnt[2] = fond;
1029 xnt[3] = (xnt[1]>0.)? -2.5*log10((double)xnt[1]): -99.;
1030 if(lp>0)
1031 cout<<"Ouv[r="<<r<<",ns="<<xnt[0]<<",nf="<<nf<<"]: fd="<<xnt[2]
1032 <<" fl="<<xnt[1]<<" mag="<<xnt[3]<<endl;
1033 ntfl->Fill(xnt);
1034}
1035delete [] p;
1036int nr = ntfl->NEntry();
1037if(lp>1) cout<<"Nombre de calculs de flux: "<<nr<<endl;
1038if(nr<=0)
1039 {cout<<"Pas de flux calcules"<<endl; delete pav; delete ntfl; delete ntex; return(-13);}
1040
1041cout<<"Ouv[rouv="<<rfin<<",npix="<<xnt[0]<<"]: fond="<<xnt[2]
1042 <<" flux="<<xnt[1]<<" mag="<<xnt[3]<<endl;
1043
1044// Extrapolation: F_mes = F_vrai + npxflux*(Fd_vrai-Fd_mes)
1045// On fit F_mes=f(npxflux) sur une fenetre glissante de npt_gliss pts.
1046bool ex = false;
1047if(nr>=npt_gliss) {
1048 ex = true;
1049 double *npix = new double[nr], *flux = new double[nr], *err = new double[nr];
1050 for(int i=0;i<nr;i++) { ntfl->GetVec(i,xnt); npix[i]=xnt[0]; flux[i]=xnt[1]; err[i]=1.;}
1051 double a,b; int n;
1052 for(int k=0;k<=nr-npt_gliss;k++) {
1053 n = npt_gliss; FitLin(&npix[k],&flux[k],&err[k],&n,&b,&a);
1054 xnt[0]=k; xnt[1]=a; xnt[2]=b; ntex->Fill(xnt);
1055 if(lp>1)
1056 cout<<"Fit["<<k<<","<<k+npt_gliss-1<<"] f0="<<xnt[2]<<" pente="<<xnt[1]<<endl;
1057 }
1058 delete [] npix; delete [] flux; delete [] err;
1059}
1060
1061// Display des resultats, fenetre 2x2 {{flux%ns,fond%ns},{f0,pente}}
1062if(ex) app->CreateGraphWin(2,2); else app->CreateGraphWin(1,2);
1063dummy = nom+"*"; omg.DelObjects(dummy);
1064
1065dummy=nom+"pav"; omg.AddObj(pav,dummy);
1066
1067string xd=""; string yd=""; string zd,exd,eyd,ezd;
1068dummy=nom+"flux"; omg.AddObj(ntfl,dummy);
1069xd="npx"; yd="flux";
1070omg.DisplayNT(omg.LastObjName(),xd,yd,zd,exd,eyd,ezd,opt);
1071if(typ_fond==2)
1072 {xd="npx"; yd="fond"; omg.DisplayNT(omg.LastObjName(),xd,yd,zd,exd,eyd,ezd,opt);}
1073if(ex) {
1074 dummy=nom+"fit"; omg.AddObj(ntex,dummy);
1075 if(typ_fond!=2)
1076 {xd="a"; yd="f0"; omg.DisplayNT(omg.LastObjName(),xd,yd,zd,exd,eyd,ezd,opt);}
1077 xd = "k"; yd = "f0";
1078 omg.DisplayNT(omg.LastObjName(),xd,yd,zd,exd,eyd,ezd,opt);
1079 xd = "k"; yd = "a";
1080 omg.DisplayNT(omg.LastObjName(),xd,yd,zd,exd,eyd,ezd,opt);
1081}
1082
1083return(0);
1084}
1085#endif
Note: See TracBrowser for help on using the repository browser.