Remove all GUI/visualization includes and code from tests

Author: dav-og

tests/regression/f3of/test_f3of_dt1.cpp

@@ -1,12 +1,9 @@
 #include <hydroc/helper.h>
 #include <hydroc/hydro_forces.h>
 
-#include <chrono/core/ChRealtimeStep.h>
 #include <chrono/physics/ChBodyEasy.h>
 #include <chrono/physics/ChSystemSMC.h>
 
-#include "chrono_postprocess/ChGnuPlot.h"
-
 #include <chrono>   // std::chrono::high_resolution_clock::now
 #include <iomanip>  // std::setprecision
 #include <vector>   // std::vector<double>
@@ -17,15 +14,8 @@ using namespace chrono;
 int main(int argc, char* argv[]) {
     std::cout << "Chrono version: " << CHRONO_VERSION << "\n\n";
 
-    // Parse CLI arguments and initialize environment
-    bool profilingOn     = true;
-    bool saveDataOn      = true;
-    bool plotOn          = true;
-    bool visualizationOn = true;
+    // Initialize environment
     std::string data_dir;
-    if (!hydroc::GetCLIArguments(argc, argv, "F3OF DT1 regression test", saveDataOn, profilingOn, plotOn,
-                                 visualizationOn, data_dir))
-        return 1;
     if (!hydroc::SetInitialEnvironment(data_dir)) return 1;
 
     // Get model file names
@@ -42,116 +32,98 @@ int main(int argc, char* argv[]) {
     system.SetGravitationalAcceleration(ChVector3d(0.0, 0.0, -9.81));
     double timestep = 0.02;
     system.SetSolverType(ChSolver::Type::SPARSE_QR);
-    ChRealtimeStepTimer realtime_timer;
     double simulationDuration = 300.0;
 
-    // some io/viz options
+    // Output timeseries
     std::vector<double> time_vector;
     std::vector<double> base_surge;
     std::vector<double> base_pitch;
     std::vector<double> fore_pitch;
     std::vector<double> aft_pitch;
 
-    // Output file names
-    std::string filename          = "CHRONO_F3OF_DT1_SURGE.txt";
-    std::string filename_duration = "CHRONO_F3OF_DT1_SURGE_DURATION.txt";
-
     // set up body from a mesh
     std::cout << "Attempting to open mesh file: " << body1_meshfame << std::endl;
-    std::shared_ptr<ChBody> base = chrono_types::make_shared<ChBodyEasyMesh>(  //
+    std::shared_ptr<ChBody> base = chrono_types::make_shared<ChBodyEasyMesh>(
         body1_meshfame,
         0,      // density
         false,  // do not evaluate mass automatically
         true,   // create visualization asset
         false   // collisions
     );
 
-    // Create a visualization material
-    auto red = chrono_types::make_shared<ChVisualMaterial>();
-    red->SetDiffuseColor(ChColor(0.3f, 0.1f, 0.1f));
-    base->GetVisualShape(0)->SetMaterial(0, red);
-
-    // define the base's initial conditions (position and rotation defined later for specific test)
+    // define the base's initial conditions
     system.Add(base);
     base->SetName("body1");
     base->SetMass(1089825.0);
     base->SetInertiaXX(ChVector3d(100000000.0, 76300000.0, 100000000.0));
 
     std::cout << "Attempting to open mesh file: " << body2_meshfame << std::endl;
-    std::shared_ptr<ChBody> flapFore = chrono_types::make_shared<ChBodyEasyMesh>(  //
+    std::shared_ptr<ChBody> flapFore = chrono_types::make_shared<ChBodyEasyMesh>(
         body2_meshfame,
         0,      // density
         false,  // do not evaluate mass automatically
         true,   // create visualization asset
         false   // collisions
     );
 
-    // Create a visualization material
-    auto blue = chrono_types::make_shared<ChVisualMaterial>();
-    blue->SetDiffuseColor(ChColor(0.3f, 0.1f, 0.6f));
-    flapFore->GetVisualShape(0)->SetMaterial(0, blue);
-
-    // define the fore flap's initial conditions (position and rotation defined later for specific tests
+    // define the fore flap's initial conditions
     system.Add(flapFore);
     flapFore->SetName("body2");
     flapFore->SetMass(179250.0);
     flapFore->SetInertiaXX(ChVector3d(100000000.0, 1300000.0, 100000000.0));
 
     std::cout << "Attempting to open mesh file: " << body3_meshfame << std::endl;
-    std::shared_ptr<ChBody> flapAft = chrono_types::make_shared<ChBodyEasyMesh>(  //
+    std::shared_ptr<ChBody> flapAft = chrono_types::make_shared<ChBodyEasyMesh>(
         body3_meshfame,
         0,      // density
         false,  // do not evaluate mass automatically
         true,   // create visualization asset
         false   // collisions
     );
 
-    // Create a visualization material
-    auto green = chrono_types::make_shared<ChVisualMaterial>();
-    green->SetDiffuseColor(ChColor(0.3f, 0.6f, 0.1f));
-    flapAft->GetVisualShape(0)->SetMaterial(0, green);
-
-    // define the aft flap's initial conditions (position and rotation defined later for specific tests
+    // define the aft flap's initial conditions
     system.Add(flapAft);
     flapAft->SetName("body3");
     flapAft->SetMass(179250.0);
     flapAft->SetInertiaXX(ChVector3d(100000000.0, 1300000.0, 100000000.0));
 
-    // ---------------- DT1 set up (surge decay, flaps locked, no waves) ------------------------------
-    // set up pos/rotations
+    // DT1 set up (surge decay, flaps locked, no waves)
     base->SetPos(ChVector3d(5.0, 0.0, -9.0));
     flapFore->SetPos(ChVector3d(5.0 + -12.5, 0.0, -9.0 + 3.5));
     flapAft->SetPos(ChVector3d(5.0 + 12.5, 0.0, -9.0 + 3.5));
+
     // set up revolute joints and lock them
     auto revoluteFore          = chrono_types::make_shared<ChLinkLockRevolute>();
     auto revoluteAft           = chrono_types::make_shared<ChLinkLockRevolute>();
-    ChQuaternion<> revoluteRot = QuatFromAngleX(CH_PI / 2.0);  // do not change
+    ChQuaternion<> revoluteRot = QuatFromAngleX(CH_PI / 2.0);
     revoluteFore->Initialize(base, flapFore, ChFramed(ChVector3d(5.0 - 12.5, 0.0, -9.0), revoluteRot));
     system.AddLink(revoluteFore);
     revoluteAft->Initialize(base, flapAft, ChFramed(ChVector3d(5.0 + 12.5, 0.0, -9.0), revoluteRot));
     system.AddLink(revoluteAft);
     revoluteFore->Lock(true);
     revoluteAft->Lock(true);
+
     // create ground
     auto ground = chrono_types::make_shared<ChBody>();
     system.AddBody(ground);
     ground->SetPos(ChVector3d(0, 0, -9.0));
     ground->SetTag(-1);
     ground->SetFixed(true);
     ground->EnableCollision(false);
+
     // add prismatic joint between the base and ground
     auto prismatic = chrono_types::make_shared<ChLinkLockPrismatic>();
     prismatic->Initialize(ground, base, ChFramed(ChVector3d(0.0, 0.0, -9.0), QuatFromAngleY(CH_PI_2)));
     system.AddLink(prismatic);
+
     // add damping to prismatic joint
     auto prismatic_pto = chrono_types::make_shared<ChLinkTSDA>();
     prismatic_pto->Initialize(ground, base, true, ChVector3d(0.0, 0.0, 0.0), ChVector3d(0.0, 0.0, 0.0));
     prismatic_pto->SetSpringCoefficient(1e5);
     prismatic_pto->SetRestLength(0.0);
     system.AddLink(prismatic_pto);
-    // ---------------- End DT specific set up, now add hydro forces ----------------------------------
 
-    // define wave parameters (not used in this demo TODO have hydroforces constructor without hydro inputs)
+    // define wave parameters
     auto default_dont_add_waves = std::make_shared<NoWave>(3);
 
     // set up hydro forces
@@ -179,53 +151,26 @@ int main(int argc, char* argv[]) {
     // for profiling
     auto end          = std::chrono::high_resolution_clock::now();
     unsigned duration = std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count();
-
-    std::string out_dir = hydroc::getTestOutDir();
-    if (profilingOn || saveDataOn) {
-        out_dir = out_dir + "/" + RESULTS_DIR_NAME;
-        std::filesystem::create_directory(std::filesystem::path(out_dir));
-    }
-
-    if (profilingOn) {
-        std::ofstream profilingFile(out_dir + "/" + RESULTS_FILE_NAME + "_duration.txt");
-        if (profilingFile.is_open()) {
-            profilingFile << duration << " ms\n";
-            profilingFile.close();
-        } else {
-            std::cout << "Error: Could not open profiling file for writing." << std::endl;
+    std::cout << "Simulation completed in " << duration << " ms" << std::endl;
+
+    // Save results
+    std::string out_dir = hydroc::getTestOutDir() + "/" + RESULTS_DIR_NAME;
+    std::filesystem::create_directories(std::filesystem::path(out_dir));
+
+    std::ofstream outputFile(out_dir + "/" + RESULTS_FILE_NAME + ".txt");
+    if (outputFile.is_open()) {
+        outputFile
+            << "Time (s)    Base Surge (m)Base Pitch (radians)Flap Fore Pitch (radians)Flap Aft Pitch (radians)"
+            << std::endl;
+        for (size_t i = 0; i < time_vector.size(); i++) {
+            outputFile << std::fixed << std::setprecision(4) << time_vector[i] << "                "
+                       << base_surge[i] << "         " << base_pitch[i] << "         " << fore_pitch[i]
+                       << "          " << aft_pitch[i] << std::endl;
         }
-    }
-
-    if (saveDataOn) {
-        std::ofstream outputFile(out_dir + "/" + RESULTS_FILE_NAME + ".txt");
-        if (outputFile.is_open()) {
-            outputFile
-                << "Time (s)    Base Surge (m)Base Pitch (radians)Flap Fore Pitch (radians)Flap Aft Pitch (radians)"
-                << std::endl;
-            for (size_t i = 0; i < time_vector.size(); i++) {
-                outputFile << std::fixed << std::setprecision(4) << time_vector[i] << "                "
-                           << base_surge[i] << "         " << base_pitch[i] << "         " << fore_pitch[i]
-                           << "          " << aft_pitch[i] << std::endl;
-            }
-            outputFile.close();
-        } else {
-            std::cout << "Error: Could not open output file for writing." << std::endl;
-            return 1;  // Return an error code
-        }
-    }
-
-    if (plotOn) {
-        postprocess::ChGnuPlot gplot(out_dir + "/f3of_dt1.gpl");
-        gplot.SetGrid();
-        gplot.SetLabelX("time (s)");
-        gplot.SetLabelY("surge");
-        gplot.SetLabelY2("pitch");
-        gplot.SetRangeX(0, simulationDuration);
-        gplot.SetTitle("F3OF DT3");
-        gplot.Plot(time_vector, base_surge, "base surge", " with lines lw 2");
-        gplot.Plot(time_vector, base_pitch, "base pitch", " axes x1y2  with lines lw 2");
-        gplot.Plot(time_vector, fore_pitch, "fore pitch", " axes x1y2  with lines lw 2");
-        gplot.Plot(time_vector, aft_pitch, "aft pitch", " axes x1y2  with lines lw 2");
+        outputFile.close();
+    } else {
+        std::cout << "Error: Could not open output file for writing." << std::endl;
+        return 1;
     }
 
     return 0;