Changeset 2816 in Sophya for trunk/SophyaPI/PIext/pawexecut.cc

Timestamp:

Jul 12, 2005, 5:22:47 PM (20 years ago)

Author:

cmv

Message:

n/copy copy de ntuple avec choix de variables cmv 12/07/05

File:

• trunk/SophyaPI/PIext/pawexecut.cc (modified) (3 diffs)

trunk/SophyaPI/PIext/pawexecut.cc

@@ -129 +129 @@
 usage += "\n   if \"ext\"=\"!\" no extension is added";
 usage += "\n   if no \"ext\" is given, automatic extension \"_0\" \"_1\" \"_2\" ...";
+piac->RegisterCommand(kw,usage,this,hgrp);
+
+kw = "n/copy";
+usage = "Copy all or some variables of a ntuple into another new ntuple";
+usage += "\n n/merge ntnew nt [vname1 vname2 vname3 ...]";
 piac->RegisterCommand(kw,usage,this,hgrp);
 
@@ -297 +302 @@
 } else if(kw == "n/merge/col") {
   n_merge_col(tokens); return(0);
+} else if(kw == "n/copy") {
+  n_copy(tokens); return(0);
 } else if(kw == "h/integ") {
   h_integ(tokens); return(0);
@@ -1046 +1053 @@
 
 /* methode */
+void PAWExecutor::n_copy(vector<string>& tokens)
+{
+ if(tokens.size()<2) {
+   cerr<<"Usage: n/copy ntnew nt [vname1 vname2 vname3 ...]"<<endl;
+   return;
+ }
+
+ NamedObjMgr omg;
+
+ // Le NTuple de depart
+ AnyDataObj* mobj = omg.GetObj(tokens[1]);
+ if(mobj==NULL) {
+   cout<<"n_copy Error: unknow object"<<tokens[1]<<endl;
+   return;
+ }
+ NTuple* nt = dynamic_cast<NTuple*>(mobj);
+ if(nt==NULL) {
+   cout<<"n_copy Error: "<<tokens[1]<<" not a NTuple"<<endl;
+   return;
+ }
+ if(nt->NEntry()==0) {
+   cout<<"n_copy Error: "<<tokens[1]<<" is empty"<<endl;
+   return;
+ }
+ if(nt->NVar()==0) {
+   cout<<"n_copy Error: "<<tokens[1]<<" has no variable"<<endl;
+   return;
+ }
+
+ // Les variables a copier
+ int nvar; vector<string> varname; vector<int> ivar;
+ if(tokens.size()==2) {
+   nvar = nt->NVar();
+   for(int i=0;i<nvar;i++) {
+     ivar.push_back(i);
+     varname.push_back(nt->NomIndex(i));
+   }
+ } else {
+   nvar = tokens.size()-2;
+   for(int i=0;i<nvar;i++) {
+     int iv = nt->IndexNom(tokens[i+2].c_str());
+     if(iv<0) {
+       cout<<"n_copy Error: unkown variable "<<tokens[i+2].c_str()<<endl;
+       return;
+     }
+     ivar.push_back(iv);
+     varname.push_back(tokens[i+2]);
+   }
+ }
+ cout<<"coping variables: ";
+ for(int i=0;i<nvar;i++) cout<<" "<<varname[i];
+ cout<<endl;
+
+ // Creation du nouveau NTuple
+ char **ntvn = new char*[nvar];
+ for(int i=0;i<nvar;i++) ntvn[i] = const_cast<char *>(varname[i].c_str());
+ NTuple ntnew(nvar,ntvn);
+ delete [] ntvn;
+
+ // Copie du NTuple
+ r_8 *xnt = new r_8[nt->NVar()];
+ r_8 *xntnew = new r_8[nvar];
+ for(int iev=0;iev<nt->NEntry();iev++) {
+   nt->GetVecD(iev,xnt);
+   for(int i=0;i<nvar;i++) xntnew[i] = xnt[ivar[i]];
+   ntnew.Fill(xntnew);
+ }
+ delete [] xnt; delete [] xntnew;
+
+ // Adding new NTuple
+ omg.AddObj(ntnew,tokens[0]);
+
+ return;
+}
+
+/* methode */
 void PAWExecutor::h_integ(vector<string>& tokens)
 // Pour remplacer le contenu d'un histo 1D par son integrale