B P?ð[”.ã@sädZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z ddl mZmZGdd „d ee ƒZGd d „d ee ƒZd d d„Zdd„Zdd„Zdd„Zdd„Zdd„ZdddgfdddgfgZdd„ZedkràeƒdS)!zA A model builder that makes use of the external 'Mace4' package. é)Úprint_functionN)Ú is_indvar)Ú ValuationÚ Expression)Ú ModelBuilderÚBaseModelBuilderCommand)ÚProver9CommandParentÚ Prover9Parentc@steZdZdZdZddd„Zedd„ƒZdd „Ze d d „ƒZ e d d „ƒZ e dd„ƒZ dd„Z dd„Zgdfdd„ZdS)Ú MaceCommandz¸ A ``MaceCommand`` specific to the ``Mace`` model builder. It contains a print_assumptions() method that is used to print the list of assumptions in multiple formats. NéôcCs4|dk rt|tƒs t‚nt|ƒ}t ||||¡dS)a• :param goal: Input expression to prove :type goal: sem.Expression :param assumptions: Input expressions to use as assumptions in the proof. :type assumptions: list(sem.Expression) :param max_models: The maximum number of models that Mace will try before simply returning false. (Use 0 for no maximum.) :type max_models: int N)Ú isinstanceÚMaceÚAssertionErrorrÚ__init__)ÚselfÚgoalÚ assumptionsÚ max_modelsZ model_builder©rú2lib/python3.7/site-packages/nltk/inference/mace.pyr!s zMaceCommand.__init__cCs | d¡S)NÚ valuation)Úmodel)Zmbcrrrr3szMaceCommand.valuationc Csº| |d¡}g}xž| d¡D]Ž}| ¡}| d¡r\t|| d¡d| d¡… ¡ƒ}q| d¡rÞ| d¡d krÞ|| d¡d| d¡… ¡}t|ƒr¤| ¡}t|| d ¡d| d ¡… ¡ƒ}|  |t   |¡f¡q| d ¡r|| d¡dd …}d|krb|d | d¡… ¡}dd„|| d ¡d| d ¡…  d¡Dƒ} |  |t   || ¡f¡q|d | d¡… ¡}t|| d ¡d| d ¡… ¡ƒ}|  ||dkf¡qWt|ƒS)z¦ Transform the output file into an NLTK-style Valuation. :return: A model if one is generated; None otherwise. :rtype: sem.Valuation ÚstandardFZinterpretationú(éú,ZfunctionÚ_éÿÿÿÿú[ú]ZrelationNcSsg|]}t| ¡ƒ‘qSr)ÚintÚstrip)Ú.0Úvrrrú Vsz,MaceCommand._convert2val..)Ú_transform_outputÚ splitlinesr!Ú startswithr ÚindexÚfindrÚupperÚappendr Ú_make_model_varÚsplitÚ_make_relation_setr) rÚ valuation_strZvaluation_standard_formatÚvalÚlineÚlÚ num_entitiesÚnameÚvalueÚvaluesrrrÚ _convert2val7s0  & $  &$zMaceCommand._convert2valc Cs>tƒ}x2dd„t|ƒDƒD]}| tt |||¡ƒ¡qW|S)a] Convert a Mace4-style relation table into a dictionary. :param num_entities: the number of entities in the model; determines the row length in the table. :type num_entities: int :param values: a list of 1's and 0's that represent whether a relation holds in a Mace4 model. :type values: list of int cSsg|]\}}|dkr|‘qS)rr)r"Úposr#rrrr$osz2MaceCommand._make_relation_set..)ÚsetÚ enumerateÚaddÚtupler Ú_make_relation_tuple)r3r6ÚrÚpositionrrrr.ds zMaceCommand._make_relation_setcCsft|ƒdkrgSt|ƒ|}||}t||ƒ}||||d|…}t |¡gt |||¡SdS)Nr)Úlenr r r,r=)r?r6r3Z sublist_sizeZ sublist_startZsublist_positionZsublistrrrr=us    z MaceCommand._make_relation_tuplecCs\ddddddddd d d d d dddddddddddddg|}|d}|dkrX|t|ƒS|S)z³ Pick an alphabetic character as identifier for an entity in the model. :param value: where to index into the list of characters :type value: int ÚaÚbÚcÚdÚeÚfÚgÚhÚiÚjÚkr2ÚmÚnÚoÚpÚqr>ÚsÚtÚur#ÚwÚxÚyÚzér)Ústr)r5ZletterÚnumrrrr,‡s: zMaceCommand._make_model_varcCs*|s|S|dkr| |¡S| ||¡SdS)a^ Print out a Mace4 model using any Mace4 ``interpformat`` format. See http://www.cs.unm.edu/~mccune/mace4/manual/ for details. :param valuation_str: str with the model builder's output :param format: str indicating the format for displaying models. Defaults to 'standard' format. :return: str rN)r7r%)rr/ÚformatrrrÚ_decorate_model®s  zMaceCommand._decorate_modelcCs&|dkr| ||g¡dStdƒ‚dS)zª Transform the output file into any Mace4 ``interpformat`` format. :param format: Output format for displaying models. :type format: str )rZ standard2ÚportableZtabularÚrawÚcookedÚxmlZtexrz#The specified format does not existN)Ú_call_interpformatÚ LookupError)rr/r[rrrr%¿s zMaceCommand._transform_outputFcCs.|jdkr|j d|¡|_|j ||j||¡S)a Call the ``interpformat`` binary with the given input. :param input_str: A string whose contents are used as stdin. :param args: A list of command-line arguments. :return: A tuple (stdout, returncode) :see: ``config_prover9`` NZ interpformat)Ú_interpformat_binZ _modelbuilderÚ _find_binaryÚ_call)rÚ input_strÚargsÚverboserrrraÔs  zMaceCommand._call_interpformat)NNr N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rcrÚpropertyrr7Ú staticmethodr.r=r,r\r%rarrrrr s  -   'r c@s2eZdZdZd dd„Zd dd„Zgdfdd „ZdS) r NéôcCs ||_dS)N)Ú _end_size)rZend_sizerrrrêsz Mace.__init__FcCs.|sg}|j| ||¡|d\}}|dk|fS)z  Use Mace4 to build a first order model. :return: ``True`` if a model was found (i.e. Mace returns value of 0), else ``False`` )rhr)Ú _call_mace4Z prover9_input)rrrrhÚstdoutÚ returncoderrrÚ _build_modelïs zMace._build_modelcCsN|jdkr| d|¡|_d}|jdkr4|d|j7}||7}| ||j||¡S)a  Call the ``mace4`` binary with the given input. :param input_str: A string whose contents are used as stdin. :param args: A list of command-line arguments. :return: A tuple (stdout, returncode) :see: ``config_prover9`` NZmace4Úrzassign(end_size, %d). )Ú _mace4_binrdrpre)rrfrgrhZupdated_input_strrrrrqþs  zMace._call_mace4)ro)NNF)rirjrkrvrrtrqrrrrr çs  r écCstd|ƒdS)Nú-)Úprint)rZrrrÚspacersrzcCsddddœ|S)zq Decode the result of model_found() :param found: The output of model_found() :type found: bool zCountermodel foundzNo countermodel foundÚNone)TFNr)ÚfoundrrrÚ decode_results r}cCsrxl|D]d\}}t |¡}dd„|Dƒ}t||dd}| ¡}x|D]}td|ƒqBWtd|t|ƒfƒqWdS)z2 Try some proofs and exhibit the results. cSsg|]}t |¡‘qSr)ÚlpÚparse)r"rArrrr$(sz$test_model_found..é2)rrz %sz |- %s: %s N)rÚ fromstringr Ú build_modelryr})Ú argumentsrrrGÚalistrLr|rArrrÚtest_model_found"s  r…cCs–t d¡}dd„dDƒ}t||d}| ¡tƒtdƒtƒx|D]}td|ƒqFWtd|t| ¡ƒfƒtƒtd ƒtƒt|jd ƒd S) z0 Try to build a ``nltk.sem.Valuation``. z all x.man(x)cSsg|]}t |¡‘qSr)rr)r"rArrrr$6sz$test_build_model..)z man(John)z man(Socrates)z man(Bill)z,some x.(-(x = John) & man(x) & sees(John,x))zsome x.(-(x = Bill) & man(x))z,all x.some y.(man(x) -> gives(Socrates,x,y)))rzAssumptions and Goalz %sz |- %s: %s rÚ N)rrr r‚rzryr}r)rƒrGr„rLrArrrÚtest_build_model0s   r‡cCsœt |d¡}dd„|dDƒ}t||d}| ¡x|D]}td|ƒq:Wtd|| ¡fƒx4dD],}tƒtd |ƒtƒt|j|d ƒqhWd S) zJ Transform the model into various Mace4 ``interpformat`` formats. rcSsg|]}t |¡‘qSr)r~r)r"rArrrr$Vsz)test_transform_output..r)rz %sz |- %s: %s )rr]r`r_zUsing '%s' format)r[N)rrr r‚ryrzr)Z argument_pairrGr„rLrAr[rrrÚtest_transform_outputQs    rˆc Cs„ttjddddgdtddgƒkƒttjddddddddddg dtdgƒkƒttjdddddddddgdtd d gƒkƒdS) Nérr)r3r6)rC)rA)rCrAé)rArBrA)rBrBrA)ryr r.r9rrrrÚtest_make_relation_setcsr‹zmortal(Socrates)zall x.(man(x) -> mortal(x))z man(Socrates)z(not mortal(Socrates))cCs ttƒttƒttdƒdS)Nr)r…rƒr‡rˆrrrrÚdemozsrŒÚ__main__)rw)rlZ __future__rÚosZtempfileZnltk.sem.logicrZnltk.semrrZnltk.inference.apirrZnltk.inference.prover9rr r r rzr}r…r‡rˆr‹rƒrŒrirrrrÚ s*  P+  !