MultiBunchDump.cpp

Go to the documentation of this file.

//
// Class MultiBunchDump
//   This class writes a SDDS file of single bunch statistics in multibunch simulations.
//
// Copyright (c) 2018 - 2019, Matthias Frey, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// Implemented as part of the PhD thesis
// "Precise Simulations of Multibunches in High Intensity Cyclotrons"
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#include "MultiBunchDump.h"
 
#include <sstream>
 
#include "Utilities/Timer.h"
#include "Utilities/Util.h"
#include "AbstractObjects/OpalData.h"
#include "Algorithms/PartBunchBase.h"
#include "Ippl.h"
 
extern Inform *gmsg;
 
MultiBunchDump::MultiBunchDump(const std::string& fname, bool restart)
    : StatBaseWriter(fname, restart)
{ }
 
 
void MultiBunchDump::fillHeader() {
 
    if (this->hasColumns()) {
        return;
    }
 
    columns_m.addColumn("t", "double", "ns", "Time");
    columns_m.addColumn("s", "double", "m", "Path length");
    columns_m.addColumn("azimuth", "double", "deg", "Azimuth in global coordinates");
    columns_m.addColumn("radius", "double", "mm", "Radius in global coordinates");
    columns_m.addColumn("numParticles", "long", "1", "Number of Macro Particles");
    columns_m.addColumn("energy", "double", "MeV", "Mean Bunch Energy");
    columns_m.addColumn("dE", "double", "MeV", "energy spread of the beam");
 
    columns_m.addColumn("rms_x", "double", "m", "RMS Beamsize in x");
    columns_m.addColumn("rms_y", "double", "m", "RMS Beamsize in y");
    columns_m.addColumn("rms_s", "double", "m", "RMS Beamsize in s");
 
    columns_m.addColumn("rms_px", "double", "1", "RMS Normalized Momenta in x");
    columns_m.addColumn("rms_py", "double", "1", "RMS Normalized Momenta in y");
    columns_m.addColumn("rms_ps", "double", "1", "RMS Normalized Momenta in s");
 
    columns_m.addColumn("emit_x", "double", "m", "Normalized Emittance x");
    columns_m.addColumn("emit_y", "double", "m", "Normalized Emittance y");
    columns_m.addColumn("emit_s", "double", "m", "Normalized Emittance s");
 
    columns_m.addColumn("mean_x", "double", "m", "Mean Beam Position in x");
    columns_m.addColumn("mean_y", "double", "m", "Mean Beam Position in y");
    columns_m.addColumn("mean_s", "double", "m", "Mean Beam Position in s");
 
    columns_m.addColumn("xpx", "double", "1", "RMS Correlation xpx");
    columns_m.addColumn("ypy", "double", "1", "RMS Correlation ypy");
    columns_m.addColumn("zpz", "double", "1", "RMS Correlation zpz");
 
    columns_m.addColumn("halo_x", "double", "1", "Halo in x");
    columns_m.addColumn("halo_y", "double", "1", "Halo in y");
    columns_m.addColumn("halo_z", "double", "1", "Halo in z");
 
    if ( mode_m == std::ios::app )
        return;
 
    OPALTimer::Timer simtimer;
    std::string dateStr(simtimer.date());
    std::string timeStr(simtimer.time());
 
    std::stringstream ss;
    ss << "Multi Bunch Statistics data '"
       << OpalData::getInstance()->getInputFn()
       << "' " << dateStr << " " << timeStr;
 
    this->addDescription(ss.str(), "multi bunch stat parameters");
 
    this->addDefaultParameters();
 
    this->addInfo("ascii", 1);
}
 
 
void MultiBunchDump::write(const PartBunchBase<double, 3>* /*beam*/,
                           const beaminfo_t& binfo) {
 
    if ( Ippl::myNode() > 0)
        return;
 
    fillHeader();
 
    this->open();
 
    this->writeHeader();
 
    columns_m.addColumnValue("t", binfo.time);
    columns_m.addColumnValue("s", binfo.pathlength);
    columns_m.addColumnValue("azimuth", binfo.azimuth);
    columns_m.addColumnValue("radius", binfo.radius);
    columns_m.addColumnValue("numParticles", binfo.nParticles);
    columns_m.addColumnValue("energy", binfo.ekin);
    columns_m.addColumnValue("dE", binfo.dEkin);
 
    columns_m.addColumnValue("rms_x", binfo.rrms[0]);
    columns_m.addColumnValue("rms_y", binfo.rrms[1]);
    columns_m.addColumnValue("rms_s", binfo.rrms[2]);
 
    columns_m.addColumnValue("rms_px", binfo.prms[0]);
    columns_m.addColumnValue("rms_py", binfo.prms[1]);
    columns_m.addColumnValue("rms_ps", binfo.prms[2]);
 
    columns_m.addColumnValue("emit_x", binfo.emit[0]);
    columns_m.addColumnValue("emit_y", binfo.emit[1]);
    columns_m.addColumnValue("emit_s", binfo.emit[2]);
 
    columns_m.addColumnValue("mean_x", binfo.mean[0]);
    columns_m.addColumnValue("mean_y", binfo.mean[1]);
    columns_m.addColumnValue("mean_s", binfo.mean[2]);
 
    columns_m.addColumnValue("xpx", binfo.correlation[0]);
    columns_m.addColumnValue("ypy", binfo.correlation[1]);
    columns_m.addColumnValue("zpz", binfo.correlation[2]);
 
    columns_m.addColumnValue("halo_x", binfo.halo[0]);
    columns_m.addColumnValue("halo_y", binfo.halo[1]);
    columns_m.addColumnValue("halo_z", binfo.halo[2]);
 
    this->writeRow();
 
    this->close();
}