source/pipeline.cpp

   1 /* $Id$
   2
   3 This file is part of libmspgl
   4 Copyright © 2009-2010  Mikko Rasa, Mikkosoft Productions
   5 Distributed under the LGPL
   6 */
   7
   8 #include "blend.h"
   9 #include "camera.h"
  10 #include "effect.h"
  11 #include "except.h"
  12 #include "framebuffer.h"
  13 #include "lighting.h"
  14 #include "pipeline.h"
  15 #include "postprocessor.h"
  16 #include "renderbuffer.h"
  17 #include "tests.h"
  18 #include "texture2d.h"
  19
  20 using namespace std;
  21
  22 namespace Msp {
  23 namespace GL {
  24
  25 Pipeline::Pipeline(unsigned w, unsigned h, bool d):
  26         camera(0),
  27         width(w),
  28         height(h),
  29         hdr(d),
  30         fbo(0),
  31         color_buf(0),
  32         depth_buf(0)
  33 { }
  34
  35 Pipeline::~Pipeline()
  36 {
  37         delete fbo;
  38         delete color_buf;
  39         delete depth_buf;
  40 }
  41
  42 void Pipeline::set_camera(const Camera *c)
  43 {
  44         camera = c;
  45 }
  46
  47 PipelinePass &Pipeline::add_pass(const Tag &tag)
  48 {
  49         if(passes.count(tag))
  50                 throw KeyError("Pass already exists");
  51
  52         PipelinePass &pass = passes[tag];
  53         pass_order.push_back(tag);
  54         return pass;
  55 }
  56
  57 PipelinePass &Pipeline::get_pass(const Tag &tag)
  58 {
  59         PassMap::iterator i = passes.find(tag);
  60         if(i==passes.end())
  61                 throw KeyError("Unknown pass");
  62         return i->second;
  63 }
  64
  65 const PipelinePass &Pipeline::get_pass(const Tag &tag) const
  66 {
  67         PassMap::const_iterator i = passes.find(tag);
  68         if(i==passes.end())
  69                 throw KeyError("Unknown pass");
  70         return i->second;
  71 }
  72
  73 void Pipeline::add_renderable(const Renderable &r)
  74 {
  75         renderables.push_back(&r);
  76 }
  77
  78 void Pipeline::add_effect(Effect &e)
  79 {
  80         effects.push_back(&e);
  81 }
  82
  83 void Pipeline::add_postprocessor(PostProcessor &pp)
  84 {
  85         postproc.push_back(&pp);
  86         if(!fbo)
  87         {
  88                 fbo = new Framebuffer;
  89                 color_buf = new Texture2D;
  90                 color_buf->set_min_filter(NEAREST);
  91                 color_buf->set_mag_filter(NEAREST);
  92                 color_buf->storage((hdr ? RGB16F : RGB), width, height, 0);
  93                 color_buf->image(0, RGB, UNSIGNED_BYTE, 0);
  94                 fbo->attach(COLOR_ATTACHMENT0, *color_buf, 0);
  95                 depth_buf = new Renderbuffer;
  96                 depth_buf->storage(DEPTH_COMPONENT, width, height);
  97                 fbo->attach(DEPTH_ATTACHMENT, *depth_buf);
  98         }
  99 }
 100
 101 void Pipeline::render(const Tag &tag) const
 102 {
 103         const PipelinePass &pass = get_pass(tag);
 104
 105         Bind bind_depth_test(pass.depth_test);
 106         Bind bind_blend(pass.blend);
 107         Bind bind_lighting(pass.lighting);
 108
 109         for(vector<Effect *>::const_iterator i=pass.effects.begin(); i!=pass.effects.end(); ++i)
 110                 (*i)->prepare();
 111
 112         for(vector<const Renderable *>::const_iterator i=renderables.begin(); i!=renderables.end(); ++i)
 113                 (*i)->render(tag);
 114
 115         for(vector<Effect *>::const_iterator i=pass.effects.end(); i!=pass.effects.begin();)
 116                 (*--i)->cleanup();
 117 }
 118
 119 void Pipeline::render_all() const
 120 {
 121         if(camera)
 122                 camera->apply();
 123
 124         if(fbo)
 125         {
 126                 fbo->bind();
 127                 fbo->clear(COLOR_BUFFER_BIT|DEPTH_BUFFER_BIT);
 128         }
 129
 130         for(vector<Effect *>::const_iterator i=effects.begin(); i!=effects.end(); ++i)
 131                 (*i)->prepare();
 132
 133         for(vector<Tag>::const_iterator i=pass_order.begin(); i!=pass_order.end(); ++i)
 134                 render(*i);
 135
 136         for(vector<Effect *>::const_iterator i=effects.end(); i!=effects.begin();)
 137                 (*--i)->cleanup();
 138
 139         if(fbo)
 140                 Framebuffer::unbind();
 141
 142         // XXX Need two color buffer textures to handle multiple post-processors correctly
 143         for(vector<PostProcessor *>::const_iterator i=postproc.begin(); i!=postproc.end(); ++i)
 144                 (*i)->render(*color_buf);
 145 }
 146
 147 } // namespace GL
 148 } // namespace Msp