Newer
Older

Silas S. Brown
committed
#!/usr/bin/env python
program_name = "Annotator Generator v0.546 (c) 2012-14 Silas S. Brown"

Silas S. Brown
committed
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# http://www.apache.org/licenses/LICENSE-2.0
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from optparse import OptionParser
parser = OptionParser()

Silas S. Brown
committed
import sys,os,os.path,tempfile,time,re

Silas S. Brown
committed
if not "mac" in sys.platform and not "darwin" in sys.platform and ("win" in sys.platform or "mingw32" in sys.platform): exe=".exe" # Windows, Cygwin, etc
else: exe=""

Silas S. Brown
committed
try: import cPickle as pickle
except:
try: import pickle
except: pickle = None

Silas S. Brown
committed
# =========== INPUT OPTIONS ==============
parser.add_option("--infile",
help="Filename of a text file (or a compressed .gz or .bz2 file) to read the input examples from. If this is not specified, standard input is used.")
parser.add_option("--incode",default="utf-8",
help="Character encoding of the input file (default %default)")
parser.add_option("--mstart",
dest="markupStart",
default="<ruby><rb>",
help="The string that starts a piece of text with annotation markup in the input examples; default %default")
parser.add_option("--mmid",
dest="markupMid",
default="</rb><rt>",

Silas S. Brown
committed
help="The string that occurs in the middle of a piece of markup in the input examples, with the word on its left and the added markup on its right (or the other way around if mreverse is set); default %default")

Silas S. Brown
committed
parser.add_option("--mend",
dest="markupEnd",
default="</rt></ruby>",
help="The string that ends a piece of annotation markup in the input examples; default %default")

Silas S. Brown
committed
parser.add_option("--mreverse",
action="store_true",default=False,
help="Specifies that the annotation markup is reversed, so the text BEFORE mmid is the annotation and the text AFTER it is the base text")

Silas S. Brown
committed
parser.add_option("--reference-sep",
help="Reference separator code used in the example input. If you want to keep example source references for each rule, you can label the input with 'references' (chapter and section numbers or whatever), and use this option to specify what keyword or other markup the input will use between each 'reference'. The name of the next reference will be whatever text immediately follows this string. Note that the reference separator, and the reference name that follows it, should not be part of the text itself and should therefore not be part of any annotation markup. If this option is not set then references will not be tracked.")
parser.add_option("--ref-name-end",default=" ",
help="Sets what the input uses to END a reference name. The default is a single space, so that the first space after the reference-sep string will end the reference name.")

Silas S. Brown
committed
parser.add_option("--ref-pri",
help="Name of a reference to be considered \"high priority\" for Yarowsky-like seed collocations (if these are in use). Normally the Yarowsky-like logic tries to identify a \"default\" annotation based on what is most common in the examples, with the exceptions indicated by collocations. If however a word is found in a high priority reference then the first annotation found in that reference will be considered the ideal \"default\" even if it's in a minority in the examples; everything else will be considered as an exception. In languages without spaces, this override should normally be used only for one-character words; if used with longer words it might have unexpected effects on rule-overlap ambiguities.")

Silas S. Brown
committed
parser.add_option("-s", "--spaces",
action="store_false",
dest="removeSpace",
default=True,
help="Set this if you are working with a language that uses whitespace in its non-markedup version (not fully tested). The default is to assume that there will not be any whitespace in the language, which is correct for Chinese and Japanese.")
parser.add_option("-c", "--capitalisation",
action="store_true",
default=False,
help="Don't try to normalise capitalisation in the input. Normally, to simplify the rules, the analyser will try to remove start-of-sentence capitals in annotations, so that the only remaining words with capital letters are the ones that are ALWAYS capitalised such as names. (That's not perfect: it's possible that some words will always be capitalised just because they happen to never occur mid-sentence in the examples.) If this option is used, the analyser will instead try to \"learn\" how to predict the capitalisation of ALL words (including start of sentence words) from their contexts.") # TODO: make the C program put the sentence capitals back

Silas S. Brown
committed
parser.add_option("-w", "--annot-whitespace",
action="store_true",
default=False,

Silas S. Brown
committed
help="Don't try to normalise the use of whitespace and hyphenation in the example annotations. Normally the analyser will try to do this, to reduce the risk of missing possible rules due to minor typographical variations.") # TODO: can this be extended to the point where the words 'try to' can be deleted ? see comments

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--glossfile",

Silas S. Brown
committed
help="Filename of an optional text file (or compressed .gz or .bz2 file) to read auxiliary \"gloss\" information. Each line of this should be of the form: word (tab) annotation (tab) gloss. When the compiled annotator generates ruby markup, it will add the gloss string as a popup title whenever that word is used with that annotation. The annotation field may be left blank to indicate that the gloss will appear for any annotation of that word. The entries in glossfile do NOT affect the annotation process itself, so it's not necessary to completely debug glossfile's word segmentation etc.")

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--manualrules",
help="Filename of an optional text file (or compressed .gz or .bz2 file) to read extra, manually-written rules. Each line of this should be a marked-up phrase (in the input format) which is to be unconditionally added as a rule. Use this sparingly, because these rules are not taken into account when generating the others and they will be applied regardless of context (although a manual rule might fail to activate if the annotator is part-way through processing a different rule).") # (or if there's a longer automatic match)

Silas S. Brown
committed
# =========== OUTPUT OPTIONS ==============

Silas S. Brown
committed
parser.add_option("--rulesFile",help="Filename of an optional auxiliary binary file to hold the accumulated rules. Adding .gz or .bz2 for compression is acceptable. If this is set then the rules will be written to it (in binary format) as well as to the output. Additionally, if the file already exists then rules will first of all be read from it before generating any new rules. This might be useful if you have made some small additions to the examples and would like these to be incorporated without a complete re-run. It might not work as well as a re-run but it should be faster. If using a rulesFile then you must keep the same input (you may make small additions etc, but it won't work properly if you delete many examples or change the format between runs) and you must keep the same ybytes-related options if any.") # You may however change whether or not a --single-words / --max-words option applies to the new examples (but hopefully shouldn't have to)

