From 8931f04037100e349027bf85523b83dbe61a8aaf Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Tue, 30 Nov 2021 19:05:17 -0500
Subject: [PATCH] redirect `print_summary` to LAMMPS log

This commit redirects `print_summary` output to LAMMPS log function, so
it will be also shown in `log.lammps`.
---
 source/lmp/pair_deepmd.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/source/lmp/pair_deepmd.cpp b/source/lmp/pair_deepmd.cpp
index 5cb058e1e8..eec4a503b5 100644
--- a/source/lmp/pair_deepmd.cpp
+++ b/source/lmp/pair_deepmd.cpp
@@ -267,6 +267,12 @@ void
 PairDeepMD::print_summary(const string pre) const
 {
   if (comm->me == 0){
+    // capture cout to a string, then call LAMMPS's utils::logmesg
+    // https://stackoverflow.com/a/4043813/9567349
+    std::stringstream buffer;
+    std::streambuf *sbuf = std::cout.rdbuf();
+    std::cout.rdbuf(buffer.rdbuf());
+
     cout << "Summary of lammps deepmd module ..." << endl;
     cout << pre << ">>> Info of deepmd-kit:" << endl;
     deep_pot.print_summary(pre);
@@ -279,6 +285,9 @@ PairDeepMD::print_summary(const string pre) const
     cout << pre << "build float prec:   " << STR_FLOAT_PREC << endl;
     cout << pre << "build with tf inc:  " << STR_TensorFlow_INCLUDE_DIRS << endl;
     cout << pre << "build with tf lib:  " << STR_TensorFlow_LIBRARY << endl;
+
+    std::cout.rdbuf(sbuf);
+    utils::logmesg(lmp, buffer.str());
   }
 }