o i @sdZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZmZd Ze jd eZGd d d e jZd d d ddd*ddZd+d d!Zd+d"d#Zd+d$d%Zd+d&d'Zd+d(d)Zd S),z'Base class for TACKY optimization tests) annotations)Path)ListOptionalSequence)basic)asmparse)OpcodeRegisterchapter_c@s8eZdZdZdddZdd d ZdddZdddZdS)TackyOptimizationTesta7Base class for TACKY (chapter 19) tests. There are two kinds of tests for these chapters. The first is the same kind of test we use in Parts I & II: we compile the test program, run it, and make sure it has the correct observable behavior. These test methods should call compile_and_run, defined in TestChapter. In the second kind of test, we still compile the program, run it, and validate its behavior, but we also inspect its assembly code to make sure it's been optimized. These test methods should call run_and_parse or run_and_parse_all, defined below. This class defines two test methods used in dead store elimination and whole pipeline tests: * store_eliminated_test: Test that stores of particular constants were eliminated * return_const_test: Test that the only thing this function does is return a specific consatnt Notes: * This class isn't designed to test intermediate stages (i.e. exit_stage should always be "run"). TODO enforce this? * There are no invalid test programs for this chapter source_filerreturndict[str, asm.AssemblyFunction]cCst|j|dd}|j|jdd|d|jdt||d}t|}t|g|g}| ||t |S)a$Compile and run a program, validate result, then return parsed assembly. The caller can then perform additional validation on the parsed assembly. Args: program_path: Absolute path to test program Returns: parsed assembly code for whole program z-s)cc_optrzcompilation of z failed with error: msgz.s) invoke_compiler assertEqual returncodestderrr print_stderr with_suffixget_libsgcc_compile_and_run validate_runsr parse_file)selfrcompile_resultasm_filelibs actual_resultr%g/home/gtoal/gtoal.com/languages/algol60/new-unicode-parser/lang/c/tests4/test_framework/tacky/common.pyrun_and_parse_all's     z'TackyOptimizationTest.run_and_parse_allasm.AssemblyFunctioncCs||dS)a:Compile and run a program, validate result, then return parsed assembly for 'target' function. The caller can then perform additional validation on the parsed assembly. Args: program_path: Absolute path to test program Returns: parsed assembly code for whole program target)r')r rr%r%r& run_and_parseEs z#TackyOptimizationTest.run_and_parseredundant_consts List[int]NonecsXdd|Ddfdd ||}fd d|jD}|j|td |||d d d S)aMake sure any stores of the form mov $const, were eliminated. The test program should contain a single 'target' function. Args: source_file: absolute path to program under test redundant_consts: any constants that were sources of mov instructions in the original program but shouldn't be after dead store elimination cSsg|]}t|qSr%)r Immediate).0cr%r%r& ]sz?TackyOptimizationTest.store_eliminated_test..i asm.AsmItemrboolcs$t|tjot|jo|jdvS)Nr) isinstancer Instructionr4operandsr2)redundant_operandsr%r& is_dead_store_s  zBTackyOptimizationTest.store_eliminated_test..is_dead_storecsg|]}|r|qSr%r%r/r2)r:r%r&r1kz1Found dead store that should have been eliminatedbad_instructions full_prog program_pathrNr2r3rr4r* instructions assertFalse build_msg)r rr+ parsed_asmr>r%)r:r9r&store_eliminated_testRs   z+TackyOptimizationTest.store_eliminated_testreturned_constintcsJd fdd ||}fdd|jD}|j|td |||d d d S)zValidate that the function doesn't do anything except return a constant. The test program should contain a single 'target' function. r2r3rr4csbt|rdS|ttjjttjjgkrdSdkr/|ttjjtjjtjjgkr/dSdS)zMWe should optimize out everything except prologue, epilogue, and mov into EAXTrF) is_prologue_or_epiloguer r6r MOVr.r AXXORr8)rHr%r&ok|sz3TackyOptimizationTest.return_const_test..okcsg|]}|s|qSr%r%r;)rNr%r&r1r<z;TackyOptimizationTest.return_const_test..z5Found instruction that should have been optimized outr=rNrArB)r rrHrFr>r%)rNrHr&return_const_testvs  z'TackyOptimizationTest.return_const_testN)rrrr)rrrr()rrr+r,rr-)rrrHrIrr-)__name__ __module__ __qualname____doc__r'r*rGrOr%r%r%r&rs    $rN)r>r?r@max_prog_disp_lengthrstrr>Optional[Sequence[asm.AsmItem]]r?Optional[asm.AssemblyFunction]r@Optional[Path]rUrIrcCsz|g}|rdd|D}|d|||r.t|j|kr%|dn |dt|g|r8|d|d|S)z?Utility function for validators to report invalid assembly codecSsg|]}t|qSr%)rVr;r%r%r&r1szbuild_msg..zBad instructions:z1Complete assembly function: zComplete assembly function:z Program:  )appendextendlenrCrVjoin)rr>r?r@rU msg_linesprinted_instructionsr%r%r&rEs     rEr2r3r4cCsvt|tjrdSt|p:|jtjtjfvo|jdt j kp:|jtj ko*|jdt j kp:|jtj ko:tdd|jDS)zIs this an instruction you might find in the function prologue or epilogue? These will be present even when everything else was optimized out.Frcss |] }|tjtjfvVqdSN)r SPBP)r/or%r%r& sz*is_prologue_or_epilogue..)r5r Labelis_retopcoder PUSHPOPr7r rdSUBrcrKallr8r%r%r&rJs   rJcCs&t|tjp|jtjtjtjtjfvSrb) r5r rgrir JMPJMPCCCALLCMOVr8r%r%r&is_control_flows rrcCst|tjo|jtjtjfvSrb)r5r r6rir RETLEAVEr8r%r%r&rhsrhcCst|tjo |jtjkSrb)r5r r6rir rKr8r%r%r&is_movsrucCs,t|tjo|jtjko|jd|jdkS)zNIs this an instruction of the form xor %reg, %reg used to zero out a register?rra)r5r r6rir rMr7r8r%r%r& is_zero_instrs  rv) rrVr>rWr?rXr@rYrUrIrrVrA)rS __future__rpathlibrtypingrrrrparserr r parser.asmr r CHAPTERTEST_DIR TestChapterrrErJrrrhrurvr%r%r%r&s*