Silas S. Brown
committed
parser.add_option("--no-input",
action="store_true",default=False,

Silas S. Brown
committed
help="Don't actually read the input, just use the rules that were previously stored in rulesFile. This can be used to increase speed if the only changes made are to the output options. You should still specify the input formatting options (which should not change), and any glossfile or manualrules options (which may change).")

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--c-filename",default=tempfile.gettempdir()+os.sep+"annotator.c",help="Where to write the C program if standard output is not connected to a pipe. Defaults to annotator.c in the system temporary directory (the program might be large, especially if Yarowsky indicators are not used, so it's best not to use a server home directory where you might have limited quota). If standard output is connected to a pipe, then this option is ignored and C code is written to the pipe instead.")

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--c-compiler",default="cc -o annotator"+exe,help="The C compiler to run if standard output is not connected to a pipe. The default is to use the \"cc\" command which usually redirects to your \"normal\" compiler. You can add options (remembering to enclose this whole parameter in quotes if it contains spaces), but if the C program is large then adding optimisation options may make the compile take a LONG time. If standard output is connected to a pipe, then this option is ignored because the C code will simply be written to the pipe. Default: %default")
# If compiling an experimental annotator quickly, you might try tcc as it compiles fast. If tcc is not available on your system then clang might compile faster than gcc.
# (BUT tcc can have problems on Raspberry Pi see http://www.raspberrypi.org/phpBB3/viewtopic.php?t=30036&p=263213; can be best to cross-compile, e.g. from a Mac use https://github.com/UnhandledException/ARMx/wiki/Sourcery-G---Lite-for-ARM-GNU-Linux-(2009q3-67)-for-Mac-OS-X and arm-none-linux-gnueabi-gcc)
# In large rulesets with --max-or-length=0 and --nested-switch, gcc takes time and gcc -Os can take a LOT longer, and CINT, Ch and picoc run out of memory. Without these options the overhead of gcc's -Os isn't so bad (and does save some room).

Silas S. Brown
committed
# clang with --max-or-length=100 and --nested-switch=False is not slowed much by -Os (slowed considerably by -O3). -Os and -Oz gave same size in my tests.
# on 64-bit distros -m32 won't always work and won't necessarily give a smaller program
parser.add_option("--max-or-length",default=100,help="The maximum number of items allowed in an OR-expression in the C code (used when ybytes is in effect). When an OR-expression becomes larger than this limit, it will be made into a function. 0 means unlimited, which works for tcc and gcc; many other compilers have limits. Default: %default")
parser.add_option("--nested-switch",
action="store_true",default=False,
help="Allow C switch() constructs to be nested to arbitrary depth. This can result in a smaller executable, but it does slow down most C compilers.") # tcc is still fast (although that doesn't generate the smallest executables anyway)

Silas S. Brown
committed
parser.add_option("--outcode",default="utf-8",
help="Character encoding to use in the generated parser and rules summary (default %default, must be ASCII-compatible i.e. not utf-16)")
parser.add_option("-S", "--summary-only",
action="store_true",default=False,
help="Don't generate a parser, just write the rules summary to standard output")

Silas S. Brown
committed
parser.add_option("--no-summary",
action="store_true",default=False,
help="Don't add a rules summary at the end of the parser")

Silas S. Brown
committed
parser.add_option("-O", "--summary-omit",
help="Filename of a text file (or a compressed .gz or .bz2 file) specifying what should be omitted from the rules summary. Each line should be a word or phrase, a tab, and its annotation (without the mstart/mmid/mend markup). If any rule in the summary exactly matches any of the lines in this text file, then that rule will be omitted from the summary (but still included in the parser). Use for example to take out of the summary any entries that correspond to things you already have in your dictionary, so you can see what's new.")
parser.add_option("--maxrefs",default=3,
help="The maximum number of example references to record in each summary line, if references are being recorded. Default is %default; 0 means unlimited.")

Silas S. Brown
committed
parser.add_option("--norefs",
action="store_true",default=False,

Silas S. Brown
committed
help="Don't write references in the rules summary. Use this if you need to specify reference-sep and ref-name-end for the ref-pri option but you don't actually want references in the summary (omitting references makes summary generation faster). This option is automatically turned on if --no-input is specified.")
parser.add_option("--newlines-reset",
action="store_false",
dest="ignoreNewlines",
default=True,
help="Have the C program reset its state on every newline byte. By default newlines do not affect state such as whether a space is required before the next word, so that if the C program is used with Web Adjuster's htmlText option (which defaults to using newline separators) the spacing should be handled sensibly when there is HTML markup in mid-sentence.")

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--obfuscate",
action="store_true",default=False,

Silas S. Brown
committed
help="Obfuscate annotation strings in C code, as a deterrent to casual snooping of the compiled binary with tools like 'strings' (does NOT stop determined reverse engineering)")
parser.add_option("--javascript",
action="store_true",default=False,
help="Instead of generating C code, generate JavaScript. This might be useful if you want to run an annotator on a device that has a JS interpreter but doesn't let you run native code. The JS will be table-driven to make it load faster (and --no-summary will also be set). See comments at the start for usage.") # but it's better to use the C version if you're in an environment where 'standard input' makes sense

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("--python",
action="store_true",default=False,
help="Instead of generating C code, generate a Python module. Similar to the Javascript option, this is for when you can't run native code.")

Silas S. Brown
committed
parser.add_option("--reannotator",
help="Shell command through which to pipe each word of the original text to obtain new annotation for that word. This might be useful as a quick way of generating a new annotator (e.g. for a different topolect) while keeping the information about word separation and/or glosses from the previous annotator, but it is limited to commands that don't need to look beyond the boundaries of each word. (If the command is prefixed by a # character, it will be given the word's existing annotation instead of its original text.) The command should treat each line of its input independently, and both its input and its output should be in the encoding specified by --outcode.") # TODO: reannotatorCode instead? (see other 'reannotatorCode' TODOs)
# (Could just get the reannotator to post-process the 1st annotator's output, but that might be slower than generating an altered annotator with it)

Silas S. Brown
committed
# =========== ANALYSIS OPTIONS ==============
parser.add_option("-o", "--allow-overlaps",
action="store_true",default=False,
help="Normally, the analyser avoids generating rules that could overlap with each other in a way that would leave the program not knowing which one to apply. If a short rule would cause overlaps, the analyser will prefer to generate a longer rule that uses more context, and if even the entire phrase cannot be made into a rule without causing overlaps then the analyser will give up on trying to cover that phrase. This option allows the analyser to generate rules that could overlap, as long as none of the overlaps would cause actual problems in the example phrases. Thus more of the examples can be covered, at the expense of a higher risk of ambiguity problems when applying the rules to other texts.")
parser.add_option("-P", "--primitive",
action="store_true",default=False,
help="Don't bother with any overlap or conflict checks at all, just make a rule for each word. The resulting parser is not likely to be useful, but the summary might be.")

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("-y","--ybytes",default=0,

Silas S. Brown
committed
help="Look for candidate Yarowsky seed-collocations within this number of bytes of the end of a word. If this is set then overlaps and rule conflicts will be allowed if the seed collocations can be used to distinguish between them. Markup examples that are completely separate (e.g. sentences from different sources) must have at least this number of bytes between them.")
parser.add_option("--ybytes-max",default=0,
help="Extend the Yarowsky seed-collocation search to check over larger ranges up to this maximum. If this is set then several ranges will be checked in an attempt to determine the best one for each word, but see also ymax-threshold.")
parser.add_option("--ymax-threshold",default=1,
help="Limits the length of word that receives the narrower-range Yarowsky search when ybytes-max is in use. For words longer than this, the search will go directly to ybytes-max. This is for languages where the likelihood of a word's annotation being influenced by its immediate neighbours more than its distant collocations increases for shorter words, and less is to be gained by comparing different ranges when processing longer words. Setting this to 0 means no limit, i.e. the full range will be explored on ALL Yarowsky checks.")
parser.add_option("--ybytes-step",default=3,
help="The increment value for the loop between ybytes and ybytes-max")
parser.add_option("--warn-yarowsky",
action="store_true",default=False,
help="Warn when absolutely no distinguishing Yarowsky seed collocations can be found for a word in the examples")
parser.add_option("--yarowsky-all",
action="store_true",default=False,
help="Accept Yarowsky seed collocations even from input characters that never occur in annotated words (this might include punctuation and example-separation markup)")
parser.add_option("--yarowsky-debug",default=1,
help="Report the details of seed-collocation false positives if there are a large number of matches and at most this number of false positives (default %default). Occasionally these might be due to typos in the corpus, so it might be worth a check.")

