summaryrefslogtreecommitdiff
path: root/main.go
diff options
context:
space:
mode:
authorGrail Finder <wohilas@gmail.com>2023-02-16 16:20:56 +0600
committerGrail Finder <wohilas@gmail.com>2023-02-16 16:20:56 +0600
commitf7fd6d7e5877f96c605137ae185626ac6f85eebc (patch)
treeb60ada208fd085fdb9789cfcb4ae74de1c384fc8 /main.go
parent84a3dd05d1623d61e694d3f763c3121be3b9b43c (diff)
Refactor: move io funcitons in separate file
Diffstat (limited to 'main.go')
-rw-r--r--main.go81
1 files changed, 2 insertions, 79 deletions
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 \