source: Sophya/trunk/SophyaPI/PIext/piafitting.cc@ 546

Last change on this file since 546 was 544, checked in by ercodmgr, 26 years ago

Documentation + ameliorations DrawStats + vector->tvector pour Planck - Reza 3/11/99

File size: 62.4 KB
RevLine 
[381]1//CMVBUG reste pb suivant sous Linux-egcs (OSF1-cxx c'est ok):
[379]2//openppf cmvwppf.ppf
3//fitw h1g g2 p:4500,45,10,900
4//delobjs h1g
5//clic sur FIT -> OK
6//clic sur FIT 2sd fois -> core dump!!!
7//
[361]8#include <stdio.h>
9#include <stdlib.h>
10#include <ctype.h>
11#include <iostream.h>
12#include <typeinfo>
13
14#include "strutil.h"
[392]15#include "nbtri.h"
[361]16#include "generalfit.h"
17#include "fct1dfit.h"
18#include "fct2dfit.h"
19#include "ntuple.h"
20#include "cimage.h"
21#include "histos.h"
22#include "histos2.h"
23#include "ntuple.h"
24#include "hisprof.h"
25
[544]26#ifdef SANS_EVOLPLANCK
27#include "matrix.h"
28#include "cvector.h"
29#else
30#include "tmatrix.h"
31#include "tvector.h"
32#endif
33
[392]34#include "piafitting.h"
[361]35#include "nobjmgr.h"
36#include "pistdimgapp.h"
37
[379]38////////////////////////////////////////////////////////////////////
39////---------- Classe de fenetre de fit interactive ------------////
40////////////////////////////////////////////////////////////////////
[361]41class PIAFitterWind : public PIWindow {
42public :
43 PIAFitterWind(PIStdImgApp* par, PIAFitter* fiter);
44 virtual ~PIAFitterWind();
45 virtual void Show();
46 virtual void Process(PIMessage msg, PIMsgHandler* sender, void* data=NULL);
[379]47 void SetState(void);
[361]48protected :
[379]49 PIStdImgApp* mDap;
50 PIAFitter* mFitter;
51 PILabel** lab; int nlab;
52 PIText** txt; int ntxt;
53 PIButton** but; int nbut;
54 PICheckBox** ckb; int nckb;
[381]55 PIOptMenu** pom; int npom;
[379]56 bool ReFillGData;
[381]57 int saveI1,saveI2, saveJ1, saveJ2;
[385]58 double ErrValue;
[361]59};
60
[379]61////////////////////////////////////////////////////////////////////////
[466]62// --- Generation automatique du "usage" pour helpfit:
63// 1-/ copier le texte ci-dessous de piafitting.cc dans toto
64// 2-/ cat toto | sed 's?^//?usage+="\\n?' | sed 's?$? ";?'
65// NE PAS EFFACER LA SUITE: c'est elle qui doit etre modifiee,
66// le usage du helpfit sera regenere a l'aide des commandes ci-dessus.
67////////////////////////////////////////////////////////////////////////
[379]68//| --------------- Fit Lineaire a 1 et 2 dimensions ---------------
69//| Syntaxe:
70//| fitlin nom pnn [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...]
71//| avec:
[466]72//| nom : cf commentaire ordre ''fit''
[379]73//| pnn : fit polynome degre nn avec classe Poly (lineaire) 1D ou 2D
[466]74//| o:aa,...,bb : cf commentaires ordre ''fit''
[379]75//|
76////////////////////////////////////////////////////////////////////////
[361]77//| --------------- Fit d'objets a 1 et 2 dimensions ---------------
[379]78//| avec interface d'aide graphique
79//| Syntaxe:
80//| fitw nom func
81//| [p:p1,p2,...,pn s:s1,s2,...,sn m:m1,m2,...,mn M:M1,mM2,...,Mn f:f1,...,fn]
82//| [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...]
[466]83//| cf commentaire ordre ''fit''
[379]84//|
85////////////////////////////////////////////////////////////////////////
86//| --------------- Fit d'objets a 1 et 2 dimensions ---------------
87//| Syntaxe:
88//| fit nom func
89//| [p:p1,p2,...,pn s:s1,s2,...,sn m:m1,m2,...,mn M:M1,mM2,...,Mn f:f1,...,fn]
90//| [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...]
91//|----- OBJET -----
[361]92//| nom : nom de l'objet qui peut etre:
93//| fit-1D: Vector,Histo1D,HProf ou GeneraFitData(1D)
94//| fit-2D: Matrix,Histo2D,Image<T> ou GeneraFitData(2D)
[379]95//|
96//|----- FUNCTION -----
97//| func : pnn : fit polynome degre nn avec GeneralFit (non-lineaire) 1D ou 2D
[361]98//| : gnn : fit gaussienne (hauteur) + polynome de degre nn 1D
99//| : g : fit gaussienne (hauteur) 1D
100//| : enn : fit exponentielle + polynome de degre nn 1D
101//| : e : fit exponentielle 1D
102//| : Gnn : fit gaussienne (volume) + polynome de degre nn 1D
103//| : G : fit gaussienne (volume) 1D
104//| : : fit gaussienne+fond (volume) 2D
105//| : Gi : fit gaussienne+fond integree (volume) 2D
106//| : d : fit DL de gaussienne+fond (volume) 2D
107//| : di : fit DL de gaussienne+fond integree (volume) 2D
108//| : D : fit DL de gaussienne+fond avec coeff variable p6 (volume) 2D
109//| : Di : fit DL de gaussienne+fond integree avec coeff variable p6 (volume) 2D
110//| : M : fit Moffat+fond (expos=p6) (volume) 2D
111//| : Mi : fit Moffat+fond integree (expos=p6) (volume) 2D
[385]112//| : Autre : fonction definie par l'utilisateur (cf commentaires ci apres)
[379]113//|
114//|----- INIT PARAMETRES ET ETAT DU FIT -----
115//| p : p1,...,pn : valeur d'initialisation des parametres (def=0)
116//| s : s1,...,sn : valeur des steps de depart (def=1)
117//| m : m1,...,mn : valeur des minima (def=1)
118//| M : M1,...,Mn : valeur des maxima (def=-1) (max<=min : pas de limite)
119//| f : f1,...,fn : si >=1 parametre fixe sinon libre (def=0)
120//| - Remarque: si pi,si,mi ou Mi = '!' la valeur correspondante n'est pas changee
121//|
122//|----- OPTIONS -----
[466]123//| o : options ''o:Eaa.b,eaa.b,f,r,caa.b,Xaa.b''
[379]124//| F : initialisation a partir des resultats et de l'etat du fit precedent
125//| (option non prioritaire sur les definitions p:,s:,m:,M:,f:o:)
[390]126//| a : initialisation automatique des parametres du fit (si disponible)
127//| (option non prioritaire sur les definitions p:,s:,m:,M:,f:o:
[466]128//| et sur l'initialisation pa l'option ''F'')
[361]129//| f : generation d'un Objet identique contenant la fonction fittee
130//| r : generation d'un Objet identique contenant les residus
131//| Xaa.b : aa.b valeur du DXi2 d'arret (def=1.e-3)
132//| Naa : aa nombre maximum d'iterations (def=100)
[466]133//| la : niveau ''a'' de print: a=niveau de print Fit1/2D
134//| da : niveau ''a'' de debug: a=niveau de GeneralFit
[379]135//| Ii1/i2 : numeros des bins X de l'histos utilises pour le fit [i1,i2]
136//|2D Jj1/j2 : numeros des bins Y de l'histos utilises pour le fit [j1,j2]
137//| - L'erreur est celle associee a l'objet (si elle existe),
138//| elle est mise a 1 sinon, sauf si E... ou e... est precise:
[361]139//| Eaa.b : si |val|>=1 erreur = aa.b*sqrt(|val|)
140//| si |val|<1 erreur = aa.b
141//| si aa.b <=0 alors aa.b=1.0
142//| E seul est equivalent a E1.0
143//| eaa.b : erreur = aa.b
144//| si aa.b <=0 alors aa.b=1.0
145//| e seul est equivalent a e1.0
146//| xaa.b : demande de centrage: on fit x-aa.b au lieu de x)
147//| x : demande de centrage: on fit x-xc au lieu de x
148//| avec xc=abscisse du milieu de l'histogramme
[379]149//| Actif pour exp+poly 1D, poly 1D
150//| Pour gauss+poly 1D, xc est le centre de la gaussienne.
[466]151//|2D yaa.b et y : idem ''xaa.b et x'' mais pour y
[385]152//|
[379]153////////////////////////////////////////////////////////////////////////
[385]154//| --------------- Def des fcts de fit par l'utilisateur ---------------
155//| Pour definir une fonction parametree utilisateur:
156//| Un exemple de fichier contenant des fonctions a fitter: PIext/userfitfunex.c
157//| > crfitfil nom_fichier nom_fonction nvar npar
158//| pour creer un squelette de fichier C contenant la fonction a fitter
159//| nom_fichier : nom du fichier ou est le code de la fonction
160//| c'est un fichier C de nom blabla.c
161//| nom_fonction : nom de la fonction ex: gauss2
162//| nvar : nombre de variables x[] (1D=1, 2D=2, etc...)
163//| npar : nombre de parametre p[]
164//| - La fonction est donc F(x[];p[]), soit par ex:
165//| double gauss2(double const* x,double const* p)
166//| - L'utilisateur a la possibilite de donner egalement la fonction
167//| retournant les derivees de la fonction par rapport aux parametres
168//| double gauss2_der(double const* x,double const* p,double* dp)
169//| ou dp[i] = dF(x[];p[])/dp[i]
[466]170//| Par convention, cette fonction a le meme nom suivi de ''_der''
[385]171//| > crfitfun nom_fichier nom_fonction nvar npar
[466]172//| Permet de compiler le fichier ''nom_fichier'' et de linker
173//| la fonction ''nom_fonction''
[385]174//| - Un meme fichier peut contenir plusieurs fonctions a fitter
175//|
176////////////////////////////////////////////////////////////////////////
[379]177
178/* --Methode-- */
179PIAFitter::PIAFitter(PIACmd *piac, PIStdImgApp* app)
180: mApp(app), FWindFit(NULL)
[385]181, mNObj(""), mObj(NULL), mGData(NULL)
[379]182, mNPar(0), mNVar(0), mNBinX(0), mNBinY(0), mNData(0)
183, mPar(1), mStep(1), mMin(1), mMax(1), mFix(1)
184, mParSave(1), mStepSave(1), mMinSave(1), mMaxSave(1), mFixSave(1)
185, mFit(NULL)
186, mV(NULL), mH(NULL), mM(NULL), mH2(NULL), mIm(NULL), mG(NULL)
[392]187, mFunction(NULL), mFName(""), mFunc(NULL)
[384]188, mDlUFunc(NULL), mNameFitFunc(""), mFitFunc(NULL), mFitFuncDer(NULL)
189, mUFNVar(0), mUFNPar(0)
[361]190{
[379]191// Init et Set des options
192mPar=0.; mStep=1.; mMin=1.; mMax=-1.; mFix=0.;
193mParSave=0.; mStepSave=1.; mMinSave=1.; mMaxSave=-1.; mFixSave=0.;
194ResetOptions();
195mOptSave = mOpt;
[361]196
[379]197// enregistrement des ordres de fit
[466]198string kw, usage, grp = "Fitting";
[379]199
[466]200kw = "helpfit";
201usage="";
202usage+="\n////////////////////////////////////////////////////////////////////// ";
203usage+="\n| --------------- Fit Lineaire a 1 et 2 dimensions --------------- ";
204usage+="\n| Syntaxe: ";
205usage+="\n| fitlin nom pnn [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...] ";
206usage+="\n| avec: ";
207usage+="\n| nom : cf commentaire ordre ''fit'' ";
208usage+="\n| pnn : fit polynome degre nn avec classe Poly (lineaire) 1D ou 2D ";
209usage+="\n| o:aa,...,bb : cf commentaires ordre ''fit'' ";
210usage+="\n| ";
211usage+="\n////////////////////////////////////////////////////////////////////// ";
212usage+="\n| --------------- Fit d'objets a 1 et 2 dimensions --------------- ";
213usage+="\n| avec interface d'aide graphique ";
214usage+="\n| Syntaxe: ";
215usage+="\n| fitw nom func ";
216usage+="\n| [p:p1,p2,...,pn s:s1,s2,...,sn m:m1,m2,...,mn M:M1,mM2,...,Mn f:f1,...,fn] ";
217usage+="\n| [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...] ";
218usage+="\n| cf commentaire ordre ''fit'' ";
219usage+="\n| ";
220usage+="\n////////////////////////////////////////////////////////////////////// ";
221usage+="\n| --------------- Fit d'objets a 1 et 2 dimensions --------------- ";
222usage+="\n| Syntaxe: ";
223usage+="\n| fit nom func ";
224usage+="\n| [p:p1,p2,...,pn s:s1,s2,...,sn m:m1,m2,...,mn M:M1,mM2,...,Mn f:f1,...,fn] ";
225usage+="\n| [o:.aa,bb,cc, o;dd,ee,ff o:gg,hh,jj,kk etc...] ";
226usage+="\n|----- OBJET ----- ";
227usage+="\n| nom : nom de l'objet qui peut etre: ";
228usage+="\n| fit-1D: Vector,Histo1D,HProf ou GeneraFitData(1D) ";
229usage+="\n| fit-2D: Matrix,Histo2D,Image<T> ou GeneraFitData(2D) ";
230usage+="\n| ";
231usage+="\n|----- FUNCTION ----- ";
232usage+="\n| func : pnn : fit polynome degre nn avec GeneralFit (non-lineaire) 1D ou 2D ";
233usage+="\n| : gnn : fit gaussienne (hauteur) + polynome de degre nn 1D ";
234usage+="\n| : g : fit gaussienne (hauteur) 1D ";
235usage+="\n| : enn : fit exponentielle + polynome de degre nn 1D ";
236usage+="\n| : e : fit exponentielle 1D ";
237usage+="\n| : Gnn : fit gaussienne (volume) + polynome de degre nn 1D ";
238usage+="\n| : G : fit gaussienne (volume) 1D ";
239usage+="\n| : : fit gaussienne+fond (volume) 2D ";
240usage+="\n| : Gi : fit gaussienne+fond integree (volume) 2D ";
241usage+="\n| : d : fit DL de gaussienne+fond (volume) 2D ";
242usage+="\n| : di : fit DL de gaussienne+fond integree (volume) 2D ";
243usage+="\n| : D : fit DL de gaussienne+fond avec coeff variable p6 (volume) 2D ";
244usage+="\n| : Di : fit DL de gaussienne+fond integree avec coeff variable p6 (volume) 2D ";
245usage+="\n| : M : fit Moffat+fond (expos=p6) (volume) 2D ";
246usage+="\n| : Mi : fit Moffat+fond integree (expos=p6) (volume) 2D ";
247usage+="\n| : Autre : fonction definie par l'utilisateur (cf commentaires ci apres) ";
248usage+="\n| ";
249usage+="\n|----- INIT PARAMETRES ET ETAT DU FIT ----- ";
250usage+="\n| p : p1,...,pn : valeur d'initialisation des parametres (def=0) ";
251usage+="\n| s : s1,...,sn : valeur des steps de depart (def=1) ";
252usage+="\n| m : m1,...,mn : valeur des minima (def=1) ";
253usage+="\n| M : M1,...,Mn : valeur des maxima (def=-1) (max<=min : pas de limite) ";
254usage+="\n| f : f1,...,fn : si >=1 parametre fixe sinon libre (def=0) ";
255usage+="\n| - Remarque: si pi,si,mi ou Mi = '!' la valeur correspondante n'est pas changee ";
256usage+="\n| ";
257usage+="\n|----- OPTIONS ----- ";
258usage+="\n| o : options ''o:Eaa.b,eaa.b,f,r,caa.b,Xaa.b'' ";
259usage+="\n| F : initialisation a partir des resultats et de l'etat du fit precedent ";
260usage+="\n| (option non prioritaire sur les definitions p:,s:,m:,M:,f:o:) ";
261usage+="\n| a : initialisation automatique des parametres du fit (si disponible) ";
262usage+="\n| (option non prioritaire sur les definitions p:,s:,m:,M:,f:o: ";
263usage+="\n| et sur l'initialisation pa l'option ''F'') ";
264usage+="\n| f : generation d'un Objet identique contenant la fonction fittee ";
265usage+="\n| r : generation d'un Objet identique contenant les residus ";
266usage+="\n| Xaa.b : aa.b valeur du DXi2 d'arret (def=1.e-3) ";
267usage+="\n| Naa : aa nombre maximum d'iterations (def=100) ";
268usage+="\n| la : niveau ''a'' de print: a=niveau de print Fit1/2D ";
269usage+="\n| da : niveau ''a'' de debug: a=niveau de GeneralFit ";
270usage+="\n| Ii1/i2 : numeros des bins X de l'histos utilises pour le fit [i1,i2] ";
271usage+="\n|2D Jj1/j2 : numeros des bins Y de l'histos utilises pour le fit [j1,j2] ";
272usage+="\n| - L'erreur est celle associee a l'objet (si elle existe), ";
273usage+="\n| elle est mise a 1 sinon, sauf si E... ou e... est precise: ";
274usage+="\n| Eaa.b : si |val|>=1 erreur = aa.b*sqrt(|val|) ";
275usage+="\n| si |val|<1 erreur = aa.b ";
276usage+="\n| si aa.b <=0 alors aa.b=1.0 ";
277usage+="\n| E seul est equivalent a E1.0 ";
278usage+="\n| eaa.b : erreur = aa.b ";
279usage+="\n| si aa.b <=0 alors aa.b=1.0 ";
280usage+="\n| e seul est equivalent a e1.0 ";
281usage+="\n| xaa.b : demande de centrage: on fit x-aa.b au lieu de x) ";
282usage+="\n| x : demande de centrage: on fit x-xc au lieu de x ";
283usage+="\n| avec xc=abscisse du milieu de l'histogramme ";
284usage+="\n| Actif pour exp+poly 1D, poly 1D ";
285usage+="\n| Pour gauss+poly 1D, xc est le centre de la gaussienne. ";
286usage+="\n|2D yaa.b et y : idem ''xaa.b et x'' mais pour y ";
287usage+="\n| ";
288usage+="\n////////////////////////////////////////////////////////////////////// ";
289usage+="\n| --------------- Def des fcts de fit par l'utilisateur --------------- ";
290usage+="\n| Pour definir une fonction parametree utilisateur: ";
291usage+="\n| Un exemple de fichier contenant des fonctions a fitter: PIext/userfitfunex.c ";
292usage+="\n| > crfitfil nom_fichier nom_fonction nvar npar ";
293usage+="\n| pour creer un squelette de fichier C contenant la fonction a fitter ";
294usage+="\n| nom_fichier : nom du fichier ou est le code de la fonction ";
295usage+="\n| c'est un fichier C de nom blabla.c ";
296usage+="\n| nom_fonction : nom de la fonction ex: gauss2 ";
297usage+="\n| nvar : nombre de variables x[] (1D=1, 2D=2, etc...) ";
298usage+="\n| npar : nombre de parametre p[] ";
299usage+="\n| - La fonction est donc F(x[];p[]), soit par ex: ";
300usage+="\n| double gauss2(double const* x,double const* p) ";
301usage+="\n| - L'utilisateur a la possibilite de donner egalement la fonction ";
302usage+="\n| retournant les derivees de la fonction par rapport aux parametres ";
303usage+="\n| double gauss2_der(double const* x,double const* p,double* dp) ";
304usage+="\n| ou dp[i] = dF(x[];p[])/dp[i] ";
305usage+="\n| Par convention, cette fonction a le meme nom suivi de ''_der'' ";
306usage+="\n| > crfitfun nom_fichier nom_fonction nvar npar ";
307usage+="\n| Permet de compiler le fichier ''nom_fichier'' et de linker ";
308usage+="\n| la fonction ''nom_fonction'' ";
309usage+="\n| - Un meme fichier peut contenir plusieurs fonctions a fitter ";
310usage+="\n| ";
311usage+="\n////////////////////////////////////////////////////////////////////// ";
312piac->RegisterHelp(kw,usage,grp);
313
[379]314kw = "fit";
315usage = "Fitting function to DataObjects (Histo, Histo2D, Vector, ...)";
316usage += "\n Usage: fit nomobj func [Options]";
317usage += "\n [p:p1,...,pn s:s1,...,sn m:m1,...,mn M:M1,...,Mn o:... o:...]";
[466]318usage += "\n Related commands: fitw fitlin crfitfun crfitfil";
319piac->RegisterCommand(kw,usage,this,grp);
[379]320
321kw = "fitw";
322usage = "Fitting function to DataObjects with Interactive Window Help";
323usage += "\n Usage: fitw nomobj func [Options]";
324usage += "\n [p:p1,...,pn s:s1,...,sn m:m1,...,mn M:M1,...,Mn o:... o:...]";
[466]325usage += "\n Related commands: fit fitlin crfitfun crfitfil";
326piac->RegisterCommand(kw,usage,this,grp);
[379]327
328kw = "fitlin";
329usage = "Linear Fitting of Polynoms to DataObjects";
330usage += "\n Usage: fitlin nomobj func [o:... o:...]";
[466]331usage += "\n Related commands: fit fitw crfitfun crfitfil";
332piac->RegisterCommand(kw,usage,this,grp);
[384]333
334kw = "crfitfun";
[385]335usage = "Creation et link de function utilisateur pour le fit";
[384]336usage += "\n Usage: crfitfun file func nvar npar";
[466]337usage += "\n Related commands: fit fitw fitlin crfitfil";
338piac->RegisterCommand(kw,usage,this,grp);
[385]339
340kw = "crfitfil";
341usage = "Creation de fichier C pour function utilisateur du fit";
342usage += "\n Usage: crfitfil file func nvar npar";
[466]343usage += "\n Related commands: fit fitw fitlin crfitfun";
344piac->RegisterCommand(kw,usage,this,grp);
[379]345}
346
347/* --Methode-- */
348PIAFitter::~PIAFitter()
349{
[385]350if(mFit != NULL) {delete mFit; mFit=NULL;}
351if(mFunction != NULL) {delete mFunction; mFunction=NULL;}
352if(mFunc != NULL) {delete mFunc; mFunc=NULL;}
353if(mGData != NULL) {delete mGData; mGData=NULL;}
354if(FWindFit != NULL) {delete FWindFit; FWindFit=NULL;}
355if (mDlUFunc != NULL) {delete mDlUFunc; mDlUFunc=NULL;}
[379]356}
357
358/* --Methode-- */
359void PIAFitter::ResetDPointer(void)
360// Reset des pointeurs sur les donnees
361{
362mNObj = ""; mObj=NULL;
363mV=NULL; mH=NULL; mM=NULL; mH2=NULL; mIm=NULL; mG=NULL;
364}
365
366/* --Methode-- */
367void PIAFitter::ResetOptions(void)
368// Reset des options
369{
370mOpt.okres = mOpt.okfun = false;
371mOpt.polcx = mOpt.polcy = 0;
372mOpt.xc = mOpt.yc = 0.;
373mOpt.stc2 = 1.e-3;
374mOpt.nstep = 100;
375mOpt.err_e = mOpt.err_E = -1.;
376mOpt.lp = 1; mOpt.lpg = 0;
377mOpt.i1 = mOpt.j1 = mOpt.i2 = mOpt.j2 = -1;
378mOpt.fromlastfit = false;
[390]379mOpt.autoinifit = false;
[379]380}
381
382/* --Methode-- */
383void PIAFitter::DecodeOptions(string opt)
384// Decodage des options a partir des chaines de caracteres
385{
386ResetOptions();
387if(opt.length()<=0) return;
388
389size_t p,q; string dum;
390opt = "," + opt + ",";
391
392if(strstr(opt.c_str(),",F,")) // init options/param a partir du fit precedent
393 {mOpt = mOptSave; mOpt.fromlastfit = true;}
[390]394if(strstr(opt.c_str(),",a,")) mOpt.autoinifit = true; // init auto par fit
[379]395if(strstr(opt.c_str(),",r,")) mOpt.okres = true; // residus
396if(strstr(opt.c_str(),",f,")) mOpt.okfun = true; // fonction fittee
397if(strstr(opt.c_str(),",x")) { // Demande de centrage (fit X=x-xc)
398 mOpt.polcx = 2; // Le centrage est calcule automatiquement
399 p = opt.find(",x"); q = opt.find_first_of(',',p+1);
400 dum = opt.substr(p,q-p);
401 if(dum.length()>2) {
402 sscanf(dum.c_str(),",x%lf",&mOpt.xc);
403 mOpt.polcx = 1; // Le centrage est fixe par la valeur lue
404 }
405}
406if(strstr(opt.c_str(),",y")) { // Demande de centrage (fit Y=y-yc)
407 mOpt.polcy = 2; // Le centrage est calcule automatiquement
408 p = opt.find(",y"); q = opt.find_first_of(',',p+1);
409 dum = opt.substr(p,q-p);
410 if(dum.length()>2) {
411 sscanf(dum.c_str(),",y%lf",&mOpt.yc);
412 mOpt.polcy = 1; // Le centrage est fixe par la valeur lue
413 }
414}
415if(strstr(opt.c_str(),",E")) { // Erreurs imposees a "sqrt(val)" ou "aa.b*sqrt(val)"
416 p = opt.find(",E"); q = opt.find_first_of(',',p+1);
417 dum = opt.substr(p,q-p);
418 if(dum.length()>2) sscanf(dum.c_str(),",E%lf",&mOpt.err_E);
419 if(mOpt.err_E<=0.) mOpt.err_E = 1.; //mOpt.err_e=-1.;
420}
421if(strstr(opt.c_str(),",e")) { // Erreurs imposees a "1" ou "aa.b"
422 p = opt.find(",e"); q = opt.find_first_of(',',p+1);
423 dum = opt.substr(p,q-p);
424 if(dum.length()>2) sscanf(dum.c_str(),",e%lf",&mOpt.err_e);
425 if(mOpt.err_e<=0.) mOpt.err_e = 1.; //mOpt.err_E=-1.;
426}
427if(strstr(opt.c_str(),",X")) { // Valeur du StopChi2
428 p = opt.find(",X"); q = opt.find_first_of(',',p+1);
429 dum = opt.substr(p,q-p);
430 if(dum.length()>2) sscanf(dum.c_str(),",X%lf",&mOpt.stc2);
431}
432if(strstr(opt.c_str(),",N")) { // Nombre maximum d'iterations
433 p = opt.find(",N"); q = opt.find_first_of(',',p+1);
434 dum = opt.substr(p,q-p);
435 if(dum.length()>2) sscanf(dum.c_str(),",N%d",&mOpt.nstep);
436}
437if(strstr(opt.c_str(),",l")) { // niveau de print
438 p = opt.find(",l"); q = opt.find_first_of(',',p+1);
439 dum = opt.substr(p,q-p);
440 if(dum.length()>2) sscanf(dum.c_str(),",l%d",&mOpt.lp);
441}
442if(strstr(opt.c_str(),",d")) { // niveau de debug du fit
443 p = opt.find(",d"); q = opt.find_first_of(',',p+1);
444 dum = opt.substr(p,q-p);
445 if(dum.length()>2) sscanf(dum.c_str(),",d%d",&mOpt.lpg);
446}
447if(strstr(opt.c_str(),",I")) { // intervalle de fit selon X
448 p = opt.find(",I"); q = opt.find_first_of(',',p+1);
449 dum = opt.substr(p,q-p);
450 if(dum.length()>2) sscanf(dum.c_str(),",I%d/%d",&mOpt.i1,&mOpt.i2);
451}
452if(strstr(opt.c_str(),",J")) { // intervalle de fit selon Y
453 p = opt.find(",J"); q = opt.find_first_of(',',p+1);
454 dum = opt.substr(p,q-p);
455 if(dum.length()>2) sscanf(dum.c_str(),",J%d/%d",&mOpt.j1,&mOpt.j2);
456}
457
458return;
459}
460
461/* --Methode-- */
462void PIAFitter::DecodeObject(string obj)
463// Decodage de l'objet a fitter
464{
[361]465NamedObjMgr omg;
[379]466AnyDataObj* mobj = omg.GetObj(obj);
467if(mobj == NULL)
468 {cout<<"PIAFitter::DecodeObject Error , Pas d'objet de nom "<<obj<<endl;
[361]469 return;}
470
[379]471// Reset
472ResetDPointer();
473mNVar = mNBinX = mNBinY = mNData = 0;
[361]474
[379]475mObj = mobj; mNObj = obj;
476string ctyp = typeid(*mObj).name();
477
478// 1D
479if (typeid(*mObj) == typeid(Vector)) {
480 mNVar = 1;
481 mV = (Vector*) mObj; mNBinX = mV->NElts(); mNBinY = 1;
[361]482 }
[379]483else if ( (typeid(*mObj) == typeid(HProf)) || (typeid(*mObj) == typeid(Histo)) ) {
484 mNVar = 1;
485 mH = (Histo*) mObj; mNBinX = mH->NBins(); mNBinY = 1;
[361]486 }
[379]487// 2D
488else if (typeid(*mObj) == typeid(Matrix)) {
489 mNVar = 2;
490 mM = (Matrix*) mObj; mNBinX = mM->NCol(); mNBinY =mM->NRows();
[361]491 }
[379]492else if (typeid(*mObj) == typeid(Histo2D)) {
493 mNVar = 2;
494 mH2 = (Histo2D*) mObj; mNBinX = mH2->NBinX(); mNBinY = mH2->NBinY();
[361]495 }
[379]496else if (typeid(*mObj) == typeid(GeneralFitData)) {
497 mG = (GeneralFitData*) mObj; mNBinX = mG->NData(); mNBinY = 1;
498 if( mG->NVar()==1) mNVar = 1;
499 else if(mG->NVar()==2) mNVar = 2;
500 else
501 {cout<<"PIAFitter::DecodeObject Error: \n"
502 <<" GeneralFitData ne peut avoir que 1 ou 2 variables d'abscisse: "
503 <<((GeneralFitData*) mObj)->NVar()<<endl;
504 return;}
[361]505 }
[493]506#ifdef SANS_EVOLPLANCK
[379]507else if (dynamic_cast<RzImage*>(mObj)) {
508 mNVar = 2;
509 mIm = (RzImage*) mObj; mNBinX = mIm->XSize(); mNBinY = mIm->YSize();
[361]510 }
[493]511#else
512//CMV_A_FAIRE else if (dynamic_cast<RzImage*>(mObj)) {
513//CMV_A_FAIRE mNVar = 2;
514//CMV_A_FAIRE mIm = (RzImage*) mObj; mNBinX = mIm->XSize(); mNBinY = mIm->YSize();
515//CMV_A_FAIRE }
516#endif
[379]517else {
518 cout<<"PIAFitter::DecodeObject Error , Objet n'est pas un "
[361]519 <<"Histo1D/HProf/Vector/Histo2D/Image/Matrix/GeneralFitData "<<ctyp<<endl;
[379]520 return;
[361]521 }
522
[379]523mNData = mNBinX*mNBinY;
524if(mNData<=0) {
525 cout<<"L'objet a "<<mNBinX<<","<<mNBinY<<" bins ("<<mNData<<")"<<endl;
526 return;
527}
[361]528
[379]529return;
[361]530}
[379]531
532/* --Methode-- */
533void PIAFitter::CheckOptions(void)
534// Check consistence des choix des options
535// ATTENTION: cette methode a besoin que l'objet a fitter soit deja lu
536{
537if(mOpt.lp<0) mOpt.lp = 0;
538if(mOpt.lpg<0) mOpt.lpg = 0;
539mOpt.i1 = (mOpt.i1<0||mOpt.i1>=mNBinX)? 0: mOpt.i1;
[381]540mOpt.i2 = (mOpt.i2<0||mOpt.i2>=mNBinX||mOpt.i2<=mOpt.i1)? mNBinX-1: mOpt.i2;
[379]541if(mNVar>=2) {
542 mOpt.j1 = (mOpt.j1<0||mOpt.j1>=mNBinY)? 0: mOpt.j1;
[381]543 mOpt.j2 = (mOpt.j2<0||mOpt.j2>=mNBinY||mOpt.j2<=mOpt.j1)? mNBinY-1: mOpt.j2;
[379]544} else mOpt.j2 = mOpt.j1 = 0;
545if(mOpt.polcx==2) {
546 if(mV||mM) mOpt.xc = (mOpt.i2-mOpt.i1+1)/2.;
547 else if(mH) mOpt.xc = (mH->XMin()+mH->XMax())/2.;
548 else if(mH2) mOpt.xc = (mH2->XMin()+mH2->XMax())/2.;
549 else if(mG) {double mini,maxi; mG->GetMinMax(2,mini,maxi); mOpt.xc=(mini+maxi)/2.;}
550 else if(mIm) {mOpt.xc = mIm->XOrg() * mIm->XPxSize()*(mOpt.i2-mOpt.i1+1)/2.;}
[361]551}
[379]552if(mOpt.polcy==2 && mNVar>=2) {
553 if(mM) mOpt.yc = (mOpt.j2-mOpt.j1+1)/2.;
554 if(mH2) mOpt.yc = (mH2->YMin()+mH2->YMax())/2.;
555 if(mG) {double mini,maxi; mG->GetMinMax(12,mini,maxi); mOpt.yc=(mini+maxi)/2.;}
556 if(mIm) {mOpt.yc = mIm->YOrg() * mIm->YPxSize()*(mOpt.j2-mOpt.j1+1)/2.;}
557}
558if(mOpt.err_e>0.) mOpt.err_E=-1.;
559if(mOpt.err_E>0.) mOpt.err_e =-1.;
560if(mOpt.stc2<=0.) mOpt.stc2 = 1.e-3;
561if(mOpt.nstep<2) mOpt.nstep = 100;
[361]562
[379]563return;
564}
[361]565
[379]566/* --Methode-- */
567void PIAFitter::PrintOptions(void)
568// Print des options
569{
570cout<<"Fit["<<mNBinX<<","<<mNBinY<<"] ("<<mNData<<") dim="<<mNVar<<":"
571 <<" Int=["<<mOpt.i1<<","<<mOpt.i2<<"],["<<mOpt.j1<<","<<mOpt.j2<<"]"<<endl
572 <<" Cent="<<mOpt.polcx<<","<<mOpt.polcy<<","<<mOpt.xc<<"+x"<<","<<mOpt.yc<<"+y"
573 <<" TypE="<<mOpt.err_e<<","<<mOpt.err_E
[385]574 <<" StpX2="<<mOpt.stc2<<" Nstep="<<mOpt.nstep<<endl
[390]575 <<" Init.LFit="<<mOpt.fromlastfit<<" AutoIni.Fit="<<mOpt.autoinifit
[379]576 <<" lp,lpg="<<mOpt.lp<<","<<mOpt.lpg<<endl;
577return;
578}
579
580/* --Methode-- */
581void PIAFitter::FillGData(void)
582// Fill generalfitdata pour le fit
583// ATTENTION: cette methode a besoin que les options soient decodees et checkees
584{
585if(mNData<=0) return;
586if(mGData!=NULL) {delete mGData; mGData=NULL;}
587mGData = new GeneralFitData(mNVar,mNData,0);
588
589for(int i=mOpt.i1;i<=mOpt.i2;i++) for(int j=mOpt.j1;j<=mOpt.j2;j++) {
590 double x,y=0.,f,e;
591
592 if(mV)
593 {x= (double) i; f=(*mV)(i); e=1.;}
594 else if(mH)
595 {x=mH->BinCenter(i); f=(*mH)(i); e=(mH->HasErrors())?mH->Error(i):1.;}
596 else if(mM)
597 {x=(double) i; y=(double) j; f=(*mM)(j,i); e=1.;}
598 else if(mH2)
599 {float xf,yf; mH2->BinCenter(i,j,xf,yf); x=(double)xf; y=(double)yf;
600 f=(*mH2)(i,j); e=(mH2->HasErrors())?mH2->Error(i,j):1.;}
601 else if(mIm)
602 {x=mIm->XOrg()+(i+0.5)*mIm->XPxSize(); y=mIm->YOrg()+(j+0.5)*mIm->YPxSize();
603 f=mIm->DValue(i,j); e=1.;}
604 else if(mG&&mNVar==1) {x= mG->X(i); f=mG->Val(i); e=mG->EVal(i);}
605 else if(mG&&mNVar==2) {x= mG->X(i); y=mG->Y(i); f=mG->Val(i); e=mG->EVal(i);}
[361]606 else x=y=f=e=0.;
607
608 // Gestion des erreurs a utiliser
[379]609 if(mOpt.err_e>0.) e=mOpt.err_e;
610 else if(mOpt.err_E>0.) {e=(f<-1.||f>1.)?mOpt.err_E*sqrt(fabs(f)):mOpt.err_E;}
[361]611
[379]612 // Remplissage de generalfitdata
613 if(mFName[0]=='p') {x -= mOpt.xc; if(mNVar>=2) y -= mOpt.yc;}
614 if(mNVar==1) mGData->AddData1(x,f,e);
615 else if(mNVar==2) mGData->AddData2(x,y,f,e);
[361]616}
617
[379]618if(mGData->NData()<=0) {
619 cout<<"Pas de donnees dans GeneralFitData: "<<mGData->NData()<<endl;
[361]620 return;
[379]621}
[361]622
[379]623if(mOpt.lpg>1) {
624 mGData->PrintStatus();
625 mGData->PrintData(0);
626 mGData->PrintData(mGData->NData()-1);
627}
628
629return;
630}
631
632/* --Methode-- */
633void PIAFitter::DecodeFunction(string func)
634// Fonction a fitter
635// ATTENTION: cette methode a besoin que les donnees soient lues
636{
[385]637if(func.length()<=0)
638 {cout<<"PIAFitter::DecodeFunction Donnez un nom de fonction a fitter."<<endl;
639 return;}
640if(mFunction!=NULL) {delete mFunction; mFunction=NULL;}
[392]641if(mFunc !=NULL) {delete mFunc; mFunc=NULL;}
642mNPar=0; mFName = "";
[379]643
[385]644if(func == mNameFitFunc) { // Fonction utilisateur
645 mFunc = new GeneralFunc(mUFNVar,mUFNPar,mFitFunc,mFitFuncDer);
646 mNPar = mUFNPar;
[386]647 cout<<"Fonction utilisateur nvar="<<mUFNVar<<", npar="<<mNPar<<endl;
[385]648} else if(func[0]=='p' && mNVar==1) { //polynome
649 int degre=0; if(func.length()>1) sscanf(func.c_str()+1,"%d",&degre);
[361]650 cout<<"Fit polynome 1D de degre "<<degre<<endl;
[379]651 Polyn1D* myf = new Polyn1D(degre,mOpt.xc);
[385]652 mFunction = myf; mNPar = mFunction->NPar(); mFName = func;
[361]653
[379]654} else if(func[0]=='e' && mNVar==1) { //exponentielle
[385]655 int degre=-1; if(func.length()>1) sscanf(func.c_str()+1,"%d",&degre);
[361]656 cout<<"Fit d'exponentielle+polynome 1D de degre "<<degre<<endl;
657 Exp1DPol* myf;
[379]658 if(degre>=0) myf = new Exp1DPol((unsigned int)degre,mOpt.xc);
659 else myf = new Exp1DPol(mOpt.xc);
[385]660 mFunction = myf; mNPar = mFunction->NPar(); mFName = func;
[361]661
[379]662} else if(func[0]=='g' && mNVar==1) { //gaussienne en hauteur
[385]663 int degre=-1; if(func.length()>1) sscanf(func.c_str()+1,"%d",&degre);
[361]664 cout<<"Fit de Gaussienne_en_hauteur+polynome 1D de degre "<<degre<<endl;
665 Gauss1DPol* myf;
[379]666 if(degre>=0) myf = new Gauss1DPol((unsigned int)degre,((mOpt.polcx)?true:false));
667 else { bool bfg = (mOpt.polcx)?true:false; myf = new Gauss1DPol(bfg); }
[385]668 mFunction = myf; mNPar = mFunction->NPar(); mFName = func;
[361]669
[379]670} else if(func[0]=='G' && mNVar==1) { //gaussienne en volume
[385]671 int degre=-1; if(func.length()>1) sscanf(func.c_str()+1,"%d",&degre);
[361]672 cout<<"Fit de Gaussienne_en_volume+polynome 1D de degre "<<degre<<endl;
673 GaussN1DPol* myf;
[379]674 if(degre>=0) myf = new GaussN1DPol((unsigned int)degre,((mOpt.polcx)?true:false));
675 else { bool bfg = (mOpt.polcx)?true:false; myf = new GaussN1DPol(bfg); }
[385]676 mFunction = myf; mNPar = mFunction->NPar(); mFName = func;
[361]677
[379]678} else if(func[0]=='p' && mNVar==2) { //polynome 2D
[385]679 int degre=0; if(func.length()>1) sscanf(func.c_str()+1,"%d",&degre);
[361]680 cout<<"Fit polynome 2D de degre "<<degre<<endl;
[379]681 Polyn2D* myf = new Polyn2D(degre,mOpt.xc,mOpt.yc);
[385]682 mFunction = myf; mNPar = mFunction->NPar(); mFName = func;
[361]683
[379]684} else if(func[0]=='G' && mNVar==2) { //gaussienne+fond en volume
[385]685 int integ=0; if(func.length()>1) if(func[1]=='i') integ=1;
[361]686 cout<<"Fit de Gaussienne+Fond 2D integ="<<integ<<endl;
[385]687 if(integ) {GauRhInt2D* myf = new GauRhInt2D; mFunction = myf;}
[392]688 else {GauRho2D* myf = new GauRho2D; mFunction = myf;}
[385]689 mNPar = mFunction->NPar(); mFName = func;
[361]690
[379]691} else if(func[0]=='d' && mNVar==2) { //DL gaussienne+fond en volume
[385]692 int integ=0; if(func.length()>1) if(func[1]=='i') integ=1;
[361]693 cout<<"Fit de DL de Gaussienne+Fond 2D integ="<<integ<<endl;
[385]694 if(integ) {GdlRhInt2D* myf = new GdlRhInt2D; mFunction = myf;}
[392]695 else {GdlRho2D* myf = new GdlRho2D; mFunction = myf;}
[385]696 mNPar = mFunction->NPar(); mFName = func;
[361]697
[379]698} else if(func[0]=='D' && mNVar==2) { //DL gaussienne+fond avec coeff variable p6 en volume
[385]699 int integ=0; if(func.length()>1) if(func[1]=='i') integ=1;
[361]700 cout<<"Fit de DL de Gaussienne+Fond avec coeff variable (p6) 2D integ="<<integ<<endl;
[385]701 if(integ) {Gdl1RhInt2D* myf = new Gdl1RhInt2D; mFunction = myf;}
[392]702 else {Gdl1Rho2D* myf = new Gdl1Rho2D; mFunction = myf;}
[385]703 mNPar = mFunction->NPar(); mFName = func;
[361]704
[379]705} else if(func[0]=='M' && mNVar==2) { //Moffat+fond (volume)
[385]706 int integ=0; if(func.length()>1) if(func[1]=='i') integ=1;
[361]707 cout<<"Fit de Moffat+Fond (expos=p6) 2D integ="<<integ<<endl;
[385]708 if(integ) {MofRhInt2D* myf = new MofRhInt2D; mFunction = myf;}
[392]709 else {MofRho2D* myf = new MofRho2D; mFunction = myf;}
[385]710 mNPar = mFunction->NPar(); mFName = func;
[361]711
712} else {
[379]713 cout<<"Fonction "<<func<<" inconnue pour la dim "<<mNVar<<endl;
[361]714 return;
715}
716
[379]717return;
[361]718}
719
[379]720/* --Methode-- */
721void PIAFitter::ReSetParam(void)
722// Pour (re)dimensionner les tableaux des parametres
723// ATTENTION: cette methode a besoin que la fonction soit connue
724{
725if(mNPar==0) return;
726mPar.Realloc(mNPar); mPar=0.;
727mStep.Realloc(mNPar); mStep=1.;
728mMin.Realloc(mNPar); mMin=1.;
729mMax.Realloc(mNPar); mMax=-1.;
730mFix.Realloc(mNPar); mFix=0.;
731return;
[361]732}
733
[379]734/* --Methode-- */
[390]735void PIAFitter::AutoIniFit(void)
736// Initialisation automatique des parametres du fit
737{
[392]738if(mFunc) { // Fonction utilisateur
[390]739 cout<<"PIAFitter::AutoIniFit : Fonction utilisateur, pas d'init auto"<<endl;
740} else if(mFName[0]=='p' && mNVar==1) { //poly 1D
741 cout<<"PIAFitter::AutoIniFit : Auto Init type polynome 1D"<<endl;
[392]742 IniFitP1ou2D();
[390]743
744} else if(mFName[0]=='g' && mNVar==1) { //gauss en haut 1D
745 cout<<"PIAFitter::AutoIniFit : Auto Init type gauss_haut+pol 1D"<<endl;
[392]746 IniFitGhP1D();
[390]747
748} else if(mFName[0]=='G' && mNVar==1) { //gauss en vol 1D
749 cout<<"PIAFitter::AutoIniFit : Auto Init type gauss_vol+pol 1D"<<endl;
[392]750 IniFitGvP1D();
[390]751
752} else if(mFName[0]=='p' && mNVar==2) { //poly 2D
753 cout<<"PIAFitter::AutoIniFit : Auto Init type polynome 2D"<<endl;
[392]754 IniFitP1ou2D();
[390]755
756} else if((mFName[0]=='G'||mFName[0]=='d') && mNVar==2) { //gauss+fond en vol
757 cout<<"PIAFitter::AutoIniFit : Auto Init type gauss_vol+fond 2D"<<endl;
[392]758 IniFitGv2D();
[390]759
760} else if(mFName[0]=='D' && mNVar==2) { //DLgauss+fond en vol 2D
761 cout<<"PIAFitter::AutoIniFit : Auto Init type DLgauss_vol+fond 2D"<<endl;
[392]762 IniFitGv2D();
[390]763 Vector dum(mNPar);
764 dum = mPar; mPar(6)=0.5; mPar(7)=dum(6);
765 dum = mStep; mStep(6)=0.05; mStep(7)=dum(6);
766 dum = mMin; mMin(6)=0.; mMin(7)=dum(6);
767 dum = mMax; mMax(6)=10.; mMax(7)=dum(6);
768 dum = mFix; mFix(6)=1.1; mFix(7)=dum(6);
769
770} else if(mFName[0]=='M' && mNVar==2) { //Moffat+fond en vol 2D
771 cout<<"PIAFitter::AutoIniFit : Auto Init type moffat_vol+fond 2D"<<endl;
[392]772 IniFitGv2D();
[390]773 Vector dum(mNPar);
774 dum = mPar; mPar(6)=2.5; mPar(7)=dum(6);
775 dum = mStep; mStep(6)=0.05; mStep(7)=dum(6);
776 dum = mMin; mMin(6)=1.2; mMin(7)=dum(6);
777 dum = mMax; mMax(6)=10.; mMax(7)=dum(6);
778 dum = mFix; mFix(6)=1.1; mFix(7)=dum(6);
779
780} else {
781 cout<<"PIAFitter::AutoIniFit : Fonction "<<mFName
782 <<", \n pas d'init auto en dim "<<mNVar<<endl;
783 return;
784}
785
786return;
787}
788
789/* --Methode-- */
[379]790void PIAFitter::InitParFromLastFit(void)
791// Initialisation du fit a partir des resultats du fit precedent
792// (Les arguments de la ligne de commande sont prioritaires)
793// ATTENTION: cette methode a besoin que la fonction soit connue
794{
795if(mNPar==0) return;
796int n = mParSave.NElts();
797if(n==0) return;
798
799for(int i=0;i<mNPar;i++) {
800 if(i>=n) break;
801 mPar(i) = mParSave(i);
802 mStep(i) = mStepSave(i);
803 mMin(i) = mMinSave(i);
804 mMax(i) = mMaxSave(i);
805 mFix(i) = mFixSave(i);
806}
807
[361]808return;
809}
810
[379]811/* --Methode-- */
812void PIAFitter::DecodeParam(string par,string step,string min,string max,string fix)
813// Decodage des parametres
814// ATTENTION: cette methode a besoin que la fonction soit connue
815// et que les options soient decodees
[361]816{
[379]817if(mNPar==0) return;
818ReSetParam();
[390]819if(mOpt.autoinifit) AutoIniFit();
[379]820if(mOpt.fromlastfit) InitParFromLastFit();
821Vector* v=NULL; string* s=NULL;
822for(int j=0;j<5;j++) {
823 if(j==0) {v=&mPar; s=&par;} else if(j==1) {v=&mStep; s=&step;}
824 else if(j==2) {v=&mMin; s=&min;} else if(j==3) {v=&mMax; s=&max;}
825 else if(j==4) {v=&mFix; s=&fix;}
826 if(s->length()>0) *s += ",";
827 for(int i=0;i<mNPar;i++) {
828 if(s->length()<=0) break;
829 if((*s)[0]!='!') sscanf(s->c_str(),"%lf",&(*v)(i));
830 size_t p = s->find_first_of(',') + 1;
831 if(p>=s->length()) *s = ""; else *s = s->substr(p);
[361]832 }
833}
[379]834return;
835}
[361]836
[379]837/* --Methode-- */
838int PIAFitter::DoFit(void)
839// Fit avec generalfit
840{
841if(mNPar==0) return -1000;
[385]842if((mFunction==NULL && mFunc==NULL) || mGData==NULL) {
843 cout<<"PIAFitter::DoFit error: mFunction,mFunc="<<mFunction<<","<<mFunc
844 <<" mGData="<<mGData<<endl;
[379]845 return -1001;
846}
847// classe GeneraFit
848if(mFit != NULL) {delete mFit; mFit= NULL;}
[392]849if(mFunction) mFit = new GeneralFit(mFunction);
850 else if(mFunc) mFit = new GeneralFit(mFunc);
851if(!mFit) return(-1002);
852
[379]853// Options et Set de GeneralFit
854mFit->SetDebug(mOpt.lpg);
855mFit->SetData(mGData);
856mFit->SetStopChi2(mOpt.stc2);
857mFit->SetMaxStep(mOpt.nstep);
[385]858{for(int i=0;i<mNPar;i++) {
[379]859 char str[10];
860 sprintf(str,"P%d",i);
861 mFit->SetParam(i,str,mPar(i),mStep(i),mMin(i),mMax(i));
862 if((int)(mFix(i)+0.001)>=1.) mFit->SetFix(i);
863}}
864if(mOpt.lp>1) mFit->PrintFit();
[361]865
[379]866// Fit
867mParSave = mPar;
868mStepSave = mStep; mMinSave = mMin; mMaxSave = mMax; mFixSave = mFix;
869mOptSave = mOpt;
870double c2r = -1.;
871int rcfit = mFit->Fit();
872if(mOpt.lp>0) mFit->PrintFit();
873if(rcfit>0) {
874 c2r = mFit->GetChi2Red();
875 cout<<"C2r_Reduit = "<<c2r<<" nstep="<<mFit->GetNStep()<<" rc="<<rcfit<<endl;
[385]876 for(int i=0;i<mNPar;i++) mParSave(i)=mFit->GetParm(i);
[379]877} else {
878 cout<<"echec Fit, rc = "<<rcfit<<" nstep="<<mFit->GetNStep()<<endl;
879 mFit->PrintFitErr(rcfit);
880}
[361]881
[379]882return rcfit;
[361]883}
[379]884
885/* --Methode-- */
886void PIAFitter::FitFunRes(void)
887// Mise a disposition des resultats (fonction fitee et residus)
888{
[385]889if(mFit && (mOpt.okres||mOpt.okfun)) {
[379]890 NamedObjMgr omg;
[384]891 string nomres = mNObj + "_res";
892 string nomfun = mNObj + "_fun";
[493]893//####################
894#ifdef SANS_EVOLPLANCK
895//####################
[379]896 if(mV) {
897 if(mOpt.okres) {Vector* ob = mV->FitResidus(*mFit); if(ob) omg.AddObj(ob,nomres);}
898 if(mOpt.okfun) {Vector* ob = mV->FitFunction(*mFit); if(ob) omg.AddObj(ob,nomfun);}
899 } else if(mH) {
900 if(mOpt.okres) {Histo* ob = mH->FitResidus(*mFit); if(ob) omg.AddObj(ob,nomres);}
901 if(mOpt.okfun) {Histo* ob = mH->FitFunction(*mFit); if(ob) omg.AddObj(ob,nomfun);}
902 } else if(mM) {
903 if(mOpt.okres) {Matrix* ob = mM->FitResidus(*mFit); if(ob) omg.AddObj(ob,nomres);}
904 if(mOpt.okfun) {Matrix* ob = mM->FitFunction(*mFit); if(ob) omg.AddObj(ob,nomfun);}
905 } else if(mH2) {
[392]906 if(mOpt.okres)
907 {Histo2D* ob = mH2->FitResidus(*mFit); if(ob) omg.AddObj(ob,nomres);}
908 if(mOpt.okfun)
909 {Histo2D* ob = mH2->FitFunction(*mFit); if(ob) omg.AddObj(ob,nomfun);}
[379]910 } else if(mIm) {
[384]911 if(mOpt.okres) {
912 RzImage* ob = mIm->FitResidus(*mFit);
913 if(ob) {ImageR4* rob = new ImageR4(*ob); omg.AddObj(rob,nomres); delete ob;}
914 }
915 if(mOpt.okfun) {
916 RzImage* ob = mIm->FitFunction(*mFit);
917 if(ob) {ImageR4* rob = new ImageR4(*ob); omg.AddObj(rob,nomfun); delete ob;}
918 }
[379]919 } else if(mG) {
[392]920 if(mOpt.okres)
921 {GeneralFitData* ob = mG->FitResidus(*mFit); if(ob) omg.AddObj(ob,nomres);}
922 if(mOpt.okfun)
923 {GeneralFitData* ob = mG->FitFunction(*mFit); if(ob) omg.AddObj(ob,nomfun);}
[361]924 }
[493]925//####################
926#else
927//####################
928 if(mV) {
929 if(mOpt.okres) {Vector* ob = new Vector(mV->FitResidus(*mFit)); if(ob) omg.AddObj(ob,nomres);}
930 if(mOpt.okfun) {Vector* ob = new Vector(mV->FitFunction(*mFit)); if(ob) omg.AddObj(ob,nomfun);}
931 } else if(mH) {
932 if(mOpt.okres) {Histo* ob = new Histo(mH->FitResidus(*mFit)); if(ob) omg.AddObj(ob,nomres);}
933 if(mOpt.okfun) {Histo* ob = new Histo(mH->FitFunction(*mFit)); if(ob) omg.AddObj(ob,nomfun);}
934 } else if(mM) {
935 if(mOpt.okres) {Matrix* ob = new Matrix(mM->FitResidus(*mFit)); if(ob) omg.AddObj(ob,nomres);}
936 if(mOpt.okfun) {Matrix* ob = new Matrix(mM->FitFunction(*mFit)); if(ob) omg.AddObj(ob,nomfun);}
937 } else if(mH2) {
938 if(mOpt.okres)
939 {Histo2D* ob = new Histo2D(mH2->FitResidus(*mFit)); if(ob) omg.AddObj(ob,nomres);}
940 if(mOpt.okfun)
941 {Histo2D* ob = new Histo2D(mH2->FitFunction(*mFit)); if(ob) omg.AddObj(ob,nomfun);}
942//CMV_A_FAIRE } else if(mIm) {
943//CMV_A_FAIRE if(mOpt.okres) {
944//CMV_A_FAIRE RzImage* ob = mIm->FitResidus(*mFit);
945//CMV_A_FAIRE if(ob) {ImageR4* rob = new ImageR4(*ob); omg.AddObj(rob,nomres); delete ob;}
946//CMV _A_FAIRE }
947//CMV_A_FAIRE if(mOpt.okfun) {
948//CMV_A_FAIRE RzImage* ob = mIm->FitFunction(*mFit);
949//CMV_A_FAIRE if(ob) {ImageR4* rob = new ImageR4(*ob); omg.AddObj(rob,nomfun); delete ob;}
950//CMV_A_FAIRE }
951 } else if(mG) {
952 if(mOpt.okres)
953 {GeneralFitData* ob = new GeneralFitData(mG->FitResidus(*mFit)); if(ob) omg.AddObj(ob,nomres);}
954 if(mOpt.okfun)
955 {GeneralFitData* ob = new GeneralFitData(mG->FitFunction(*mFit)); if(ob) omg.AddObj(ob,nomfun);}
956 }
957//####################
958#endif
959//####################
[361]960}
[379]961return;
[361]962}
[379]963
964/* --Methode-- */
965void PIAFitter::LinFit(void)
966// Fit lineaire de polynomes
967{
968if(mFName.length()<=0)
969 {cout<<"PIAFitter::LinFit Donnez un nom de fonction a fitter."<<endl;
970 return;}
971if(!mGData)
972 {cout<<"PIAFitter::LinFit pas de Data."<<endl; return;}
973if(mFName[0]!='p')
974 {cout<<"PIAFitter::LinFit fonction non prevue "<<mFName<<endl;
975 return;}
976int degre = 0;
977if(mFName.length()>1) sscanf(mFName.c_str()+1,"%d",&degre);
978
979if(mNVar==1) { // Fit lineaire de polynome 1D
980 cout<<"Fit (lineaire) 1D polynome de degre "<<degre<<endl;
981 Poly p1(0);
982 double c2rl = mGData->PolFit(0,p1,degre);
983 cout<<"C2r_lineaire = "<<c2rl<<endl;
984 if(mOpt.lp>0) cout<<p1<<endl;
985} else if(mNVar==2) { // Fit lineaire de polynome 2D
986 cout<<"Fit (lineaire) polynome 2D de degre "<<degre<<endl;
987 Poly2 p2(0);
988 double c2rl = mGData->PolFit(0,1,p2,degre);
989 cout<<"C2r_lineaire = "<<c2rl<<endl;
990 if(mOpt.lp>0) cout<<p2<<endl;
[361]991}
[379]992
993return;
[361]994}
[379]995
996/* --Methode-- */
[385]997void PIAFitter::LinkFitFunc(string const& fname, string const& funcname
998 ,int nvar,int npar)
999// Compilation et link dynamique de la fonction a fitter
[384]1000{
[385]1001if(nvar<1 || npar<1) return;
1002if(mDlUFunc != NULL) {delete mDlUFunc; mDlUFunc=NULL;}
1003mFitFunc = NULL; mFitFuncDer = NULL;
1004mUFNVar= mUFNPar = 0; mNameFitFunc = "";
[384]1005
1006// On compile le fichier C, on fabrique un .so et on l'ouvre
1007mDlUFunc = PDynLinkMgr::BuildFromCFile(fname);
[385]1008if(mDlUFunc == NULL)
1009 {cerr<<"PIAFitter::LinkFitFunc() Erreur building .so "<<endl; return;}
[384]1010
1011// On recupere la fonction de fit
1012mFitFunc = (DlUserFitFunc) mDlUFunc->GetFunction(funcname);
[385]1013if(mFitFunc == NULL) // Probleme, on n'a pas trouve etc ...
1014 {cerr<<"PIAFitter::LinkFitFunc() Erreur linking function"<<funcname<<endl;
1015 delete mDlUFunc; mDlUFunc = NULL; return;}
[384]1016
[385]1017// On recupere la fonction des derivees
[384]1018string fnameder = funcname+"_der";
1019mFitFuncDer = (DlUserFitFuncDer) mDlUFunc->GetFunction(fnameder);
[385]1020if(mFitFuncDer == NULL)
1021 {cout<<"PIAFitter::LinkFitFunc() No derive for"<<fnameder<<endl;}
[384]1022
[385]1023// Variables et pointeurs definissant le link a la fonction utilisateur
[384]1024mNameFitFunc = funcname;
1025mUFNVar = nvar;
1026mUFNPar = npar;
[385]1027cout<<"PIAFitter::LinkFitFunc() UserFitFunc="<<mNameFitFunc
1028 <<" NVar="<<mUFNVar<<" NPar="<<mUFNPar<<endl;
[384]1029
[385]1030return;
1031}
[384]1032
1033/* --Methode-- */
[385]1034void PIAFitter::PrepareFitFuncCFile(string const& fname, string const& funcname
1035 , int nvar, int npar)
1036// Pour ecrire un squelette des fichier C avec pour une fonction a fitter
[384]1037{
[385]1038if(fname.length()<=0 || funcname.length()<=0 || nvar<=0 || npar <=0) {
1039 cerr<<"PIAFitter::PrepareFitFuncCFile: error "
1040 <<" nvar="<<nvar<<" npar="<<npar<<endl
1041 <<" file_name="<<fname<<" func_name="<<funcname<<endl
1042 <<" ---- No File Created ! ----"<<endl;
1043 return;
[384]1044}
1045
[385]1046FILE* fip = fopen(fname.c_str(),"w");
1047if(fip==NULL)
1048 {cerr<<"PIAFitter::PrepareFitFuncCFile: error opening file for writing\n"
1049 <<fname<<endl; return;}
1050
1051fprintf(fip,"#include <stdio.h>\n#include <stdlib.h>\n#include <math.h>\n\n");
1052fprintf(fip,"/* - many functions can be written in this file: */\n");
1053fprintf(fip,"/* here is an exemple for creation of function %s */\n\n"
1054 ,funcname.c_str());
1055fprintf(fip,"double %s(double const* x,double const* p)\n",funcname.c_str());
1056fprintf(fip,"/* Function : F(x[0-%d];p[0-%d]) */\n",nvar-1,npar-1);
1057fprintf(fip,"/* variables x1=x[0],x2=x[1],...,x%d=x[%d] */\n",nvar,nvar-1);
1058fprintf(fip,"/* parameters p1=p[0],p2=p[1],...,p%d=p[%d] */\n",nvar,nvar-1);
1059fprintf(fip,"{\n val = F(x[];p[]);\n return(val);\n}\n\n");
1060string fder = funcname + "_der";
1061fprintf(fip,"/* That function is optional */\n");
1062fprintf(fip,"/*\n");
1063fprintf(fip,"double %s(double const* x,double const* p,double* dp)\n",fder.c_str());
1064fprintf(fip,"{\n");
1065for(int i=0;i<npar;i++) fprintf(fip," dp[%d] = dF(x[];p[])/dp[%d]\n",i,i);
1066fprintf(fip," val = F(x[];p[]);\n return(val);\n}\n\n");
1067fprintf(fip,"*/\n");
1068
1069fclose(fip);
1070}
1071
[384]1072/* --Methode-- */
[379]1073int PIAFitter::Execute(string& kw, vector<string>& tokens)
1074// Execution des commandes de fit : "fit" et "fitw"
1075{
[385]1076////// Les commandes pour creer les fonctions de fit utilisateur
1077if(kw == "crfitfil") {
1078 if (tokens.size() < 4)
1079 {cout<<"Usage:crfitfil filename funcname nvar npar"<<endl;return(-1);}
1080 int nvar = atoi(tokens[2].c_str()), npar = atoi(tokens[3].c_str());
1081 PrepareFitFuncCFile(tokens[0],tokens[1],nvar,npar);
1082 return(0);
1083} else if (kw == "crfitfun") {
1084 if (tokens.size() < 4)
1085 {cout<<"Usage:crfitfun filename funcname nvar npar"<<endl;return(-1);}
1086 int nvar = atoi(tokens[2].c_str()), npar = atoi(tokens[3].c_str());
1087 LinkFitFunc(tokens[0],tokens[1],nvar,npar);
1088 return(0);
1089}
1090
1091////// Les commandes pour fitter
[379]1092if(tokens.size() < 2) {
1093 cout
1094 <<"Usage:fit nomobj func [Options]\n"
1095 <<" ou fitw nomobj func [Options]\n"
1096 <<" avec Options: [p:p1,...,pn s:s1,...,sn m:m1,...,mn \n"
1097 <<" M:M1,...,Mn f:f1,...,fn o:.,.,. o:.,.,.]\n"
1098 <<" ou fitlin nomobj func [o:.,.,. o:.,.,.]\n";
1099 return(-1);
[361]1100}
[392]1101
[379]1102// decodage des arguments de la commande
1103string mSp,mSs,mSm,mSM,mSf,mSo;
1104mSp=""; mSs=""; mSm=""; mSM=""; mSf=""; mSo="";
1105if(tokens.size()>2)
1106 for(int ip=2;ip<(int)tokens.size();ip++) {
1107 if(tokens[ip].length()<=2) continue;
1108 const char *c = tokens[ip].c_str();
1109 if(c[1]!=':') continue;
1110 if(c[0]=='p') mSp=c+2;
1111 else if(c[0]=='s') mSs=c+2;
1112 else if(c[0]=='m') mSm=c+2;
1113 else if(c[0]=='M') mSM=c+2;
1114 else if(c[0]=='f') mSf=c+2;
1115 else if(c[0]=='o') {mSo += ","; mSo += c+2;}
1116 }
[392]1117
[379]1118// Execution des commandes
1119DecodeOptions(mSo);
1120DecodeObject(tokens[0]);
1121CheckOptions();
1122if(mOpt.lp>0) PrintOptions();
1123FillGData();
1124DecodeFunction(tokens[1]);
1125DecodeParam(mSp,mSs,mSm,mSM,mSf);
1126if (kw == "fit") {
1127 int rc = DoFit();
1128 if(rc>=0) FitFunRes();
1129} else if (kw == "fitlin") {
1130 LinFit();
1131} else if (kw == "fitw") {
1132 if(FWindFit!=NULL) {delete FWindFit; FWindFit=NULL;}
1133 FWindFit = new PIAFitterWind(mApp,this);
1134 FWindFit->Show();
[361]1135}
[379]1136
1137return(0);
[361]1138}
1139
[392]1140/* --Methode-- */
1141void PIAFitter::IniFitP1ou2D(void)
1142{
1143if(mNPar==0) return;
1144mPar=0.; mStep=1.; mMin=1.; mMax=-1.; mFix=0.;
1145if(!mGData) return;
1146double mean,sigma;
1147int rc = mGData->GetMeanSigma(0,mean,sigma);
1148if(rc<=0) return;
1149mPar(0) = mean;
1150return;
1151}
1152
1153void PIAFitter::IniFitGhP1D(void)
1154{
1155if(mNPar==0) return;
1156mPar=0.; mStep=1.; mMin=1.; mMax=-1.; mFix=0.;
1157if(mNVar!=1 || mNPar<3) return;
1158double h,v,x0,y0,sx,sy,f;
1159int rc = IniFitGaus(*mGData,h,v,x0,y0,sx,sy,f);
1160if(rc!=0) return;
1161mPar(0)=h; mStep(0)=h/1000.;
1162mPar(1)=x0; mStep(1)=sx/10.;
1163mPar(2)=sx; mStep(2)=sx/10.; mMin(2)=0.; mMax(2)=20.*sx;
1164if(mNPar>3) {mPar(3)=f; mStep(3)=f/1000.;} else mPar(0)=h+f;
1165}
1166
1167void PIAFitter::IniFitGvP1D(void)
1168{
1169if(mNPar==0) return;
1170mPar=0.; mStep=1.; mMin=1.; mMax=-1.; mFix=0.;
1171if(mNVar!=1 || mNPar<3) return;
1172double h,v,x0,y0,sx,sy,f;
1173int rc = IniFitGaus(*mGData,h,v,x0,y0,sx,sy,f);
1174if(rc!=0) return;
1175mPar(0)=v; mStep(0)=v/1000.;
1176mPar(1)=x0; mStep(1)=sx/10.;
1177mPar(2)=sx; mStep(2)=sx/10.; mMin(2)=0.; mMax(2)=20.*sx;
1178if(mNPar>3) {mPar(3)=f; mStep(3)=f/1000.;} else mPar(0)=v+mNData*f;
1179}
1180
1181void PIAFitter::IniFitGv2D(void)
1182{
1183if(mNPar==0) return;
1184mPar=0.; mStep=1.; mMin=1.; mMax=-1.; mFix=0.;
1185if(mNVar!=2 || mNPar<6) return;
1186double h,v,x0,y0,sx,sy,f;
1187int rc = IniFitGaus(*mGData,h,v,x0,y0,sx,sy,f);
1188if(rc!=0) return;
1189mPar(0)=v; mStep(0)=v/1000.;
1190mPar(1)=x0; mStep(1)=sx/10.;
1191mPar(2)=y0; mStep(2)=sy/10.;
1192mPar(3)=sx; mStep(3)=sx/10.; mMin(3)=0.; mMax(3)=20.*sx;
1193mPar(4)=sy; mStep(4)=sy/10.; mMin(4)=0.; mMax(4)=20.*sy;
1194mPar(5)=0.; mStep(5)=0.005; mMin(5)=-1.; mMax(5)=1.;
1195if(mNPar>6) {mPar(6)=f; mStep(6)=f/1000.;} else mPar(0)=v+mNData*f;
1196}
1197
1198int PIAFitter::IniFitGaus(GeneralFitData& g,double& h,double& v
1199 ,double& x0,double& y0,double& sx,double& sy,double& f)
1200// Pour initialiser une gaussienne 1D ou 2D
1201// return code =0 si OK
1202{
1203h=v=x0=y0=sx=sy=f=0.;
1204if(!&g) return(-1);
1205int nvar = g.NVar();
1206int ndata = g.NData();
1207if(nvar<1 || 2<nvar || ndata<=0) return(-2);
1208
1209// X et Y ordonnees croissant
1210double *X=NULL, *Y=NULL;
1211X = new double[ndata];
1212if(nvar==2) Y = new double[ndata];
1213{for(int i=0;i<ndata;i++) {X[i] = g.X(i); if(Y) Y[i] = g.Y(i);}}
1214qsort(X,(size_t) ndata,sizeof(double),qSort_Dble);
1215if(Y) qsort(Y,(size_t) ndata,sizeof(double),qSort_Dble);
1216
1217// determination des zones externes=[0,1/3] et [2/3,1]
1218// et des zones internes=[1/3,2/3]
1219int k1=ndata/3,k2=2*ndata/3;
1220double xmin,xmax, ymin=-1.,ymax=1.;
1221xmin = X[k1]; xmax = X[k2];
1222if(Y) {ymin = Y[k1]; ymax = Y[k2];}
1223
1224// recherche des parametres
1225double nb=0, nc=0;
1226{for(int i=0;i<ndata;i++) {
1227 double val,x,y=0.;
1228 x = g.X(i); if(Y) y = g.Y(i); val = g.Val(i);
1229 if(xmin<x && x<xmax && ymin<y && y<ymax) { // c'est le centre
1230 if(nc==0 || val>h) {h=val; x0=x; y0=y;}
1231 v += val; nc++;
1232 } else {f += val; nb++;} // c'est le bord
1233
1234}}
1235if(nb==0 || nc==0) {delete [] X; if(Y) delete [] Y; return(-3);}
1236
1237// Calcul des parametres
1238f /= (double) nb; h -= f; v -= f*nc;
1239sx = (xmax-xmin)/2.; if(Y) sy = (ymax-ymin)/2.;
1240
1241delete [] X; if(Y) delete [] Y;
1242return(0);
1243}
1244
1245
[379]1246////////////////////////////////////////////////////////////////////
1247////---------- Classe de fenetre de fit interactive ------------////
1248////////////////////////////////////////////////////////////////////
[361]1249
1250/* --Methode-- */
1251PIAFitterWind::PIAFitterWind(PIStdImgApp* par, PIAFitter* fiter)
[385]1252: PIWindow((PIMsgHandler*)par, "PIAFitter", PIWK_normal, 240, 240, 150, 150)
1253, mDap(par), mFitter(fiter), ReFillGData(false)
1254, saveI1(0), saveI2(0), saveJ1(0), saveJ2(0), ErrValue(1.)
[361]1255{
[495]1256#ifdef SANS_EVOLPLANCK
[386]1257COMPATIBLE_ASSERT(mFitter);
[495]1258#else
1259ASSERT(mFitter);
1260#endif
1261
[379]1262int npar = mFitter->mNPar;
[361]1263
[379]1264// Alloc de la taille
[390]1265nlab = 13+npar; lab = new PILabel*[nlab]; {for(int i=0;i<nlab;i++) lab[i]=NULL;}
[385]1266ntxt = 10+4*npar; txt = new PIText*[ntxt]; {for(int i=0;i<ntxt;i++) txt[i]=NULL;}
[390]1267nbut = 6; but = new PIButton*[nbut]; {for(int i=0;i<nbut;i++) but[i]=NULL;}
[385]1268nckb = 2+npar; ckb = new PICheckBox*[nckb]; {for(int i=0;i<nckb;i++) ckb[i]=NULL;}
1269npom = 3; pom = new PIOptMenu*[npom]; {for(int i=0;i<npom;i++) pom[i]=NULL;}
[379]1270
1271// On definit la taille a partir de la taille par defaut des composantes
1272// bsx,bsy = environ 6 lettres
1273int bsx,bsy;
[361]1274par->PrefCompSz(bsx, bsy);
[379]1275int spx = (bsx>=10) ? bsx/10 : 1; // intervalle entre lettres X
1276int spy = (bsy>=6) ? bsy/6 : 1; // intervalle entre lettres Y
[361]1277
[379]1278int wszx = (5*bsx+5*spx)+spx;
[386]1279int wszy = 8*(bsy+spy)+npar*(bsy+spy)+2*spy;
[361]1280SetSize(wszx, wszy);
[379]1281int cpx,cpy;
[361]1282
[381]1283// new ligne
[385]1284cpx = spx; cpy = spy;
[390]1285string dum = "Object: "+mFitter->mNObj+" , Fun: ";
[392]1286if(mFitter->mFunc) dum+=mFitter->mNameFitFunc; else dum+=mFitter->mFName;
[386]1287lab[0] = new PILabel(this,dum.c_str(),4*bsx,bsy,cpx,cpy);
[385]1288lab[0]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1289
1290// new ligne
1291cpx = spx; cpy += bsy+spy;
[379]1292but[0] = new PIButton(this,"Fr.Last",111,bsx,bsy,cpx,cpy);
1293but[0]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1294cpx += bsx+spx;
1295but[1] = new PIButton(this,"Default",222,bsx,bsy,cpx,cpy);
1296but[1]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1297cpx += bsx+spx;
[379]1298ckb[0] = new PICheckBox(this,"Gen Fun",1001,bsx,bsy,cpx,cpy);
1299ckb[0]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1300cpx += bsx+2*spx;
1301ckb[1] = new PICheckBox(this,"Res",1002,bsx,bsy,cpx,cpy);
1302ckb[1]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[361]1303
[381]1304// new ligne
[385]1305cpx = spx; cpy += bsy+spy;
1306pom[0] = new PIOptMenu(this,"Err def",1.5*bsx,bsy,cpx,cpy);
1307 pom[0]->AppendItem("Err def",2101);
1308 pom[0]->AppendItem("Err cste",2102);
1309 pom[0]->AppendItem("Err sqrt",2103);
[386]1310 pom[0]->SetValue(2101);
[381]1311pom[0]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1312cpx += int(1.5*bsx+spx);
[379]1313txt[0] = new PIText(this,"",bsx,bsy,cpx,cpy);
1314txt[0]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1315cpx += bsx+spx;
1316lab[1] = new PILabel(this,"Prt,dbg",bsx,bsy,cpx,cpy);
1317lab[1]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1318cpx += bsx+spx;
[379]1319txt[1] = new PIText(this,"",bsx,bsy,cpx,cpy);
1320txt[1]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[361]1321
[381]1322// new ligne
[385]1323cpx = spx; cpy += bsy+spy;
1324lab[2] = new PILabel(this,"Stop Xi2",1.25*bsx,bsy,cpx,cpy);
[381]1325lab[2]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1326cpx += int(1.25*bsx+spx);
[379]1327txt[2] = new PIText(this,"",bsx,bsy,cpx,cpy);
1328txt[2]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1329cpx += bsx+spx;
1330lab[3] = new PILabel(this,"Iter",bsx, bsy, cpx, cpy);
[381]1331lab[3]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1332cpx += bsx+spx;
[379]1333txt[3] = new PIText(this,"",bsx,bsy,cpx,cpy);
1334txt[3]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[361]1335
[381]1336// new ligne
[385]1337cpx = spx; cpy += bsy+spy;
1338lab[4] = new PILabel(this,"Range X",bsx,bsy,cpx,cpy);
[381]1339lab[4]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1340cpx += bsx+spx;
1341txt[4] = new PIText(this,"",0.75*bsx,bsy,cpx,cpy);
[379]1342txt[4]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1343cpx += int(0.75*bsx+spx);
1344txt[5] = new PIText(this,"",0.75*bsx,bsy,cpx,cpy);
1345txt[5]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1346cpx += int(0.75*bsx+spx);
1347lab[5] = new PILabel(this,"Y",bsx/2,bsy,cpx,cpy);
[381]1348lab[5]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1349cpx += int(bsx/2.+spx);
1350txt[6] = new PIText(this,"",0.75*bsx,bsy,cpx,cpy);
[379]1351txt[6]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1352cpx += int(0.75*bsx+spx);
1353txt[7] = new PIText(this,"",0.75*bsx,bsy,cpx,cpy);
[379]1354txt[7]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[361]1355
[381]1356// new ligne
[385]1357cpx = spx; cpy += bsy+spy;
1358lab[6] = new PILabel(this,"Cent",bsx/2,bsy,cpx,cpy);
1359lab[6]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1360cpx += int(bsx/2.+spx);
1361pom[1] = new PIOptMenu(this,"X.No",1.25*bsx,bsy,cpx,cpy);
1362 pom[1]->AppendItem("X.No",2001);
1363 pom[1]->AppendItem("X.Value",2002);
1364 pom[1]->AppendItem("X.Auto",2003);
[386]1365 pom[1]->SetValue(2001);
[385]1366pom[1]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1367cpx += int(1.25*bsx+spx);
1368txt[8] = new PIText(this,"",bsx,bsy,cpx,cpy);
[379]1369txt[8]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1370cpx += bsx+spx;
1371pom[2] = new PIOptMenu(this,"Y.No",1.25*bsx,bsy,cpx,cpy);
1372 pom[2]->AppendItem("Y.No",2011);
1373 pom[2]->AppendItem("Y.Value",2012);
1374 pom[2]->AppendItem("Y.Auto",2013);
[386]1375 pom[2]->SetValue(2011);
[385]1376pom[2]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1377cpx += int(1.25*bsx+spx);
1378txt[9] = new PIText(this,"",bsx,bsy,cpx,cpy);
[379]1379txt[9]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[368]1380
[381]1381// new ligne
[385]1382cpx = spx; cpy += bsy+spy;
[379]1383but[2] = new PIButton(this, "Dismiss",777,bsx,bsy,cpx,cpy);
1384but[2]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1385cpx += bsx+spx;
[385]1386but[3] = new PIButton(this, "Check",666,bsx,bsy,cpx,cpy);
[379]1387but[3]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1388cpx += bsx+spx;
1389but[4] = new PIButton(this, "Fit",333,bsx,bsy,cpx,cpy);
1390but[4]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1391cpx += bsx+spx;
[390]1392but[5] = new PIButton(this, "Ini Auto",444,bsx,bsy,cpx,cpy);
1393but[5]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1394
1395// new ligne
1396cpx = spx; cpy += bsy+spy;
[390]1397lab[7] = new PILabel(this,"Par",bsx/2,bsy,cpx,cpy);
1398lab[7]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1399cpx += int(bsx/2.+spx);
1400lab[8] = new PILabel(this,"Fx",bsx/3,bsy,cpx,cpy);
[385]1401lab[8]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[390]1402cpx += int(bsx/3.+spx);
1403lab[9] = new PILabel(this,"Init",bsx,bsy,cpx,cpy);
[385]1404lab[9]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[390]1405cpx += bsx+spx;
1406lab[10] = new PILabel(this,"Step",bsx,bsy,cpx,cpy);
[381]1407lab[10]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1408cpx += bsx+spx;
[390]1409lab[11] = new PILabel(this,"Min",bsx,bsy,cpx,cpy);
[381]1410lab[11]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1411cpx += bsx+spx;
[390]1412lab[12] = new PILabel(this,"Max",bsx,bsy,cpx,cpy);
[381]1413lab[12]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1414
1415// new lines: Parametres (npar lignes)
[379]1416if(npar>0) {
1417 for(int i=0;i<npar;i++) {
[385]1418 cpx = spx; cpy += bsy+spy;
[379]1419 char str[8]; sprintf(str,"P%d",i);
[390]1420 lab[13+i] = new PILabel(this,str,bsx/2,bsy,cpx,cpy);
1421 lab[13+i]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
[385]1422 cpx += int(bsx/2.+spx);
[379]1423 // Attention pas de message entre 1500 et 2000 (permet 500 parametres!)
[390]1424 ckb[2+i] = new PICheckBox(this,"",1500+i,bsx/3,bsy,cpx,cpy+spy);
[385]1425 ckb[2+i]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1426 cpx+=int(bsx/3.+spx);
[379]1427 for(int j=0;j<4;j++) {
[385]1428 txt[10+4*i+j] = new PIText(this,"",bsx,bsy,cpx,cpy);
1429 txt[10+4*i+j]->SetBinding(PIBK_elastic,PIBK_elastic, PIBK_elastic,PIBK_elastic);
1430 cpx += bsx+spx;
[379]1431 }
1432 }
[361]1433}
[368]1434
[379]1435return;
1436}
1437
[361]1438/* --Methode-- */
1439PIAFitterWind::~PIAFitterWind()
1440{
[368]1441int i;
[379]1442for(i=0;i<nlab;i++) if(lab[i]) {delete lab[i]; lab[i]=NULL;}
1443for(i=0;i<ntxt;i++) if(txt[i]) {delete txt[i]; txt[i]=NULL;}
1444for(i=0;i<nbut;i++) if(but[i]) {delete but[i]; but[i]=NULL;}
1445for(i=0;i<nckb;i++) if(ckb[i]) {delete ckb[i]; ckb[i]=NULL;}
[381]1446for(i=0;i<npom;i++) if(pom[i]) {delete pom[i]; pom[i]=NULL;}
[379]1447if(lab) {delete lab; lab=NULL; nlab=0;}
1448if(txt) {delete txt; txt=NULL; ntxt=0;}
1449if(but) {delete but; but=NULL; nbut=0;}
1450if(ckb) {delete ckb; ckb=NULL; nckb=0;}
[381]1451if(pom) {delete pom; pom=NULL; npom=0;}
[361]1452}
[368]1453
[361]1454/* --Methode-- */
[379]1455void PIAFitterWind::Show()
[368]1456{
[379]1457//mDap->SetBlocked();
1458SetState();
1459PIWindow::Show();
1460return;
1461}
[368]1462
[379]1463/* --Methode-- */
1464void PIAFitterWind::SetState(void)
1465// Ecriture et positionnement des ckb et txt
1466{
1467int npar = mFitter->mNPar;
[368]1468
[379]1469// Les check-boxes
1470ckb[0]->SetState(mFitter->mOpt.okfun);
1471ckb[1]->SetState(mFitter->mOpt.okres);
1472if(npar>0)
[385]1473 for(int i=0;i<npar;i++) ckb[2+i]->SetState((mFitter->mFix(i)>=1.)?true:false);
[368]1474
[379]1475// Les champs textes
1476char str[128]; string dum;
[385]1477sprintf(str,"%f",ErrValue); dum=str; txt[0]->SetText(dum);
1478sprintf(str,"%d,%d",mFitter->mOpt.lp,mFitter->mOpt.lpg); dum=str; txt[1]->SetText(dum);
1479sprintf(str,"%f",mFitter->mOpt.stc2); dum=str; txt[2]->SetText(dum);
1480sprintf(str,"%d",mFitter->mOpt.nstep); dum=str; txt[3]->SetText(dum);
1481sprintf(str,"%d",mFitter->mOpt.i1); dum=str; txt[4]->SetText(dum);
1482sprintf(str,"%d",mFitter->mOpt.i2); dum=str; txt[5]->SetText(dum);
1483sprintf(str,"%d",mFitter->mOpt.j1); dum=str; txt[6]->SetText(dum);
1484sprintf(str,"%d",mFitter->mOpt.j2); dum=str; txt[7]->SetText(dum);
1485sprintf(str,"%f",mFitter->mOpt.xc); dum=str; txt[8]->SetText(dum);
1486sprintf(str,"%f",mFitter->mOpt.yc); dum=str; txt[9]->SetText(dum);
[379]1487if(npar>0) {
1488 for(int i=0;i<npar;i++) {
[385]1489 sprintf(str,"%f",mFitter->mPar(i)); dum=str; txt[10+4*i+0]->SetText(dum);
1490 sprintf(str,"%f",mFitter->mStep(i)); dum=str; txt[10+4*i+1]->SetText(dum);
1491 sprintf(str,"%f",mFitter->mMin(i)); dum=str; txt[10+4*i+2]->SetText(dum);
1492 sprintf(str,"%f",mFitter->mMax(i)); dum=str; txt[10+4*i+3]->SetText(dum);
[368]1493 }
1494}
1495
[379]1496return;
[361]1497}
1498
1499/* --Methode-- */
1500void PIAFitterWind::Process(PIMessage msg, PIMsgHandler* sender, void* data)
1501{
[381]1502int lp = (mFitter->mOpt.lp>2)?1:0;;
[379]1503int npar = mFitter->mNPar;
[361]1504
[385]1505// Decodage des messages, init et dismiss
1506msg = UserMsg(msg);
1507if(msg ==777) {
1508 if(lp) cout<<"Dismiss"<<endl;
1509 //mDap->SetReady();
1510 Hide();
1511 return;
1512}
1513
[379]1514// L'objet existe t-il encore?
1515NamedObjMgr omg;
1516if(omg.GetObj(mFitter->mNObj) == NULL) {
[381]1517 cout<<"PIAFitterWind::Process Error , Pas d'objet de nom "
1518 <<mFitter->mNObj<<endl;
[379]1519 mFitter->ResetDPointer(); return;
1520}
[368]1521
[379]1522// On recupere les champs textes
1523string dum; char str[128];
[385]1524// Attention gestion du type d'erreur
[379]1525dum=txt[0]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1526 sscanf(str,"%lf",&ErrValue); if(ErrValue<=0.) ErrValue=1.;
1527 if(mFitter->mOpt.err_e>0. ) mFitter->mOpt.err_e = ErrValue;
1528 if(mFitter->mOpt.err_E>0. ) mFitter->mOpt.err_E = ErrValue;
[379]1529dum=txt[1]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1530 sscanf(str,"%d,%d",&(mFitter->mOpt.lp),&(mFitter->mOpt.lpg));
[379]1531dum=txt[2]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1532 sscanf(str,"%lf",&(mFitter->mOpt.stc2));
[379]1533dum=txt[3]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1534 sscanf(str,"%d",&(mFitter->mOpt.nstep));
1535// Attention si les valeurs de mOpt.i1/.i2/.j1/.j2 ont ete changees
[379]1536dum=txt[4]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1537 sscanf(str,"%d",&(mFitter->mOpt.i1));
[379]1538dum=txt[5]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1539 sscanf(str,"%d",&(mFitter->mOpt.i2));
[379]1540dum=txt[6]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1541 sscanf(str,"%d",&(mFitter->mOpt.j1));
[379]1542dum=txt[7]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1543 sscanf(str,"%d",&(mFitter->mOpt.j2));
1544 if(saveI1!=mFitter->mOpt.i1 || saveI2!=mFitter->mOpt.i2 ||
1545 saveJ1!=mFitter->mOpt.j1 || saveJ2!=mFitter->mOpt.j2 ) {
1546 if(lp)
1547 cout<<"Range changed: I("<<saveI1<<","<<saveI2
1548 <<") -> I("<<mFitter->mOpt.i1<<","<<mFitter->mOpt.i2
1549 <<") ou J("<<saveJ1<<","<<saveJ2
1550 <<") -> J("<<mFitter->mOpt.j1<<","<<mFitter->mOpt.j2
1551 <<")"<<endl;
1552 ReFillGData = true;
1553 saveI1=mFitter->mOpt.i1; saveI2=mFitter->mOpt.i2;
1554 saveJ1=mFitter->mOpt.j1; saveJ2=mFitter->mOpt.j2;
1555 }
[379]1556dum=txt[8]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1557 sscanf(str,"%lf",&(mFitter->mOpt.xc));
[379]1558dum=txt[9]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[385]1559 sscanf(str,"%lf",&(mFitter->mOpt.yc));
[381]1560
[379]1561if(npar>0) {
1562 for(int i=0;i<npar;i++) {
[385]1563 dum=txt[10+4*i+0]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[379]1564 sscanf(str,"%lf",&(mFitter->mPar(i)));
[385]1565 dum=txt[10+4*i+1]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[379]1566 sscanf(str,"%lf",&(mFitter->mStep(i)));
[385]1567 dum=txt[10+4*i+2]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[379]1568 sscanf(str,"%lf",&(mFitter->mMin(i)));
[385]1569 dum=txt[10+4*i+3]->GetText(); strcpy(str,dum.c_str()); strip(str,'B',' ');
[379]1570 sscanf(str,"%lf",&(mFitter->mMax(i)));
[368]1571 }
[379]1572}
[361]1573
[385]1574// Decodage des messages, suite
1575if(msg ==666) {
1576 if(lp) cout<<"Check"<<endl;
[381]1577 mFitter->CheckOptions();
[379]1578 if(ReFillGData) mFitter->FillGData();
1579 ReFillGData = false;
[386]1580 if(mFitter->mFit) mFitter->mFit->PrintStatus();
[379]1581}
[381]1582else if(msg ==111) {
[379]1583 if(lp) cout<<"Update from last"<<endl;
1584 mFitter->ResetOptions();
1585 mFitter->mOpt = mFitter->mOptSave;
1586 mFitter->ReSetParam();
1587 mFitter->InitParFromLastFit();
1588 ReFillGData = true;
1589}
[381]1590else if(msg ==222) {
[379]1591 if(lp) cout<<"Default options/par"<<endl;
1592 mFitter->ResetOptions();
1593 mFitter->ReSetParam();
1594 ReFillGData = true;
1595}
[381]1596else if(msg ==333) {
[379]1597 if(lp) cout<<"Do the fit"<<endl;
[381]1598 mFitter->CheckOptions();
[379]1599 if(ReFillGData) mFitter->FillGData();
1600 ReFillGData = false;
1601 int rc = mFitter->DoFit();
1602 if(rc>=0) mFitter->FitFunRes();
1603}
[390]1604else if(msg ==444) {
1605 if(lp) cout<<"Automatic initialisation"<<endl;
1606 mFitter->AutoIniFit();
1607}
[381]1608else if(msg==1001) {
[379]1609 if(ckb[0]->GetState()) {
1610 if(lp) cout<<"Gen fitted function"<<endl;
1611 mFitter->mOpt.okfun = true;
1612 } else if(!ckb[0]->GetState()) {
1613 if(lp) cout<<"No Gen fitted function"<<endl;
1614 mFitter->mOpt.okfun = false;
[361]1615 }
[379]1616}
[381]1617else if(msg==1002) {
[379]1618 if(ckb[1]->GetState()) {
1619 if(lp) cout<<"Gen fitted residus"<<endl;
1620 mFitter->mOpt.okres = true;
1621 } else if(!ckb[1]->GetState()) {
1622 if(lp) cout<<"No Gen fitted residus"<<endl;
1623 mFitter->mOpt.okres = false;
1624 }
1625}
[385]1626else if(2101<=msg && msg<=2103) {
1627 mFitter->mOpt.err_e = mFitter->mOpt.err_E = -1.;
1628 if(msg==2102) mFitter->mOpt.err_e = ErrValue;
1629 else if(msg==2103) mFitter->mOpt.err_E = ErrValue;
1630 if(lp) cout<<"Err_cste="<<mFitter->mOpt.err_e
1631 <<" Err_sqrt="<<mFitter->mOpt.err_E
1632 <<" ErrValue="<<ErrValue<<endl;
1633 ReFillGData = true;
1634}
[381]1635else if(2001<=msg && msg<=2003) {
1636 mFitter->mOpt.polcx = msg-2001;
1637 if(lp) cout<<"Centrage X polcx="<<mFitter->mOpt.polcx
1638 <<" xc="<<mFitter->mOpt.xc<<endl;
[379]1639 ReFillGData = true;
1640}
[381]1641else if(2011<=msg && msg<=2013) {
[392]1642 mFitter->mOpt.polcy = msg-2011;
[381]1643 if(lp) cout<<"Centrage Y polcy="<<mFitter->mOpt.polcy
1644 <<" yc="<<mFitter->mOpt.yc<<endl;
[379]1645 ReFillGData = true;
1646}
[381]1647else if(msg>=1500 && msg<2000) {
[379]1648 int ip = msg-1500;
[385]1649 if(ckb[2+ip]->GetState()) {
[379]1650 if(lp) cout<<"Fix Param "<<ip<<endl;
1651 mFitter->mFix(ip) = 1.001;
[385]1652 } else if(!ckb[2+ip]->GetState()) {
[379]1653 if(lp) cout<<"Not Fix Param "<<ip<<endl;
1654 mFitter->mFix(ip) = 0.;
1655 }
1656}
[361]1657
[379]1658mFitter->CheckOptions();
1659SetState();
1660if(mFitter->mOpt.lp>1) mFitter->PrintOptions();
1661
1662return;
[361]1663}
Note: See TracBrowser for help on using the repository browser.