Silas S. Brown
committed
parser.add_option("--single-words",

Silas S. Brown
committed
action="store_true",default=False,

Silas S. Brown
committed
help="Do not consider any rule longer than 1 word, although it can still have Yarowsky seed collocations if -y is set. This speeds up the search, but at the expense of thoroughness. You might want to use this in conjuction with -y to make a parser quickly. It is like -P (primitive) but without removing the conflict checks.")

Silas S. Brown
committed
parser.add_option("--max-words",default=0,
help="Limits the number of words in a rule; rules longer than this are not considered. 0 means no limit. --single-words is equivalent to --max-words=1. If you need to limit the search time, and are using -y, it should suffice to use --single-words for a quick annotator or --max-words=5 for a more thorough one.")
# TODO: optionally (especially if NOT using Yarowsky) do an additional pass (after discovering all other rules) and turn whole phrases that are not completely covered by other rules into whole-phrase rules, if it doesn't conflict 1 phrase w. anothr of equal priority; shld be ok if no overlap, overlaps wld *sometimes* be ok suggest a len threshold

Silas S. Brown
committed
parser.add_option("--checkpoint",help="Periodically save checkpoint files in the specified directory. These files can save time when starting again after a reboot (and it's easier than setting up Condor etc). As well as a protection against random reboots, this can be used for scheduled reboots: if file called ExitASAP appears in the checkpoint directory, annogen will checkpoint, remove the ExitASAP file, and exit. After a run has completed, the checkpoint directory should be removed, unless you want to re-do the last part of the run for some reason.")
# (Condor can checkpoint an application on Win/Mac/Linux but is awkward to set up. Various Linux and BSD application checkpoint approaches also exist; another option is virtualisation.)

Silas S. Brown
committed

Silas S. Brown
committed
parser.add_option("-d","--diagnose",help="Output some diagnostics for the specified word. Use this option to help answer \"why doesn't it have a rule for...?\" issues. This option expects the word without markup and uses the system locale (UTF-8 if it cannot be detected).")

Silas S. Brown
committed
parser.add_option("--diagnose-limit",default=10,help="Maximum number of phrases to print diagnostics for (0 means unlimited); can be useful when trying to diagnose a common word in rulesFile without re-evaluating all phrases that contain it. Default: %default")

Silas S. Brown
committed
parser.add_option("--time-estimate",
action="store_true",default=False,
help="Estimate time to completion. The code to do this is unreliable and is prone to underestimate. If you turn it on, its estimate is displayed at the end of the status line as days, hours or minutes.") # Unreliable because the estimate assumes 'phrases per minute' will remain constant on average, whereas actually it will decrease because the more complex phrases are processed last

Silas S. Brown
committed
sys.stderr.write(program_name+"\n") # not sys.stdout, because may or may not be showing --help (and anyway might want to process the help text for website etc)
options, args = parser.parse_args()
globals().update(options.__dict__)

Silas S. Brown
committed
sys.setcheckinterval(32767) # won't be using threads or signals, so don't have to check for them very often
import gc ; gc.disable() # should be OK if we don't create cycles (TODO: run gc.collect() manually after init, just in case?)

Silas S. Brown
committed
if primitive and ybytes: sys.stderr.write("Warning: primitive will override ybytes\n")
if ybytes: ybytes=int(ybytes)

Silas S. Brown
committed
if ybytes_max: ybytes_max=int(ybytes_max)
else: ybytes_max = ybytes
if yarowsky_debug: yarowsky_debug=int(yarowsky_debug)
else: yarowsky_debug = 0

Silas S. Brown
committed
ybytes_step = int(ybytes_step)

Silas S. Brown
committed
maxrefs = int(maxrefs)

Silas S. Brown
committed
ymax_threshold = int(ymax_threshold)

Silas S. Brown
committed
def errExit(msg):
sys.stderr.write(msg+"\n") ; sys.exit(1)
if ref_pri and not (reference_sep and ref_name_end): errExit("ref-pri option requires reference-sep and ref-name-end to be set")

Silas S. Brown
committed
if javascript or python:
if javascript and python: errExit("Outputting both Javascript and Python on the same run is not yet implemented")
if not outcode=="utf-8": errExit("outcode must be utf-8 when using Javascript or Python")
if obfuscate: errExit("obfuscate not yet implemented for the Javascript or Python version") # (and it would probably slow down the JS far too much if it were)
if c_filename.endswith(".c"):
if javascript: c_filename = c_filename[:-2]+".js"
else: c_filename = c_filename[:-2]+".py"
try:
import locale
terminal_charset = locale.getdefaultlocale()[1]
except: terminal_charset = "utf-8"
if diagnose: diagnose=diagnose.decode(terminal_charset)

Silas S. Brown
committed
diagnose_limit = int(diagnose_limit)

Silas S. Brown
committed
max_words = int(max_words)
if single_words: max_words = 1

Silas S. Brown
committed
def nearCall(conds,subFuncs,subFuncL):
# returns what to put in the if() for ybytes near() lists
if not max_or_length or len(conds) <= max_or_length:

Silas S. Brown
committed
return " || ".join("near(\""+c_escape(c,0)+"\")" for c in conds)

Silas S. Brown
committed
return subFuncCall("static int NewFunc() {\n"+"\n".join("if("+nearCall(conds[i:j],subFuncs,subFuncL)+") return 1;" for i,j in zip(range(0,len(conds),max_or_length),range(max_or_length,len(conds),max_or_length)+[len(conds)]))+"\nreturn 0;}",subFuncs,subFuncL)
def subFuncCall(newFunc,subFuncs,subFuncL):
if newFunc in subFuncs:
# we generated an identical one before
subFuncName=subFuncs[newFunc]
else:
subFuncName="match%d" % len(subFuncs)
subFuncs[newFunc]=subFuncName
subFuncL.append(newFunc.replace("NewFunc",subFuncName,1))
return subFuncName+"()" # the call (without a semicolon)

Silas S. Brown
committed

Silas S. Brown
committed
def stringSwitch(byteSeq_to_action_dict,subFuncL,funcName="topLevelMatch",subFuncs={}): # ("topLevelMatch" is also mentioned in the C code)

Silas S. Brown
committed
# make a function to switch on a large number of variable-length string cases without repeated lookahead for each case
# (may still backtrack if no words or no suffices match)

