set attributes

Hi-PACE · Dec 17, 2023 · 11b9188 · 11b9188
1 parent 15a6818
commit 11b9188
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 13 deletions.
diff --git a/src/Hipace.cpp b/src/Hipace.cpp
@@ -995,13 +995,13 @@ Hipace::WriteDiagnostics (const int step)
 #ifdef HIPACE_USE_OPENPMD
     if (m_diags.hasAnyFieldOutput(step, m_max_step, m_physical_time, m_max_time)) {
         m_openpmd_writer.WriteDiagnostics(m_diags.getFieldData(), m_multi_beam,
-                        m_physical_time, step, getDiagBeamNames(),
+                        m_multi_laser, m_physical_time, step, getDiagBeamNames(),
                         m_3D_geom, OpenPMDWriterCallType::fields);
     }
 
     if (m_diags.hasBeamOutput(step, m_max_step, m_physical_time, m_max_time)) {
         m_openpmd_writer.WriteDiagnostics(m_diags.getFieldData(), m_multi_beam,
-                        m_physical_time, step, getDiagBeamNames(),
+                        m_multi_laser, m_physical_time, step, getDiagBeamNames(),
                         m_3D_geom, OpenPMDWriterCallType::beams);
     }
 #else

diff --git a/src/diagnostics/OpenPMDWriter.H b/src/diagnostics/OpenPMDWriter.H
@@ -67,9 +67,11 @@ private:
     /** \brief writing openPMD field data
      *
      * \param[in] fd field diagnostic data
+     * \param[in] a_multi_laser multi laser to get the central wavelength
      * \param[in,out] iteration openPMD iteration to which the data is written
      */
-    void WriteFieldData (const FieldDiagnosticData& fd, openPMD::Iteration iteration);
+    void WriteFieldData (const FieldDiagnosticData& fd, const MultiLaser& a_multi_laser,
+                         openPMD::Iteration iteration);
 
     /** Named Beam SoA attributes per particle as defined in BeamIdx
      */
@@ -124,7 +126,7 @@ public:
      */
     void WriteDiagnostics (
         const amrex::Vector<FieldDiagnosticData>& field_diag, MultiBeam& a_multi_beam,
-        const amrex::Real physical_time, const int output_step,
+        const MultiLaser& a_multi_laser, const amrex::Real physical_time, const int output_step,
         const amrex::Vector< std::string > beamnames,
         amrex::Vector<amrex::Geometry> const& geom3D,
         const OpenPMDWriterCallType call_type);

diff --git a/src/diagnostics/OpenPMDWriter.cpp b/src/diagnostics/OpenPMDWriter.cpp
@@ -64,7 +64,7 @@ OpenPMDWriter::InitDiagnostics ()
 void
 OpenPMDWriter::WriteDiagnostics (
     const amrex::Vector<FieldDiagnosticData>& field_diag, MultiBeam& a_multi_beam,
-    const amrex::Real physical_time, const int output_step,
+    const MultiLaser& a_multi_laser, const amrex::Real physical_time, const int output_step,
     const amrex::Vector< std::string > beamnames,
     amrex::Vector<amrex::Geometry> const& geom3D,
     const OpenPMDWriterCallType call_type)
@@ -77,14 +77,15 @@ OpenPMDWriter::WriteDiagnostics (
     } else if (call_type == OpenPMDWriterCallType::fields) {
         for (const auto& fd : field_diag) {
             if (fd.m_has_field) {
-                WriteFieldData(fd.m_F, fd.m_geom_io, fd.m_slice_dir, fd.m_comps_output, iteration);
+                WriteFieldData(fd, a_multi_laser, iteration);
             }
         }
     }
 }
 
 void
-OpenPMDWriter::WriteFieldData (const FieldDiagnosticData& fd, openPMD::Iteration iteration)
+OpenPMDWriter::WriteFieldData (
+    const FieldDiagnosticData& fd, const MultiLaser& a_multi_laser, openPMD::Iteration iteration)
 {
     HIPACE_PROFILE("OpenPMDWriter::WriteFieldData()");
 
@@ -94,13 +95,13 @@ OpenPMDWriter::WriteFieldData (const FieldDiagnosticData& fd, openPMD::Iteration
     amrex::Vector<std::string> varnames = fd.m_comps_output;
 
     if (fd.m_do_laser) {
-        varnames.push_back("laser_envelope");
+        varnames.push_back("laserEnvelope");
     }
 
     // loop over field components
     for ( int icomp = 0; icomp < varnames.size(); ++icomp )
     {
-        const bool is_laser_comp = varnames[icomp] == "laser_envelope";
+        const bool is_laser_comp = varnames[icomp] == "laserEnvelope";
 
         //                      "B"                "x" (todo)
         //                      "Bx"               ""  (just for now)
@@ -123,8 +124,8 @@ OpenPMDWriter::WriteFieldData (const FieldDiagnosticData& fd, openPMD::Iteration
         const amrex::IntVect box_offset {0, 0, data_box.smallEnd(2) - geom.Domain().smallEnd(2)};
         openPMD::Offset chunk_offset = utils::getReversedVec(box_offset);
         openPMD::Extent chunk_size = utils::getReversedVec(data_box.size());
-        if (d.m_slice_dir >= 0) {
-            const int remove_dir = 2 - d.m_slice_dir;
+        if (fd.m_slice_dir >= 0) {
+            const int remove_dir = 2 - fd.m_slice_dir;
             // User requested slice IO
             // remove the slicing direction in position, label, resolution, offset
             relative_cell_pos.erase(relative_cell_pos.begin() + remove_dir);
@@ -141,14 +142,20 @@ OpenPMDWriter::WriteFieldData (const FieldDiagnosticData& fd, openPMD::Iteration
         field.setGridGlobalOffset(offWindow);
 
         openPMD::Datatype datatype = is_laser_comp ?
-            openPMD::determineDatatype< amrex::GpuComplex<amrex::Real> >() :
+            openPMD::determineDatatype< std::complex<amrex::Real> >() :
             openPMD::determineDatatype< amrex::Real >();
         // set data type and global size of the simulation
         openPMD::Dataset dataset(datatype, global_size);
         field_comp.resetDataset(dataset);
 
         if (is_laser_comp) {
-            field_comp.storeChunkRaw(fd.m_F_laser.dataPtr(icomp), chunk_offset, chunk_size);
+            // set laser attributes and store laser
+            field.setAttribute("envelopeField", "normalized_vector_potential");
+            field.setAttribute("angularFrequency",
+                double(2.) * MathConst::pi * PhysConstSI::c / a_multi_laser.GetLambda0());
+            std::vector< std::complex<double> > polarization {{1., 0.}, {0., 0.}};
+            field.setAttribute("polarization", polarization);
+            field_comp.storeChunkRaw(fd.m_F_laser.dataPtr(), chunk_offset, chunk_size);
         } else {
             field_comp.storeChunkRaw(fd.m_F.dataPtr(icomp), chunk_offset, chunk_size);
         }

diff --git a/src/laser/MultiLaser.H b/src/laser/MultiLaser.H
@@ -192,6 +192,9 @@ public:
      */
     void InitLaserSlice (const amrex::Geometry& geom, const int islice, const int comp);
 
+    /** Get the central wavelength */
+    amrex::Real GetLambda0 () const { return m_lambda0; }
+
     bool m_use_laser {false}; /**< whether a laser is used or not */
 
 private: