git.sesse.net Git - mlt/blob - src/modules/opengl/filter_glsl_manager.cpp

   1 /*
   2  * filter_glsl_manager.cpp
   3  * Copyright (C) 2011-2012 Christophe Thommeret <hftom@free.fr>
   4  * Copyright (C) 2013 Dan Dennedy <dan@dennedy.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19  */
  20
  21 #include <stdlib.h>
  22 #include <string>
  23 #include "glsl_manager.h"
  24 #include <movit/init.h>
  25 #include <movit/effect_chain.h>
  26 #include "mlt_movit_input.h"
  27 #include "mlt_flip_effect.h"
  28 #include <mlt++/MltEvent.h>
  29 #include <mlt++/MltProducer.h>
  30
  31 extern "C" {
  32 #include <framework/mlt_factory.h>
  33 }
  34
  35 void deleteManager(GlslManager *p)
  36 {
  37         delete p;
  38 }
  39
  40 GlslManager::GlslManager()
  41         : Mlt::Filter( mlt_filter_new() )
  42         , pbo(0)
  43         , initEvent(0)
  44         , closeEvent(0)
  45 {
  46         mlt_filter filter = get_filter();
  47         if ( filter ) {
  48                 // Set the mlt_filter child in case we choose to override virtual functions.
  49                 filter->child = this;
  50                 mlt_properties_set_data(mlt_global_properties(), "glslManager", this, 0,
  51                         (mlt_destructor) deleteManager, NULL);
  52
  53                 mlt_events_register( get_properties(), "init glsl", NULL );
  54                 mlt_events_register( get_properties(), "close glsl", NULL );
  55                 initEvent = listen("init glsl", this, (mlt_listener) GlslManager::onInit);
  56                 closeEvent = listen("close glsl", this, (mlt_listener) GlslManager::onClose);
  57         }
  58 }
  59
  60 GlslManager::~GlslManager()
  61 {
  62         mlt_log_debug(get_service(), "%s\n", __FUNCTION__);
  63         cleanupContext();
  64         delete initEvent;
  65         delete closeEvent;
  66 }
  67
  68 GlslManager* GlslManager::get_instance()
  69 {
  70         return (GlslManager*) mlt_properties_get_data(mlt_global_properties(), "glslManager", 0);
  71 }
  72
  73 glsl_fbo GlslManager::get_fbo(int width, int height)
  74 {
  75         lock();
  76         for (int i = 0; i < fbo_list.count(); ++i) {
  77                 glsl_fbo fbo = (glsl_fbo) fbo_list.peek(i);
  78                 if (!fbo->used && (fbo->width == width) && (fbo->height == height)) {
  79                         fbo->used = 1;
  80                         unlock();
  81                         return fbo;
  82                 }
  83         }
  84         unlock();
  85
  86         GLuint fb = 0;
  87         glGenFramebuffers(1, &fb);
  88         if (!fb)
  89                 return NULL;
  90
  91         glsl_fbo fbo = new glsl_fbo_s;
  92         if (!fbo) {
  93                 glDeleteFramebuffers(1, &fb);
  94                 return NULL;
  95         }
  96         fbo->fbo = fb;
  97         fbo->width = width;
  98         fbo->height = height;
  99         fbo->used = 1;
 100         lock();
 101         fbo_list.push_back(fbo);
 102         unlock();
 103         return fbo;
 104 }
 105
 106 void GlslManager::release_fbo(glsl_fbo fbo)
 107 {
 108         fbo->used = 0;
 109 }
 110
 111 glsl_texture GlslManager::get_texture(int width, int height, GLint internal_format)
 112 {
 113         lock();
 114         for (int i = 0; i < texture_list.count(); ++i) {
 115                 glsl_texture tex = (glsl_texture) texture_list.peek(i);
 116                 if (!tex->used && (tex->width == width) && (tex->height == height) && (tex->internal_format == internal_format)) {
 117                         glBindTexture(GL_TEXTURE_2D, tex->texture);
 118                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
 119                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
 120                         glBindTexture( GL_TEXTURE_2D, 0);
 121                         tex->used = 1;
 122                         unlock();
 123                         return tex;
 124                 }
 125         }
 126         unlock();
 127
 128         GLuint tex = 0;
 129         glGenTextures(1, &tex);
 130         if (!tex)
 131                 return NULL;
 132
 133         glsl_texture gtex = new glsl_texture_s;
 134         if (!gtex) {
 135                 glDeleteTextures(1, &tex);
 136                 return NULL;
 137         }
 138         glBindTexture( GL_TEXTURE_2D, tex );
 139         glTexImage2D( GL_TEXTURE_2D, 0, internal_format, width, height, 0, internal_format, GL_UNSIGNED_BYTE, NULL );
 140     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE );
 141     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE );
 142     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
 143     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
 144     glBindTexture( GL_TEXTURE_2D, 0 );
 145
 146         gtex->texture = tex;
 147         gtex->width = width;
 148         gtex->height = height;
 149         gtex->internal_format = internal_format;
 150         gtex->used = 1;
 151         lock();
 152         texture_list.push_back(gtex);
 153         unlock();
 154         return gtex;
 155 }
 156
 157 void GlslManager::release_texture(glsl_texture texture)
 158 {
 159         texture->used = 0;
 160 }
 161
 162 glsl_pbo GlslManager::get_pbo(int size)
 163 {
 164         lock();
 165         if (!pbo) {
 166                 GLuint pb = 0;
 167                 glGenBuffers(1, &pb);
 168                 if (!pb) {
 169                         unlock();
 170                         return NULL;
 171                 }
 172
 173                 pbo = new glsl_pbo_s;
 174                 if (!pbo) {
 175                         glDeleteBuffers(1, &pb);
 176                         unlock();
 177                         return NULL;
 178                 }
 179                 pbo->pbo = pb;
 180                 pbo->size = 0;
 181         }
 182         if (size > pbo->size) {
 183                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, pbo->pbo);
 184                 glBufferData(GL_PIXEL_UNPACK_BUFFER_ARB, size, NULL, GL_STREAM_DRAW);
 185                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, 0);
 186                 pbo->size = size;
 187         }
 188         unlock();
 189         return pbo;
 190 }
 191
 192 void GlslManager::cleanupContext()
 193 {
 194         lock();
 195         while (fbo_list.peek_back()) {
 196                 glsl_fbo fbo = (glsl_fbo) fbo_list.pop_back();
 197                 glDeleteFramebuffers(1, &fbo->fbo);
 198                 delete fbo;
 199         }
 200         while (texture_list.peek_back()) {
 201                 glsl_texture texture = (glsl_texture) texture_list.pop_back();
 202                 glDeleteTextures(1, &texture->texture);
 203                 delete texture;
 204         }
 205         if (pbo) {
 206                 glDeleteBuffers(1, &pbo->pbo);
 207                 delete pbo;
 208                 pbo = 0;
 209         }
 210         unlock();
 211 }
 212
 213 void GlslManager::onInit( mlt_properties owner, GlslManager* filter )
 214 {
 215         mlt_log_debug( filter->get_service(), "%s\n", __FUNCTION__ );
 216 #ifdef WIN32
 217         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("\\share\\movit");
 218 #elif defined(__DARWIN__) && defined(RELOCATABLE)
 219         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("/share/movit");
 220 #else
 221         std::string path = std::string(getenv("MLT_MOVIT_PATH") ? getenv("MLT_MOVIT_PATH") : SHADERDIR);
 222 #endif
 223         ::init_movit( path, mlt_log_get_level() == MLT_LOG_DEBUG? MOVIT_DEBUG_ON : MOVIT_DEBUG_OFF );
 224         filter->set( "glsl_supported", movit_initialized );
 225 }
 226
 227 void GlslManager::onClose( mlt_properties owner, GlslManager *filter )
 228 {
 229         filter->cleanupContext();
 230 }
 231
 232 void GlslManager::onServiceChanged( mlt_properties owner, mlt_service aservice )
 233 {
 234         Mlt::Service service( aservice );
 235         service.lock();
 236         service.set( "movit chain", NULL, 0 );
 237         service.set( "movit input", NULL, 0 );
 238         // Destroy the effect list.
 239         GlslManager::get_instance()->set( service.get( "_unique_id" ), NULL, 0 );
 240         service.unlock();
 241 }
 242
 243 void GlslManager::onPropertyChanged( mlt_properties owner, mlt_service service, const char* property )
 244 {
 245         if ( property && std::string( property ) == "disable" )
 246                 onServiceChanged( owner, service );
 247 }
 248
 249 extern "C" {
 250
 251 mlt_filter filter_glsl_manager_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
 252 {
 253         GlslManager* g = GlslManager::get_instance();
 254         if (g)
 255                 g->inc_ref();
 256         else
 257                 g = new GlslManager();
 258         return g->get_filter();
 259 }
 260
 261 } // extern "C"
 262
 263 Mlt::Properties GlslManager::effect_list( Mlt::Service& service )
 264 {
 265         char *unique_id =  service.get( "_unique_id" );
 266         mlt_properties properties = (mlt_properties) get_data( unique_id );
 267         if ( !properties ) {
 268                 properties = mlt_properties_new();
 269                 set( unique_id, properties, 0, (mlt_destructor) mlt_properties_close );
 270         }
 271         Mlt::Properties p( properties );
 272         return p;
 273 }
 274
 275 static void deleteChain( EffectChain* chain )
 276 {
 277         delete chain;
 278 }
 279
 280 bool GlslManager::init_chain( mlt_service aservice )
 281 {
 282         bool error = true;
 283         Mlt::Service service( aservice );
 284         EffectChain* chain = (EffectChain*) service.get_data( "movit chain" );
 285         if ( !chain ) {
 286                 mlt_profile profile = mlt_service_profile( aservice );
 287                 Input* input = new MltInput( profile->width, profile->height );
 288                 chain = new EffectChain( profile->display_aspect_num, profile->display_aspect_den );
 289                 chain->add_input( input );
 290                 service.set( "movit chain", chain, 0, (mlt_destructor) deleteChain );
 291                 service.set( "movit input", input, 0 );
 292                 service.set( "_movit finalized", 0 );
 293                 service.listen( "service-changed", aservice, (mlt_listener) GlslManager::onServiceChanged );
 294                 service.listen( "property-changed", aservice, (mlt_listener) GlslManager::onPropertyChanged );
 295                 error = false;
 296         }
 297         return error;
 298 }
 299
 300 EffectChain* GlslManager::get_chain( mlt_service service )
 301 {
 302         return (EffectChain*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit chain", NULL );
 303 }
 304
 305 MltInput *GlslManager::get_input( mlt_service service )
 306 {
 307         return (MltInput*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit input", NULL );
 308 }
 309
 310 void GlslManager::reset_finalized( mlt_service service )
 311 {
 312         mlt_properties_set_int( MLT_SERVICE_PROPERTIES(service), "_movit finalized", 0 );
 313 }
 314
 315 Effect* GlslManager::get_effect( mlt_filter filter, mlt_frame frame )
 316 {
 317         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 318         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 319         return (Effect*) GlslManager::get_instance()->effect_list( producer ).get_data( unique_id );
 320 }
 321
 322 Effect* GlslManager::add_effect( mlt_filter filter, mlt_frame frame, Effect* effect )
 323 {
 324         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 325         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 326         chain->add_effect( effect );
 327         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 328         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 329         return effect;
 330 }
 331
 332 Effect* GlslManager::add_effect( mlt_filter filter, mlt_frame frame, Effect* effect, Effect* input_b )
 333 {
 334         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 335         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 336         chain->add_effect( effect, chain->last_added_effect(),
 337                 input_b? input_b : chain->last_added_effect() );
 338         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 339         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 340         return effect;
 341 }
 342
 343 void GlslManager::render( mlt_service service, void* chain, GLuint fbo, int width, int height )
 344 {
 345         EffectChain* effect_chain = (EffectChain*) chain;
 346         mlt_properties properties = MLT_SERVICE_PROPERTIES( service );
 347         if ( !mlt_properties_get_int( properties, "_movit finalized" ) ) {
 348                 mlt_properties_set_int( properties, "_movit finalized", 1 );
 349                 effect_chain->add_effect( new Mlt::VerticalFlip() );
 350                 effect_chain->finalize();
 351         }
 352         effect_chain->render_to_fbo( fbo, width, height );
 353 }
 354
 355 void GlslManager::lock_service( mlt_frame frame )
 356 {
 357         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 358         producer.lock();
 359 }
 360
 361 void GlslManager::unlock_service( mlt_frame frame )
 362 {
 363         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 364         producer.unlock();
 365 }