Silas S. Brown
committed
# byteSeq_to_action_dict is really a byte sequence to [(action, OR-list of Yarowsky-like indicators which are still in Unicode)], the latter will be c_escape()d
# can also be byte seq to [(action,(OR-list,nbytes))] but only if OR-list is not empty, so value[1] will always be false if OR-list is empty
allBytes = set(b[0] for b in byteSeq_to_action_dict.iterkeys() if b)

Silas S. Brown
committed
ret = []
if funcName:

Silas S. Brown
committed
ret.append("static void %s() {" % funcName)

Silas S. Brown
committed
savePos = len(ret)
ret.append("{ SAVEPOS;")
elif "" in byteSeq_to_action_dict and len(byteSeq_to_action_dict) > 1:
# no funcName, but might still want to come back here as there's a possible action at this level
savePos = len(ret)
ret.append("{ SAVEPOS;")
else: savePos = None

Silas S. Brown
committed
called_subswitch = False

Silas S. Brown
committed
if "" in byteSeq_to_action_dict and len(byteSeq_to_action_dict) > 1 and len(byteSeq_to_action_dict[""])==1 and not byteSeq_to_action_dict[""][0][1] and all((len(a)==1 and a[0][0].startswith(byteSeq_to_action_dict[""][0][0]) and not a[0][1]) for a in byteSeq_to_action_dict.itervalues()):
# there's an action in common for this and all subsequent matches, and no Yarowsky-like indicators, so we can do the common action up-front
ret.append(byteSeq_to_action_dict[""][0][0])
l = len(byteSeq_to_action_dict[""][0][0])
byteSeq_to_action_dict = dict((x,[(y[l:],z)]) for x,[(y,z)] in byteSeq_to_action_dict.iteritems())

Silas S. Brown
committed
# and, since we'll be returning no matter what,
# we can put the inner switch in a new function

Silas S. Brown
committed
# (even if not re-used, this helps compiler speed)

Silas S. Brown
committed
# + DON'T save/restore pos around it (it itself
# will do any necessary save/restore pos)
del ret[savePos] ; savePos = None
del byteSeq_to_action_dict[""]

Silas S. Brown
committed
newFunc = "\n".join(stringSwitch(byteSeq_to_action_dict,subFuncL,"NewFunc",subFuncs))

Silas S. Brown
committed
byteSeq_to_action_dict[""] = [("",[])] # for the end of this func

Silas S. Brown
committed
ret.append(subFuncCall(newFunc,subFuncs,subFuncL)+"; return;")

Silas S. Brown
committed
elif allBytes:
# deal with all actions except "" first

Silas S. Brown
committed
use_if = (len(allBytes)==1)
if not use_if: ret.append("switch(NEXTBYTE) {")

Silas S. Brown
committed
for case in sorted(allBytes):
if 32<=ord(case)<127 and case!="'": cstr="'%c'" % case
else: cstr=ord(case)

Silas S. Brown
committed
if use_if: ret.append("if(NEXTBYTE==%s) {" % cstr)
else: ret.append("case %s:" % cstr)

Silas S. Brown
committed
inner = stringSwitch(dict([(k[1:],v) for k,v in byteSeq_to_action_dict.iteritems() if k and k[0]==case]),subFuncL,None,subFuncs)

Silas S. Brown
committed
need_break = not (use_if or inner[-1].endswith("return;"))
if nested_switch or not inner[0].startswith("switch"): ret += [" "+x for x in inner]
else:
# Put the inner switch into a different function
# which returns 1 if we should return.
# (TODO: this won't catch cases where there's a SAVEPOS before the inner switch; will still nest in that case. But it shouldn't lead to big nesting in practice.)
myFunc=["static int NewFunc() {"]
for x in inner:
if x.endswith("return;"): x=x[:-len("return;")]+"return 1;"
myFunc.append(" "+x)
ret.append(" if("+subFuncCall("\n".join(myFunc)+" return 0;\n}",subFuncs,subFuncL)+") return;")
called_subswitch=True # as it'll include more NEXTBYTE calls which are invisible to the code below
if need_break: ret.append(" break;")

Silas S. Brown
committed
ret.append("}") # end of switch or if

Silas S. Brown
committed
if not savePos==None:

Silas S. Brown
committed
if len(' '.join(ret).split('NEXTBYTE'))==2 and not called_subswitch:

Silas S. Brown
committed
# only 1 NEXTBYTE after the SAVEPOS - just
# do a PREVBYTE instead
# (note however that splitting on NEXTBYTE
# does not necessarily give a reliable value
# for max amount of lookahead required if
# there's more than 1. We use max rule len
# as an upper bound for that instead.)
del ret[savePos]
ret.append("PREVBYTE;")
else: ret.append("RESTOREPOS; }")
if funcName: ret.append("}")
elif "" in byteSeq_to_action_dict:
# if the C code gets to this point, no return; happened - no suffices

Silas S. Brown
committed
# so execute one of the "" actions and return
# (which one, if any, depends on the Yarowsky-like indicators; there should be at most one "default" action without indicators)
default_action = ""
for action,conds in byteSeq_to_action_dict[""]:
if conds:
assert action, "conds without action in "+repr(byteSeq_to_action_dict[""])
if type(conds)==tuple:
conds,nbytes = conds
ret.append("setnear(%d);" % nbytes)

Silas S. Brown
committed
ret.append("if ("+nearCall(conds,subFuncs,subFuncL)+") {")

Silas S. Brown
committed
ret.append((action+" return;").strip())
ret.append("}")
else:
if default_action: sys.stderr.write("WARNING! More than one default action in "+repr(byteSeq_to_action_dict[""])+" - earlier one discarded!\n(This might indicate invalid markup in the corpus)\n") # see TODO in yarowsky_indicators. (TODO: we sometimes also get this if an incremental run has updated the annotation; why doesn't remove_old_rules prevent this? see also the TODO there re yBytesRet)

Silas S. Brown
committed
default_action = action
if default_action or not byteSeq_to_action_dict[""]: ret.append((default_action+" return;").strip()) # (return only if there was a default action, OR if an empty "" was in the dict with NO conditional actions (e.g. from the common-case optimisation above). Otherwise, if there were conditional actions but no default, we didn't "match" anything if none of the conditions were satisfied.)

Silas S. Brown
committed
return ret # caller does '\n'.join

Silas S. Brown
committed
if obfuscate:
import random ; pad=[]
for i in xrange(32): pad.append(random.randint(128,255))
unobfusc_func=r"""
void OutWriteDecode(const char *s) {
static const char pad[]={%s}; int i=0;
while(*s) {
int t=pad[i++]; if(i==sizeof(pad)) i=0;
if(*s==t) OutWriteByte(t); else OutWriteByte((*s)^t); s++;
}
}""" % repr(pad)[1:-1]
def encodeOutstr(s):
i = 0 ; r = []
for c in s:
t = pad[i] ; i = (i+1) % len(pad)
if ord(c) == t: toApp = t
else: toApp = ord(c)^t
if toApp==ord("\\"): r.append(r'\\')
elif toApp==ord('"'): r.append(r'\"')
elif toApp&0x80 or toApp<32: r.append(r'\x%x" "' % toApp)
else: r.append(chr(toApp))
return ''.join(r)
else: unobfusc_func = ""

