Changeset 442 in PSPA for Interface_Web/trunk/pspaWT/sources/controler/src/particleBeam.cc

Timestamp:

Dec 11, 2013, 9:33:35 AM (11 years ago)

Author:

lemeur

Message:

ajout traitement utilisateur

File:

• Interface_Web/trunk/pspaWT/sources/controler/src/particleBeam.cc (modified) (2 diffs)

Interface_Web/trunk/pspaWT/sources/controler/src/particleBeam.cc

@@ -5 +5 @@
 #include "mathematicalTools.h"
 #include "mixedTools.h"
+
+#include "UAP/UAPUtilities.hpp"
+#include "AML/AMLReader.hpp"
 
 #include <stdio.h>
@@ -203 +206 @@
   sortie << endl;
   return sortie.str();
+}
+
+
+void particleBeam::writeToAMLFile(string fileName) {
+  UAPNode* uap_root = new UAPNode("UAP_root");
+  UAPNode* rep = uap_root->addChild("AML_representation");
+
+  // root node in the hierarchy
+  UAPNode* lab = rep->addChild("laboratory");
+  lab->addAttribute("name","PSPA lab");
+
+
+
+  // root node in the hierarchy
+  UAPNode* par = lab->addChild("particles");
+  par->addAttribute("name","PSPA particles");
+
+  unsigned int k;
+  if ( particleRepresentationOk_ ) {
+  for ( k = 0 ; k < relativePartic_.size(); k++)
+    {
+      TRIDVECTOR& position = relativePartic_.at(k).getReferenceToPosition();
+      TRIDVECTOR& betgam = relativePartic_.at(k).getReferenceToBetaGamma();
+      UAPNode* partic = par->addChild("particle");
+      string txt;
+      txt = mixedTools::doubleToString(position.getComponent(0));
+      partic->addAttribute("x",txt);
+      txt = mixedTools::doubleToString(position.getComponent(1));
+      partic->addAttribute("y",txt);
+      txt = mixedTools::doubleToString(position.getComponent(2));
+      partic->addAttribute("cdeltat",txt);
+      txt = mixedTools::doubleToString(betgam.getComponent(0));
+      partic->addAttribute("betagamma_x",txt);
+      txt = mixedTools::doubleToString(betgam.getComponent(1));
+      partic->addAttribute("betagamma_y",txt);
+      txt = mixedTools::doubleToString(betgam.getComponent(2));
+      partic->addAttribute("betagamma_z",txt);
+    }
+
+  } else {
+    cout << " particleBeam::writeToAMLFile : representation 'particules' indisponible" << endl;
+  }
+
+  cout << "!Create the AML particle file ---------------------------" << endl;
+  AMLReader reader;
+  reader.AMLRepToAMLFile (uap_root, fileName);
+
+}
+
+bool particleBeam::readFromAMLFile(string fileName) {
+
+  cout << "!read the AML particle file ---------------------------" << endl;
+  AMLReader reader;
+  UAPNode* uap_root = NULL;
+  uap_root = reader.AMLFileToAMLRep (fileName);
+  if ( !uap_root ) {
+    cout << " particleBeam::readFromAMLFile ERREUR lecture fichier particules " << endl;
+    return false;
+  }
+
+
+  NodeVec lesParticules = uap_root->getSubNodesByName(string("particles"));
+
+  NodeVec& listPart = (*lesParticules.begin())->getChildren();
+
+  if (  !listPart.size() ) {
+    cout << " particleBeam::readFromAMLFile ERREUR lecture fichier : pas de particules ? " << endl;
+    return false;
+  }
+
+  vector<bareParticle> vecteurParticules;
+  bareParticle reference;
+  for (NodeVecIter iter = listPart.begin(); iter != listPart.end(); iter++) {
+    TRIDVECTOR position, betagamma;
+    UAPAttribute* att = NULL;
+    att = (*iter)->getAttribute("x");
+    if ( att ) att->getDouble(position.component(0));
+    att = (*iter)->getAttribute("y");
+    if ( att ) att->getDouble(position.component(1));
+    att = (*iter)->getAttribute("cdeltat");
+    if ( att ) att->getDouble(position.component(2));
+
+    att = (*iter)->getAttribute("betagamma_x");
+    if ( att ) att->getDouble(betagamma.component(0));
+    att = (*iter)->getAttribute("betagamma_y");
+    if ( att ) att->getDouble(betagamma.component(1));
+    att = (*iter)->getAttribute("betagamma_z");
+    if ( att ) att->getDouble(betagamma.component(2));
+    
+    if ( fabs(position.component(0)) < PETIT &&  fabs(position.component(1)) < PETIT && fabs(position.component(2)) < PETIT ) {
+      reference = bareParticle(position, betagamma);
+    }
+    vecteurParticules.push_back(bareParticle(position, betagamma));
+  }
+  vector<double> centroid(6,0.0);
+  setWithParticles(centroid, reference, vecteurParticules);
+  return true;
 }