FieldProbe.H

Go to the documentation of this file.

/* Copyright 2021 Lorenzo Giacomel, Neil Zaim, Yinjian Zhao
 *                Elisa Rheaume, Axel Huebl
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
 
#ifndef WARPX_DIAGNOSTICS_REDUCEDDIAGS_FIELDPROBE_H_
#define WARPX_DIAGNOSTICS_REDUCEDDIAGS_FIELDPROBE_H_
 
#include "ReducedDiags.H"
#include "FieldProbeParticleContainer.H"
 
#include <AMReX.H>
#include <AMReX_Vector.H>
 
#include <unordered_map>
#include <string>
#include <vector>

enum struct DetectorGeometry
{
    Point = 0,
    Line,
    Plane
};

class FieldProbe : public ReducedDiags
{
public:

    FieldProbe (const std::string& rd_name);

    void InitData () final;

    void LoadBalance () final;

    void ComputeDiags (int step) final;
 
    /*
     * Define constants used throughout FieldProbe
     */

    static const int noutputs = 11;
 
private:
    amrex::Real x_probe = 0.;
    amrex::Real y_probe = 0.;
    amrex::Real x1_probe = 0.;
    amrex::Real y1_probe = 0.;
    amrex::Real target_normal_x = 0.;
    amrex::Real target_normal_y = 1.;
    amrex::Real target_normal_z = 0.;
    amrex::Real target_up_x = 0.;
    amrex::Real target_up_y = 0.;
    amrex::Real target_up_z = 1.;
    amrex::Real z_probe, z1_probe;
    amrex::Real detector_radius;

    long m_valid_particles {0};

    int m_last_compute_step = 0;

    DetectorGeometry m_probe_geometry = DetectorGeometry::Point;

    int m_resolution = 0;

    amrex::Vector<amrex::Real> m_data;

    amrex::Vector<amrex::Real> m_data_out;

    FieldProbeParticleContainer m_probe;

    bool m_field_probe_integrate = false;

    int interp_order = 1;

    bool do_moving_window_FP = false;

    void WriteToFile (int step) const override;

    bool ProbeInDomain () const;

    void normalize(amrex::Real &AMREX_RESTRICT x, amrex::Real &AMREX_RESTRICT y,
                   amrex::Real &AMREX_RESTRICT z){
        const amrex::Real mag = std::sqrt(x*x + y*y + z*z);
        x /= mag;
        y /= mag;
        z /= mag;
    }
};
 
#endif // WARPX_DIAGNOSTICS_REDUCEDDIAGS_FIELDPROBE_H_