Add action to export analysis results from DB to files.

Issue #20

src/analyse.rs

@@ -482,6 +482,13 @@ pub fn read_tags(db_path: &str, mpaths: &Vec<PathBuf>) {
     db.close();
 }
 
+pub fn export(db_path: &str, mpaths: &Vec<PathBuf>) {
+    let db = db::Db::new(&String::from(db_path));
+    db.init();
+    db.export(&mpaths);
+    db.close();
+}
+
 pub fn update_ignore(db_path: &str, ignore_path: &PathBuf) {
     let file = File::open(ignore_path).unwrap();
     let reader = BufReader::new(file);

src/db.rs

@@ -29,6 +29,11 @@ pub struct FileMetadata {
     pub duration: u32,
 }
 
+struct AnalysisResults {
+    pub file: String,
+    pub analysis: Analysis,
+}
+
 #[derive(Default, PartialEq)]
 pub struct Metadata {
     pub title: String,
@@ -338,4 +343,50 @@ impl Db {
             }
         }
     }
+
+    pub fn export(&self, mpaths: &Vec<PathBuf>) {
+        let total = self.get_track_count();
+        if total > 0 {
+            let progress = ProgressBar::new(total.try_into().unwrap()).with_style(
+                ProgressStyle::default_bar()
+                    .template(
+                        "[{elapsed_precise}] [{bar:25}] {percent:>3}% {pos:>6}/{len:6} {wide_msg}",
+                    )
+                    .progress_chars("=> "),
+            );
+
+            let mut stmt = self.conn.prepare("SELECT File, Tempo, Zcr, MeanSpectralCentroid, StdDevSpectralCentroid, MeanSpectralRolloff, StdDevSpectralRolloff, MeanSpectralFlatness, StdDevSpectralFlatness, MeanLoudness, StdDevLoudness, Chroma1, Chroma2, Chroma3, Chroma4, Chroma5, Chroma6, Chroma7, Chroma8, Chroma9, Chroma10 FROM Tracks ORDER BY File ASC;").unwrap();
+            let track_iter = stmt
+                .query_map([], |row| {
+                    Ok(AnalysisResults {
+                        file: row.get(0)?,
+                        analysis: Analysis::new([row.get(1)?, row.get(2)?, row.get(3)?, row.get(4)?, row.get(5)?, row.get(6)?, row.get(7)?, row.get(8)?, row.get(9)?, row.get(10)?, row.get(11)?, row.get(12)?, row.get(13)?, row.get(14)?, row.get(15)?, row.get(16)?, row.get(17)?, row.get(18)?, row.get(19)?, row.get(20)?]),
+                    })
+                })
+                .unwrap();
+
+            let mut updated = 0;
+            for tr in track_iter {
+                let dbtags = tr.unwrap();
+                if !dbtags.file.contains(CUE_MARKER) {
+                    progress.set_message(format!("{}", dbtags.file));
+
+                    for mpath in mpaths {
+                        let track_path = mpath.join(&dbtags.file);
+                        if track_path.exists() {
+                            let spath = String::from(track_path.to_string_lossy());
+                            let meta = tags::read(&spath, true);
+                            if  meta.is_empty() || meta.analysis.is_none() || meta.analysis.unwrap()!=dbtags.analysis {
+                                tags::write_analysis(&spath, &dbtags.analysis);
+                                updated+=1;
+                            }
+                            break;
+                        }
+                    }
+                }
+                progress.inc(1);
+            }
+            progress.finish_with_message(format!("{} Updated.", updated))
+        }
+    }
 }

src/main.rs

@@ -76,7 +76,7 @@ fn main() {
         arg_parse.refer(&mut max_num_files).add_option(&["-n", "--numfiles"], Store, "Maximum number of files to analyse");
         arg_parse.refer(&mut max_threads).add_option(&["-t", "--threads"], Store, "Maximum number of threads to use for analysis");
         arg_parse.refer(&mut use_tags).add_option(&["-T", "--tags"], StoreTrue, "Read/write analysis results from/to source files");
-        arg_parse.refer(&mut task).add_argument("task", Store, "Task to perform; analyse, tags, ignore, upload, stopmixer.");
+        arg_parse.refer(&mut task).add_argument("task", Store, "Task to perform; analyse, tags, ignore, upload, export, stopmixer.");
         arg_parse.parse_args_or_exit();
     }
 
@@ -94,7 +94,7 @@ fn main() {
     builder.init();
 
     if task.is_empty() {
-        log::error!("No task specified, please choose from; analyse, tags, ignore, upload");
+        log::error!("No task specified, please choose from; analyse, tags, ignore, upload, export, stopmixer");
         process::exit(-1);
     }
 
@@ -206,6 +206,8 @@ fn main() {
                     process::exit(-1);
                 }
                 analyse::update_ignore(&db_path, &ignore_path);
+            } else if task.eq_ignore_ascii_case("export") {
+                analyse::export(&db_path, &music_paths);
             } else {
                 let ignore_path = PathBuf::from(&ignore_file);
                 analyse::analyse_files(&db_path, &music_paths, dry_run, keep_old, max_num_files, max_threads, &ignore_path, use_tags);