├── LICENSE ├── README.md └── dx-shader-decompiler.py /LICENSE: -------------------------------------------------------------------------------- 1 | GNU GENERAL PUBLIC LICENSE 2 | Version 2, June 1991 3 | 4 | Copyright (C) 1989, 1991 Free Software Foundation, Inc., 5 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 6 | Everyone is permitted to copy and distribute verbatim copies 7 | of this license document, but changing it is not allowed. 8 | 9 | Preamble 10 | 11 | The licenses for most software are designed to take away your 12 | freedom to share and change it. By contrast, the GNU General Public 13 | License is intended to guarantee your freedom to share and change free 14 | software--to make sure the software is free for all its users. This 15 | General Public License applies to most of the Free Software 16 | Foundation's software and to any other program whose authors commit to 17 | using it. (Some other Free Software Foundation software is covered by 18 | the GNU Lesser General Public License instead.) You can apply it to 19 | your programs, too. 20 | 21 | When we speak of free software, we are referring to freedom, not 22 | price. Our General Public Licenses are designed to make sure that you 23 | have the freedom to distribute copies of free software (and charge for 24 | this service if you wish), that you receive source code or can get it 25 | if you want it, that you can change the software or use pieces of it 26 | in new free programs; and that you know you can do these things. 27 | 28 | To protect your rights, we need to make restrictions that forbid 29 | anyone to deny you these rights or to ask you to surrender the rights. 30 | These restrictions translate to certain responsibilities for you if you 31 | distribute copies of the software, or if you modify it. 32 | 33 | For example, if you distribute copies of such a program, whether 34 | gratis or for a fee, you must give the recipients all the rights that 35 | you have. You must make sure that they, too, receive or can get the 36 | source code. And you must show them these terms so they know their 37 | rights. 38 | 39 | We protect your rights with two steps: (1) copyright the software, and 40 | (2) offer you this license which gives you legal permission to copy, 41 | distribute and/or modify the software. 42 | 43 | Also, for each author's protection and ours, we want to make certain 44 | that everyone understands that there is no warranty for this free 45 | software. If the software is modified by someone else and passed on, we 46 | want its recipients to know that what they have is not the original, so 47 | that any problems introduced by others will not reflect on the original 48 | authors' reputations. 49 | 50 | Finally, any free program is threatened constantly by software 51 | patents. We wish to avoid the danger that redistributors of a free 52 | program will individually obtain patent licenses, in effect making the 53 | program proprietary. To prevent this, we have made it clear that any 54 | patent must be licensed for everyone's free use or not licensed at all. 55 | 56 | The precise terms and conditions for copying, distribution and 57 | modification follow. 58 | 59 | GNU GENERAL PUBLIC LICENSE 60 | TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION 61 | 62 | 0. This License applies to any program or other work which contains 63 | a notice placed by the copyright holder saying it may be distributed 64 | under the terms of this General Public License. The "Program", below, 65 | refers to any such program or work, and a "work based on the Program" 66 | means either the Program or any derivative work under copyright law: 67 | that is to say, a work containing the Program or a portion of it, 68 | either verbatim or with modifications and/or translated into another 69 | language. (Hereinafter, translation is included without limitation in 70 | the term "modification".) Each licensee is addressed as "you". 71 | 72 | Activities other than copying, distribution and modification are not 73 | covered by this License; they are outside its scope. The act of 74 | running the Program is not restricted, and the output from the Program 75 | is covered only if its contents constitute a work based on the 76 | Program (independent of having been made by running the Program). 77 | Whether that is true depends on what the Program does. 78 | 79 | 1. You may copy and distribute verbatim copies of the Program's 80 | source code as you receive it, in any medium, provided that you 81 | conspicuously and appropriately publish on each copy an appropriate 82 | copyright notice and disclaimer of warranty; keep intact all the 83 | notices that refer to this License and to the absence of any warranty; 84 | and give any other recipients of the Program a copy of this License 85 | along with the Program. 86 | 87 | You may charge a fee for the physical act of transferring a copy, and 88 | you may at your option offer warranty protection in exchange for a fee. 89 | 90 | 2. You may modify your copy or copies of the Program or any portion 91 | of it, thus forming a work based on the Program, and copy and 92 | distribute such modifications or work under the terms of Section 1 93 | above, provided that you also meet all of these conditions: 94 | 95 | a) You must cause the modified files to carry prominent notices 96 | stating that you changed the files and the date of any change. 97 | 98 | b) You must cause any work that you distribute or publish, that in 99 | whole or in part contains or is derived from the Program or any 100 | part thereof, to be licensed as a whole at no charge to all third 101 | parties under the terms of this License. 102 | 103 | c) If the modified program normally reads commands interactively 104 | when run, you must cause it, when started running for such 105 | interactive use in the most ordinary way, to print or display an 106 | announcement including an appropriate copyright notice and a 107 | notice that there is no warranty (or else, saying that you provide 108 | a warranty) and that users may redistribute the program under 109 | these conditions, and telling the user how to view a copy of this 110 | License. (Exception: if the Program itself is interactive but 111 | does not normally print such an announcement, your work based on 112 | the Program is not required to print an announcement.) 113 | 114 | These requirements apply to the modified work as a whole. If 115 | identifiable sections of that work are not derived from the Program, 116 | and can be reasonably considered independent and separate works in 117 | themselves, then this License, and its terms, do not apply to those 118 | sections when you distribute them as separate works. But when you 119 | distribute the same sections as part of a whole which is a work based 120 | on the Program, the distribution of the whole must be on the terms of 121 | this License, whose permissions for other licensees extend to the 122 | entire whole, and thus to each and every part regardless of who wrote it. 123 | 124 | Thus, it is not the intent of this section to claim rights or contest 125 | your rights to work written entirely by you; rather, the intent is to 126 | exercise the right to control the distribution of derivative or 127 | collective works based on the Program. 128 | 129 | In addition, mere aggregation of another work not based on the Program 130 | with the Program (or with a work based on the Program) on a volume of 131 | a storage or distribution medium does not bring the other work under 132 | the scope of this License. 133 | 134 | 3. You may copy and distribute the Program (or a work based on it, 135 | under Section 2) in object code or executable form under the terms of 136 | Sections 1 and 2 above provided that you also do one of the following: 137 | 138 | a) Accompany it with the complete corresponding machine-readable 139 | source code, which must be distributed under the terms of Sections 140 | 1 and 2 above on a medium customarily used for software interchange; or, 141 | 142 | b) Accompany it with a written offer, valid for at least three 143 | years, to give any third party, for a charge no more than your 144 | cost of physically performing source distribution, a complete 145 | machine-readable copy of the corresponding source code, to be 146 | distributed under the terms of Sections 1 and 2 above on a medium 147 | customarily used for software interchange; or, 148 | 149 | c) Accompany it with the information you received as to the offer 150 | to distribute corresponding source code. (This alternative is 151 | allowed only for noncommercial distribution and only if you 152 | received the program in object code or executable form with such 153 | an offer, in accord with Subsection b above.) 154 | 155 | The source code for a work means the preferred form of the work for 156 | making modifications to it. For an executable work, complete source 157 | code means all the source code for all modules it contains, plus any 158 | associated interface definition files, plus the scripts used to 159 | control compilation and installation of the executable. However, as a 160 | special exception, the source code distributed need not include 161 | anything that is normally distributed (in either source or binary 162 | form) with the major components (compiler, kernel, and so on) of the 163 | operating system on which the executable runs, unless that component 164 | itself accompanies the executable. 165 | 166 | If distribution of executable or object code is made by offering 167 | access to copy from a designated place, then offering equivalent 168 | access to copy the source code from the same place counts as 169 | distribution of the source code, even though third parties are not 170 | compelled to copy the source along with the object code. 171 | 172 | 4. You may not copy, modify, sublicense, or distribute the Program 173 | except as expressly provided under this License. Any attempt 174 | otherwise to copy, modify, sublicense or distribute the Program is 175 | void, and will automatically terminate your rights under this License. 176 | However, parties who have received copies, or rights, from you under 177 | this License will not have their licenses terminated so long as such 178 | parties remain in full compliance. 179 | 180 | 5. You are not required to accept this License, since you have not 181 | signed it. However, nothing else grants you permission to modify or 182 | distribute the Program or its derivative works. These actions are 183 | prohibited by law if you do not accept this License. Therefore, by 184 | modifying or distributing the Program (or any work based on the 185 | Program), you indicate your acceptance of this License to do so, and 186 | all its terms and conditions for copying, distributing or modifying 187 | the Program or works based on it. 188 | 189 | 6. Each time you redistribute the Program (or any work based on the 190 | Program), the recipient automatically receives a license from the 191 | original licensor to copy, distribute or modify the Program subject to 192 | these terms and conditions. You may not impose any further 193 | restrictions on the recipients' exercise of the rights granted herein. 194 | You are not responsible for enforcing compliance by third parties to 195 | this License. 196 | 197 | 7. If, as a consequence of a court judgment or allegation of patent 198 | infringement or for any other reason (not limited to patent issues), 199 | conditions are imposed on you (whether by court order, agreement or 200 | otherwise) that contradict the conditions of this License, they do not 201 | excuse you from the conditions of this License. If you cannot 202 | distribute so as to satisfy simultaneously your obligations under this 203 | License and any other pertinent obligations, then as a consequence you 204 | may not distribute the Program at all. For example, if a patent 205 | license would not permit royalty-free redistribution of the Program by 206 | all those who receive copies directly or indirectly through you, then 207 | the only way you could satisfy both it and this License would be to 208 | refrain entirely from distribution of the Program. 209 | 210 | If any portion of this section is held invalid or unenforceable under 211 | any particular circumstance, the balance of the section is intended to 212 | apply and the section as a whole is intended to apply in other 213 | circumstances. 214 | 215 | It is not the purpose of this section to induce you to infringe any 216 | patents or other property right claims or to contest validity of any 217 | such claims; this section has the sole purpose of protecting the 218 | integrity of the free software distribution system, which is 219 | implemented by public license practices. Many people have made 220 | generous contributions to the wide range of software distributed 221 | through that system in reliance on consistent application of that 222 | system; it is up to the author/donor to decide if he or she is willing 223 | to distribute software through any other system and a licensee cannot 224 | impose that choice. 225 | 226 | This section is intended to make thoroughly clear what is believed to 227 | be a consequence of the rest of this License. 228 | 229 | 8. If the distribution and/or use of the Program is restricted in 230 | certain countries either by patents or by copyrighted interfaces, the 231 | original copyright holder who places the Program under this License 232 | may add an explicit geographical distribution limitation excluding 233 | those countries, so that distribution is permitted only in or among 234 | countries not thus excluded. In such case, this License incorporates 235 | the limitation as if written in the body of this License. 236 | 237 | 9. The Free Software Foundation may publish revised and/or new versions 238 | of the General Public License from time to time. Such new versions will 239 | be similar in spirit to the present version, but may differ in detail to 240 | address new problems or concerns. 241 | 242 | Each version is given a distinguishing version number. If the Program 243 | specifies a version number of this License which applies to it and "any 244 | later version", you have the option of following the terms and conditions 245 | either of that version or of any later version published by the Free 246 | Software Foundation. If the Program does not specify a version number of 247 | this License, you may choose any version ever published by the Free Software 248 | Foundation. 249 | 250 | 10. If you wish to incorporate parts of the Program into other free 251 | programs whose distribution conditions are different, write to the author 252 | to ask for permission. For software which is copyrighted by the Free 253 | Software Foundation, write to the Free Software Foundation; we sometimes 254 | make exceptions for this. Our decision will be guided by the two goals 255 | of preserving the free status of all derivatives of our free software and 256 | of promoting the sharing and reuse of software generally. 257 | 258 | NO WARRANTY 259 | 260 | 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY 261 | FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN 262 | OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES 263 | PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED 264 | OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 265 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS 266 | TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE 267 | PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING, 268 | REPAIR OR CORRECTION. 269 | 270 | 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING 271 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR 272 | REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, 273 | INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING 274 | OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED 275 | TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY 276 | YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER 277 | PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE 278 | POSSIBILITY OF SUCH DAMAGES. 279 | 280 | END OF TERMS AND CONDITIONS 281 | 282 | How to Apply These Terms to Your New Programs 283 | 284 | If you develop a new program, and you want it to be of the greatest 285 | possible use to the public, the best way to achieve this is to make it 286 | free software which everyone can redistribute and change under these terms. 287 | 288 | To do so, attach the following notices to the program. It is safest 289 | to attach them to the start of each source file to most effectively 290 | convey the exclusion of warranty; and each file should have at least 291 | the "copyright" line and a pointer to where the full notice is found. 292 | 293 | {description} 294 | Copyright (C) {year} {fullname} 295 | 296 | This program is free software; you can redistribute it and/or modify 297 | it under the terms of the GNU General Public License as published by 298 | the Free Software Foundation; either version 2 of the License, or 299 | (at your option) any later version. 300 | 301 | This program is distributed in the hope that it will be useful, 302 | but WITHOUT ANY WARRANTY; without even the implied warranty of 303 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 304 | GNU General Public License for more details. 305 | 306 | You should have received a copy of the GNU General Public License along 307 | with this program; if not, write to the Free Software Foundation, Inc., 308 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 309 | 310 | Also add information on how to contact you by electronic and paper mail. 311 | 312 | If the program is interactive, make it output a short notice like this 313 | when it starts in an interactive mode: 314 | 315 | Gnomovision version 69, Copyright (C) year name of author 316 | Gnomovision comes with ABSOLUTELY NO WARRANTY; for details type `show w'. 317 | This is free software, and you are welcome to redistribute it 318 | under certain conditions; type `show c' for details. 319 | 320 | The hypothetical commands `show w' and `show c' should show the appropriate 321 | parts of the General Public License. Of course, the commands you use may 322 | be called something other than `show w' and `show c'; they could even be 323 | mouse-clicks or menu items--whatever suits your program. 324 | 325 | You should also get your employer (if you work as a programmer) or your 326 | school, if any, to sign a "copyright disclaimer" for the program, if 327 | necessary. Here is a sample; alter the names: 328 | 329 | Yoyodyne, Inc., hereby disclaims all copyright interest in the program 330 | `Gnomovision' (which makes passes at compilers) written by James Hacker. 331 | 332 | {signature of Ty Coon}, 1 April 1989 333 | Ty Coon, President of Vice 334 | 335 | This General Public License does not permit incorporating your program into 336 | proprietary programs. If your program is a subroutine library, you may 337 | consider it more useful to permit linking proprietary applications with the 338 | library. If this is what you want to do, use the GNU Lesser General 339 | Public License instead of this License. -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | dx-shader-decompiler 2 | ==================== 3 | 4 | A decompiler for DirectX 9 shaders 5 | 6 | This currently supports decompiling pixel shader and vertex shader 3_0 7 | Reference for the assembly language: 8 | http://msdn.microsoft.com/en-us/library/bb219840%28v=vs.85%29.aspx 9 | 10 | Reference for the machine language: http://msdn.microsoft.com/en-us/library/windows/hardware/ff552891%28v=vs.85%29.aspx 11 | 12 | Compile: 13 | fxc.exe fx_examples/PP_ColorBloomH.fx /E PostProcessPS /T ps_3_0 /Fx test.txt 14 | (note, this produces a text file containing the hex of the compiled shaders, but binary works as input also) 15 | 16 | Decompile: 17 | python dx-shader-decompiler.py test.txt 18 | -------------------------------------------------------------------------------- /dx-shader-decompiler.py: -------------------------------------------------------------------------------- 1 | """ 2 | Decompiler for DirectX shader code. 3 | 4 | Copyright (c) 2013-2014 CiiNOW Inc. 5 | Written by Alex Izvorski , 6 | 7 | This program is free software; you can redistribute it and/or 8 | modify it under the terms of the GNU General Public License 9 | as published by the Free Software Foundation; either version 2 10 | of the License, or (at your option) any later version. 11 | 12 | This program is distributed in the hope that it will be useful, 13 | but WITHOUT ANY WARRANTY; without even the implied warranty of 14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15 | GNU General Public License for more details. 16 | 17 | You should have received a copy of the GNU General Public License 18 | along with this program; if not, write to the Free Software 19 | Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. 20 | """ 21 | 22 | 23 | """ 24 | This is currently supports decompiling pixel shader and vertex shader 3_0 25 | 26 | Reference for the assembly language: http://msdn.microsoft.com/en-us/library/bb219840%28v=vs.85%29.aspx 27 | Reference for the machine language: http://msdn.microsoft.com/en-us/library/windows/hardware/ff552891%28v=vs.85%29.aspx 28 | 29 | Compile: 30 | fxc.exe fx_examples/PP_ColorBloomH.fx /E PostProcessPS /T ps_3_0 /Fx test.txt 31 | (note, this produces a text file containing the hex of the compiled shaders, but binary works as input also) 32 | 33 | Decompile: 34 | python dx-shader-decompiler.py test.txt 35 | """ 36 | 37 | 38 | from bitstring import * 39 | import pdb 40 | import sys 41 | import re 42 | import fileinput 43 | 44 | opcodes_list = [ 45 | 'nop', 46 | 'mov', 47 | 'add', 48 | 'sub', 49 | 'mad', 50 | 'mul', 51 | 'rcp', 52 | 'rsq', 53 | 'dp3', 54 | 'dp4', 55 | 'min', 56 | 'max', 57 | 'slt', 58 | 'sge', 59 | 'exp', 60 | 'log', 61 | 'lit', 62 | 'dst', 63 | 'lrp', 64 | 'frc', 65 | 'm4x4', 66 | 'm4x3', 67 | 'm3x4', 68 | 'm3x3', 69 | 'm3x2', 70 | 'call', 71 | 'callnz', 72 | 'loop', 73 | 'ret', 74 | 'endloop', 75 | 'label', 76 | 'dcl', 77 | 'pow', 78 | 'crs', 79 | 'sgn', 80 | 'abs', 81 | 'nrm', 82 | 'sincos', 83 | 'rep', 84 | 'endrep', 85 | 'if', 86 | 'ifc', 87 | 'else', 88 | 'endif', 89 | 'break', 90 | 'breakc', 91 | 'mova', 92 | 'defb', 93 | 'defi', 94 | None, 95 | None, 96 | None, 97 | None, 98 | None, 99 | None, 100 | None, 101 | None, 102 | None, 103 | None, 104 | None, 105 | None, 106 | None, 107 | None, 108 | None, 109 | 'texcoord', # 64 110 | 'texkill', 111 | 'texld', # tex in ps < 2_0 112 | 'texbem', 113 | 'texbeml', 114 | 'texreg2ar', 115 | 'texreg2gb', 116 | 'texm3x2pad', 117 | 'texm3x2tex', 118 | 'texm3x3pad', 119 | 'texm3x3tex', 120 | 'reserved0', 121 | 'texm3x3spec', 122 | 'texm3x3vspec', 123 | 'expp', 124 | 'logp', 125 | 'cnd', 126 | 'def', 127 | 'texreg2rgb', 128 | 'texdp3tex', 129 | 'texm3x2depth', 130 | 'texdp3', 131 | 'texm3x3', 132 | 'texdepth', 133 | 'cmp', 134 | 'bem', 135 | 'dp2add', 136 | 'dsx', 137 | 'dsy', 138 | 'texldd', 139 | 'setp', 140 | 'texldl', 141 | 'breakp', 142 | ] 143 | 144 | 145 | # http://msdn.microsoft.com/en-us/library/ff569707%28v=vs.85%29.aspx 146 | register_type_list = [ 147 | 'temp', 148 | 'input', 149 | 'const', 150 | 'texture', 151 | #'addr', # Duplicate of texture 152 | 'rastout', 153 | 'attrout', 154 | #'texcrdout', # Duplicate of output 155 | 'output', 156 | 'constint', 157 | 'colorout', 158 | 'depthout', 159 | 'sampler', 160 | 'const2', 161 | 'const3', 162 | 'const4', 163 | 'constbool', 164 | 'loop', 165 | 'tempfloat16', 166 | 'misctype', 167 | 'label', 168 | 'predicate', 169 | ] 170 | 171 | register_type_mnemonic_list = [ 172 | 'r', 173 | 'v', 174 | 'c', 175 | 't', 176 | #'a', # Duplicate of texture 177 | 'rastout', # o ? 178 | 'attrout', # o ? 179 | #'oT', # Duplicate of output 180 | 'o', 181 | 'i', 182 | 'oC', 183 | 'oDepth', 184 | 's', 185 | 'const2', 186 | 'const3', 187 | 'const4', 188 | 'b', 189 | 'aL', 190 | 'tempfloat16', 191 | 'misctype', 192 | 'label', 193 | 'predicate', 194 | ] 195 | 196 | source_modifier_mnemonic_list = [ 197 | None, # 0x0 None 198 | '-R', # 0x1 Negate 199 | None, # 0x2 Bias 200 | None, # 0x3 Bias and negate 201 | None, # 0x4 Sign (bx2) 202 | None, # 0x5 Sign (bx2) and negate 203 | None, # 0x6 Complement 204 | None, # 0x7 x2 (PS 1_4) 205 | None, # 0x8 x2 and negate (PS 1_4) 206 | None, # 0x9 dz (divide through by Z component PS 1_4) 207 | None, # 0xa dw (divide through by W component PS 1_4) 208 | 'abs(R)', # 0xb abs(x) compute absolute value 209 | None, # 0xc -abs(x) compute absolute value and negate 210 | None, # 0xd NOT. Applied only to the predication register, which is BOOL. Therefore, it is logical NOT. 211 | ] 212 | 213 | result_modifier_mnemonic_list = [ 214 | None, 215 | '_sat', 216 | '_pp', 217 | '_centroid', 218 | ] 219 | 220 | texture_type_mnemonic_list = [ 221 | '', # None, 222 | '', # Unused 223 | '2d', 224 | 'cube', 225 | 'volume', 226 | ] 227 | 228 | # http://msdn.microsoft.com/en-us/library/windows/desktop/bb172534(v=vs.85).aspx 229 | usage_type_list = [ 230 | 'position', 231 | 'blendweight', 232 | 'blendindices', 233 | 'normal', 234 | 'psize', 235 | 'texcoord', 236 | 'tangent', 237 | 'binormal', 238 | 'tessfactor', 239 | 'positiont', 240 | 'color', 241 | 'fog', 242 | 'depth', 243 | 'sample', 244 | ] 245 | 246 | class OpCodeToken(): 247 | # Instruction: 248 | # [15:00] opcode 249 | # [23:16] specific controls 250 | # [27:24] ps >= 2_0: dword size 251 | # [28] ps >= 2_0: predicated instruction 252 | # [29] 0 253 | # [30] ps < 2_0: co issue bit 254 | # [31] 0 255 | 256 | def __init__(self, token): 257 | self.token = token 258 | 259 | bs = BitStream(uint=token, length=32) 260 | junk, self.is_predicated, self.instruction_size, self.specific_controls, self.operation_code = bs.unpack('uint:3, bool, uint:4, uint:8, uint:16') 261 | 262 | # HACK comments: should be better way to handle? 263 | if self.operation_code == 0xFFFE: 264 | junk, self.instruction_size, self.operation_code = bs.unpack('uint:1, uint:15, uint:16') 265 | 266 | # print "operation_code=%d (%s) instruction_size=%d specific_controls=%d is_predicated=%d" % (operation_code, operation_code_name, instruction_size, specific_controls, is_predicated) 267 | 268 | def get_name(self): 269 | s = '' 270 | if self.operation_code < len(opcodes_list): 271 | s += opcodes_list[ self.operation_code ] 272 | elif self.operation_code == 0xFFFE: 273 | s += 'comment' 274 | else: 275 | s += "?" 276 | return s 277 | 278 | def __str__(self): 279 | return "\n" + self.get_name() 280 | 281 | class SourceParamToken(): 282 | # Source: 283 | # [10:0] Reg Number 284 | # [12:11] Register Type bits 3,4 285 | # [13] relative addressing bit (0 for all versions except vs3.0) 286 | # [15:14] MBZ 287 | # [23:16] Swizzle [17:16] -x, [19:18] - y etc (0 means X, 1 means Y, 2 means Z, 3-W) 288 | # [27:24] Modifier - 1=Negate, 0xb=Abs(), 0xd=Logical BOOL on predicate reg 289 | # [30:28] Register Type bits 0,1,2 290 | # [31] 1 291 | 292 | def __init__(self, token): 293 | self.token = token 294 | 295 | bs = BitStream(uint=token, length=32) 296 | junk1, regtype1, self.source_modifier, self.swizzle, junk2, self.is_relative_addressing, regtype2, self.register_number = bs.unpack('uint:1, uint:3, uint:4, uint:8, uint:2, bool, uint:2, uint:11') 297 | 298 | self.register_type = regtype2 << 3 | regtype1 299 | 300 | # print "src register_number=%d register_type=%d (%s) swizzle=%d source_modifier=%d is_relative_addressing=%d" % (register_number, register_type, register_type_list[register_type], swizzle, source_modifier, is_relative_addressing) 301 | 302 | def swizzle_to_str(self, swizzle): 303 | s = '' 304 | for i in range(4): 305 | sc = (swizzle & (0x03<<(i*2))) >> (i*2) 306 | if sc == 0x00: s += 'x' 307 | elif sc == 0x01: s += 'y' 308 | elif sc == 0x02: s += 'z' 309 | elif sc == 0x03: s += 'w' 310 | else: s += '?' 311 | if s == 'xxxx': s = 'x' 312 | if s == 'yyyy': s = 'y' 313 | if s == 'zzzz': s = 'z' 314 | if s == 'wwww': s = 'w' 315 | return s 316 | 317 | def __str__(self): 318 | s = '' 319 | if self.register_type < len( register_type_mnemonic_list ): 320 | s = register_type_mnemonic_list[ self.register_type ] + str(self.register_number) 321 | sws = self.swizzle_to_str( self.swizzle ) 322 | if sws != 'xyzw': s += '.' + sws 323 | if self.source_modifier < len( source_modifier_mnemonic_list ): 324 | if source_modifier_mnemonic_list[ self.source_modifier ] == '-R': s = '-' + s 325 | if source_modifier_mnemonic_list[ self.source_modifier ] == 'abs(R)': s = 'abs(' + s + ')' 326 | return s 327 | 328 | class DestParamToken(): 329 | # Destination: 330 | # [10:0] Reg Number 331 | # [12:11] Register Type bits 3,4 332 | # [13] relative addressing bit (0 for all versions except vs3.0) 333 | # [15:14] MBZ 334 | # [19:16] WriteMask 335 | # [23:20] Modifier 1-sat, 2-pp, 3-centroid 336 | # [27:24] MBZ (used for ps < 2.0) 337 | # [30:28] Register Type bits 0,1,2 338 | # [31] 1 339 | 340 | def __init__(self, token): 341 | self.token = token 342 | 343 | bs = BitStream(uint=token, length=32) 344 | junk1, regtype1, junk2, self.result_modifier, self.write_mask, junk3, self.is_relative_addressing, regtype2, self.register_number = bs.unpack('uint:1, uint:3, uint:4, uint:4, uint:4, uint:2, bool, uint:2, uint:11') 345 | 346 | self.register_type = regtype2 << 3 | regtype1 347 | 348 | # print "dest register_number=%d register_type=%d (%s) write_mask=%d (%s) is_relative_addressing=%d" % (register_number, register_type, register_type_list[register_type], write_mask, write_mask_to_str(write_mask), is_relative_addressing) 349 | 350 | def write_mask_to_str(self, write_mask): 351 | s = '' 352 | if write_mask & 1<<0: s += 'x' 353 | if write_mask & 1<<1: s += 'y' 354 | if write_mask & 1<<2: s += 'z' 355 | if write_mask & 1<<3: s += 'w' 356 | return s 357 | 358 | def __str__(self): 359 | s = register_type_mnemonic_list[ self.register_type ] + str(self.register_number) 360 | wms = self.write_mask_to_str( self.write_mask ) 361 | if wms != 'xyzw': s += '.' + wms 362 | # TODO these usually are part of the instruction mnemonic 363 | # if self.result_modifier < len( result_modifier_mnemonic_list ): s = result_modifier_mnemonic_list[ self.result_modifier ] + ' ' + s 364 | return s 365 | 366 | class VersionToken: 367 | def __init__(self, token): 368 | self.token = token 369 | bs = BitStream(uint=token, length=32) 370 | junk1, self.major, self.minor = bs.unpack('uint:16, uint:8, uint:8') 371 | 372 | def __str__(self): 373 | # FIXME could be ps or vs 374 | return 'ps_%d_%d' % (self.major, self.minor) 375 | 376 | class EndToken: 377 | def __init__(self, token): self.token = token 378 | def __str__(self): return "\nend" 379 | 380 | class CommentToken: 381 | def __init__(self, token): 382 | self.token = token 383 | bs = BitStream(uint=token, length=32) 384 | c1, c2, c3, c4 = bs.unpack('uint:8, uint:8, uint:8, uint:8') 385 | self.comment = chr(c1) + chr(c2) + chr(c3) + chr(c4) 386 | def __str__(self): 387 | return "" 388 | # return self.comment 389 | 390 | class DclInfoToken: 391 | def __init__(self, token): 392 | self.token = token 393 | bs = BitStream(uint=token, length=32) 394 | junk1, self.sampler_texture_type, junk2, self.usage_index, junk3, self.usage = bs.unpack("uint:1, uint:4, uint:7, uint:4, uint:12, uint:4") 395 | 396 | def __str__(self): 397 | if(self.sampler_texture_type == 0): 398 | return "" + usage_type_list[ self.usage ] + repr( self.usage_index ) 399 | else: 400 | return "" + texture_type_mnemonic_list[ self.sampler_texture_type ] 401 | 402 | # return "%08x" % self.token 403 | 404 | class ConstFloatToken: 405 | def __init__(self, token): 406 | self.token = token 407 | bs = BitStream(uint=token, length=32) 408 | self.floatval, = bs.unpack('floatbe:32') 409 | 410 | def __str__(self): return "%f" % self.floatval 411 | 412 | class Shader(): 413 | def __init__(self, tokens): 414 | self.tokens = [] 415 | self.parsed_tokens = [] 416 | 417 | opcode = None 418 | opcode_idx = 0 419 | 420 | for i in range(len( tokens )): 421 | token = tokens[ i ] 422 | 423 | if i == 0: 424 | if token & 0xFFFF0000 == 0xFFFF0000: 425 | self.parsed_tokens.append( VersionToken( token ) ) 426 | continue 427 | # else: throw 428 | 429 | if opcode and i <= opcode_idx + opcode.instruction_size: 430 | if opcode.get_name() == 'comment': 431 | self.parsed_tokens.append( CommentToken( token ) ) 432 | elif opcode.get_name() == 'dcl' and i == opcode_idx + 1: 433 | self.parsed_tokens.append( DclInfoToken( token ) ) 434 | elif opcode.get_name() == 'dcl' and i == opcode_idx + 2: 435 | self.parsed_tokens.append( DestParamToken( token ) ) 436 | elif opcode.get_name() == 'def' and i != opcode_idx + 1: 437 | self.parsed_tokens.append( ConstFloatToken( token ) ) 438 | elif opcode.get_name() != 'ifc' and i == opcode_idx + 1: 439 | self.parsed_tokens.append( DestParamToken( token ) ) 440 | else: 441 | self.parsed_tokens.append( SourceParamToken( token ) ) 442 | continue 443 | 444 | if token == 0x0000FFFF: 445 | self.parsed_tokens.append( EndToken( token ) ) 446 | # if i != len( tokens )-1: throw 447 | break 448 | 449 | opcode = OpCodeToken( token ) 450 | opcode_idx = i 451 | self.parsed_tokens.append( opcode ) 452 | 453 | 454 | def __str__(self): 455 | s = '' 456 | for i in range(len( self.parsed_tokens )): 457 | s += str( self.parsed_tokens[ i ] ) + " " 458 | return s 459 | 460 | 461 | 462 | # parse format produced by fxc.exe /Fx 463 | # asm_code = [] 464 | # for line in fileinput.input(): 465 | # m = re.search('// [0-9a-f]{4}: (\w{8}) (\w{8})? (\w{8})? (\w{8})?', line) 466 | # if m: 467 | # t1, t2, t3, t4 = m.groups() 468 | # if t1: asm_code.append( t1 ) 469 | # if t2: asm_code.append( t2 ) 470 | # if t3: asm_code.append( t3 ) 471 | # if t4: asm_code.append( t4 ) 472 | 473 | for line in fileinput.input(): 474 | sys.stdout.write("Tokens: " + line) 475 | asm_code = re.findall('[0-9a-f]{8}', line) 476 | asm_code = map(lambda x: int(x, 16), asm_code) 477 | sys.stdout.write("Assembly: \n") 478 | sh = Shader(asm_code) 479 | sys.stdout.write( str( sh ) ) 480 | sys.stdout.write("\n\n") 481 | sys.stdout.flush() 482 | --------------------------------------------------------------------------------