use antlr listener
This commit is contained in:
parent
9b46389801
commit
c678e928c3
|
@ -0,0 +1,162 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import antlr4
|
||||
from CommentMacroParser import CommentMacroParser
|
||||
from CommentMacroListener import CommentMacroListener
|
||||
from pythonosc import osc_message_builder
|
||||
from pythonosc import udp_client
|
||||
from time import sleep
|
||||
|
||||
|
||||
def button_delay(time=0.05):
|
||||
sleep(time)
|
||||
|
||||
|
||||
def _master_go(self, expr):
|
||||
if (expr.master().getText() == "*"):
|
||||
print("Main GO")
|
||||
self.server.osc.send_message("/hog/hardware/maingo", 1) # button down
|
||||
button_delay()
|
||||
self.server.osc.send_message("/hog/hardware/maingo", 0) # button up
|
||||
return 1
|
||||
|
||||
print("GO Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_halt(self, expr):
|
||||
print("Halt Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_release(self, expr):
|
||||
print("Release Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_fade(self, expr):
|
||||
print("Fade Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_fade_grand(self, expr):
|
||||
try:
|
||||
level = float(expr.number().getText())
|
||||
except ValueError:
|
||||
print("Expected a number...")
|
||||
return -1
|
||||
|
||||
if (level < 0 or level > 100):
|
||||
print("Level must be between 0 and 100.")
|
||||
return -1
|
||||
|
||||
print("Fading Grand Master to " + str(level) + "%")
|
||||
level *= 255 / 100 # percent in Macro, 0>255 in OSC
|
||||
self.server.osc.send_message("/hog/hardware/fader/0", level)
|
||||
return 1
|
||||
|
||||
|
||||
def _master_choose(self, expr):
|
||||
try:
|
||||
master = int(expr.number().getText())
|
||||
except ValueError:
|
||||
print("Expected a number...")
|
||||
return -1
|
||||
|
||||
if (master < 0):
|
||||
print("Master must be greater than 0.")
|
||||
return -1
|
||||
|
||||
print("Choose Master " + str(master))
|
||||
self.server.osc.send_message("/hog/hardware/choose/" + str(master), 1) # down
|
||||
button_delay()
|
||||
self.server.osc.send_message("/hog/hardware/choose/" + str(master), 0) # up
|
||||
return 1
|
||||
|
||||
|
||||
def _list_go(self, expr):
|
||||
print("Go List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _list_halt(self, expr):
|
||||
print("Halt List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _list_release(self, expr):
|
||||
print("Release List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_go(self, expr):
|
||||
print("Go Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_halt(self, expr):
|
||||
print("Halt Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_release(self, expr):
|
||||
print("Release Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
command = {"GM": _master_go,
|
||||
"HM": _master_halt,
|
||||
"RM": _master_release,
|
||||
"FM": _master_fade,
|
||||
"FGM": _master_fade_grand,
|
||||
"CM": _master_choose,
|
||||
"GL": _list_go,
|
||||
"HL": _list_halt,
|
||||
"RL": _list_release,
|
||||
"GS": _scene_go,
|
||||
"HS": _scene_halt,
|
||||
"RS": _scene_release
|
||||
}
|
||||
|
||||
|
||||
# https://raw.githubusercontent.com/jszheng/py3antlr4book/master/bin/pygrun
|
||||
# this is a python version of TestRig
|
||||
def beautify_lisp_string(in_string):
|
||||
__author__ = 'jszheng'
|
||||
indent_size = 2
|
||||
add_indent = ' ' * indent_size
|
||||
out_string = in_string[0] # no indent for 1st (
|
||||
indent = ''
|
||||
for i in range(1, len(in_string) - 1):
|
||||
if in_string[i] == '(' and in_string[i + 1] != ' ':
|
||||
indent += add_indent
|
||||
out_string += "\n" + indent + '('
|
||||
elif in_string[i] == ')':
|
||||
out_string += ')'
|
||||
if len(indent) > 0:
|
||||
indent = indent.replace(add_indent, '', 1)
|
||||
else:
|
||||
out_string += in_string[i]
|
||||
return out_string
|
||||
|
||||
|
||||
class OscCommentMacroListener(CommentMacroListener):
|
||||
|
||||
def __init__(self, parser, server):
|
||||
self.parser = parser
|
||||
self.server = server
|
||||
|
||||
def exitMacro(self, ctx: CommentMacroParser.MacroContext):
|
||||
lisp_tree_str = ctx.toStringTree(recog=self.parser)
|
||||
print(beautify_lisp_string(lisp_tree_str))
|
||||
|
||||
name = ctx.children[0].getText()
|
||||
try:
|
||||
command[name](self, ctx)
|
||||
except KeyError:
|
||||
print(name + " macro is not compatable with OSC.")
|
||||
return -1
|
||||
|
||||
print("Exiting Macro")
|
||||
return 1
|
150
comment.py
150
comment.py
|
@ -23,9 +23,8 @@ import sys
|
|||
|
||||
from CommentMacroLexer import CommentMacroLexer
|
||||
from CommentMacroParser import CommentMacroParser
|
||||
from pythonosc import osc_message_builder
|
||||
from OscCommentMacroListener import OscCommentMacroListener
|
||||
from pythonosc import udp_client
|
||||
from time import sleep
|
||||
|
||||
|
||||
class hog4_osc:
|
||||
|
@ -46,134 +45,16 @@ hog4 = hog4_osc(server.get("ip", "10.0.0.1"),
|
|||
server.getint("net", 1))
|
||||
|
||||
|
||||
def _master_go(expr):
|
||||
if (expr.master().getText() == "*"):
|
||||
print("Main GO")
|
||||
hog4.osc.send_message("/hog/hardware/maingo", 1) # button down
|
||||
button_delay()
|
||||
hog4.osc.send_message("/hog/hardware/maingo", 0) # button up
|
||||
return 1
|
||||
|
||||
print("GO Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_halt(expr):
|
||||
print("Halt Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_release(expr):
|
||||
print("Release Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_fade(expr):
|
||||
print("Fade Master is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _master_fade_grand(expr):
|
||||
try:
|
||||
level = float(expr.number().getText())
|
||||
except ValueError:
|
||||
print("Expected a number...")
|
||||
return -1
|
||||
|
||||
if (level < 0 or level > 100):
|
||||
print("Level must be between 0 and 100.")
|
||||
return -1
|
||||
|
||||
print("Fading Grand Master to " + str(level) + "%")
|
||||
level *= 255 / 100 # percent in Macro, 0>255 in OSC
|
||||
hog4.osc.send_message("/hog/hardware/fader/0", level)
|
||||
return 1
|
||||
|
||||
|
||||
def _master_choose(expr):
|
||||
try:
|
||||
master = int(expr.number().getText())
|
||||
except ValueError:
|
||||
print("Expected a number...")
|
||||
return -1
|
||||
|
||||
if (master < 0):
|
||||
print("Master must be greater than 0.")
|
||||
return -1
|
||||
|
||||
print("Choose Master " + str(master))
|
||||
hog4.osc.send_message("/hog/hardware/choose/" + str(master), 1) # down
|
||||
button_delay()
|
||||
hog4.osc.send_message("/hog/hardware/choose/" + str(master), 0) # up
|
||||
return 1
|
||||
|
||||
|
||||
def _list_go(expr):
|
||||
print("Go List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _list_halt(expr):
|
||||
print("Halt List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _list_release(expr):
|
||||
print("Release List is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_go(expr):
|
||||
print("Go Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_halt(expr):
|
||||
print("Halt Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
def _scene_release(expr):
|
||||
print("Release Scene is a stub")
|
||||
return -1
|
||||
|
||||
|
||||
command = {"GM": _master_go,
|
||||
"HM": _master_halt,
|
||||
"RM": _master_release,
|
||||
"FM": _master_fade,
|
||||
"FGM": _master_fade_grand,
|
||||
"CM": _master_choose,
|
||||
"GL": _list_go,
|
||||
"HL": _list_halt,
|
||||
"RL": _list_release,
|
||||
"GS": _scene_go,
|
||||
"HS": _scene_halt,
|
||||
"RS": _scene_release
|
||||
}
|
||||
|
||||
|
||||
def button_delay(time=0.05):
|
||||
sleep(time)
|
||||
|
||||
|
||||
def comment(text):
|
||||
input_stream = antlr4.InputStream(text)
|
||||
lexer = CommentMacroLexer(input_stream)
|
||||
stream = antlr4.CommonTokenStream(lexer)
|
||||
parser = CommentMacroParser(stream)
|
||||
tree = parser.macro()
|
||||
tree = parser.prog()
|
||||
|
||||
lisp_tree_str = tree.toStringTree(recog=parser)
|
||||
print(beautify_lisp_string(lisp_tree_str))
|
||||
|
||||
name = tree.children[0].getText()
|
||||
try:
|
||||
command[name](tree)
|
||||
except KeyError:
|
||||
print(name + " macro is not compatable with OSC.")
|
||||
return -1
|
||||
return 1
|
||||
oscMacro = OscCommentMacroListener(parser, hog4)
|
||||
walker = antlr4.ParseTreeWalker()
|
||||
walker.walk(oscMacro, tree)
|
||||
|
||||
|
||||
def orderly_exit(signal=None, frame=None):
|
||||
|
@ -181,27 +62,6 @@ def orderly_exit(signal=None, frame=None):
|
|||
sys.exit(0)
|
||||
|
||||
|
||||
# https://raw.githubusercontent.com/jszheng/py3antlr4book/master/bin/pygrun
|
||||
# this is a python version of TestRig
|
||||
def beautify_lisp_string(in_string):
|
||||
__author__ = 'jszheng'
|
||||
indent_size = 2
|
||||
add_indent = ' ' * indent_size
|
||||
out_string = in_string[0] # no indent for 1st (
|
||||
indent = ''
|
||||
for i in range(1, len(in_string) - 1):
|
||||
if in_string[i] == '(' and in_string[i + 1] != ' ':
|
||||
indent += add_indent
|
||||
out_string += "\n" + indent + '('
|
||||
elif in_string[i] == ')':
|
||||
out_string += ')'
|
||||
if len(indent) > 0:
|
||||
indent = indent.replace(add_indent, '', 1)
|
||||
else:
|
||||
out_string += in_string[i]
|
||||
return out_string
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
signal.signal(signal.SIGINT, orderly_exit)
|
||||
if len(sys.argv) > 1:
|
||||
|
|
Loading…
Reference in New Issue