Silas S. Brown
committed
c_start = "/* -*- coding: "+outcode+r""" -*- */

Silas S. Brown
committed
#include <stdio.h>
#include <string.h>
/* To include this code in another program,
define the ifndef'd macros below + define Omit_main */

Silas S. Brown
committed
enum { ybytes = %%YBYTES%% }; /* for Yarowsky matching, minimum readahead */
static int nearbytes = ybytes;
#define setnear(n) (nearbytes = (n))

Silas S. Brown
committed
#ifndef NEXTBYTE

Silas S. Brown
committed
/* Default definition of NEXTBYTE etc is to read input
from stdin and write output to stdout. */

Silas S. Brown
committed
enum { Half_Bufsize = %%LONGEST_RULE_LEN%% };
static unsigned char lookahead[Half_Bufsize*2];

Silas S. Brown
committed
static size_t readPtr=0,writePtr=0,bufStart=0,bufLen=0;

Silas S. Brown
committed
static int nextByte() {

Silas S. Brown
committed
if (readPtr-bufStart +ybytes >= bufLen) {

Silas S. Brown
committed
if (bufLen == Half_Bufsize * 2) {
memmove(lookahead,lookahead+Half_Bufsize,Half_Bufsize);
bufStart += Half_Bufsize; bufLen -= Half_Bufsize;
}
bufLen += fread(lookahead+bufLen,1,Half_Bufsize*2-bufLen,stdin);

Silas S. Brown
committed
if (readPtr-bufStart == bufLen) return EOF;

Silas S. Brown
committed
}

Silas S. Brown
committed
return lookahead[(readPtr++)-bufStart];

Silas S. Brown
committed
}
static int near(char* string) {

Silas S. Brown
committed
/* for Yarowsky-like matching */

Silas S. Brown
committed
size_t offset = readPtr-bufStart, l=strlen(string),

Silas S. Brown
committed
maxPos = bufLen;

Silas S. Brown
committed
if (maxPos >= l) maxPos -= l; else return 0; // can't possibly start after maxPos-l
if (offset+nearbytes>l) {
if (maxPos > offset+nearbytes-l)
maxPos = offset+nearbytes-l;
} else maxPos = 0; // (don't let it go below 0, as size_t is usually unsigned)

Silas S. Brown
committed
if (offset>nearbytes) offset-=nearbytes; else offset = 0;
// can use strnstr(haystack,needle,n) if on a BSD system

Silas S. Brown
committed
while (offset <= maxPos) {

Silas S. Brown
committed
if(!strncmp((char*)lookahead+offset,string,l)) return 1;

Silas S. Brown
committed
offset++;
}
return 0;
}
#define NEXTBYTE nextByte()

Silas S. Brown
committed
#define NEXT_COPY_BYTE lookahead[(writePtr++)-bufStart]
#define COPY_BYTE_SKIP writePtr++

Silas S. Brown
committed
#define POSTYPE size_t

Silas S. Brown
committed
#define THEPOS readPtr /* or get it via a function */

Silas S. Brown
committed
#define SAVEPOS POSTYPE oldPos=THEPOS

Silas S. Brown
committed
#define RESTOREPOS readPtr=oldPos /* or set via a func */
#define PREVBYTE readPtr--
#define FINISHED (feof(stdin) && readPtr-bufStart == bufLen)

Silas S. Brown
committed
#define OutWriteStr(s) fputs(s,stdout)
#define OutWriteByte(c) putchar(c)
#endif
#ifndef Default_Annotation_Mode
#define Default_Annotation_Mode ruby_markup
#endif
enum {
annotations_only,
ruby_markup,
brace_notation} annotation_mode = Default_Annotation_Mode;

Silas S. Brown
committed
static int needSpace=0;
static void s() {
if (needSpace) OutWriteByte(' ');
else needSpace=1; /* for after the word we're about to write (if no intervening bytes cause needSpace=0) */

Silas S. Brown
committed
}""" + unobfusc_func + r"""

Silas S. Brown
committed

Silas S. Brown
committed
static void o(int numBytes,const char *annot) {

Silas S. Brown
committed
s();

Silas S. Brown
committed
switch (annotation_mode) {

Silas S. Brown
committed
case annotations_only: OutWriteDecode(annot); break;
case ruby_markup:
OutWriteStr("<ruby><rb>");
for(;numBytes;numBytes--)
OutWriteByte(NEXT_COPY_BYTE);
OutWriteStr("</rb><rt>"); OutWriteDecode(annot);
OutWriteStr("</rt></ruby>"); break;
case brace_notation:
OutWriteByte('{');
for(;numBytes;numBytes--)
OutWriteByte(NEXT_COPY_BYTE);
OutWriteByte('|'); OutWriteDecode(annot);
OutWriteByte('}'); break;

Silas S. Brown
committed
}
}

Silas S. Brown
committed
static void o2(int numBytes,const char *annot,const char *title) {

Silas S. Brown
committed
if (annotation_mode == ruby_markup) {
s();

Silas S. Brown
committed
OutWriteStr("<ruby title=\""); OutWriteDecode(title);
OutWriteStr("\"><rb>");
for(;numBytes;numBytes--)
OutWriteByte(NEXT_COPY_BYTE);
OutWriteStr("</rb><rt>"); OutWriteDecode(annot);
OutWriteStr("</rt></ruby>");
} else o(numBytes,annot);

Silas S. Brown
committed
}

Silas S. Brown
committed
"""

Silas S. Brown
committed
if not obfuscate: c_start = c_start.replace("OutWriteDecode","OutWriteStr")

Silas S. Brown
committed
c_end = """
void matchAll() {
while(!FINISHED) {
POSTYPE oldPos=THEPOS;

Silas S. Brown
committed
topLevelMatch();

Silas S. Brown
committed
if (oldPos==THEPOS) { needSpace=0; OutWriteByte(NEXTBYTE); COPY_BYTE_SKIP; }

Silas S. Brown
committed
}
}
#ifndef Omit_main

Silas S. Brown
committed
int main(int argc,char*argv[]) {

Silas S. Brown
committed
int i; for(i=1; i<argc; i++) {
if(!strcmp(argv[i],"--help")) {
puts("Use --ruby to output ruby markup (default)");
puts("Use --raw to output just the annotations without the base text");
puts("Use --braces to output as {base-text|annotation}");
return 0;
} else if(!strcmp(argv[i],"--ruby")) {
annotation_mode = ruby_markup;
} else if(!strcmp(argv[i],"--raw")) {
annotation_mode = annotations_only;
} else if(!strcmp(argv[i],"--braces")) {
annotation_mode = brace_notation;
}
}
matchAll();
}
#endif
"""

