improving cli application

Signed-off-by: Guillaume W. Bres <[email protected]>

Author: gwbres

cggtts-cli/src/main.rs

@@ -6,7 +6,9 @@ use plot::PlotContext;
 
 mod processing;
 
+use std::io::Write;
 use std::path::Path;
+use std::process::Command;
 
 #[macro_use]
 extern crate log;
@@ -16,9 +18,7 @@ use walkdir::WalkDir;
 
 use cggtts::prelude::CGGTTS;
 
-use std::io::Write;
-
-use std::process::Command;
+use itertools::Itertools;
 
 #[cfg(target_os = "linux")]
 pub fn open_with_web_browser(path: &str) {
@@ -109,16 +109,30 @@ pub fn main() {
 
     if cli.identification() {
         for p in pool {
-            info!("station: {}", p.station);
-            info!("number of tracks: {}", p.tracks.len());
+            info!("STATION          \"{}\"", p.station);
+            info!("NUMBER OF TRACKS  {}", p.tracks.len());
+            info!(
+                "CODES            {:?}",
+                p.tracks()
+                    .map(|trk| trk.frc.clone())
+                    .unique()
+                    .collect::<Vec<_>>()
+            );
+            info!(
+                "SV               {:?}",
+                p.tracks()
+                    .map(|trk| trk.sv.to_string())
+                    .unique()
+                    .collect::<Vec<_>>()
+            );
         }
         return;
     }
 
     if pool.len() == 1 {
         processing::single_clock(&pool[0], &mut plot_ctx);
     } else {
-        processing::clock_comparison(&pool, &mut plot_ctx);
+        processing::clock_comparison(&workspace_path, &pool, &mut plot_ctx);
     }
 
     /*

cggtts-cli/src/processing.rs

@@ -2,6 +2,9 @@ use cggtts::prelude::{Epoch, CGGTTS};
 use itertools::Itertools;
 use plotly::common::Mode;
 use std::collections::HashMap;
+use std::fs::File;
+use std::io::Write;
+use std::path::Path;
 
 use crate::plot::{
     //build_timedomain_plot,
@@ -180,7 +183,7 @@ pub fn single_clock(cggtts: &CGGTTS, ctx: &mut PlotContext) {
     }
 }
 
-pub fn clock_comparison(pool: &Vec<CGGTTS>, ctx: &mut PlotContext) {
+pub fn clock_comparison(workspace: &Path, pool: &Vec<CGGTTS>, ctx: &mut PlotContext) {
     let ref_clock = &pool[0];
     info!("{} is considered reference clock", ref_clock.station);
 
@@ -241,4 +244,42 @@ pub fn clock_comparison(pool: &Vec<CGGTTS>, ctx: &mut PlotContext) {
             }
         }
     }
+
+    let mut fd = File::create(workspace.join(&pool[0].station))
+        .expect("failed to create textfile: permission denied");
+
+    writeln!(fd, "{}", "t, CLOCK(A), CLOCK(B), SV, (elev[°], azi[°]) @REF, (elev[°], azi[°]) @CLOCK, SIGNAL, CLOCK(A) - CLOCK(B) [s]")
+        .expect("failed to generate textfile");
+
+    for trk in ref_clock.tracks() {
+        let ref_t = trk.epoch;
+        let ref_sv = trk.sv;
+        let (ref_elev, ref_azim) = (trk.elevation, trk.azimuth);
+        let ref_frc = &trk.frc;
+        for i in 1..pool.len() {
+            let track = pool[i]
+                .tracks()
+                .filter(|trk| trk.epoch == ref_t && trk.sv == ref_sv && trk.frc == *ref_frc)
+                .reduce(|trk, _| trk);
+            if let Some(b_trk) = track {
+                let (b_elev, b_azim) = (b_trk.elevation, b_trk.azimuth);
+                let dt = b_trk.data.refsys - trk.data.refsys;
+                writeln!(
+                    fd,
+                    "{:?}, {}, {}, {}, ({:.2E}, {:.2E}), ({:.2E}, {:.2E}), {}, {:.3E}",
+                    ref_t,
+                    pool[i].station,
+                    pool[0].station,
+                    ref_sv,
+                    ref_elev,
+                    ref_azim,
+                    b_elev,
+                    b_azim,
+                    ref_frc,
+                    dt
+                )
+                .expect("failed to generate textfile");
+            }
+        }
+    }
 }