Compute jxb_gradp (and related quantities) even if VMEC did not converge.

Author: jons-pf

src/vmecpp/cpp/vmecpp/vmec/output_quantities/output_quantities.cc

@@ -1250,9 +1250,11 @@ vmecpp::OutputQuantities vmecpp::ComputeOutputQuantities(
   output_quantities.vmec_internal_results = GatherDataFromThreads(
       sign_of_jacobian, s, fc, constants, radial_partitioning, decomposed_x,
       models_from_threads, radial_profiles);
+  output_quantities.vmec_internal_results.return_outputs_even_if_not_converged = indata.return_outputs_even_if_not_converged;
 
   if (vmec_status == VmecStatus::NORMAL_TERMINATION ||
-      vmec_status == VmecStatus::SUCCESSFUL_TERMINATION) {
+      vmec_status == VmecStatus::SUCCESSFUL_TERMINATION ||
+      output_quantities.vmec_internal_results.return_outputs_even_if_not_converged) {
     MeshBledingBSubZeta(
         s, fc,
         /*m_vmec_internal_results=*/output_quantities.vmec_internal_results);
@@ -2599,7 +2601,7 @@ vmecpp::JxBOutFileContents vmecpp::ComputeJxBOutputFileContents(
     jxbout.jperp2[jF] = dnorm1 * tjnorm * average_jperp2;
 
     // Some quantities are only computed if VMEC++ actually converged.
-    if (vmec_status == VmecStatus::SUCCESSFUL_TERMINATION) {
+    if (vmec_status == VmecStatus::SUCCESSFUL_TERMINATION || vmec_internal_results.return_outputs_even_if_not_converged) {
       // normalized toroidal magnetic flux
       jxbout.phin[jF] = vmec_internal_results.phiF[jF] /
                         vmec_internal_results.phiF[fc.ns - 1];
@@ -2636,7 +2638,7 @@ vmecpp::JxBOutFileContents vmecpp::ComputeJxBOutputFileContents(
   }  // jF
 
   // Some quantities are only computed if VMEC++ actually converged.
-  if (vmec_status == VmecStatus::SUCCESSFUL_TERMINATION) {
+  if (vmec_status == VmecStatus::SUCCESSFUL_TERMINATION || vmec_internal_results.return_outputs_even_if_not_converged) {
     // The loop in jxbforce.f90:594 goes over js=2,ns1,
     // which means that the last half-grid point is not touched.
     for (int jH = 0; jH < vmec_internal_results.num_half - 1; ++jH) {

src/vmecpp/cpp/vmecpp/vmec/output_quantities/output_quantities.h

@@ -25,6 +25,11 @@ namespace vmecpp {
 // This is the data from inside VMEC, gathered from all threads,
 // that form the basis of computing the output quantities.
 struct VmecInternalResults {
+  // copy of corresponding VmecINDATA variable
+  // if true, compute full outputs even if VMEC did not converge
+  // defaults to false
+  bool return_outputs_even_if_not_converged;
+
   int sign_of_jacobian;
 
   // total number of full-grid points