From 247f205a44692cc5cb4c94b728c549ba437d2fa8 Mon Sep 17 00:00:00 2001
From: chn <chn@chn.moe>
Date: Thu, 19 Sep 2024 01:35:40 +0800
Subject: [PATCH] packages.ufo: add OutputRawDataFile

---
 packages/ufo/src/plot.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/ufo/src/plot.cpp b/packages/ufo/src/plot.cpp
index d1faa19c..c756cdc2 100644
--- a/packages/ufo/src/plot.cpp
+++ b/packages/ufo/src/plot.cpp
@@ -307,8 +307,10 @@ void ufo::plot_point(std::string config_file)
     std::optional<std::vector<std::pair<double, std::string>>> XTicks;
     // 是否输出图片
     std::optional<std::string> OutputPictureFile;
-    // 是否输出数据，可以进一步使用 matplotlib 画图
+    // 是否输出插值后数据，可以进一步使用 matplotlib 画图
     std::optional<std::string> OutputDataFile;
+    // 是否输出插值前数据，可以配合 phonopy 结果深入研究
+    std::optional<std::string> OutputRawDataFile;
   };
 
   // 根据 q 点路径, 搜索要使用的 q 点，返回的是 q 点在 QpointData 中的索引
@@ -424,4 +426,6 @@ void ufo::plot_point(std::string config_file)
       .write("XTickLabels", x_ticklabels)
       .write("InterpolationResolution", input.InterpolationResolution)
       .write("FrequencyRange", input.FrequencyRange);
+  if (input.OutputRawDataFile)
+    std::ofstream(*input.OutputRawDataFile) << YAML::Node(unfolded_data.QpointData[qpoint_index]);
 }