package fun import "image" import "git.tebibyte.media/sashakoshka/tomo" import "git.tebibyte.media/sashakoshka/tomo/input" import "git.tebibyte.media/sashakoshka/tomo/artist" import "git.tebibyte.media/sashakoshka/tomo/elements/core" import "git.tebibyte.media/sashakoshka/tomo/default/theme" import "git.tebibyte.media/sashakoshka/tomo/default/config" import "git.tebibyte.media/sashakoshka/tomo/elements/fun/music" const pianoKeyWidth = 18 type pianoKey struct { image.Rectangle music.Note } // Piano is an element that can be used to input midi notes. type Piano struct { *core.Core *core.FocusableCore core core.CoreControl focusableControl core.FocusableCoreControl low, high music.Octave config config.Wrapped theme theme.Wrapped flatKeys []pianoKey sharpKeys []pianoKey contentBounds image.Rectangle pressed *pianoKey keynavPressed map[music.Note] bool onPress func (music.Note) onRelease func (music.Note) } // NewPiano returns a new piano element with a lowest and highest octave, // inclusive. If low is greater than high, they will be swapped. func NewPiano (low, high music.Octave) (element *Piano) { if low > high { temp := low low = high high = temp } element = &Piano { low: low, high: high, keynavPressed: make(map[music.Note] bool), } element.theme.Case = tomo.C("tomo", "piano") element.Core, element.core = core.NewCore (element, func () { element.recalculate() element.draw() }) element.FocusableCore, element.focusableControl = core.NewFocusableCore(element.core, element.redo) element.updateMinimumSize() return } // OnPress sets a function to be called when a key is pressed. func (element *Piano) OnPress (callback func (note music.Note)) { element.onPress = callback } // OnRelease sets a function to be called when a key is released. func (element *Piano) OnRelease (callback func (note music.Note)) { element.onRelease = callback } func (element *Piano) HandleMouseDown (x, y int, button input.Button) { element.Focus() if button != input.ButtonLeft { return } element.pressUnderMouseCursor(image.Pt(x, y)) } func (element *Piano) HandleMouseUp (x, y int, button input.Button) { if button != input.ButtonLeft { return } if element.onRelease != nil && element.pressed != nil { element.onRelease((*element.pressed).Note) } element.pressed = nil element.redo() } func (element *Piano) HandleMotion (x, y int) { if element.pressed == nil { return } element.pressUnderMouseCursor(image.Pt(x, y)) } func (element *Piano) pressUnderMouseCursor (point image.Point) { // find out which note is being pressed newKey := (*pianoKey)(nil) for index, key := range element.flatKeys { if point.In(key.Rectangle) { newKey = &element.flatKeys[index] break } } for index, key := range element.sharpKeys { if point.In(key.Rectangle) { newKey = &element.sharpKeys[index] break } } if newKey == nil { return } if newKey != element.pressed { // release previous note if element.pressed != nil && element.onRelease != nil { element.onRelease((*element.pressed).Note) } // press new note element.pressed = newKey if element.onPress != nil { element.onPress((*element.pressed).Note) } element.redo() } } var noteForKey = map[input.Key] music.Note { 'a': 46, 'z': 47, 'x': 48, 'd': 49, 'c': 50, 'f': 51, 'v': 52, 'b': 53, 'h': 54, 'n': 55, 'j': 56, 'm': 57, 'k': 58, ',': 59, '.': 60, ';': 61, '/': 62, '\'': 63, '1': 56, 'q': 57, '2': 58, 'w': 59, 'e': 60, '4': 61, 'r': 62, '5': 63, 't': 64, 'y': 65, '7': 66, 'u': 67, '8': 68, 'i': 69, '9': 70, 'o': 71, 'p': 72, '-': 73, '[': 74, '=': 75, ']': 76, '\\': 77, } func (element *Piano) HandleKeyDown (key input.Key, modifiers input.Modifiers) { if !element.Enabled() { return } note, exists := noteForKey[key] if !exists { return } if !element.keynavPressed[note] { element.keynavPressed[note] = true if element.onPress != nil { element.onPress(note) } element.redo() } } func (element *Piano) HandleKeyUp (key input.Key, modifiers input.Modifiers) { note, exists := noteForKey[key] if !exists { return } _, pressed := element.keynavPressed[note] if !pressed { return } delete(element.keynavPressed, note) if element.onRelease != nil { element.onRelease(note) } element.redo() } // SetTheme sets the element's theme. func (element *Piano) SetTheme (new tomo.Theme) { if new == element.theme.Theme { return } element.theme.Theme = new element.updateMinimumSize() element.recalculate() element.redo() } // SetConfig sets the element's configuration. func (element *Piano) SetConfig (new tomo.Config) { if new == element.config.Config { return } element.config.Config = new element.updateMinimumSize() element.recalculate() element.redo() } func (element *Piano) updateMinimumSize () { padding := element.theme.Padding(tomo.PatternPinboard) element.core.SetMinimumSize ( pianoKeyWidth * 7 * element.countOctaves() + padding.Horizontal(), 64 + padding.Vertical()) } func (element *Piano) countOctaves () int { return int(element.high - element.low + 1) } func (element *Piano) countFlats () int { return element.countOctaves() * 8 } func (element *Piano) countSharps () int { return element.countOctaves() * 5 } func (element *Piano) redo () { if element.core.HasImage() { element.draw() element.core.DamageAll() } } func (element *Piano) recalculate () { element.flatKeys = make([]pianoKey, element.countFlats()) element.sharpKeys = make([]pianoKey, element.countSharps()) padding := element.theme.Padding(tomo.PatternPinboard) bounds := padding.Apply(element.Bounds()) dot := bounds.Min note := element.low.Note(0) limit := element.high.Note(12) flatIndex := 0 sharpIndex := 0 for note < limit { if note.IsSharp() { element.sharpKeys[sharpIndex].Rectangle = image.Rect ( -(pianoKeyWidth * 3) / 7, 0, (pianoKeyWidth * 3) / 7, (bounds.Dy() * 5) / 8).Add(dot) element.sharpKeys[sharpIndex].Note = note sharpIndex ++ } else { element.flatKeys[flatIndex].Rectangle = image.Rect ( 0, 0, pianoKeyWidth, bounds.Dy()).Add(dot) dot.X += pianoKeyWidth element.flatKeys[flatIndex].Note = note flatIndex ++ } note ++ } element.contentBounds = image.Rectangle { bounds.Min, image.Pt(dot.X, bounds.Max.Y), } } func (element *Piano) draw () { state := tomo.State { Focused: element.Focused(), Disabled: !element.Enabled(), } for _, key := range element.flatKeys { _, keynavPressed := element.keynavPressed[key.Note] element.drawFlat ( key.Rectangle, element.pressed != nil && (*element.pressed).Note == key.Note || keynavPressed, state) } for _, key := range element.sharpKeys { _, keynavPressed := element.keynavPressed[key.Note] element.drawSharp ( key.Rectangle, element.pressed != nil && (*element.pressed).Note == key.Note || keynavPressed, state) } pattern := element.theme.Pattern(tomo.PatternPinboard, state) artist.DrawShatter ( element.core, pattern, element.Bounds(), element.contentBounds) } func (element *Piano) drawFlat ( bounds image.Rectangle, pressed bool, state tomo.State, ) { state.Pressed = pressed pattern := element.theme.Theme.Pattern ( tomo.PatternButton, state, tomo.C("fun", "piano", "flatKey")) pattern.Draw(element.core, bounds) } func (element *Piano) drawSharp ( bounds image.Rectangle, pressed bool, state tomo.State, ) { state.Pressed = pressed pattern := element.theme.Theme.Pattern ( tomo.PatternButton, state, tomo.C("fun", "piano", "sharpKey")) pattern.Draw(element.core, bounds) }