add FTP, RSP, and FTK results to the output file

1 files changed, 32 insertions, 0 deletions

diff --git a/src/fit.c b/src/fit.c
index adaaaef..4e04c0f 100644
--- a/src/fit.c
+++ b/src/fit.c
@@ -5872,6 +5872,9 @@ int main(int argc, char **argv)
     zebraBank bmast, bmc, bmcgn, mctk, b;
     int rv;
     EVBank bev;
+    FTPVBank bftpv;
+    FTXKBank bftxk;
+    RSPBank bftxr;
     MCTKBank bmctk;
     MCVXBank bmcvx;
     event ev = {0};
@@ -6192,6 +6195,35 @@ skip_mc:
                 fprintf(fout, "        dc: 0x%08x\n", get_dc_word(&ev, f, &bmast, &b));
             }
 
+            if (fout) {
+                if (get_ftpv(f,&b,&bftpv)) {
+                    fprintf(stderr, "%s\n", zdab_err);
+                } else {
+                    fprintf(fout, "        ftp:\n");
+                    fprintf(fout, "            x: %.2f\n", bftpv.x);
+                    fprintf(fout, "            y: %.2f\n", bftpv.y);
+                    fprintf(fout, "            z: %.2f\n", bftpv.z);
+                }
+            }
+
+            if (fout) {
+                if (get_ftxk(f,&b,&bftxk)) {
+                    fprintf(stderr, "%s\n", zdab_err);
+                } else {
+                    fprintf(fout, "        ftk:\n");
+                    fprintf(fout, "            energy: %.2f\n", bftxk.energy);
+                }
+            }
+
+            if (fout) {
+                if (get_rsp(f,&b,&bftxr)) {
+                    fprintf(stderr, "%s\n", zdab_err);
+                } else {
+                    fprintf(fout, "        rsp:\n");
+                    fprintf(fout, "            energy: %.2f\n", bftxr.ene);
+                }
+            }
+
             if (nhit < min_nhit) goto skip_event;
 
             if (fout) fprintf(fout, "        fit:\n");