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         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
  76                 if(i->renderable==&r)
  77                 {
  78                         i->passes.clear();
  79                         return;
  80                 }
  81
  82         renderables.push_back(&r);
  83 }
  84
  85 void Pipeline::add_renderable_for_pass(const Renderable &r, const Tag &tag)
  86 {
  87         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
  88                 if(i->renderable==&r)
  89                 {
  90                         i->passes.insert(tag);
  91                         return;
  92                 }
  93
  94         renderables.push_back(&r);
  95         renderables.back().passes.insert(tag);
  96 }
  97
  98 void Pipeline::remove_renderable(const Renderable &r)
  99 {
 100         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
 101                 if(i->renderable==&r)
 102                 {
 103                         renderables.erase(i);
 104                         return;
 105                 }
 106 }
 107
 108 void Pipeline::add_effect(Effect &e)
 109 {
 110         effects.push_back(&e);
 111 }
 112
 113 void Pipeline::add_postprocessor(PostProcessor &pp)
 114 {
 115         postproc.push_back(&pp);
 116         if(!fbo)
 117         {
 118                 fbo = new Framebuffer;
 119                 color_buf = new Texture2D;
 120                 color_buf->set_min_filter(NEAREST);
 121                 color_buf->set_mag_filter(NEAREST);
 122                 color_buf->storage((hdr ? RGB16F : RGB), width, height);
 123                 color_buf->image(0, RGB, UNSIGNED_BYTE, 0);
 124                 fbo->attach(COLOR_ATTACHMENT0, *color_buf, 0);
 125                 depth_buf = new Renderbuffer;
 126                 depth_buf->storage(DEPTH_COMPONENT, width, height);
 127                 fbo->attach(DEPTH_ATTACHMENT, *depth_buf);
 128         }
 129 }
 130
 131 void Pipeline::render(const Tag &tag) const
 132 {
 133         const PipelinePass &pass = get_pass(tag);
 134
 135         Bind bind_depth_test(pass.depth_test);
 136         Bind bind_blend(pass.blend);
 137         Bind bind_lighting(pass.lighting);
 138
 139         for(vector<Effect *>::const_iterator i=pass.effects.begin(); i!=pass.effects.end(); ++i)
 140                 (*i)->prepare();
 141
 142         for(vector<Slot>::const_iterator i=renderables.begin(); i!=renderables.end(); ++i)
 143                 if(i->passes.empty() || i->passes.count(tag))
 144                         i->renderable->render(tag);
 145
 146         for(vector<Effect *>::const_iterator i=pass.effects.end(); i!=pass.effects.begin();)
 147                 (*--i)->cleanup();
 148 }
 149
 150 void Pipeline::render_all() const
 151 {
 152         if(camera)
 153                 camera->apply();
 154
 155         if(fbo)
 156         {
 157                 fbo->bind();
 158                 fbo->clear(COLOR_BUFFER_BIT|DEPTH_BUFFER_BIT);
 159         }
 160
 161         for(vector<Effect *>::const_iterator i=effects.begin(); i!=effects.end(); ++i)
 162                 (*i)->prepare();
 163
 164         for(vector<Tag>::const_iterator i=pass_order.begin(); i!=pass_order.end(); ++i)
 165                 render(*i);
 166
 167         for(vector<Effect *>::const_iterator i=effects.end(); i!=effects.begin();)
 168                 (*--i)->cleanup();
 169
 170         if(fbo)
 171                 Framebuffer::unbind();
 172
 173         // XXX Need two color buffer textures to handle multiple post-processors correctly
 174         for(vector<PostProcessor *>::const_iterator i=postproc.begin(); i!=postproc.end(); ++i)
 175                 (*i)->render(*color_buf);
 176 }
 177
 178
 179 Pipeline::Slot::Slot(const Renderable *r):
 180         renderable(r)
 181 { }
 182
 183 } // namespace GL
 184 } // namespace Msp