aboutsummaryrefslogtreecommitdiffstats
path: root/main.go
blob: 740e4bedcf79c76ee03c04446945978804dbc016 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
package main

import (
	"fmt"
	"os"
	"sync"

	"github.com/faiface/mainthread"
	"volute/gui"
	"volute/gui/win"
)

const (
	WIDTH  = 800
	HEIGHT = 600

	POINTS = 6

	R = 8314.3 // gas constant
	M = 28.962 // molar mass of air

	WIDEST_LABEL = "mass flow (kg/min)"
)

func run() {
	wg := new(sync.WaitGroup)
	defer wg.Wait()

	focus := NewFocus([]int{1, POINTS, POINTS, POINTS, POINTS})
	defer focus.Close()

	displacementChan := make(chan uint)
	displacementBroadcast := NewBroadcast(displacementChan)
	defer displacementBroadcast.Wait()

	var (
		rpmChan  [POINTS]chan uint
		veChan   [POINTS]chan uint
		imapChan [POINTS]chan uint
		actChan  [POINTS]chan uint

		flowChan [POINTS]chan float64
	)
	makeChans(rpmChan[:], veChan[:], imapChan[:], actChan[:])
	makeChans(flowChan[:])

	w, err := win.New(win.Title("volute"), win.Size(WIDTH, HEIGHT))
	if err != nil {
		fmt.Println("error creating window:", err)
		os.Exit(1)
	}
	mux, env := gui.NewMux(w)
	defer close(env.Draw())

	spawnWidgets(
		displacementChan,
		rpmChan, veChan, imapChan, actChan,
		flowChan,
		&focus, mux, wg,
	)

	for i := 0; i < POINTS; i++ {
		wg.Add(1)
		go calculateFlow(
			flowChan[i],
			displacementBroadcast.AddDestination(),
			rpmChan[i], veChan[i], actChan[i], imapChan[i],
			wg,
		)
	}

	focus.Focus(true)
	eventLoop(env, &focus)
}

func eventLoop(env gui.Env, focus *Focus) {
	for event := range env.Events() {
		switch event := event.(type) {
		case win.WiClose:
			return
		case win.KbType:
			switch event.Rune {
			case 'q':
				return
			case 'h':
				focus.Left()
			case 'j':
				focus.Down()
			case 'k':
				focus.Up()
			case 'l':
				focus.Right()
			}
		}
	}
}

func makeChans[T any](chanss ...[]chan T) {
	for i := range chanss {
		for j := range chanss[i] {
			chanss[i][j] = make(chan T)
		}
	}
}

func calculateFlow(
	flow chan<- float64,
	displacementChan, rpmChan, veChan, actChan, imapChan <-chan uint,
	wg *sync.WaitGroup,
) {
	defer wg.Done()
	defer close(flow)

	var (
		displacement Volume
		rpm          uint
		ve           uint
		act          Temperature
		imap         Pressure

		v  uint
		ok bool
	)

	for {
		select {
		case v, ok = <-displacementChan:
			displacement = Volume(v) * CubicCentimetre
		case rpm, ok = <-rpmChan:
		case ve, ok = <-veChan:
		case v, ok = <-actChan:
			act = Temperature{float64(v), Celcius}
		case v, ok = <-imapChan:
			imap = Pressure(v) * Millibar
		}
		if !ok {
			return
		}
		flow <- massFlow(displacement, rpm, ve, act, imap)
	}
}

func massFlow(displacement Volume, rpm, ve uint, act Temperature, imap Pressure) float64 {
	density := (M / R) * float64(imap/Pascal) / act.AsUnit(Kelvin)                          // kg/m3
	volumeFlow := float64(displacement/CubicMetre) * float64(rpm/2) * (float64(ve) / 100.0) // m3/min
	return density * volumeFlow
}

func main() {
	mainthread.Run(run)
}