-from os.path import isfile, dirname
-from panda3d.core import AmbientLight, DirectionalLight, PointLight, \
- Spotlight, LVector4f, LVector3f, Vec3, Shader, TextureStage, \
- TexMatrixAttrib
-from direct.filter.FilterManager import FilterManager
-from ya2.lib.builder import LibP3d
-
-
-def load_shader(vert, frag):
-
- def is_file(path):
- joinchar = '/' if LibP3d.runtime() and not path.startswith('/') else ''
- dpath = LibP3d.runtime() and dirname(__file__)
- return isfile((dpath or '') + joinchar + path)
- if is_file(vert) and is_file(frag):
- shader = Shader.load(Shader.SLGLSL, vert, frag)
- else: shader = Shader.make(Shader.SLGLSL, vert, frag)
- return shader
-
-
-class P3dShaderMgr:
-
- def __init__(self, shaders, gamma):
- self.filter_mgr = None
- self.gamma, self.buffer, self.lcam, self.lights = gamma, None, None, []
- if shaders: self.setup_post_fx()
-
- def __set_lgt(self, lgt, col):
- if type(col) in [int, float]: lgt.set_color_temperature(col)
- else: lgt.set_color(col)
- self.lights += [render.attach_new_node(lgt)]
- render.set_light(self.lights[-1])
-
- def set_amb_lgt(self, col):
- self.__set_lgt(AmbientLight('ambient light'), col)
-
- def set_dir_lgt(self, col, direction):
- self.__set_lgt(DirectionalLight('directional light'), col)
- self.lights[-1].set_hpr(*direction)
-
- def set_shadow_lgt(self, direction):
- self.__set_lgt(DirectionalLight('directional light'), (1, 1, 1, 1))
- self.lights[-1].node().set_shadow_caster(True, 8192, 8192)
- self.lights[-1].node().get_lens().set_film_size(2048, 2048)
- self.lights[-1].node().get_lens().set_near_far(1, 2048)
- #self.lights[-1].node().show_frustum()
- self.lights[-1].set_hpr(*direction)
- return self.lights[-1]
-
- def set_pnt_lgt(self, col, pos):
- self.__set_lgt(PointLight('point light'), col)
- self.lights[-1].set_pos(*pos)
-
- def set_spotlight(self, col, exp, cutoff, pos, look_at):
- self.__set_lgt(Spotlight('spotlight'), col)
- self.lights[-1].set_exponent(exp)
- self.lights[-1].get_lens().set_fov(cutoff, cutoff)
- self.lights[-1].set_pos(*pos)
- self.lights[-1].look_at(*look_at)
-
- @staticmethod
- def set_default_args(idx):
- pref = 'lights[%s].' % idx
- args = [(pref + 'pos', LVector4f(0, 0, 0, 1)),
- (pref + 'amb', LVector3f(0, 0, 0)),
- (pref + 'diff', LVector3f(0, 0, 0)),
- (pref + 'spec', LVector3f(0, 0, 0)),
- (pref + 'dir', LVector3f(0, 0, 0)),
- (pref + 'exp', .0),
- (pref + 'cutoff', .0)]
- list(map(lambda _args: render.set_shader_input(*_args), args))
-
- def set_lgt_args(self, idx, lgt):
- self.set_default_args(idx)
- ShaderSetter.build(lgt).set('lights[%s].' % idx, lgt)
-
- def clear_lights(self):
- for lgt in self.lights:
- base.render.clear_light(lgt)
- lgt.removeNode()
- self.lights = []
-
- def setup_post_fx(self):
- self.filter_mgr = FilterManager(base.win, base.cam)
- # rendered_scene = Texture()
- # aa_scene = Texture()
- # filtered_scene = Texture()
- # filter_quad = self.filter_mgr.renderQuadInto(colortex=filtered_scene)
- # aa_quad = self.filter_mgr.renderQuadInto(colortex=aa_scene)
- # final_quad = self.filter_mgr.renderSceneInto(colortex=rendered_scene)
- # filter_quad.set_shader(self.__load_shader('filter', 'sobel_filter'))
- # filter_quad.set_shader_input('in_tex', rendered_scene)
- # aa_quad.set_shader(self.__load_shader('fxaa', 'fxaa'))
- # aa_quad.set_shader_input('in_tex', filtered_scene)
- # final_quad.set_shader(self.__load_shader('filter', 'pass'))
- # final_quad.set_shader_input('gamma', self.gamma)
- # final_quad.set_shader_input('in_tex', aa_scene)
-
- @staticmethod
- def __load_shader(vshad, fshad):
- with open('assets/shaders/%s.vert' % vshad) as vfile:
- fvert = vfile.read()
- with open('assets/shaders/%s.frag' % fshad) as ffile:
- ffrag = ffile.read()
- return load_shader(fvert, ffrag)
-
- def apply(self):
- # winprops = WindowProperties.size(2048, 2048)
- # props = FrameBufferProperties()
- # props.set_rgb_color(1)
- # props.set_alpha_bits(1)
- # props.set_depth_bits(1)
- # lbuffer = base.graphicsEngine.make_output(
- # base.pipe, 'offscreen buffer', -2, props, winprops,
- # GraphicsPipe.BFRefuseWindow, base.win.getGsg(), base.win)
- # self.buffer = lbuffer
- # ldepthmap = Texture()
- # lbuffer.addRenderTexture(ldepthmap, GraphicsOutput.RTMBindOrCopy,
- # GraphicsOutput.RTPDepthStencil)
- # ldepthmap.set_minfilter(Texture.FTShadow)
- # ldepthmap.set_magfilter(Texture.FTShadow)
-
- # base.camLens.set_near_far(1.0, 10000)
- # base.camLens.set_fov(75)
-
- # self.lcam = base.makeCamera(lbuffer)
- # self.lcam.node().set_scene(render)
- # self.lcam.node().get_lens().set_fov(45)
- # self.lcam.node().get_lens().set_near_far(1, 100)
-
- # render.set_shader_input('light', self.lcam)
- # render.set_shader_input('depthmap', ldepthmap)
- # render.set_shader_input('ambient', .15, .15, .15, 1.0)
-
- # lci = NodePath(PandaNode('light camera initializer'))
- # lci.set_shader(self.__load_shader('caster', 'caster'))
- # self.lcam.node().set_initial_state(lci.get_state())
-
- # mci = NodePath(PandaNode('main camera initializer'))
- # # use PTALVecBaseX instead
- # # setShaderInput('vec3argname', PTALVecBase3(((0, 0, 0), (1, 1, 1))))
- render.set_shader(self.__main_shader())
- # render.set_shader_input('num_lights', len(self.lights))
- # self.set_shader_pars(render)
- # list(map(
- # lambda lgt: self.set_lgt_args(*lgt), enumerate(self.lights)))
- # mci.setShader(self.__main_shader())
- # base.cam.node().set_initial_state(mci.getState())
-
- # self.lcam.set_pos(15, 30, 45)
- # self.lcam.look_at(0, 15, 0)
- # self.lcam.node().get_lens().set_near_far(1, 100)
-
- def __main_shader(self):
- with open('assets/shaders/main.vert') as fvert:
- vert = fvert.read()
- with open('assets/shaders/main.frag') as ffrag:
- frag = ffrag.read()
- frag = frag.replace('<LIGHTS>', str(len(self.lights)))
- return load_shader(vert, frag)
-
- def toggle_shader(self):
- if render.get_shader():
- render.set_shader_off()
- render.set_shader_auto()
- return
- self.apply()
-
- def set_shader_pars(self, model):
- texture_stages = model.find_all_texture_stages()
- model.set_shader_input('gloss_slot', 0)
- model.set_shader_input('detail_slot', 0)
- model.set_shader_input('detail_scale', (1, 1))
- for tstage in texture_stages:
- if tstage.getSort() == 0: continue
- self.__set_slots(tstage, model, 1 if tstage.getSort() == 10 else 2)
-
- @staticmethod
- def __set_slots(tstage, model, slot):
- if tstage.getMode() == TextureStage.MGloss:
- model.set_shader_input('gloss_slot', slot)
- else:
- model.set_shader_input('detail_slot', slot)
- attrib_type = TexMatrixAttrib.get_class_type()
- for geom_np in model.find_all_matches('**/+GeomNode'):
- geom_node = geom_np.node()
- for i in range(geom_node.get_num_geoms()):
- state = geom_node.get_geom_state(i)
- if state.has_attrib(attrib_type):
- attrib = state.get_attrib(attrib_type)
- for j in range(attrib.get_num_stages()):
- stage = attrib.get_stage(j)
- scale = attrib.get_transform(stage).get_scale()
- model.set_shader_input('detail_scale', scale)
-
- def destroy(self): self.clear_lights()
-
-
-class ShaderSetter:
-
- @staticmethod
- def build(lgt):
- cls2sett = {
- AmbientLight: ShaderSetterAmbient,
- PointLight: ShaderSetterPointLight,
- DirectionalLight: ShaderSetterDirectionalLight,
- Spotlight: ShaderSetterSpotlight}
- return cls2sett[lgt.node().__class__]()
-
- @staticmethod
- def _set_pars(pref, lgt_pos, lgt):
- render.set_shader_input(pref + 'pos', lgt_pos)
- render.set_shader_input(pref + 'diff', lgt.node().get_color())
- render.set_shader_input(pref + 'spec', lgt.node().get_color())
-
-
-class ShaderSetterAmbient(ShaderSetter):
-
- @staticmethod
- def set(pref, lgt):
- render.set_shader_input(pref + 'amb', lgt.node().get_color())
-
-
-class ShaderSetterPointLight(ShaderSetter):
-
- @staticmethod
- def set(pref, lgt):
- lgt_pos = lgt.get_mat(base.cam).xform(LVector4f(0, 0, 0, 1))
- ShaderSetter._set_pars(pref, lgt_pos, lgt)
-
-
-class ShaderSetterDirectionalLight(ShaderSetter):
-
- @staticmethod
- def set(pref, lgt):
- lgt_vec = -render.get_relative_vector(lgt, Vec3(0, 1, 0))
- lgt_pos = LVector4f(lgt_vec[0], lgt_vec[1], lgt_vec[2], 0)
- ShaderSetter._set_pars(pref, lgt_pos, lgt)
-
-
-class ShaderSetterSpotlight(ShaderSetter):
-
- @staticmethod
- def set(pref, lgt):
- lgt_vec = base.cam.get_relative_vector(lgt, Vec3(0, 1, 0))
- lgt_pos = lgt.get_mat(base.cam).xform(LVector4f(0, 0, 0, 1))
- ShaderSetter._set_pars(pref, lgt_pos, lgt)
- render.set_shader_input(pref + 'dir', lgt_vec)
- render.set_shader_input(pref + 'exp', lgt.node().get_exponent())
- cutoff = lgt.node().get_lens().get_fov()[0]
- render.set_shader_input(pref + 'cutoff', cutoff)
+# from os.path import isfile, dirname
+# from panda3d.core import AmbientLight, DirectionalLight, PointLight, \
+# Spotlight, LVector4f, LVector3f, Vec3, Shader, TextureStage, \
+# TexMatrixAttrib
+# from direct.filter.FilterManager import FilterManager
+# from ya2.lib.builder import LibP3d
+
+
+# def load_shader(vert, frag):
+
+# def is_file(path):
+# joinchar = '/' if LibP3d.runtime() and not path.startswith('/') else ''
+# dpath = LibP3d.runtime() and dirname(__file__)
+# return isfile((dpath or '') + joinchar + path)
+# if is_file(vert) and is_file(frag):
+# shader = Shader.load(Shader.SLGLSL, vert, frag)
+# else: shader = Shader.make(Shader.SLGLSL, vert, frag)
+# return shader
+
+
+# class P3dShaderMgr:
+
+# def __init__(self, shaders, gamma):
+# self.filter_mgr = None
+# self.gamma, self.buffer, self.lcam, self.lights = gamma, None, None, []
+# if shaders: self.setup_post_fx()
+
+# def __set_lgt(self, lgt, col):
+# if type(col) in [int, float]: lgt.set_color_temperature(col)
+# else: lgt.set_color(col)
+# self.lights += [render.attach_new_node(lgt)]
+# render.set_light(self.lights[-1])
+
+# def set_amb_lgt(self, col):
+# self.__set_lgt(AmbientLight('ambient light'), col)
+
+# def set_dir_lgt(self, col, direction):
+# self.__set_lgt(DirectionalLight('directional light'), col)
+# self.lights[-1].set_hpr(*direction)
+
+# def set_shadow_lgt(self, direction):
+# self.__set_lgt(DirectionalLight('directional light'), (1, 1, 1, 1))
+# self.lights[-1].node().set_shadow_caster(True, 8192, 8192)
+# self.lights[-1].node().get_lens().set_film_size(2048, 2048)
+# self.lights[-1].node().get_lens().set_near_far(1, 2048)
+# #self.lights[-1].node().show_frustum()
+# self.lights[-1].set_hpr(*direction)
+# return self.lights[-1]
+
+# def set_pnt_lgt(self, col, pos):
+# self.__set_lgt(PointLight('point light'), col)
+# self.lights[-1].set_pos(*pos)
+
+# def set_spotlight(self, col, exp, cutoff, pos, look_at):
+# self.__set_lgt(Spotlight('spotlight'), col)
+# self.lights[-1].set_exponent(exp)
+# self.lights[-1].get_lens().set_fov(cutoff, cutoff)
+# self.lights[-1].set_pos(*pos)
+# self.lights[-1].look_at(*look_at)
+
+# @staticmethod
+# def set_default_args(idx):
+# pref = 'lights[%s].' % idx
+# args = [(pref + 'pos', LVector4f(0, 0, 0, 1)),
+# (pref + 'amb', LVector3f(0, 0, 0)),
+# (pref + 'diff', LVector3f(0, 0, 0)),
+# (pref + 'spec', LVector3f(0, 0, 0)),
+# (pref + 'dir', LVector3f(0, 0, 0)),
+# (pref + 'exp', .0),
+# (pref + 'cutoff', .0)]
+# list(map(lambda _args: render.set_shader_input(*_args), args))
+
+# def set_lgt_args(self, idx, lgt):
+# self.set_default_args(idx)
+# ShaderSetter.build(lgt).set('lights[%s].' % idx, lgt)
+
+# def clear_lights(self):
+# for lgt in self.lights:
+# base.render.clear_light(lgt)
+# lgt.removeNode()
+# self.lights = []
+
+# def setup_post_fx(self):
+# self.filter_mgr = FilterManager(base.win, base.cam)
+# # rendered_scene = Texture()
+# # aa_scene = Texture()
+# # filtered_scene = Texture()
+# # filter_quad = self.filter_mgr.renderQuadInto(colortex=filtered_scene)
+# # aa_quad = self.filter_mgr.renderQuadInto(colortex=aa_scene)
+# # final_quad = self.filter_mgr.renderSceneInto(colortex=rendered_scene)
+# # filter_quad.set_shader(self.__load_shader('filter', 'sobel_filter'))
+# # filter_quad.set_shader_input('in_tex', rendered_scene)
+# # aa_quad.set_shader(self.__load_shader('fxaa', 'fxaa'))
+# # aa_quad.set_shader_input('in_tex', filtered_scene)
+# # final_quad.set_shader(self.__load_shader('filter', 'pass'))
+# # final_quad.set_shader_input('gamma', self.gamma)
+# # final_quad.set_shader_input('in_tex', aa_scene)
+
+# @staticmethod
+# def __load_shader(vshad, fshad):
+# with open('assets/shaders/%s.vert' % vshad) as vfile:
+# fvert = vfile.read()
+# with open('assets/shaders/%s.frag' % fshad) as ffile:
+# ffrag = ffile.read()
+# return load_shader(fvert, ffrag)
+
+# def apply(self):
+# # winprops = WindowProperties.size(2048, 2048)
+# # props = FrameBufferProperties()
+# # props.set_rgb_color(1)
+# # props.set_alpha_bits(1)
+# # props.set_depth_bits(1)
+# # lbuffer = base.graphicsEngine.make_output(
+# # base.pipe, 'offscreen buffer', -2, props, winprops,
+# # GraphicsPipe.BFRefuseWindow, base.win.getGsg(), base.win)
+# # self.buffer = lbuffer
+# # ldepthmap = Texture()
+# # lbuffer.addRenderTexture(ldepthmap, GraphicsOutput.RTMBindOrCopy,
+# # GraphicsOutput.RTPDepthStencil)
+# # ldepthmap.set_minfilter(Texture.FTShadow)
+# # ldepthmap.set_magfilter(Texture.FTShadow)
+
+# # base.camLens.set_near_far(1.0, 10000)
+# # base.camLens.set_fov(75)
+
+# # self.lcam = base.makeCamera(lbuffer)
+# # self.lcam.node().set_scene(render)
+# # self.lcam.node().get_lens().set_fov(45)
+# # self.lcam.node().get_lens().set_near_far(1, 100)
+
+# # render.set_shader_input('light', self.lcam)
+# # render.set_shader_input('depthmap', ldepthmap)
+# # render.set_shader_input('ambient', .15, .15, .15, 1.0)
+
+# # lci = NodePath(PandaNode('light camera initializer'))
+# # lci.set_shader(self.__load_shader('caster', 'caster'))
+# # self.lcam.node().set_initial_state(lci.get_state())
+
+# # mci = NodePath(PandaNode('main camera initializer'))
+# # # use PTALVecBaseX instead
+# # # setShaderInput('vec3argname', PTALVecBase3(((0, 0, 0), (1, 1, 1))))
+# render.set_shader(self.__main_shader())
+# # render.set_shader_input('num_lights', len(self.lights))
+# # self.set_shader_pars(render)
+# # list(map(
+# # lambda lgt: self.set_lgt_args(*lgt), enumerate(self.lights)))
+# # mci.setShader(self.__main_shader())
+# # base.cam.node().set_initial_state(mci.getState())
+
+# # self.lcam.set_pos(15, 30, 45)
+# # self.lcam.look_at(0, 15, 0)
+# # self.lcam.node().get_lens().set_near_far(1, 100)
+
+# def __main_shader(self):
+# with open('assets/shaders/main.vert') as fvert:
+# vert = fvert.read()
+# with open('assets/shaders/main.frag') as ffrag:
+# frag = ffrag.read()
+# frag = frag.replace('<LIGHTS>', str(len(self.lights)))
+# return load_shader(vert, frag)
+
+# def toggle_shader(self):
+# if render.get_shader():
+# render.set_shader_off()
+# render.set_shader_auto()
+# return
+# self.apply()
+
+# def set_shader_pars(self, model):
+# texture_stages = model.find_all_texture_stages()
+# model.set_shader_input('gloss_slot', 0)
+# model.set_shader_input('detail_slot', 0)
+# model.set_shader_input('detail_scale', (1, 1))
+# for tstage in texture_stages:
+# if tstage.getSort() == 0: continue
+# self.__set_slots(tstage, model, 1 if tstage.getSort() == 10 else 2)
+
+# @staticmethod
+# def __set_slots(tstage, model, slot):
+# if tstage.getMode() == TextureStage.MGloss:
+# model.set_shader_input('gloss_slot', slot)
+# else:
+# model.set_shader_input('detail_slot', slot)
+# attrib_type = TexMatrixAttrib.get_class_type()
+# for geom_np in model.find_all_matches('**/+GeomNode'):
+# geom_node = geom_np.node()
+# for i in range(geom_node.get_num_geoms()):
+# state = geom_node.get_geom_state(i)
+# if state.has_attrib(attrib_type):
+# attrib = state.get_attrib(attrib_type)
+# for j in range(attrib.get_num_stages()):
+# stage = attrib.get_stage(j)
+# scale = attrib.get_transform(stage).get_scale()
+# model.set_shader_input('detail_scale', scale)
+
+# def destroy(self): self.clear_lights()
+
+
+# class ShaderSetter:
+
+# @staticmethod
+# def build(lgt):
+# cls2sett = {
+# AmbientLight: ShaderSetterAmbient,
+# PointLight: ShaderSetterPointLight,
+# DirectionalLight: ShaderSetterDirectionalLight,
+# Spotlight: ShaderSetterSpotlight}
+# return cls2sett[lgt.node().__class__]()
+
+# @staticmethod
+# def _set_pars(pref, lgt_pos, lgt):
+# render.set_shader_input(pref + 'pos', lgt_pos)
+# render.set_shader_input(pref + 'diff', lgt.node().get_color())
+# render.set_shader_input(pref + 'spec', lgt.node().get_color())
+
+
+# class ShaderSetterAmbient(ShaderSetter):
+
+# @staticmethod
+# def set(pref, lgt):
+# render.set_shader_input(pref + 'amb', lgt.node().get_color())
+
+
+# class ShaderSetterPointLight(ShaderSetter):
+
+# @staticmethod
+# def set(pref, lgt):
+# lgt_pos = lgt.get_mat(base.cam).xform(LVector4f(0, 0, 0, 1))
+# ShaderSetter._set_pars(pref, lgt_pos, lgt)
+
+
+# class ShaderSetterDirectionalLight(ShaderSetter):
+
+# @staticmethod
+# def set(pref, lgt):
+# lgt_vec = -render.get_relative_vector(lgt, Vec3(0, 1, 0))
+# lgt_pos = LVector4f(lgt_vec[0], lgt_vec[1], lgt_vec[2], 0)
+# ShaderSetter._set_pars(pref, lgt_pos, lgt)
+
+
+# class ShaderSetterSpotlight(ShaderSetter):
+
+# @staticmethod
+# def set(pref, lgt):
+# lgt_vec = base.cam.get_relative_vector(lgt, Vec3(0, 1, 0))
+# lgt_pos = lgt.get_mat(base.cam).xform(LVector4f(0, 0, 0, 1))
+# ShaderSetter._set_pars(pref, lgt_pos, lgt)
+# render.set_shader_input(pref + 'dir', lgt_vec)
+# render.set_shader_input(pref + 'exp', lgt.node().get_exponent())
+# cutoff = lgt.node().get_lens().get_fov()[0]
+# render.set_shader_input(pref + 'cutoff', cutoff)
Ya2 / projects / pmachines.git / blobdiff