From: Wolfgang Bangerth <bangerth@colostate.edu>
Date: Wed, 8 Nov 2017 03:56:39 +0000 (-0700)
Subject: Output both real and imaginary parts of the solution.
X-Git-Tag: v9.0.0-rc1~798^2~7
X-Git-Url: https://gitweb.dealii.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ac7940ad652c348cfd766ce6f46f2ee1751f1902;p=dealii.git

Output both real and imaginary parts of the solution.
---

diff --git a/source/numerics/data_out.cc b/source/numerics/data_out.cc
index b485c6936f..f8a3a827ae 100644
--- a/source/numerics/data_out.cc
+++ b/source/numerics/data_out.cc
@@ -226,23 +226,50 @@ build_one_patch
             // reasons.
             if (n_components == 1)
               {
+                // first output the real part of the solution vector
                 this->dof_data[dataset]->get_function_values (this_fe_patch_values,
                                                               internal::DataOut::ComponentExtractor::real_part,
                                                               scratch_data.patch_values_scalar.solution_values);
                 for (unsigned int q=0; q<n_q_points; ++q)
                   patch.data(offset,q) = scratch_data.patch_values_scalar.solution_values[q];
+
+                // and if there is one, also output the imaginary part
+                if (this->dof_data[dataset]->is_complex_valued() == true)
+                  {
+                    this->dof_data[dataset]->get_function_values (this_fe_patch_values,
+                                                                  internal::DataOut::ComponentExtractor::imaginary_part,
+                                                                  scratch_data.patch_values_scalar.solution_values);
+                    for (unsigned int q=0; q<n_q_points; ++q)
+                      patch.data(offset+1,q) = scratch_data.patch_values_scalar.solution_values[q];
+                  }
               }
             else
               {
                 scratch_data.resize_system_vectors(n_components);
+
+                // same as above: first the real part
+                const unsigned int stride = (this->dof_data[dataset]->is_complex_valued() ? 2 : 1);
                 this->dof_data[dataset]->get_function_values (this_fe_patch_values,
                                                               internal::DataOut::ComponentExtractor::real_part,
                                                               scratch_data.patch_values_system.solution_values);
                 for (unsigned int component=0; component<n_components;
                      ++component)
                   for (unsigned int q=0; q<n_q_points; ++q)
-                    patch.data(offset+component,q) =
+                    patch.data(offset+component*stride,q) =
                       scratch_data.patch_values_system.solution_values[q](component);
+
+                // and if there is one, also output the imaginary part
+                if (this->dof_data[dataset]->is_complex_valued() == true)
+                  {
+                    this->dof_data[dataset]->get_function_values (this_fe_patch_values,
+                                                                  internal::DataOut::ComponentExtractor::imaginary_part,
+                                                                  scratch_data.patch_values_system.solution_values);
+                    for (unsigned int component=0; component<n_components;
+                         ++component)
+                      for (unsigned int q=0; q<n_q_points; ++q)
+                        patch.data(offset+component*stride+1,q) =
+                          scratch_data.patch_values_system.solution_values[q](component);
+                  }
               }
 
           // increment the counter for the actual data record