From f7fd6d7e5877f96c605137ae185626ac6f85eebc Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Thu, 16 Feb 2023 16:20:56 +0600 Subject: Refactor: move io funcitons in separate file --- main.go | 81 ++--------------------------------------------------------------- 1 file changed, 2 insertions(+), 79 deletions(-) (limited to 'main.go') diff --git a/main.go b/main.go index a2eae60..13729b9 100644 --- a/main.go +++ b/main.go @@ -1,18 +1,15 @@ package main import ( - "bufio" - "encoding/csv" - "encoding/json" "flag" "fmt" - "io/ioutil" "os" "path" "path/filepath" "strings" ) +// TODO: to config file const ( subExt = ".vtt" outdir = "data/utterances" @@ -76,7 +73,7 @@ func linesToUtterances(lines []string, fd *FileData) []*Utterance { RightTime: strings.TrimSpace(splitted[1]), FD: fd, } - u.OutPath = fmt.Sprintf("%s/%s_%s_%s.wav", outdir, fd.AudioBase, + u.OutPath = fmt.Sprintf("%s/%s_%s_%s.opus", outdir, fd.AudioBase, u.LeftTime, u.RightTime) if u.LeftTime == u.RightTime { @@ -88,80 +85,6 @@ func linesToUtterances(lines []string, fd *FileData) []*Utterance { return resp } -func readLines(filepath string) []string { - file, err := os.Open(filepath) - if err != nil { - panic(err) - } - defer file.Close() - - resp := []string{} - scanner := bufio.NewScanner(file) - for scanner.Scan() { - resp = append(resp, scanner.Text()) - } - - if err := scanner.Err(); err != nil { - panic(err) - } - return resp -} - -// writeLines writes the lines to the given file. -func writeLines(lines []string, path string) error { - file, err := os.Create(path) - // file, err := os.OpenFile(path, - // os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0666) - if err != nil { - return err - } - defer file.Close() - - w := bufio.NewWriter(file) - for _, line := range lines { - fmt.Fprintln(w, line) - } - return w.Flush() -} - -func readJson(filepath string) map[string]string { - data := make(map[string]string) - plan, err := ioutil.ReadFile(filepath) - if err != nil { - return data - } - err = json.Unmarshal(plan, &data) - if err != nil { - panic(err) - } - return data -} - -func writeJson(data map[string]string) { - metadataJson, _ := json.MarshalIndent(data, "", " ") - err := ioutil.WriteFile(metadataPath, metadataJson, 0644) - if err != nil { - panic(err) - } -} - -func writeCSV(data [][]string) { - f, err := os.Create(metadataPathCSV) - defer f.Close() - - if err != nil { - panic(err) - } - - w := csv.NewWriter(f) - w.Comma = '\t' - defer w.Flush() - - if err := w.WriteAll(data); err != nil { - panic(err) - } -} - func buildFFmpegCall(ut *Utterance) string { return fmt.Sprintf( `yes no | ffmpeg -i "%s" -ss %s -to %s \ -- cgit v1.2.3