source: Sophya/trunk/SophyaPI/PIext/pawexecut.cc@ 544

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

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

File size: 14.5 KB
Line 
1#include <stdio.h>
2#include <stdlib.h>
3#include <ctype.h>
4#include <iostream.h>
5#include <typeinfo>
6
7#include "strutil.h"
8#include "histos.h"
9#include "histos.h"
10#include "histos2.h"
11#include "hisprof.h"
12#include "ntuple.h"
13
14#include "pawexecut.h"
15#include "nobjmgr.h"
16#include "servnobjm.h"
17#include "pistdimgapp.h"
18
19#ifdef SANS_EVOLPLANCK
20#include "cvector.h"
21#include "matrix.h"
22#else
23#include "tmatrix.h"
24#include "tvector.h"
25
26#endif
27
28/* Reza + cmv 13/10/99 */
29
30/* methode */
31PAWExecutor::PAWExecutor(PIACmd *piac, PIStdImgApp* app)
32: mApp(app)
33{
34string kw, usage;
35string hgrp = "pawCmd";
36
37kw = "reset";
38usage = "Reset histograms vectors or matrix";
39usage += "\n reset nameobj";
40piac->RegisterCommand(kw,usage,this,hgrp);
41
42kw = "n/plot";
43usage = "Plot NTuple variables a la paw";
44usage += "\n n/plot nameobj.x_exp [cut] [w_exp] [loop] [gratt]";
45usage += "\n n/plot nameobj.y_exp%x_exp [cut] [loop] [gratt]";
46usage += "\n n/plot nameobj.z_exp%y_exp%x_exp [cut] [loop] [gratt]";
47usage += "\n for default use ! , loop=i1[:i2[:di]]";
48usage += "\n Related commands: plot2dw plot3d";
49piac->RegisterCommand(kw,usage,this,hgrp);
50
51kw = "n/proj";
52usage = "Project NTuple in histogram (1D or 2D) a la paw";
53usage += "\n n/proj nameproj nameobj.x_exp [cut] [w_exp] [loop] [gratt]";
54usage += "\n n/proj nameproj nameobj.y_exp%x_exp [cut] [w_exp] [loop] [gratt]";
55usage += "\n for default use ! , loop=i1[:i2[:di]]";
56usage += "\n Related commands: projh1d projh2d projprof";
57piac->RegisterCommand(kw,usage,this,hgrp);
58
59kw = "h/integ";
60usage = "Integrate a 1D histogram or profile";
61usage += "\n h/integ nameh1d [norm]";
62usage += "\n Related commands: h/deriv";
63piac->RegisterCommand(kw,usage,this,hgrp);
64
65kw = "h/deriv";
66usage = "Derivate a 1D histogram or profile";
67usage += "\n h/deriv nameh1d";
68usage += "\n Related commands: h/integ";
69piac->RegisterCommand(kw,usage,this,hgrp);
70
71kw = "h/rebin";
72usage = "Rebin a 1D histogram or profile";
73usage += "\n h/rebin nbin";
74piac->RegisterCommand(kw,usage,this,hgrp);
75
76kw = "h/cadd";
77usage = "Add a constant to an histogramme";
78usage += "\n h/cadd val";
79usage += "\n Related commands: h/cmult";
80piac->RegisterCommand(kw,usage,this,hgrp);
81
82kw = "h/cmult";
83usage = "Multiply an histogramme by a constant";
84usage += "\n h/cmult val";
85usage += "\n Related commands: h/cadd";
86piac->RegisterCommand(kw,usage,this,hgrp);
87
88kw = "h/plot/2d";
89usage = "Specific plot for 2D histogrammes";
90usage += "\n h/plot/2d nameh2d show : infos on 2D histogramme";
91usage += "\n h/plot/2d nameh2d h [dopt] : plot 2D histogramme";
92usage += "\n h/plot/2d nameh2d px [dopt] : plot X projection";
93usage += "\n h/plot/2d nameh2d py [dopt] : plot Y projection";
94usage += "\n h/plot/2d nameh2d bx n [dopt] : plot X band number n";
95usage += "\n h/plot/2d nameh2d by n [dopt] : plot Y band number n";
96usage += "\n h/plot/2d nameh2d sx n [dopt] : plot X slice number n";
97usage += "\n h/plot/2d nameh2d sy n [dopt] : plot Y slice number n";
98usage += "\n n < 0 means Show Info";
99piac->RegisterCommand(kw,usage,this,hgrp);
100}
101
102/* methode */
103PAWExecutor::~PAWExecutor()
104{
105}
106
107/* methode */
108int PAWExecutor::Execute(string& kw, vector<string>& tokens)
109{
110if(kw == "reset") {
111 reset(tokens); return(0);
112} else if(kw == "n/plot") {
113 n_plot(tokens); return(0);
114} else if(kw == "n/proj") {
115 n_proj(tokens); return(0);
116} else if(kw == "h/integ") {
117 h_integ(tokens); return(0);
118} else if(kw == "h/deriv") {
119 h_deriv(tokens); return(0);
120} else if(kw == "h/rebin") {
121 h_rebin(tokens); return(0);
122} else if(kw == "h/cadd") {
123 h_cadd(tokens); return(0);
124} else if(kw == "h/cmult") {
125 h_cmult(tokens); return(0);
126} else if(kw == "h/plot/2d") {
127 h_plot_2d(tokens); return(0);
128} else return(1);
129}
130
131/* methode */
132void PAWExecutor::reset(vector<string>& tokens)
133// Reset d'histogrammes, vecteurs et matrices
134{
135if(tokens.size() < 1)
136 {cout<<"Usage: reset nameobj"<<endl; return;}
137NamedObjMgr omg;
138AnyDataObj* mobj = omg.GetObj(tokens[0]);
139if(mobj == NULL)
140 {cout<<"PAWExecutor::reset Error , Pas d'objet de nom "<<tokens[0]<<endl;
141 return;}
142string ctyp = typeid(*mobj).name();
143
144if(typeid(*mobj)==typeid(Vector)) {Vector* ob=(Vector*) mobj; ob->Zero();}
145else if(typeid(*mobj)==typeid(Matrix)) {Matrix* ob=(Matrix*) mobj; ob->Zero();}
146else if(typeid(*mobj)==typeid(Histo)) {Histo* ob=(Histo*) mobj; ob->Zero();}
147else if(typeid(*mobj)==typeid(HProf)) {HProf* ob=(HProf*) mobj; ob->Zero();}
148else if(typeid(*mobj)==typeid(Histo2D)) {Histo2D* ob=(Histo2D*)mobj; ob->Zero();}
149else {
150 cout<<"PAWExecutor::reset Error , No reset possible on "<<ctyp<<endl;
151 return;
152}
153
154return;
155}
156
157/* methode */
158void PAWExecutor::n_plot(vector<string>& tokens)
159// Equivalent n/plot de paw
160// Plot 1D
161// n/plot nameobj.x_exp [cut] [w_exp] [gratt]
162// Plot 2D (plot2dw)
163// n/plot nameobj.y_exp%x_exp [cut] [w_exp] [gratt]
164// Plot 3D (plot3d)
165// n/plot nameobj.z_exp%y_exp%x_exp [cut] [gratt]
166{
167if(tokens.size() < 1) {
168 cout
169 <<"Usage: n/plot nameobj.x_exp [cut] [w_exp] [loop] [gratt]"<<endl
170 <<" n/plot nameobj.y_exp%x_exp [cut] [loop] [gratt]"<<endl
171 <<" n/plot nameobj.z_exp%y_exp%x_exp [cut] [loop] [gratt]"<<endl
172 <<" for default use ! , loop=i1[:i2[:di]]"<<endl;
173 return;
174}
175string nameobj,expx,expy,expz;
176int nvar = decodepawstring(tokens[0],nameobj,expx,expy,expz);
177string expcut = "1"; string expwt = "1."; string loop = ""; string dopt = "";
178if(tokens.size()>=2) expcut = tokens[1]; if(expcut=="!") expcut="1";
179
180NamedObjMgr omg;
181Services2NObjMgr* srvo = omg.GetServiceObj();
182
183if(nvar<=0) {
184 cout<<"PAWExecutor::n_plot Error: bad coding "<<tokens[0]<<endl;
185} else if(nvar==1) { // c'est un plot 1D
186 if(tokens.size()>=3) expwt = tokens[2]; if(expwt=="!") expwt="1.";
187 if(tokens.size()>=4) loop = tokens[3]; if(loop=="!") loop="";
188 if(tokens.size()>=5) dopt = tokens[4];
189 string nameproj = "/autoc/paw_n_plot1D";
190 AnyDataObj* mobj = omg.GetObj(nameproj);
191 if(mobj!=NULL) omg.DelObj(nameproj);
192 srvo->ProjectH1(nameobj,expx,expwt,expcut,nameproj,dopt,loop);
193} else if(nvar==2) { // c'est un plot 2D
194 if(tokens.size()>=3) loop = tokens[2]; if(loop=="!") loop="";
195 if(tokens.size()>=4) dopt = tokens[3];
196 string err = "";
197 srvo->DisplayPoints2D(nameobj,expx,expy,err,err,expcut,dopt,loop);
198} else { // c'est un plot 3D
199 if(tokens.size()>=3) loop = tokens[2]; if(loop=="!") loop="";
200 if(tokens.size()>=4) dopt = tokens[3];
201 srvo->DisplayPoints3D(nameobj,expx,expy,expz,expcut,dopt,loop);
202}
203
204return;
205}
206
207/* methode */
208void PAWExecutor::n_proj(vector<string>& tokens)
209// Equivalent n/proj de paw
210// Project NTuple in histogram a la paw
211// Dans un Histo 1D
212// n/proj nameproj nameobj.x_exp [cut] [w_exp] [gratt]
213// Dans un Histo 2D ou un HProf (dans ce cas nameproj doit etre cree).
214// n/proj nameproj nameobj.y_exp%x_exp [cut] [w_exp] [gratt]
215{
216if(tokens.size()<2)
217 {cout<<"Usage: n/proj nameproj nameobj.[y_exp%]x_exp [cut] [w_exp] [loop] [gratt]"<<endl
218 <<" for default use ! , loop=i1[:i2[:di]]"<<endl; return;}
219string nameproj = tokens[0];
220string nameobj,expx,expy,expz;
221int nvar = decodepawstring(tokens[1],nameobj,expx,expy,expz);
222string expcut = "1"; string expwt = "1."; string loop = ""; string dopt = "";
223if(tokens.size()>=3) expcut = tokens[2]; if(expcut=="!") expcut="1";
224if(tokens.size()>=4) expwt = tokens[3]; if(expwt=="!") expwt="1.";
225if(tokens.size()>=5) loop = tokens[4]; if(loop=="!") loop="";
226if(tokens.size()>=6) dopt = tokens[5];
227
228NamedObjMgr omg;
229Services2NObjMgr* srvo = omg.GetServiceObj();
230
231if(nvar<=0) {
232 cout<<"PAWExecutor::n_proj Error: bad coding "<<tokens[1]<<endl;
233} else if(nvar==1) {
234 // c'est une projection dans un histo 1D
235 srvo->ProjectH1(nameobj,expx,expwt,expcut,nameproj,dopt,loop);
236} else {
237 // c'est une projection dans un histo2D
238 // OU un HProf si nameproj est HProf un deja defini
239 AnyDataObj* mobj = omg.GetObj(nameproj);
240 if(mobj==NULL)
241 srvo->ProjectH2(nameobj,expx,expy,expwt,expcut,nameproj,dopt,loop);
242 else if(dynamic_cast<HProf*>(mobj))
243 srvo->ProjectHProf(nameobj,expx,expy,expwt,expcut,nameproj,dopt,loop);
244 else
245 srvo->ProjectH2(nameobj,expx,expy,expwt,expcut,nameproj,dopt,loop);
246}
247
248return;
249}
250
251/* methode */
252void PAWExecutor::h_integ(vector<string>& tokens)
253// Pour remplacer le contenu d'un histo 1D par son integrale
254{
255if(tokens.size()<1)
256 {cout<<"Usage: h/integ nameh1d [norm]"<<endl; return;}
257NamedObjMgr omg;
258AnyDataObj* mobj = omg.GetObj(tokens[0]);
259if(mobj==NULL)
260 {cout<<"PAWExecutor::h_integ Error: unknow object"<<tokens[0]<<endl;
261 return;}
262double norm = 1.;
263if(tokens.size()>=2) norm = atof(tokens[1].c_str());
264Histo* h1 = dynamic_cast<Histo*>(mobj);
265// Le HProf se fait tout seul avec Histo: dynamic_cast<Histo*>(HProf)=Vrai!
266//HProf* hp = dynamic_cast<HProf*>(mobj);
267if(h1) h1->HInteg(norm);
268//else if(hp) hp->HInteg(norm);
269else cout<<"PAWExecutor::h_integ Error: "<<tokens[0]<<" not an Histo/HProf"<<endl;
270}
271
272/* methode */
273void PAWExecutor::h_deriv(vector<string>& tokens)
274// Pour remplacer le contenu d'un histo 1D par sa derivee
275{
276if(tokens.size()<1)
277 {cout<<"Usage: h/deriv nameh1d"<<endl; return;}
278NamedObjMgr omg;
279AnyDataObj* mobj = omg.GetObj(tokens[0]);
280if(mobj==NULL)
281 {cout<<"PAWExecutor::h_deriv Error: unknow object"<<tokens[0]<<endl;
282 return;}
283Histo* h1 = dynamic_cast<Histo*>(mobj);
284//HProf* hp = dynamic_cast<HProf*>(mobj);
285if(h1) h1->HDeriv();
286//else if(hp) hp->HDeriv();
287else cout<<"PAWExecutor::h_deriv Error: "<<tokens[0]<<" not an Histo/HProf"<<endl;
288}
289
290/* methode */
291void PAWExecutor::h_rebin(vector<string>& tokens)
292// Pour re-binner un histogramme 1D
293{
294if(tokens.size()<2)
295 {cout<<"Usage: h/rebin nameh1d nbin"<<endl; return;}
296NamedObjMgr omg;
297AnyDataObj* mobj = omg.GetObj(tokens[0]);
298if(mobj==NULL)
299 {cout<<"PAWExecutor::h_rebin Error: unknow object"<<tokens[0]<<endl;
300 return;}
301int nbin = atoi(tokens[1].c_str());
302Histo* h1 = dynamic_cast<Histo*>(mobj);
303//HProf* hp = dynamic_cast<HProf*>(mobj);
304if(h1) h1->HRebin(nbin);
305//else if(hp) hp->HRebin(nbin);
306else cout<<"PAWExecutor::h_rebin Error: "<<tokens[0]<<" not an Histo/HProf"<<endl;
307}
308
309/* methode */
310void PAWExecutor::h_cadd(vector<string>& tokens)
311// Additionne une constante a un histogramme
312{
313if(tokens.size()<2)
314 {cout<<"Usage: h/cadd nameh1d val"<<endl; return;}
315NamedObjMgr omg;
316AnyDataObj* mobj = omg.GetObj(tokens[0]);
317if(mobj==NULL)
318 {cout<<"PAWExecutor::h_cadd Error: unknow object"<<tokens[0]<<endl;
319 return;}
320double val = atof(tokens[1].c_str());
321Histo* h1 = dynamic_cast<Histo*>(mobj);
322Histo2D* h2 = dynamic_cast<Histo2D*>(mobj);
323if(h1) *h1 += val;
324else if(h2) *h2 += val;
325else cout<<"PAWExecutor::h_cadd Error: "<<tokens[0]<<" not an Histo/HProf/Histo2D"<<endl;
326}
327
328/* methode */
329void PAWExecutor::h_cmult(vector<string>& tokens)
330// Multiplie un histogramme par une constante
331{
332if(tokens.size()<2)
333 {cout<<"Usage: h/cmult nameh1d val"<<endl; return;}
334NamedObjMgr omg;
335AnyDataObj* mobj = omg.GetObj(tokens[0]);
336if(mobj==NULL)
337 {cout<<"PAWExecutor::h_cmult Error: unknow object"<<tokens[0]<<endl;
338 return;}
339double val = atof(tokens[1].c_str());
340Histo* h1 = dynamic_cast<Histo*>(mobj);
341Histo2D* h2 = dynamic_cast<Histo2D*>(mobj);
342if(h1) *h1 *= val;
343else if(h2) *h2 *= val;
344else cout<<"PAWExecutor::h_cmult Error: "<<tokens[0]
345 <<" not an Histo/HProf/Histo2D"<<endl;
346}
347
348/* methode */
349void PAWExecutor::h_plot_2d(vector<string>& tokens)
350// plot for 2D histogramme: plot histo, bandx/y, slicex/y or projx/y
351{
352if(tokens.size()<2)
353 {cout<<"Usage: h/plot/2d nameh2d to_plot [n/s] [dopt]"<<endl; return;}
354NamedObjMgr omg;
355AnyDataObj* mobj = omg.GetObj(tokens[0]);
356if(mobj==NULL)
357 {cout<<"PAWExecutor::h_plot_2d Error: unknow object"<<tokens[0]<<endl;
358 return;}
359Histo2D* h2 = dynamic_cast<Histo2D*>(mobj);
360if(!h2)
361 {cout<<"PAWExecutor::h_plot_2d Error: "<<tokens[0]<<" not an Histo2D"<<endl;
362 return;}
363
364Histo* h1p = NULL; string nametoplot = "/autoc/h_plot_2d_h1";
365AnyDataObj* mobjh1 = omg.GetObj(nametoplot);
366if(mobjh1!=NULL) omg.DelObj(nametoplot);
367
368string dopt = ""; if(tokens.size()>=3) dopt = tokens[2];
369if(tokens[1] == "show") {
370 h2->ShowProj();
371 h2->ShowBand(2);
372 h2->ShowSli(2);
373 return;
374} else if(tokens[1] == "h") {
375 nametoplot = tokens[0];
376} else if(tokens[1] == "px") {
377 if((h1p=h2->HProjX())) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
378 else {h2->ShowProj(); return;}
379} else if(tokens[1] == "py") {
380 if((h1p=h2->HProjY())) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
381 else {h2->ShowProj(); return;}
382} else {
383 if(tokens.size()<3)
384 {cout<<"Usage: h/plot/2d nameh2d bx/by/sx/sy n [dopt]"<<endl; return;}
385 int n = atoi(tokens[2].c_str());
386 dopt = ""; if(tokens.size()>=4) dopt = tokens[3];
387 if(tokens[1] == "bx") {
388 if((h1p=h2->HBandX(n))) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
389 else {h2->ShowBand(); return;}
390 } else if(tokens[1] == "by") {
391 if((h1p=h2->HBandY(n))) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
392 else {h2->ShowBand(); return;}
393 } else if(tokens[1] == "sx") {
394 if((h1p=h2->HSliX(n))) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
395 else {h2->ShowSli(); return;}
396 } else if(tokens[1] == "sy") {
397 if((h1p=h2->HSliY(n))) {Histo* h1=new Histo(*h1p); omg.AddObj(h1,nametoplot);}
398 else {h2->ShowSli(); return;}
399 }
400}
401
402omg.DisplayObj(nametoplot,dopt);
403}
404
405/* methode */
406int PAWExecutor::decodepawstring(string tokens,string& nameobj
407 ,string& xexp,string& yexp,string& zexp)
408// Decodage general de "nameobj.xexp"
409// "nameobj.yexp%xexp"
410// "nameobj.zexp%yexp%xexp"
411// Return: nombre de variables trouvees, -1 si probleme
412{
413nameobj = ""; xexp= ""; yexp= ""; zexp= "";
414
415int lt = (int) tokens.length();
416if(lt<=0) return -1;
417
418// decodage de la chaine de type PAW.
419char *str = new char[lt+2];
420strcpy(str,tokens.c_str()); strip(str,'B',' '); lt = strlen(str);
421//cout<<"chaine1["<<lt<<"] :"<<str<<":"<<endl;
422char *c[3] = {NULL,NULL,NULL};
423int i, np=0; bool namefound = false;
424for(int i=0;i<lt;i++) {
425 if(!namefound && str[i]=='.') {
426 str[i]='\0';
427 namefound=true;
428 c[np] = str+i+1; np++;
429 }
430 if( namefound && str[i]=='%') {
431 str[i]='\0';
432 if(np<3) {c[np] = str+i+1; np++;}
433 }
434}
435//cout<<"chaine2 :"; for(i=0;i<lt;i++) cout<<str[i]; cout<<":"<<endl;
436
437// Remplissage du nom et des variables
438nameobj = str;
439if(np==1) xexp=c[0];
440if(np==2) {yexp=c[0]; xexp=c[1];}
441if(np==3) {zexp=c[0]; yexp=c[1]; xexp=c[2];}
442//cout<<"pawstring str,c[0-2] "<<str<<" "<<c[0]<<" "<<c[1]<<" "<<c[2]<<endl;
443delete [] str;
444
445// Comptage des variables
446np = -1;
447if(nameobj.length()>0)
448 {np = 0; if(xexp.length()>0)
449 {np++; if(yexp.length()>0)
450 {np++; if(zexp.length()>0) np++;}}}
451cout<<"pawstring["<<np<<"] name="<<nameobj
452 <<" xexp="<<xexp<<" yexp="<<yexp<<" zexp="<<zexp<<endl;
453return np;
454}
Note: See TracBrowser for help on using the repository browser.