source: PSPA/Interface_Web/trunk/pspaWT/sources/controler/src/elementBeam.cc @ 359

#include "elementBeam.h"
#include "beam2Moments.h"
#include "mixedTools.h"

elementBeam::elementBeam() :  abstractElement() {
    setDefaultValues();
    setDefaults();
    elementName_ = nomdElements("beam");
    nbParam_ = elementName_.getElementNbParameters();
    parametersString_ = new string[nbParam_+1];
    //  transportOk_ = true;
    registerAcceptableSoftware(nomDeLogiciel::transport, TBoolOk);
}

elementBeam::~elementBeam() {;}

void elementBeam::setDefaultValues() {
    defaultLabel_ = "beam";
    xDef_ = 0.0;
    xpDef_ = 0.0;
    yDef_ = 0.0;
    ypDef_ = 0.0;
    dlDef_ = 0.0;
    delDef_ = 0.0;
    p0Def_ = 0.0;
}

void elementBeam::setDefaults() {
    label_ = defaultLabel_;
    x_ = xDef_;
    xp_ = xpDef_;
    y_ = yDef_;
    yp_ = ypDef_;
    dl_ = dlDef_;
    del_ = delDef_;
    p0_ = p0Def_;
}


string* elementBeam::getParametersString() const {
    int compteur = -1;
    parametersString_[++compteur] = mixedTools::intToString(nbParam_);
    parametersString_[++compteur] = label_;
    parametersString_[++compteur] = mixedTools::doubleToString(x_);
    parametersString_[++compteur] = mixedTools::doubleToString(xp_);
    parametersString_[++compteur] = mixedTools::doubleToString(y_);
    parametersString_[++compteur] = mixedTools::doubleToString(yp_);
    parametersString_[++compteur] = mixedTools::doubleToString(dl_);
    parametersString_[++compteur] = mixedTools::doubleToString(del_);
    parametersString_[++compteur] = mixedTools::doubleToString(p0_);
    if ( compteur != nbParam_ ) {
        cerr << " elementBeam::getParametersString() : ERROR nr pf parameters doesnt match " << endl;
        return NULL;
    }
    return parametersString_;
}

void elementBeam::setParametersString(string* param) {
    if ( param == NULL )
    {
        cerr << "  elementBeam::setParametersString parameters empty parameter set";
        return;
    }
    int compteur = -1;
    int nbparam = atoi(param[++compteur].c_str());
    if ( nbparam != nbParam_ )
    {
        cerr << "  elementBeam::setParametersString parameters do not match for a BEAM";
        return;
    }
    
    label_ = param[++compteur];
    x_ = atof(param[++compteur].c_str());
    xp_ = atof(param[++compteur].c_str());
    y_ = atof(param[++compteur].c_str());
    yp_ = atof(param[++compteur].c_str());
    dl_ = atof(param[++compteur].c_str());
    del_ = atof(param[++compteur].c_str());
    p0_ = atof(param[++compteur].c_str());
}

string elementBeam::parmelaOutputFlow() const
{
    ostringstream sortie;
    cout << " BEAM sortie parmela non programmee " << endl;
    return sortie.str();
}

string elementBeam::transportOutputFlow() const {
    ostringstream sortie;
    beam2Moments moments(x_, xp_, y_, yp_, dl_, del_);
    cout << " elementBeam::transportOutputFlow(), p0 = " << p0_ << endl;
    sortie << label_ << ":" << moments.writeToTransportInput(p0_);
    return sortie.str();
}

string elementBeam::FileOutputFlow() const {
    ostringstream sortie;
    // sortie << elementName_.getElementType() << endl;
    sortie << elementName_.getElementLabel() << endl;
    sortie  << label_ << endl;
    sortie << x_ << "  " << xp_ << " " <<  y_  << " " <<  yp_  << endl;
    sortie << dl_ << "  " << del_ << " " <<  p0_  << endl;
    cout << " elementBeam::FileOutputFlow, p0 = " << p0_ << endl;
    return sortie.str();
}

void elementBeam::elementBeam::FileInput(ifstream& ifs) {
    ifs >> label_;
    ifs >> x_ >> xp_ >>  y_ >> yp_;
    ifs >> dl_ >> del_ >> p0_;
    cout << " elementBeam::FileFileInput, p0 = " << p0_ << endl;
    lenghtElem_ = getLenghtOfElement();
    cout << " elementBeam::FileInput calcule longueur = " << lenghtElem_ << endl;
}


string elementBeam::print() {
    string txt = "";
    txt += label_;
    txt += "\n 1/2 horizontal beam extend rms (cm) : ";
    txt += mixedTools::doubleToString(x_);
    txt += "\n 1/2 horizontal beam divergence rms (mrad) : ";
    txt += mixedTools::doubleToString(xp_);
    txt += "\n1/2 vertical beam extend rms (cm) : ";
    txt += mixedTools::doubleToString(y_);
    txt += "\n1/2 horizontal beam divergence rms (mrad) : ";
    txt += mixedTools::doubleToString(yp_);
    txt += "\n1/2 longitudinal beam extend rms (cm) : ";
    txt += mixedTools::doubleToString(dl_);
    txt += "\n1/2 momentum spread rms (mrad) : ";
    txt += mixedTools::doubleToString(del_);
    txt += "\nmomentum of the central trajectory (GeV/c) : ";
    txt += mixedTools::doubleToString(p0_);
    
    return txt;
}