Silas S. Brown
committed
class BytecodeAssembler:

Silas S. Brown
committed
# Bytecode for a virtual machine run by the Javascript version etc

Silas S. Brown
committed
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
def __init__(self):
self.l = []
self.d2l = {}
self.lastLabelNo = 0
self.addingPosStack = []
def addOpcode(self,opcode):
self.addBytes({
'jump': 50, # params: address
'call': 51, # params: function address
'return': 52, # (or 'end program' if top level)
'switchbyte': 60, # switch(NEXTBYTE) (params: numBytes-1, bytes (sorted, TODO take advantage of this), addresses, default address)
'copyBytes':71,'o':72,'o2':73, # (don't change these numbers, they're hard-coded below)
'savepos':80, # local to the function
'restorepos':81,
'neartest':90, # params: true-label, false-label, byte nbytes, addresses of conds strings until true-label reached
}[opcode])
def addBytes(self,bStr):
if type(bStr)==int: self.l.append(chr(bStr))
elif type(bStr)==str: self.l.append(bStr)
else: raise Exception("unspported bytes type")
def startAddingFunction(self):
self.addingPosStack.append((len(self.l),self.lastLabelNo))
self.lastLabelNo = 0
def finishFunctionAndAddCall(self):
# make sure to add a return instruction before this!
fPtr, self.lastLabelNo = self.addingPosStack[-1]
del self.addingPosStack[-1]
fBody = tuple(self.l[fPtr:]) ; self.l=self.l[:fPtr]
if not fBody in self.d2l: # not a duplicate
self.d2l[fBody] = (-len(self.d2l)-1,)
self.addOpcode('call')
self.l.append(self.d2l[fBody])
def addByteswitch(self,byteArray,labelArray):
assert len(byteArray) + 1 == len(labelArray)
# labelArray has the default case added also (TODO: could re-organize code so the bytes immediately after the switch are either the default or one of the items, saving 1 address)
if not len(byteArray): return # empty switch = no-op
self.addOpcode('switchbyte')
self.addBytes(len(byteArray)-1) # num of bytes in list - 1 (so all 256 values can be accounted for if needed)
self.addBytes("".join(byteArray))
for i in labelArray: self.addRef(i)
def addActions(self,actionList):
for a in actionList:
assert 1 <= len(a) <= 3
assert type(a[0])==int and 1 <= a[0] <= 255, "bytecode currently supports markup or copy between 1 and 255 bytes only (but 0 is reserved for expansion)"
self.addBytes(70+len(a)) # 71=copyBytes 72=o() 73=o2
self.addBytes(a[0])
for i in a[1:]: self.addRefToString(i)
def addActionDictSwitch(self,byteSeq_to_action_dict,isFunc=True,labelToJump=None):
# a modified stringSwitch for the bytecode
# Actions aren't strings: they list tuples of either
# 1, 2 or 3 items for copyBytes, o(), o2()
# labelToJump is a jump to insert afterwards if not isFunc and if we don't emit an unconditional 'return'. Otherwise, will ALWAYS end up with a 'return' (even if not isFunc i.e. the main program)
allBytes = set(b[0] for b in byteSeq_to_action_dict.iterkeys() if b)
if isFunc:
self.startAddingFunction()
savePos = len(self.l)
self.addOpcode('savepos')
elif ("" in byteSeq_to_action_dict and len(byteSeq_to_action_dict) > 1) or not labelToJump: # ('not labelToJump' and 'not isFunc' == main program)
savePos = len(self.l)
self.addOpcode('savepos')
else: savePos = None
if "" in byteSeq_to_action_dict and len(byteSeq_to_action_dict) > 1 and len(byteSeq_to_action_dict[""])==1 and not byteSeq_to_action_dict[""][0][1] and all((len(a)==1 and a[0][0][:len(byteSeq_to_action_dict[""][0][0])]==byteSeq_to_action_dict[""][0][0] and not a[0][1]) for a in byteSeq_to_action_dict.itervalues()):
self.addActions(byteSeq_to_action_dict[""][0][0])
l = len(byteSeq_to_action_dict[""][0][0])
byteSeq_to_action_dict = dict((x,[(y[l:],z)]) for x,[(y,z)] in byteSeq_to_action_dict.iteritems())
del self.l[savePos] ; savePos = None
del byteSeq_to_action_dict[""]
self.addActionDictSwitch(byteSeq_to_action_dict) # as a subfunction (ends up adding the call to it)
byteSeq_to_action_dict[""] = [("",[])] # for the end of this func
self.addOpcode('return')
elif allBytes:
allBytes = list(allBytes)
labels = [self.makeLabel() for b in allBytes+[0]]
self.addByteswitch(allBytes,labels)
for case in allBytes:
self.addLabelHere(labels[0]) ; del labels[0]
self.addActionDictSwitch(dict([(k[1:],v) for k,v in byteSeq_to_action_dict.iteritems() if k and k[0]==case]),False,labels[-1])
self.addLabelHere(labels[0])
if not savePos==None: self.addOpcode('restorepos')
if isFunc:
self.addOpcode('return')
if self.l[-1]==self.l[-2]: del self.l[-1] # double return
return self.finishFunctionAndAddCall()
elif "" in byteSeq_to_action_dict:
default_action = ""
for action,conds in byteSeq_to_action_dict[""]:
if conds:
if type(conds)==tuple: conds,nbytes = conds
else: nbytes = ybytes_max
assert 1 <= nbytes <= 255, "bytecode supports only single-byte nbytes (but nbytes=0 is reserved for expansion)"
trueLabel,falseLabel = self.makeLabel(),self.makeLabel()
self.addOpcode('neartest')
self.addRef(trueLabel)
self.addRef(falseLabel)
assert type(nbytes)==int
self.addBytes(nbytes)
for c in conds: self.addRefToString(c.encode(outcode))
self.addLabelHere(trueLabel)
self.addActions(action)
self.addOpcode('return')
self.addLabelHere(falseLabel)
else: default_action = action
if default_action or not byteSeq_to_action_dict[""]:
self.addActions(default_action)
self.addOpcode('return') ; return
if labelToJump:
self.addOpcode('jump')
self.addRef(labelToJump)
else: self.addOpcode('return')
def makeLabel(self):
self.lastLabelNo += 1
return self.lastLabelNo
def addLabelHere(self,labelNo):
assert type(labelNo)==int
assert labelNo, "label 0 not allowed"
self.l.append(labelNo)
def addRef(self,labelNo):
assert type(labelNo)==int
self.l.append(-labelNo)
def addRefToString(self,string):
# prepends with a length hint if possible (or if not,
# prepends with 0 and null-terminates it)
assert type(string)==str
if 1 <= len(string) < 256:
string = chr(len(string))+string
else: string = chr(0)+string+chr(0)
if not string in self.d2l:
self.d2l[string] = (-len(self.d2l)-1,)
self.l.append(self.d2l[string])
def link(self): # returns resulting bytes
# (add an 'end program' instruction before calling)
def f(*args): raise Exception("Must call link() only once")
self.link = f
sys.stderr.write("Linking... ")
for dat,ref in self.d2l.iteritems():
assert type(ref)==tuple and type(ref[0])==int
self.l.append((-ref[0],)) # the label
if type(dat)==str:
self.l.append(dat) ; continue
# otherwise it's a function, and non-reserved labels are local, so we need to rename them
l2l = {}
for i in dat:
if type(i)==int:
if i>0: j=i
else: j=-i
if not j in l2l:
l2l[j] = self.makeLabel()
if i>0: self.addLabelHere(l2l[j])
else: self.addRef(l2l[j])
else: self.l.append(i) # str or tuple just cp
del self.d2l
# elements of self.l are now:
# - byte strings (just copied in)
# - positive integers (labels)
# - negative integers (references to labels)
# - +ve or -ve integers in tuples (reserved labels)
# 1st byte of o/p is num bytes needed per address
class TooNarrow(Exception): pass
for numBytes in xrange(1,256):
sys.stderr.write("(%d-bit) " % (numBytes*8))
try:
lDic = {} ; r = [chr(numBytes)]
for P in [1,2]:
ll = 1
for i in self.l:
if type(i) in [int,tuple]:
if type(i)==int: i2,iKey = i,-i
else: i2,iKey = i[0],(-i[0],)
assert type(i2)==int
if i2 > 0: # label going in here
if i in lDic: assert lDic[i] == ll, "changing label %s from %d to %d, P=%d" % (repr(i),lDic[i],ll,P)
else: lDic[i] = ll
continue
elif iKey in lDic: # known label
i = lDic[iKey]
shift = 8*numBytes
if (i >> shift): raise TooNarrow()
j = []
for b in xrange(numBytes):
# MSB-LSB (easier to do in JS)
shift -= 8
j.append(chr((i>>shift)&0xFF))
i = "".join(j)
assert len(i)==numBytes
else: # as-yet unknown label
assert P==1, "undefined label %d" % -i
ll += numBytes ; continue
if P==2: r.append(i)
ll += len(i)
sys.stderr.write("%d bytes\n" % ll)
return "".join(r)
except TooNarrow: pass
assert 0, "can't even assemble it with 255-byte addressing !?!"
js_start = r"""/* Javascript generated by """+program_name[:program_name.index("(c)")].strip()+r"""
Usage:
- You could just include this code and then call the
annotate() function i.e. var result = annotate(input)
- Or you could use (and perhaps extend) the Annotator
object, and call its annotate() method. If you have
Backbone.JS, Annotator will instead be a generator
(extending Backbone.Model) which you will have to
instantiate yourself (possibly after extending it).
The Annotator object/class is also what will be
exported by this module if you're using Common.JS.
- On Unix systems with Node.JS, you can run this file in
"node" to annotate standard input as a simple test.
*/
var Annotator={
"""
js_end = r"""
annotate: function(input) {
/* TODO: if input is a whole html doc, insert css in head
(e.g. from annoclip and/or adjuster), and hope there's
no stuff that's not to be annotated (form fields...) */
input = unescape(encodeURIComponent(input)); // to UTF-8
var data = this.data;
var addrLen = data.charCodeAt(0);
var dPtr;
var inputLength = input.length;
var p = 0; // read-ahead pointer
var copyP = 0; // copy pointer
var output = new Array();
var needSpace = 0;
function readAddr() {
var i,addr=0;
for (i=addrLen; i; i--) addr=(addr << 8) | data.charCodeAt(dPtr++);
return addr;
}
function readRefStr() {
var a = readAddr(); var l=data.charCodeAt(a);
if (l != 0) return data.slice(a+1,a+l+1);
else return data.slice(a+1,data.indexOf('\x00',a+1));
}
function s() {
if (needSpace) output.push(" ");
else needSpace=1; // for after the word we're about to write (if no intervening bytes cause needSpace=0)
}
function readData() {
var sPos = new Array();
while(1) {
switch(data.charCodeAt(dPtr++)) {
case 50: dPtr = readAddr(); break;
case 51: {
var f = readAddr(); var dO=dPtr;
dPtr = f; readData() ; dPtr = dO;
break; }
case 52: return;
case 60: {
var nBytes = data.charCodeAt(dPtr++)+1;
var i = data.slice(dPtr,dPtr+nBytes).indexOf(input.charAt(p++));
if (i==-1) i = nBytes;
dPtr += (nBytes + i * addrLen);
dPtr = readAddr(); break; }
case 71: {
var numBytes = data.charCodeAt(dPtr++);
output.push(input.slice(copyP,copyP+numBytes));
copyP += numBytes; break; }
case 72: {
var numBytes = data.charCodeAt(dPtr++);
var annot = readRefStr();
s();
output.push("<ruby><rb>");
output.push(input.slice(copyP,copyP+numBytes));
copyP += numBytes;
output.push("</rb><rt>"); output.push(annot);
output.push("</rt></ruby>"); break; }
case 73: {
var numBytes = data.charCodeAt(dPtr++);
var annot = readRefStr();
var title = readRefStr();
s();
output.push("<ruby title=\""); output.push(title);
output.push("\"><rb>");
output.push(input.slice(copyP,copyP+numBytes));
copyP += numBytes;
output.push("</rb><rt>"); output.push(annot);
output.push("</rt></ruby>"); break; }
case 80: sPos.push(p); break;
case 81: p=sPos.pop(); break;
case 90: {
var tPtr = readAddr();
var fPtr = readAddr();
var nearbytes = data.charCodeAt(dPtr++);
var o=p;
if (o > nearbytes) o -= nearbytes; else o = 0;
var max = p + nearbytes;
if (max > inputLength) max = inputLength;
var tStr = input.slice(o,max);
var found = 0;
while (dPtr < tPtr) if (tStr.indexOf(readRefStr()) != -1) { found = 1; break; }
dPtr = found ? tPtr : fPtr; break;
}
default: throw("corrupt data table at "+(dPtr-1)+" ("+data.charCodeAt(dPtr-1)+")");
}
}
}
while(p < inputLength) {
var oldPos=p;
dPtr=1;readData();
if (oldPos==p) { needSpace=0; output.push(input.charAt(p++)); copyP++; }
}
return decodeURIComponent(escape(output.join(""))); // from UTF-8 back to Unicode
} // end of annotate function
};
function annotate(input) { return Annotator.annotate(input); }
if (typeof Backbone != "undefined" && Backbone.Model) { Annotator = Backbone.Model.extend(Annotator); annotate=function(input) { return new Annotator().annotate(input) } }
if (typeof require != "undefined" && typeof module != "undefined" && require.main === module) {
// Node.js command-line test
fs=require('fs');
process.stdout.write(annotate(fs.readFileSync('/dev/stdin').toString()));
} else if (typeof module != "undefined" && module.exports) { // Common.js
module.exports = Annotator;
}
"""

