o rZh+"@s~dZddlZddlmZdddZdddZdd d Zdd d Zd dZdddZ ddZ ddZ e dkr=e e dSdS)z Utility functions for batch-processing sentences: parsing and extraction of the semantic representation of the root node of the the syntax tree, followed by evaluation of the semantic representation in a first-order model. Nevaluatec Csnddlm}ddlm}m}t||r||}n|||d}g}|D]}|} t|| } | | q"|S)a Convert input sentences into syntactic trees. :param inputs: sentences to be parsed :type inputs: list(str) :param grammar: ``FeatureGrammar`` or name of feature-based grammar :type grammar: nltk.grammar.FeatureGrammar :rtype: list(nltk.tree.Tree) or dict(list(str)): list(Tree) :return: a mapping from input sentences to a list of ``Tree`` instances. rFeatureGrammar)FeatureChartParser load_parsertrace) nltk.grammarrZ nltk.parserr isinstancesplitlistparseappend) inputsgrammarr rrrcpZparsessenttokenssyntreesrVsz.interpret_sents...r)r$rr&rrr'Usz#interpret_sents..r)r)rrr!r rr&rinterpret_sentsIs r(csfddt||DS)a Add the truth-in-a-model value to each semantic representation for each syntactic parse of each input sentences. :param inputs: a list of sentences :type inputs: list(str) :param grammar: ``FeatureGrammar`` or name of feature-based grammar :type grammar: nltk.grammar.FeatureGrammar :return: a mapping from sentences to lists of triples (parse-tree, semantic-representations, evaluation-in-model) :rtype: list(list(tuple(nltk.tree.Tree, nltk.sem.logic.ConstantExpression, bool or dict(str): bool))) cs"g|] }fdd|DqS)c s*g|]\}}||jd|dfqS)z%srr)r$r%sem assignmentmodelr rrr'hsz-evaluate_sents...r)r$Zinterpretationsr*rrr'gs z"evaluate_sents..)r()rrr,r+r rr*revaluate_sents[s r-cCsddddddddd hfd d d hfd ddhfdddhfdhdfdhdfdhdfdhdfdhdfg}t|}|j}t||at|adS)N)Zjohnb1)Zmaryg1)Zsuzieg2)Zfidod1)Ztessd2)Znoosangirlr/r0boyr.b2dogr1r2Zbarkwalk>r0r.r1Zchase>r.r/)r/r1)r6r/)r0r2Zsee>)r6r2)r0r3)r2r.r/r.r9in>)r2r3)r6r3)r.r3with>r9)r1r.r:)r.r1)rZ ValuationdomainZModelm0Z Assignmentg0)vvaldomrrr demo_model0ps,          rCutf8cCs\t|d|}dd|D}Wdn1swYdd|D}dd|D}|S)NrcSsg|]}|qSr)rstripr$lrrrr'szread_sents..cSsg|] }t|dkr|qSr)lenrGrrrr'cSsg|] }|ddks|qS)r#rrGrrrr'rK)codecsopen)filenameencodingfpsentsrrr read_sentss rScCs`ddlm}|d}td|tdtdg|ddD]}|d\}}ttd |qd S) z Check that interpret_sents() is compatible with legacy grammars that use a lowercase 'sem' feature. Define 'test.fcfg' to be the following rrz1 % start S S[sem=] -> 'hello' zReading grammar: %sz********************Zhellor)r&zoutput: N)r rZ fromstringrr()rgreadingr%r)rrrdemo_legacy_grammars    rVc Csddl}ddlm}d}||d}|jdddddddd|jd d d d d d|jdddddd|jdddddd|jdddddd|jddd d!d"d#|jd$d%d d&d'd#|jd(d)d*d+d,d#|jd-d.d*d/d0d#|\}}d1}tgd2}d3}|jr|j} |jr|j}|j rt d4|j |durt | }t } t } |jrt||| | |jd5} nt|||jd5} t|D]N\}}d6}td7|t||jr| |D] \}}}t|trt|}td8||ft||d67}qq| |D]\}}td8||f|d67}qqdS)9Nr) OptionParserz, Parse and evaluate some sentences. ) descriptionTdefault)rbetasyntracesemtracedemor sentencesz-dz--demor^zBchoose demo D; omit this for the default demo, or specify 'chat80'D)desthelpmetavarz-gz--gramrzread in grammar GGz-mz--modelr,z"import model M (omit '.py' suffix)Mz-sz --sentencesr_z"read in a file of test sentences SSz-ez --no-eval store_falserzjust do a syntactic analysis)actionrarbz-bz--no-beta-reductionr[zdon't carry out beta-reductionz-tz --syntracecountr\z.set syntactic tracing on; requires '-e' optionz-Tz --semtracer]zset semantic tracing onz------------------------------)zFido sees a boy with MaryzJohn sees Maryzevery girl chases a dogzevery boy chases a girlzJohn walks with a girl in Noosaz who walksz"grammars/sample_grammars/sem2.fcfgzimport %s as modelrz Sentence: %sz%d: %s)sysoptparserW set_defaults add_option parse_argsrCr_rr,execrSr>r?rr-r]r(r\ enumeraterr dictsetkeys)rkrWrXoptsoptionsargsZSPACERrRZgramfileZ sentsfiler,rTZ evaluationsZsemrepsirr3r Zsemrepvaluerrrr^s           r^__main__rI)r)rr)rD) __doc__rMZnltk.semrrr#r(r-rCrSrVr^__name__rrrrs      ! ~