Created
November 1, 2018 09:23
-
-
Save benjamintanweihao/458928a93254418d83f74a87e1be5f52 to your computer and use it in GitHub Desktop.
This only works on DirectX 10!
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# RenderDoc Python scripts, powered by IronPython 2.7.4.1000 | |
# The 'renderdoc' object is the Core class instance. | |
config = {} | |
config['py_lib_dir'] = 'C:\\Python27\\Lib\\' | |
config['save_dir'] = 'C:\\workspace\\playing-for-data\\data\\processed\\' # where we store extraction results | |
# creates a prefixes for files and directories from logfilename | |
config['dir_prefix'] = lambda logFilename: '' | |
config['file_prefix'] = lambda logFilename: basename(logFilename)[:-4] + '_' | |
# The following values depend on the game and need to be specified. | |
# Hence, we do not set default values. | |
# For extracting mesh, shader, and texture ids we need to reliably identify the G-buffer pass. | |
# By 'G-buffer' pass we mean the pass that renders all objects of interest. Hence, this pass will reference | |
# all meshes, shaders, and textures. | |
# We identify the G-buffer pass by its render targets (number of color targets, whether there is depth) | |
# If multiple passes fulfil the specified conditions, we only process the first one. | |
# You need to change this behaviour for some games. | |
config[ | |
'gbufferpass_colortargets'] = 4 # number of color targets that identify the G-buffer pass, most probably 4 as games want to use all available render targets | |
config['gbufferpass_hasdepth'] = True # whether G-buffer pass has a depth target | |
# Our names for the color buffers | |
config['gbuffer_names'] = ['gbuffer1', 'gbuffer2', 'gbuffer3', | |
'gbuffer4'] # the number should match the config['gbufferpass_colortargets'] | |
# We want to extract clean images without any UI elements. Hence, we need to identify the pass | |
# where the HUD is rendered. | |
config['hudpass_colortargets'] = 4 # number of color targets that identify the HUD pass | |
config['hudpass_hasdepth'] = True # whether HUD pass has a depth target | |
config['hudpass_drawcallname'] = '' # name of drawcall event | |
# Add python libraries | |
import sys | |
sys.path.append(config['py_lib_dir']) | |
# Load frame info | |
frameInfo = renderdoc.FrameInfo | |
assert (len(frameInfo) == 1, 'expected only one frame.') | |
frameId = renderdoc.CurFrame | |
print('Extracting from frame %d' % frameId) | |
# Get prefix and set up directory | |
from os import mkdir | |
from os.path import dirname, basename, exists | |
dirPrefix = config['dir_prefix'](renderdoc.LogFileName) | |
filePrefix = config['file_prefix'](renderdoc.LogFileName) | |
saveDir = '%s/%s/' % (config['save_dir'], dirPrefix) | |
if not exists(saveDir): | |
mkdir(saveDir) | |
pass | |
print('Output directory is %s' % saveDir) | |
print('File prefix is %s' % filePrefix) | |
# Get drawcalls | |
drawcalls = renderdoc.GetDrawcalls(frameId) | |
print('Found %d drawcalls.' % len(drawcalls)) | |
def containsTargets(drawcallName, numColorTargets, hasDepthTarget): | |
""" Determines if drawcall has multiple render targets by checking its name. | |
The name is defined by renderdoc and contains information about render targets.""" | |
if hasDepthTarget: | |
result = drawcallName.find('(%d Targets + Depth)' % numColorTargets) >= 0 | |
return result | |
else: | |
result = drawcallName.find('(%d Targets)' % numColorTargets) >= 0 | |
return result | |
pass | |
def findGbufferPass(numColorTargets, hasDepthTarget): | |
""" Identifies the G-buffer pass. """ | |
gbufferEnd = 0 | |
gbufferIds = [i for i, call in enumerate(drawcalls) if containsTargets(call.name, numColorTargets, hasDepthTarget)] | |
if len(gbufferIds) == 1: | |
gbufferId = gbufferIds[0] | |
gbufferCalls = drawcalls[gbufferId].children | |
print('G-buffer pass has %d drawcalls.' % len(gbufferCalls)) | |
gbufferEnd = drawcalls[gbufferId].children[-1].eventID # last drawcall of the G-buffer pass | |
pass | |
assert (gbufferEnd > 0, 'Did not find any drawcall with the specified G-buffer settings.') | |
return gbufferId, gbufferEnd | |
def findFinalPass(numColorTargets, hasDepthTarget, drawCallName): | |
""" Returns the EventID of the pass that draws the final image (before HUD). """ | |
# Find last drawcall before HUD | |
potentialHudIds = [i for i, call in enumerate(drawcalls) if | |
containsTargets(call.name, numColorTargets, hasDepthTarget)] | |
print('Found %d potential HUD passes.' % len(potentialHudIds)) | |
colorpassIds = [i for i, call in enumerate(drawcalls) if containsTargets(call.name, 1, False)] | |
a = colorpassIds[:] | |
a.extend(potentialHudIds[:-3]) | |
assert (len(a) > 0, 'Found not enough potential final passes.') | |
firstPotentialFinalId = max(a) | |
finalPassId = 0 | |
return [call.eventID for call in drawcalls[-1:0:-1] if call.name.find(drawCallName) >= 0][1] | |
def getColorBuffers(frameId, eventId): | |
""" Sets the pipeline to eventId and returns the ids of bound render targets. """ | |
renderdoc.SetEventID(None, frameId, eventId) | |
commonState = renderdoc.CurPipelineState | |
outputTargets = commonState.GetOutputTargets() | |
return [t for t in outputTargets if str(t) <> '0'] | |
def initIDRendering(frameId, gbufferId, gbufferEnd): | |
""" Initializes ID rendering. """ | |
gbufferStart = drawcalls[gbufferId].children[0].eventID | |
renderdoc.SetEventID(None, frameId, gbufferStart) | |
renderdoc.SetIDRenderingEvents(frameId, gbufferStart, gbufferEnd) | |
renderdoc.SetIDRendering(True) | |
pass | |
gbufferId, gbufferEnd = findGbufferPass(config['gbufferpass_colortargets'], config['gbufferpass_hasdepth']) | |
print('G-buffer pass is done at EID %d.' % gbufferEnd) | |
bufferIds = getColorBuffers(frameId, gbufferEnd) | |
# Save color targets | |
for i, bid in enumerate(bufferIds): | |
renderdoc.SaveTexture(bid, '{0}/{1}_{2}.png'.format(saveDir, filePrefix, config['gbuffer_names'][i])) | |
# Save depth target | |
depthTarget = renderdoc.CurPipelineState.GetDepthTarget() | |
renderdoc.SaveTexture(depthTarget, '{0}/{1}_depth.exr'.format(saveDir, filePrefix)) | |
finalPassId = findFinalPass(config['hudpass_colortargets'], config['hudpass_hasdepth'], config['hudpass_drawcallname']) | |
assert (finalPassId == 0, 'Found not enough potential final passes.') | |
bufferIds = getColorBuffers(frameId, finalPassId) | |
assert (len(bufferIds) == 1, 'Found %d potential final render targets.' % len(bufferIds)) | |
renderdoc.SaveTexture(bufferIds[0], '{0}/{1}_final.png'.format(saveDir, filePrefix)) | |
# now do the id rendering | |
print('Rendering ids...') | |
initIDRendering(frameId, gbufferId, gbufferEnd) | |
bufferIds = getColorBuffers(frameId, gbufferEnd) | |
bufferNames = ['texture', 'mesh', 'shader', 'overflow'] | |
for i, bid in enumerate(bufferIds): | |
renderdoc.SaveTexture(bid, '{0}/{1}_{2}.png'.format(saveDir, filePrefix, bufferNames[i])) | |
renderdoc.HashTextures('{0}/{1}_tex.txt'.format(saveDir, filePrefix)) | |
renderdoc.HashBuffers('{0}/{1}_mesh.txt'.format(saveDir, filePrefix)) | |
renderdoc.HashShaders('{0}/{1}_shader.txt'.format(saveDir, filePrefix)) | |
print('done.') | |
# close renderdoc | |
# renderdoc.AppWindow.Close() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello,actually sometimes I run it successfully and produced right png files and txt files,but only for one logfile,and when I open another logfile and run the same program,it doesn't work with the following error:
EnvironmentError: System.Runtime.InteropServices.SEHException (0x80004005): 外部组件发生异常。
And this error always happens if I go on running the program with a certain logfile,unless I restart my computer then it can work only with the first logfile,and if I go on running the same error happens. Do you know why?