Forcing the output file (if any) to close when finishing the bencnhmarks

Author: Nacho Valladares

include/celero/ResultTable.h

@@ -45,6 +45,10 @@ namespace celero
 		///
 		void setFileName(const std::string& x);
 
+        ///
+        /// Force the output file (if any) to close
+        ///
+        void closeFile();
 		///
 		/// Add a new result to the result table.
 		///

src/Celero.cpp

@@ -139,13 +139,15 @@ void celero::Run(int argc, char** argv)
 	}
 
 	// Has a result output file been specified?
+    bool must_close_file = false;
 	auto argument = args.get<std::string>("outputTable");
 	if(argument.empty() == false)
 	{
 		std::cout << "Writing results to: " << argument << std::endl;
 		celero::ResultTable::Instance().setFileName(argument);
 
 		celero::AddExperimentResultCompleteFunction([](std::shared_ptr<celero::Result> p) { celero::ResultTable::Instance().add(p); });
+        must_close_file = true;
 	}
 
 	// Has a result output file been specified?
@@ -185,6 +187,9 @@ void celero::Run(int argc, char** argv)
 		executor::RunAll();
 	}
 
+    if (must_close_file) {
+        celero::ResultTable::Instance().closeFile();
+    }
 	// Final output.
 	std::cout << "Complete.\n";
 }

src/ResultTable.cpp

@@ -38,6 +38,16 @@ class celero::ResultTable::Impl
 	{
 	}
 
+    ~Impl() {
+        closeFile();
+    }
+
+    void closeFile() {
+        if (this->ofs.is_open() == true) {
+            this->ofs.close();
+        }
+    }
+
 	void setFileName(const std::string& x)
 	{
 		if(this->ofs.is_open() == true)
@@ -88,6 +98,10 @@ void ResultTable::setFileName(const std::string& x)
 	this->pimpl->setFileName(x);
 }
 
+void ResultTable::closeFile() {
+    this->pimpl->closeFile();
+}
+
 void ResultTable::add(std::shared_ptr<Result> x)
 {
 	if(this->pimpl->ofs.is_open() == true)