Silas S. Brown
committed
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
py_start = r"""# Python generated by """+program_name[:program_name.index("(c)")].strip()+r"""
# You can import this module and call annotate(utf8 bytes)
# (from multiple threads if desired),
# or you can run from the command line on standard input.
# annotate has an optional second argument, which can be
# 'ruby' (default), 'raw' (annotation only) or 'braces'.
"""
py_end = r"""
class Annotator:
def __call__(self,inStr,aType):
if aType=="ruby": self.startA,self.midA,self.endA = "<ruby><rb>","</rb><rt>","</rt></ruby>"
elif aType=="raw": self.startA=self.midA=self.endA = ""
elif aType=="braces": self.startA,self.midA,self.endA = "{","|","}"
else: raise Exception("Unrecognised annotation type "+repr(aType))
assert type(inStr)==str
self.inStr = inStr
self.addrLen = ord(data[0])
self.inputLength = len(inStr)
self.p = 0 # read-ahead pointer
self.copyP = 0 # copy pointer
self.output = []
self.needSpace = 0 ; out = self.output
while self.p < self.inputLength:
oldPos = self.p
self.dPtr = 1 ; self.readData()
if oldPos == self.p:
self.needSpace=0
out.append(inStr[self.p])
self.p += 1 ; self.copyP += 1
return "".join(self.output)
def readAddr(self):
addr = 0
for i in range(self.addrLen):
addr=(addr << 8) | ord(data[self.dPtr])
self.dPtr += 1
return addr
def readRefStr(self):
a = self.readAddr(); l=ord(data[a])
if l: return data[a+1:a+l+1]
else: return data[a+1:data.index('\x00',a+1)]
def s(self):
if self.needSpace: self.output.append(" ")
else: self.needSpace=1
def readData(self):
sPos = [] ; out = self.output
while True:
d = ord(data[self.dPtr]) ; self.dPtr += 1
if d==50: self.dPtr = self.readAddr()
elif d==51:
func = self.readAddr() ; dO = self.dPtr
self.dPtr = func ; self.readData() ; self.dPtr = dO
elif d==52: return
elif d==60:
nBytes = ord(data[self.dPtr])+1 ; self.dPtr += 1
i = data[self.dPtr:self.dPtr+nBytes].find(self.inStr[self.p]) ; self.p += 1
if i==-1: i = nBytes
self.dPtr += (nBytes + i * self.addrLen)
self.dPtr = self.readAddr()
elif d==71:
numBytes = ord(data[self.dPtr]) ; self.dPtr += 1
out.append(self.inStr[self.copyP:self.copyP+numBytes])
self.copyP += numBytes
elif d==72:
numBytes = ord(data[self.dPtr]) ; self.dPtr += 1
annot = self.readRefStr()
self.s()
if self.startA:
out.append(self.startA)
out.append(self.inStr[self.copyP:self.copyP+numBytes])
self.copyP += numBytes
out.append(self.midA) ; out.append(annot)
out.append(self.endA)
elif d==73:
numBytes = ord(data[self.dPtr]) ; self.dPtr += 1
annot = self.readRefStr()
title = self.readRefStr()
self.s()
if self.startA=="{": # omit title in braces mode
out.append(self.startA)
out.append(self.inStr[self.copyP:self.copyP+numBytes])
elif self.startA:
out.append("<ruby title=\"");out.append(title)
out.append("\"><rb>");
out.append(self.inStr[self.copyP:self.copyP+numBytes])
self.copyP += numBytes
out.append(self.midA) ; out.append(annot)
out.append(self.endA)
elif d==80: sPos.append(self.p)
elif d==81: self.p = sPos.pop()
elif d==90:
tPtr = self.readAddr()
fPtr = self.readAddr()
nearbytes = ord(data[self.dPtr]) ; self.dPtr += 1
o = max(self.p-nearbytes,0)
maxx = min(self.p+nearbytes,self.inputLength)
tStr = self.inStr[o:maxx]
found = 0
while self.dPtr < tPtr:
if self.readRefStr() in tStr:
found = 1 ; break
if found: self.dPtr = tPtr
else: self.dPtr = fPtr
else: raise Exception("corrupt data table at "+str(self.dPtr-1)+" ("+str(ord(data[self.dPtr-1]))+")")
def annotate(inStr,p="ruby"): return Annotator()(inStr,p)
def main():
import sys
if sys.argv[-1].startswith("--"): param=sys.argv[-1][2:]
else: param = "ruby"
sys.stdout.write(annotate(sys.stdin.read(),param))
if __name__=="__main__": main()
"""

