DumpEMFields.h

Go to the documentation of this file.

//
// Class DumpEMFields
//   DumpEMFields dumps the dynamically changing fields of a Ring in a user-
//   defined grid.
//
// Copyright (c) 2017, Chris Rogers
// All rights reserved
//
// 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/>.
//
#ifndef OPAL_BASICACTIONS_DUMPEMFIELDS_HH
#define OPAL_BASICACTIONS_DUMPEMFIELDS_HH

#include "AbsBeamline/Component.h"
#include "AbstractObjects/Action.h"

#include <string>
#include <unordered_set>
#include <vector>

namespace interpolation {
  class NDGrid;
}
class Component;

class DumpEMFields: public Action {

public:
    enum {
        FILE_NAME,
        COORDINATE_SYSTEM,
        X_START,
        DX,
        X_STEPS,
        Y_START,
        DY,
        Y_STEPS,
        Z_START,
        DZ,
        Z_STEPS,
        T_START,
        DT,
        T_STEPS,
        R_START,
        DR,
        R_STEPS,
        PHI_START,
        DPHI,
        PHI_STEPS,
        SIZE
    };

    DumpEMFields();

    DumpEMFields(const std::string& name, DumpEMFields* parent);

    virtual ~DumpEMFields();

    virtual DumpEMFields* clone(const std::string& name);

    virtual void execute();

    static void writeFields(Component* field);

    void print(std::ostream& os) const;

private:

    enum class CoordinateSystem: unsigned short {
        CARTESIAN,
        CYLINDRICAL
    };

    virtual void writeFieldThis(Component* field);
    virtual void buildGrid();
    void parseCoordinateSystem();
    void writeHeader(std::ofstream& fout) const;
    void writeFieldLine(Component* field,
                        const Vector_t& point,
                        const double& time,
                        std::ofstream& fout) const;

    interpolation::NDGrid* grid_m;
    std::string filename_m;

    CoordinateSystem coordinates_m = CoordinateSystem::CARTESIAN;

    static std::unordered_set<DumpEMFields*> dumpsSet_m;

    DumpEMFields(const DumpEMFields& dump);  // disabled
    DumpEMFields& operator=(const DumpEMFields& dump);  // disabled
};

inline std::ostream& operator<<(std::ostream& os, const DumpEMFields& b) {
    b.print(os);
    return os;
}

#endif // ifdef OPAL_DUMPFIELDS_HH