mirror of
https://github.com/fumiama/ahsai.git
synced 2026-06-11 02:50:26 +08:00
add cmd
This commit is contained in:
81
cmd/main.go
Normal file
81
cmd/main.go
Normal file
@@ -0,0 +1,81 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"strconv"
|
||||
|
||||
"github.com/faiface/beep/wav"
|
||||
"github.com/fumiama/ahsai"
|
||||
)
|
||||
|
||||
func main() {
|
||||
f := flag.String("f", "", "line-separated text to read")
|
||||
o := flag.String("o", "out.wav", "output wav file path")
|
||||
n := flag.String("n", "民安ともえ", "specify speaker")
|
||||
v := flag.Float64("v", 1.0, "volume")
|
||||
s := flag.Float64("s", 1.0, "speed")
|
||||
p := flag.Float64("p", 1.0, "pitch")
|
||||
r := flag.Float64("r", 1.0, "range")
|
||||
a := flag.Float64("a", 0, "anger")
|
||||
d := flag.Float64("d", 0, "sadness")
|
||||
j := flag.Float64("j", 0, "joy")
|
||||
b := flag.Uint("b", 2048, "border slience sample lenth")
|
||||
h := flag.Bool("h", false, "display this help")
|
||||
flag.Parse()
|
||||
if *h {
|
||||
flag.Usage()
|
||||
return
|
||||
}
|
||||
if *f == "" {
|
||||
panic("parameter -f must be specified")
|
||||
}
|
||||
spk := ahsai.Speaker{Volume: float32(*v), Speed: float32(*s), Pitch: float32(*p), Range: float32(*r), Anger: float32(*a), Sadness: float32(*d), Joy: float32(*j)}
|
||||
err := spk.SetName(*n)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
txt, err := os.Open(*f)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
sc := bufio.NewScanner(txt)
|
||||
lst := make([]string, 0, 128)
|
||||
i := 0
|
||||
for sc.Scan() {
|
||||
t := sc.Text()
|
||||
i++
|
||||
if len([]rune(t)) > 100 {
|
||||
panic("line " + strconv.Itoa(i) + ": too long (> 100 chars)")
|
||||
}
|
||||
lst = append(lst, t)
|
||||
}
|
||||
err = txt.Close()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
for i, t := range lst {
|
||||
u, err := spk.Speak(t)
|
||||
if err != nil {
|
||||
panic("line " + strconv.Itoa(i) + "error: " + err.Error())
|
||||
}
|
||||
lst[i] = u
|
||||
fmt.Print("\rread: ", (i+1)*100/len(lst), " %")
|
||||
}
|
||||
sm, format, err := ahsai.ComposeStream(*b, func(p int) { fmt.Print("\rcompose: ", p, " %") }, lst...)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
out, err := os.Create(*o)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer out.Close()
|
||||
err = wav.Encode(out, sm, format)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Println("\rall process succeed")
|
||||
}
|
||||
Reference in New Issue
Block a user