You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
319 lines
7.6 KiB
Go
319 lines
7.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"image"
|
|
"image/color"
|
|
"image/draw"
|
|
"image/png"
|
|
"log"
|
|
"os/exec"
|
|
"time"
|
|
|
|
"github.com/aarzilli/nucular"
|
|
"github.com/aarzilli/nucular/label"
|
|
"github.com/lawl/pulseaudio"
|
|
)
|
|
|
|
type ntcontext struct {
|
|
inputList []input
|
|
noiseSupressorState int
|
|
paClient *pulseaudio.Client
|
|
librnnoise string
|
|
sourceListColdWidthIndex int
|
|
useBuiltinRNNoise bool
|
|
config *config
|
|
loadingScreen bool
|
|
licenseScreen bool
|
|
versionScreen bool
|
|
licenseTextArea nucular.TextEditor
|
|
masterWindow *nucular.MasterWindow
|
|
update updateui
|
|
reloadRequired bool
|
|
}
|
|
|
|
var green = color.RGBA{34, 187, 69, 255}
|
|
var red = color.RGBA{255, 70, 70, 255}
|
|
var orange = color.RGBA{255, 140, 0, 255}
|
|
|
|
var patreonImg *image.RGBA
|
|
|
|
func updatefn(ctx *ntcontext, w *nucular.Window) {
|
|
|
|
if !ctx.paClient.Connected() {
|
|
connectScreen(ctx, w)
|
|
return
|
|
}
|
|
|
|
if ctx.loadingScreen {
|
|
loadingScreen(ctx, w)
|
|
return
|
|
}
|
|
|
|
if ctx.licenseScreen {
|
|
licenseScreen(ctx, w)
|
|
return
|
|
}
|
|
|
|
if ctx.versionScreen {
|
|
versionScreen(ctx, w)
|
|
return
|
|
}
|
|
|
|
w.MenubarBegin()
|
|
|
|
w.Row(10).Dynamic(1)
|
|
if w := w.Menu(label.TA("About", "LC"), 120, nil); w != nil {
|
|
w.Row(10).Dynamic(1)
|
|
if w.MenuItem(label.T("Licenses")) {
|
|
ctx.licenseScreen = true
|
|
}
|
|
w.Row(10).Dynamic(1)
|
|
if w.MenuItem(label.T("Source code")) {
|
|
exec.Command("xdg-open", "https://github.com/lawl/NoiseTorch").Run()
|
|
}
|
|
if w.MenuItem(label.T("Version")) {
|
|
ctx.versionScreen = true
|
|
}
|
|
}
|
|
|
|
w.MenubarEnd()
|
|
|
|
w.Row(25).Dynamic(2)
|
|
if patreonImg == nil {
|
|
patreonImg = loadPatreonImg()
|
|
}
|
|
|
|
if imageButton(w, patreonImg) {
|
|
exec.Command("xdg-open", "https://patreon.com/lawl").Run()
|
|
}
|
|
|
|
if ctx.noiseSupressorState == loaded {
|
|
w.LabelColored("NoiseTorch active", "RC", green)
|
|
} else if ctx.noiseSupressorState == unloaded {
|
|
w.LabelColored("NoiseTorch inactive", "RC", red)
|
|
} else if ctx.noiseSupressorState == inconsistent {
|
|
w.LabelColored("Inconsistent state, please unload first.", "RC", orange)
|
|
}
|
|
|
|
if ctx.update.available && !ctx.update.triggered {
|
|
w.Row(20).Ratio(0.9, 0.1)
|
|
w.LabelColored("Update available! Click to install version: "+ctx.update.serverVersion, "LC", green)
|
|
if w.ButtonText("Update") {
|
|
ctx.update.triggered = true
|
|
go update(ctx)
|
|
(*ctx.masterWindow).Changed()
|
|
}
|
|
}
|
|
|
|
if ctx.update.triggered {
|
|
w.Row(20).Dynamic(1)
|
|
w.Label(ctx.update.updatingText, "CC")
|
|
}
|
|
|
|
if w.TreePush(nucular.TreeTab, "Settings", true) {
|
|
w.Row(15).Dynamic(2)
|
|
if w.CheckboxText("Display Monitor Sources", &ctx.config.DisplayMonitorSources) {
|
|
ctx.sourceListColdWidthIndex++ //recompute the with because of new elements
|
|
go writeConfig(ctx.config)
|
|
}
|
|
|
|
w.Spacing(1)
|
|
|
|
w.Row(25).Ratio(0.5, 0.45, 0.05)
|
|
w.Label("Voice Activation Threshold", "LC")
|
|
if w.Input().Mouse.HoveringRect(w.LastWidgetBounds) {
|
|
w.Tooltip("If you have a decent microphone, you can usually turn this all the way up.")
|
|
}
|
|
if w.SliderInt(0, &ctx.config.Threshold, 95, 1) {
|
|
go writeConfig(ctx.config)
|
|
ctx.reloadRequired = true
|
|
}
|
|
w.Label(fmt.Sprintf("%d%%", ctx.config.Threshold), "RC")
|
|
|
|
if ctx.reloadRequired {
|
|
w.Row(20).Dynamic(1)
|
|
w.LabelColored("Reloading NoiseTorch is required to apply these changes.", "LC", orange)
|
|
}
|
|
|
|
w.TreePop()
|
|
}
|
|
if w.TreePush(nucular.TreeTab, "Select Device", true) {
|
|
w.Row(15).Dynamic(1)
|
|
w.Label("Select an input device below:", "LC")
|
|
|
|
for i := range ctx.inputList {
|
|
el := &ctx.inputList[i]
|
|
|
|
if el.isMonitor && !ctx.config.DisplayMonitorSources {
|
|
continue
|
|
}
|
|
w.Row(15).Static()
|
|
w.LayoutFitWidth(0, 0)
|
|
if w.CheckboxText("", &el.checked) {
|
|
ensureOnlyOneInputSelected(&ctx.inputList, el)
|
|
}
|
|
|
|
w.LayoutFitWidth(ctx.sourceListColdWidthIndex, 0)
|
|
if el.dynamicLatency {
|
|
w.Label(el.Name, "LC")
|
|
} else {
|
|
w.LabelColored("(incompatible?) "+el.Name, "LC", orange)
|
|
}
|
|
}
|
|
|
|
w.Row(30).Dynamic(1)
|
|
w.Spacing(1)
|
|
|
|
w.Row(25).Dynamic(2)
|
|
if ctx.noiseSupressorState != unloaded {
|
|
if w.ButtonText("Unload NoiseTorch") {
|
|
ctx.loadingScreen = true
|
|
ctx.reloadRequired = false
|
|
go func() { // don't block the UI thread, just display a working screen so user can't run multiple loads/unloads
|
|
if err := unloadSupressor(ctx.paClient); err != nil {
|
|
log.Println(err)
|
|
}
|
|
//wait until PA reports it has actually loaded it, timeout at 10s
|
|
for i := 0; i < 20; i++ {
|
|
if supressorState(ctx.paClient) != unloaded {
|
|
time.Sleep(time.Millisecond * 500)
|
|
}
|
|
}
|
|
ctx.loadingScreen = false
|
|
(*ctx.masterWindow).Changed()
|
|
}()
|
|
}
|
|
} else {
|
|
w.Spacing(1)
|
|
}
|
|
txt := "Load NoiseTorch"
|
|
if ctx.noiseSupressorState == loaded {
|
|
txt = "Reload NoiseTorch"
|
|
}
|
|
if inp, ok := inputSelection(ctx); ok && ctx.noiseSupressorState != inconsistent {
|
|
if w.ButtonText(txt) {
|
|
ctx.loadingScreen = true
|
|
ctx.reloadRequired = false
|
|
go func() { // don't block the UI thread, just display a working screen so user can't run multiple loads/unloads
|
|
if ctx.noiseSupressorState == loaded {
|
|
if err := unloadSupressor(ctx.paClient); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
if err := loadSupressor(ctx, inp); err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
//wait until PA reports it has actually loaded it, timeout at 10s
|
|
for i := 0; i < 20; i++ {
|
|
if supressorState(ctx.paClient) != loaded {
|
|
time.Sleep(time.Millisecond * 500)
|
|
}
|
|
}
|
|
ctx.config.LastUsedInput = inp.ID
|
|
go writeConfig(ctx.config)
|
|
ctx.loadingScreen = false
|
|
(*ctx.masterWindow).Changed()
|
|
}()
|
|
}
|
|
} else {
|
|
w.Spacing(1)
|
|
}
|
|
w.TreePop()
|
|
}
|
|
}
|
|
|
|
func ensureOnlyOneInputSelected(inps *[]input, current *input) {
|
|
if current.checked != true {
|
|
return
|
|
}
|
|
for i := range *inps {
|
|
el := &(*inps)[i]
|
|
el.checked = false
|
|
}
|
|
current.checked = true
|
|
}
|
|
|
|
func inputSelection(ctx *ntcontext) (input, bool) {
|
|
for _, in := range ctx.inputList {
|
|
if in.checked {
|
|
return in, true
|
|
}
|
|
}
|
|
return input{}, false
|
|
}
|
|
|
|
func loadingScreen(ctx *ntcontext, w *nucular.Window) {
|
|
w.Row(50).Dynamic(1)
|
|
w.Label("Working...", "CB")
|
|
w.Row(50).Dynamic(1)
|
|
w.Label("(this may take a few seconds)", "CB")
|
|
}
|
|
|
|
func licenseScreen(ctx *ntcontext, w *nucular.Window) {
|
|
w.Row(255).Dynamic(1)
|
|
field := &ctx.licenseTextArea
|
|
field.Flags |= nucular.EditMultiline
|
|
if len(field.Buffer) < 1 {
|
|
field.Buffer = []rune(licenseString)
|
|
}
|
|
field.Edit(w)
|
|
|
|
w.Row(20).Dynamic(2)
|
|
w.Spacing(1)
|
|
if w.ButtonText("OK") {
|
|
ctx.licenseScreen = false
|
|
}
|
|
}
|
|
|
|
func versionScreen(ctx *ntcontext, w *nucular.Window) {
|
|
w.Row(50).Dynamic(1)
|
|
w.Label("Version", "CB")
|
|
w.Row(50).Dynamic(1)
|
|
w.Label(version, "CB")
|
|
w.Row(50).Dynamic(1)
|
|
w.Spacing(1)
|
|
w.Row(20).Dynamic(2)
|
|
w.Spacing(1)
|
|
if w.ButtonText("OK") {
|
|
ctx.versionScreen = false
|
|
}
|
|
}
|
|
|
|
func connectScreen(ctx *ntcontext, w *nucular.Window) {
|
|
w.Row(50).Dynamic(1)
|
|
w.Label("Connecting to pulseaudio...", "CB")
|
|
}
|
|
|
|
func resetUI(ctx *ntcontext) {
|
|
ctx.loadingScreen = false
|
|
|
|
if ctx.masterWindow != nil {
|
|
(*ctx.masterWindow).Changed()
|
|
}
|
|
}
|
|
|
|
func loadPatreonImg() *image.RGBA {
|
|
var pat *image.RGBA
|
|
img, _ := png.Decode(bytes.NewReader(patreonPNG))
|
|
pat = image.NewRGBA(img.Bounds())
|
|
draw.Draw(pat, img.Bounds(), img, image.Point{}, draw.Src)
|
|
return pat
|
|
}
|
|
|
|
func imageButton(w *nucular.Window, img *image.RGBA) bool {
|
|
style := w.Master().Style()
|
|
origButtonStyle := style.Button
|
|
style.Button.Border = 0
|
|
style.Button.Normal.Data.Color = style.NormalWindow.Background
|
|
style.Button.Hover.Data.Color = style.NormalWindow.Background
|
|
style.Button.Active.Data.Color = style.NormalWindow.Background
|
|
|
|
defer (func() { style.Button = origButtonStyle })()
|
|
|
|
return w.Button(label.I(patreonImg), false)
|
|
|
|
}
|