source/pipeline.cpp

   1 #include <msp/core/maputils.h>
   2 #include "blend.h"
   3 #include "camera.h"
   4 #include "framebuffer.h"
   5 #include "lighting.h"
   6 #include "pipeline.h"
   7 #include "postprocessor.h"
   8 #include "renderbuffer.h"
   9 #include "renderer.h"
  10 #include "tests.h"
  11 #include "texture2d.h"
  12 #include "view.h"
  13
  14 using namespace std;
  15
  16 namespace Msp {
  17 namespace GL {
  18
  19 Pipeline::Pipeline(unsigned w, unsigned h, bool d)
  20 {
  21         init(w, h);
  22         hdr = d;
  23 }
  24
  25 Pipeline::Pipeline(const View &view)
  26 {
  27         init(view.get_width(), view.get_height());
  28 }
  29
  30 Pipeline::Pipeline(const Framebuffer &fbo)
  31 {
  32         init(fbo.get_width(), fbo.get_height());
  33 }
  34
  35 void Pipeline::init(unsigned w, unsigned h)
  36 {
  37         camera = 0;
  38         width = w;
  39         height = h;
  40         hdr = false;
  41         alpha = false;
  42         samples = 0;
  43         target_ms = 0;
  44         target[0] = 0;
  45         target[1] = 0;
  46 }
  47
  48 Pipeline::~Pipeline()
  49 {
  50         delete target[0];
  51         delete target[1];
  52         delete target_ms;
  53 }
  54
  55 void Pipeline::set_hdr(bool h)
  56 {
  57         if(h==hdr)
  58                 return;
  59
  60         bool old_hdr= hdr;
  61         hdr = h;
  62         try
  63         {
  64                 create_targets(2);
  65         }
  66         catch(...)
  67         {
  68                 hdr = old_hdr;
  69                 throw;
  70         }
  71 }
  72
  73 void Pipeline::set_alpha(bool a)
  74 {
  75         if(a==alpha)
  76                 return;
  77
  78         bool old_alpha = alpha;
  79         alpha = a;
  80         try
  81         {
  82                 create_targets(2);
  83         }
  84         catch(...)
  85         {
  86                 alpha = old_alpha;
  87                 throw;
  88         }
  89 }
  90
  91 void Pipeline::set_multisample(unsigned s)
  92 {
  93         if(s==samples)
  94                 return;
  95
  96         unsigned old_samples = samples;
  97         samples = s;
  98         try
  99         {
 100                 create_targets(1);
 101         }
 102         catch(...)
 103         {
 104                 samples = old_samples;
 105                 throw;
 106         }
 107 }
 108
 109 void Pipeline::set_camera(const Camera *c)
 110 {
 111         camera = c;
 112 }
 113
 114 Pipeline::Pass &Pipeline::add_pass(const Tag &tag)
 115 {
 116         passes.push_back(Pass(tag, 0));
 117         return passes.back();
 118 }
 119
 120 void Pipeline::add_renderable(Renderable &r)
 121 {
 122         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
 123                 if(i->renderable==&r)
 124                 {
 125                         i->passes.clear();
 126                         return;
 127                 }
 128
 129         renderables.push_back(&r);
 130 }
 131
 132 void Pipeline::add_renderable_for_pass(Renderable &r, const Tag &tag)
 133 {
 134         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
 135                 if(i->renderable==&r)
 136                 {
 137                         i->passes.insert(tag);
 138                         return;
 139                 }
 140
 141         renderables.push_back(&r);
 142         renderables.back().passes.insert(tag);
 143 }
 144
 145 void Pipeline::remove_renderable(Renderable &r)
 146 {
 147         for(vector<Slot>::iterator i=renderables.begin(); i!=renderables.end(); ++i)
 148                 if(i->renderable==&r)
 149                 {
 150                         renderables.erase(i);
 151                         return;
 152                 }
 153 }
 154
 155 Pipeline::Pass &Pipeline::add_pass(const Tag &tag, Renderable &r)
 156 {
 157         passes.push_back(Pass(tag, &r));
 158         return passes.back();
 159 }
 160
 161 void Pipeline::add_postprocessor(PostProcessor &pp)
 162 {
 163         add_postprocessor(&pp, true);
 164 }
 165
 166 void Pipeline::add_postprocessor_owned(PostProcessor *pp)
 167 {
 168         add_postprocessor(pp, false);
 169 }
 170
 171 void Pipeline::add_postprocessor(PostProcessor *pp, bool keep)
 172 {
 173         postproc.push_back(pp);
 174         if(keep)
 175                 postproc.back().keep();
 176         try
 177         {
 178                 create_targets(0);
 179         }
 180         catch(...)
 181         {
 182                 postproc.pop_back();
 183                 throw;
 184         }
 185 }
 186
 187 void Pipeline::setup_frame(Renderer &renderer)
 188 {
 189         for(PassList::const_iterator i=passes.begin(); i!=passes.end(); ++i)
 190                 if(Renderable *renderable = i->get_renderable())
 191                         renderable->setup_frame(renderer);
 192         for(vector<Slot>::const_iterator i=renderables.begin(); i!=renderables.end(); ++i)
 193                 i->renderable->setup_frame(renderer);
 194 }
 195
 196 void Pipeline::finish_frame()
 197 {
 198         for(PassList::const_iterator i=passes.begin(); i!=passes.end(); ++i)
 199                 if(Renderable *renderable = i->get_renderable())
 200                         renderable->finish_frame();
 201         for(vector<Slot>::const_iterator i=renderables.begin(); i!=renderables.end(); ++i)
 202                 i->renderable->finish_frame();
 203 }
 204
 205 void Pipeline::render(Renderer &renderer, const Tag &tag) const
 206 {
 207         if(tag.id)
 208                 return;
 209
 210         const Framebuffer *out_fbo = Framebuffer::current();
 211         // These are no-ops but will ensure the related state gets restored
 212         BindRestore restore_fbo(out_fbo);
 213         BindRestore restore_depth_test(DepthTest::current());
 214         BindRestore restore_blend(Blend::current());
 215
 216         if(target[0])
 217         {
 218                 Framebuffer &fbo = (samples ? target_ms : target[0])->get_framebuffer();
 219                 fbo.bind();
 220                 fbo.clear(COLOR_BUFFER_BIT|DEPTH_BUFFER_BIT);
 221         }
 222
 223         for(PassList::const_iterator i=passes.begin(); i!=passes.end(); ++i)
 224         {
 225                 if(const DepthTest *dt = i->get_depth_test())
 226                         dt->bind();
 227                 else
 228                         DepthTest::unbind();
 229
 230                 if(const Blend *b = i->get_blend())
 231                         b->bind();
 232                 else
 233                         Blend::unbind();
 234
 235                 renderer.set_lighting(i->get_lighting());
 236                 renderer.set_clipping(i->get_clipping());
 237
 238                 if(const Renderable *renderable = i->get_renderable())
 239                         renderer.render(*renderable, i->get_tag());
 240
 241                 for(vector<Slot>::const_iterator j=renderables.begin(); j!=renderables.end(); ++j)
 242                         if(j->passes.empty() || j->passes.count(i->get_tag()))
 243                                 renderer.render(*j->renderable, i->get_tag());
 244         }
 245
 246         if(target[0])
 247         {
 248                 DepthTest::unbind();
 249                 Blend::unbind();
 250
 251                 if(samples)
 252                         target[0]->blit_from(*target_ms);
 253
 254                 for(unsigned i=0; i<postproc.size(); ++i)
 255                 {
 256                         unsigned j = i%2;
 257                         if(i+1<postproc.size())
 258                                 target[1-j]->get_framebuffer().bind();
 259                         else
 260                                 out_fbo->bind();
 261                         const Texture2D &color = target[j]->get_target_texture(RENDER_COLOR);
 262                         const Texture2D &depth = target[j]->get_target_texture(RENDER_DEPTH);
 263                         postproc[i]->render(renderer, color, depth);
 264                 }
 265         }
 266 }
 267
 268 void Pipeline::create_targets(unsigned recreate)
 269 {
 270         if(recreate>=2)
 271         {
 272                 delete target[0];
 273                 delete target[1];
 274                 target[0] = 0;
 275                 target[1] = 0;
 276         }
 277         if(recreate>=1)
 278         {
 279                 delete target_ms;
 280                 target_ms = 0;
 281         }
 282
 283         PixelFormat color_pf = (hdr ? (alpha ? RGBA16F : RGB16F) : (alpha ? RGBA : RGB));
 284         RenderTargetFormat fmt = (RENDER_COLOR,color_pf, RENDER_DEPTH);
 285         if(!postproc.empty() || samples)
 286         {
 287                 if(!target[0])
 288                         target[0] = new RenderTarget(width, height, fmt);
 289                 if(!target[1] && postproc.size()>1)
 290                         target[1] = new RenderTarget(width, height, fmt);
 291         }
 292
 293         if(!target_ms && samples)
 294                 target_ms = new RenderTarget(width, height, samples, fmt);
 295 }
 296
 297
 298 Pipeline::Pass::Pass(const Tag &t, Renderable *r):
 299         tag(t),
 300         lighting(0),
 301         depth_test(0),
 302         blend(0),
 303         clipping(0),
 304         renderable(r)
 305 { }
 306
 307 void Pipeline::Pass::set_lighting(const Lighting *l)
 308 {
 309         lighting = l;
 310 }
 311
 312 void Pipeline::Pass::set_depth_test(const DepthTest *d)
 313 {
 314         depth_test = d;
 315 }
 316
 317 void Pipeline::Pass::set_blend(const Blend *b)
 318 {
 319         blend = b;
 320 }
 321
 322 void Pipeline::Pass::set_clipping(const Clipping *c)
 323 {
 324         clipping =c;
 325 }
 326
 327
 328 Pipeline::Slot::Slot(Renderable *r):
 329         renderable(r)
 330 { }
 331
 332 } // namespace GL
 333 } // namespace Msp