Silas S. Brown
committed
def splitWords(text,phrases=False):
# split text into words, ignoring anything between markupStart and markupEnd
# if phrases = True, instead of words, split on any non-whitespace char outside markupStart..markupEnd

Silas S. Brown
committed
i=start=0

Silas S. Brown
committed
text = text.replace(markupEnd+markupStart, markupEnd+' '+markupStart) # force at least one breakpoint between each marked-up phrase (otherwise get problems later - much code assumes each item returned by splitWords contains at most 1 markup)

Silas S. Brown
committed
def isSplitpoint():
isspace = not text[i].split()
if phrases: return not isspace
else: return isspace
while i<len(text):
if text[i:i+len(markupStart)]==markupStart:
i = text.find(markupEnd,i+len(markupStart))
if i==-1: i=len(text)
else: i += len(markupEnd)
elif isSplitpoint():

Silas S. Brown
committed
if i>start: yield text[start:i]

Silas S. Brown
committed
if phrases:
# can skip to text markupStart
i=text.find(markupStart,i)
if i==-1: i=len(text)
start = i
else:
i += 1 # just after the 1st splitter
while i<len(text) and isSplitpoint(): i += 1
start = i # 1st non-splitter char
else: i += 1

Silas S. Brown
committed
if i>start: yield text[start:i]

Silas S. Brown
committed

Silas S. Brown
committed
markupPattern = re.compile(re.escape(markupStart)+"(.*?)"+re.escape(markupMid)+"(.*?)"+re.escape(markupEnd))
whitespacePattern = re.compile(r"\s+")

Silas S. Brown
committed
def annotationOnly(text):
ret = []

Silas S. Brown
committed
for w in re.finditer(markupPattern,text):
if mreverse: ret.append(w.group(1))
else: ret.append(w.group(2))

Silas S. Brown
committed
return ' '.join(ret)
def markDown(text):
# Return just the original text, without markup

Silas S. Brown
committed
if mreverse: group=r"\2"
else: group=r"\1"
text = re.sub(markupPattern,group,text)
if removeSpace: text=re.sub(whitespacePattern,"",text)
return text

Silas S. Brown
committed

Silas S. Brown
committed
def markUp(text,annotation):
if mreverse: text,annotation = annotation,text
return markupStart + text + markupMid + annotation + markupEnd

Silas S. Brown
committed
def checkpoint_exit(doIt=1):
if not checkpoint: return
try: open(checkpoint+os.sep+"ExitASAP")
except: return
if doIt:
os.remove(checkpoint+os.sep+"ExitASAP")
sys.stderr.write("\nExitASAP found: exit\n")
raise SystemExit