Output element coordinates in csv

Author: gyalla

libmfpic/ParticleOperations.cpp

@@ -291,15 +291,25 @@ void ParticleOperations::writeMomentsCSVsPerSpecies(const ParticleContainer& par
     out << std::setprecision(17);
 
     if (need_header) {
-      out << "step,time,elem,number_density,temperature,bulk_velocity_0,bulk_velocity_1,bulk_velocity_2\n";
+      out << "step,time,elem,x,y,z,number_density,temperature,bulk_velocity_0,bulk_velocity_1,bulk_velocity_2\n";
     }
 
     for (int e = 0; e < nelem; ++e) {
+      mfem::ElementTransformation *element_transform = mesh.GetElementTransformation(e);
+      mfem::IntegrationPoint ip = mfem::IntRules.Get(mesh.GetElementBaseGeometry(e), 1).IntPoint(0);
+      element_transform->SetIntPoint(&ip);
+      mfem::Vector element_point(3);
+      element_point = 0.0;
+      const int dim = mesh.SpaceDimension();
+      mfem::Vector element_point_view(element_point.GetData(), dim);   
+      element_transform->Transform(ip, element_point_view);
+
       const mfem::Vector bulk_velocity_in_element(particle_bulk_velocity_(s).GetColumn(e), 3);
 
       out << step << ","
           << time << ","
           << e << ","
+          << element_point(0) << "," << element_point(1) << "," << element_point(2) << ","
           << particle_number_density_(e, s) << ","
           << particle_temperature_(e, s) << ","
           << bulk_velocity_in_element(0) << "," << bulk_velocity_in_element(1) << "," << bulk_velocity_in_element(2) << "\n";

libmfpic/ParticleOperations.test.cpp

@@ -1166,17 +1166,17 @@ TEST(ParticleOperationsCSV, CSVCorrectForOneSpecies)
       for (int e = 0; e < num_elems; ++e) {
         const std::string& line = lines[1 + step*num_elems + e];
         auto toks = splitCsv(line);
-        ASSERT_EQ(toks.size(), static_cast<size_t>(8));
+        ASSERT_EQ(toks.size(), static_cast<size_t>(11));
 
         EXPECT_EQ(stringToInt(toks[0]), step);
         EXPECT_DOUBLE_EQ(stringToDouble(toks[1]), dt*step);
         EXPECT_EQ(stringToInt(toks[2]), e);
 
-        const double csv_number_density = stringToDouble(toks[3]);
-        const double csv_temperature = stringToDouble(toks[4]);
-        const double csv_bulk_velocity_0 = stringToDouble(toks[5]);
-        const double csv_bulk_velocity_1 = stringToDouble(toks[6]);
-        const double csv_bulk_velocity_2 = stringToDouble(toks[7]);
+        const double csv_number_density = stringToDouble(toks[6]);
+        const double csv_temperature = stringToDouble(toks[7]);
+        const double csv_bulk_velocity_0 = stringToDouble(toks[8]);
+        const double csv_bulk_velocity_1 = stringToDouble(toks[9]);
+        const double csv_bulk_velocity_2 = stringToDouble(toks[10]);
 
         EXPECT_NEAR(csv_number_density, computed_number_density(e, s), 1e-12);
         EXPECT_NEAR(csv_temperature,  computed_temperature(e, s),  1e-12);
@@ -1262,17 +1262,17 @@ TEST(ParticleOperationsCSV, CSVCorrectForTwoSpecies)
       for (int e = 0; e < num_elems; ++e) {
         const std::string& line = lines[1 + step*num_elems + e];
         auto toks = splitCsv(line);
-        ASSERT_EQ(toks.size(), static_cast<size_t>(8));
+        ASSERT_EQ(toks.size(), static_cast<size_t>(11));
 
         EXPECT_EQ(stringToInt(toks[0]), step);
         EXPECT_DOUBLE_EQ(stringToDouble(toks[1]), dt*step);
         EXPECT_EQ(stringToInt(toks[2]), e);
 
-        const double csv_number_density = stringToDouble(toks[3]);
-        const double csv_temperature = stringToDouble(toks[4]);
-        const double csv_bulk_velocity_0 = stringToDouble(toks[5]);
-        const double csv_bulk_velocity_1 = stringToDouble(toks[6]);
-        const double csv_bulk_velocity_2 = stringToDouble(toks[7]);
+        const double csv_number_density = stringToDouble(toks[6]);
+        const double csv_temperature = stringToDouble(toks[7]);
+        const double csv_bulk_velocity_0 = stringToDouble(toks[8]);
+        const double csv_bulk_velocity_1 = stringToDouble(toks[9]);
+        const double csv_bulk_velocity_2 = stringToDouble(toks[10]);
 
         EXPECT_NEAR(csv_number_density, computed_number_density(e, s), 1e-12);
         EXPECT_NEAR(csv_temperature,  computed_temperature(e, s),  1e-12);