Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Inline the switch() function for performance.

Browse Source
This commit is contained in:
Charles Haley 2022-07-23 17:16:10 +01:00
parent 0f0d330ba3
commit 7b6c56befe
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/calibre/utils/formatter.py
View File

@ -52,6 +52,7 @@ class Node:
NODE_LOCAL_FUNCTION_DEFINE = 28
NODE_LOCAL_FUNCTION_CALL = 29
NODE_RANGE = 30
NODE_SWITCH = 31
def __init__(self, line_number, name):
self.my_line_number = line_number
@ -278,6 +279,13 @@ class FirstNonEmptyNode(Node):
self.expression_list = expression_list
class SwitchNode(Node):
def __init__(self, line_number, expression_list):
Node.__init__(self, line_number, 'first_non_empty()')
self.node_type = self.NODE_SWITCH
self.expression_list = expression_list
class ContainsNode(Node):
def __init__(self, line_number, arguments):
Node.__init__(self, line_number, 'contains()')
@ -670,6 +678,8 @@ class _Parser:
lambda ln, args: IfNode(ln, args[0], (args[1],), (args[2],))),
'first_non_empty': (lambda args: len(args) >= 1,
lambda ln, args: FirstNonEmptyNode(ln, args)),
'switch': (lambda args: len(args) >= 3 and (len(args) %2) == 0,
lambda ln, args: SwitchNode(ln, args)),
'assign': (lambda args: len(args) == 2 and len(args[0]) == 1 and args[0][0].node_type == Node.NODE_RVALUE,
lambda ln, args: AssignNode(ln, args[0][0].name, args[1])),
'contains': (lambda args: len(args) == 4,
@ -1148,6 +1158,20 @@ class _Interpreter:
self.break_reporter(prog.node_name, '', prog.line_number)
return ''
def do_node_switch(self, prog):
val = self.expr(prog.expression_list[0])
for i in range(1, len(prog.expression_list)-1, 2):
v = self.expr(prog.expression_list[i])
if re.search(v, val, flags=re.I):
res = self.expr(prog.expression_list[i+1])
if self.break_reporter:
self.break_reporter(prog.node_name, res, prog.line_number)
return res
res = self.expr(prog.expression_list[-1])
if (self.break_reporter):
self.break_reporter(prog.node_name, res, prog.line_number)
return res
def do_node_strcat(self, prog):
res = ''.join([self.expr(expr) for expr in prog.expression_list])
if self.break_reporter:
@ -1363,6 +1387,7 @@ class _Interpreter:
Node.NODE_ARGUMENTS: do_node_arguments,
Node.NODE_CALL_STORED_TEMPLATE: do_node_stored_template_call,
Node.NODE_FIRST_NON_EMPTY: do_node_first_non_empty,
Node.NODE_SWITCH: do_node_switch,
Node.NODE_FOR: do_node_for,
Node.NODE_RANGE: do_node_range,
Node.NODE_GLOBALS: do_node_globals,