Files
vimix/Session.cpp
brunoherbelin 56dfbc737d Deep redesign of frame grabbers to prevent multiple frame captures
(slow) and optimize distribution of gabbed frames to multiple recording
and streaming pipelines (e.g. record H264 *and* share stream on the
network). New implementation of Loopback source for LINUX using
v4l2loopback (experimental).
2020-11-09 23:56:41 +01:00

386 lines
8.8 KiB
C++

#include <algorithm>
#include "defines.h"
#include "Settings.h"
#include "FrameBuffer.h"
#include "Session.h"
#include "GarbageVisitor.h"
#include "FrameGrabber.h"
#include "SessionCreator.h"
#include "Log.h"
Session::Session() : failedSource_(nullptr), active_(true), fading_target_(0.f)
{
filename_ = "";
config_[View::RENDERING] = new Group;
config_[View::RENDERING]->scale_ = FrameBuffer::getResolutionFromParameters(Settings::application.render.ratio, Settings::application.render.res);
config_[View::GEOMETRY] = new Group;
config_[View::GEOMETRY]->scale_ = Settings::application.views[View::GEOMETRY].default_scale;
config_[View::GEOMETRY]->translation_ = Settings::application.views[View::GEOMETRY].default_translation;
config_[View::LAYER] = new Group;
config_[View::LAYER]->scale_ = Settings::application.views[View::LAYER].default_scale;
config_[View::LAYER]->translation_ = Settings::application.views[View::LAYER].default_translation;
config_[View::MIXING] = new Group;
config_[View::MIXING]->scale_ = Settings::application.views[View::MIXING].default_scale;
config_[View::MIXING]->translation_ = Settings::application.views[View::MIXING].default_translation;
}
Session::~Session()
{
// delete all sources
for(auto it = sources_.begin(); it != sources_.end(); ) {
// erase this source from the list
it = deleteSource(*it);
}
delete config_[View::RENDERING];
delete config_[View::GEOMETRY];
delete config_[View::LAYER];
delete config_[View::MIXING];
}
void Session::setActive (bool on)
{
if (active_ != on) {
active_ = on;
for(auto it = sources_.begin(); it != sources_.end(); it++) {
(*it)->setActive(active_);
}
}
}
// update all sources
void Session::update(float dt)
{
failedSource_ = nullptr;
// pre-render of all sources
for( SourceList::iterator it = sources_.begin(); it != sources_.end(); it++){
if ( (*it)->failed() ) {
failedSource_ = (*it);
}
else {
// render the source
(*it)->render();
// update the source
(*it)->update(dt);
}
}
// apply fading (smooth dicotomic reaching)
float f = render_.fading();
if ( ABS_DIFF(f, fading_target_) > EPSILON) {
render_.setFading( f + ( fading_target_ - f ) / 2.f);
}
// update the scene tree
render_.update(dt);
// draw render view in Frame Buffer
render_.draw();
// grab frames to recorders & streamers
FrameGrabbing::manager().grabFrame(render_.frame(), dt);
// // send frame to recorders
// std::list<FrameGrabber *>::iterator iter = iter=grabbers_.begin();
// // if there is at least once frame grabber
// if (iter != grabbers_.end()) {
// // grab a frame (once for all recorders)
// FrameGrabber::Buffer buf = FrameGrabber::grabFrame(render_.frame());
// // give the frame to all recorders
// while (iter != grabbers_.end())
// {
// FrameGrabber *rec = *iter;
// rec->addFrame(buf, dt);
// if (rec->finished()) {
// iter = grabbers_.erase(iter);
// delete rec;
// }
// else {
// iter++;
// }
// }
// gst_buffer_unref(buf.buffer);
// // gst_clear_buffer(&buf.buffer);
// }
}
SourceList::iterator Session::addSource(Source *s)
{
// lock before change
access_.lock();
// find the source
SourceList::iterator its = find(s);
// ok, its NOT in the list !
if (its == sources_.end()) {
// insert the source in the rendering
render_.scene.ws()->attach(s->group(View::RENDERING));
// insert the source to the beginning of the list
sources_.push_front(s);
}
// unlock access
access_.unlock();
// return the iterator to the source created at the beginning
return sources_.begin();
}
SourceList::iterator Session::deleteSource(Source *s)
{
// lock before change
access_.lock();
// find the source
SourceList::iterator its = find(s);
// ok, its in the list !
if (its != sources_.end()) {
// remove Node from the rendering scene
render_.scene.ws()->detatch( s->group(View::RENDERING) );
// erase the source from the update list & get next element
its = sources_.erase(its);
// delete the source : safe now
delete s;
}
// unlock access
access_.unlock();
// return end of next element
return its;
}
void Session::removeSource(Source *s)
{
// lock before change
access_.lock();
// find the source
SourceList::iterator its = find(s);
// ok, its in the list !
if (its != sources_.end()) {
// remove Node from the rendering scene
render_.scene.ws()->detatch( s->group(View::RENDERING) );
// erase the source from the update list & get next element
sources_.erase(its);
}
// unlock access
access_.unlock();
}
Source *Session::popSource()
{
Source *s = nullptr;
SourceList::iterator its = sources_.begin();
if (its != sources_.end())
{
s = *its;
// remove Node from the rendering scene
render_.scene.ws()->detatch( s->group(View::RENDERING) );
// erase the source from the update list & get next element
sources_.erase(its);
}
return s;
}
void Session::setResolution(glm::vec3 resolution)
{
render_.setResolution(resolution);
config_[View::RENDERING]->scale_ = resolution;
}
void Session::setFading(float f, bool forcenow)
{
if (forcenow)
render_.setFading( f );
fading_target_ = CLAMP(f, 0.f, 1.f);
}
SourceList::iterator Session::begin()
{
return sources_.begin();
}
SourceList::iterator Session::end()
{
return sources_.end();
}
SourceList::iterator Session::find(Source *s)
{
return std::find(sources_.begin(), sources_.end(), s);
}
SourceList::iterator Session::find(uint64_t id)
{
return std::find_if(sources_.begin(), sources_.end(), Source::hasId(id));
}
SourceList::iterator Session::find(std::string namesource)
{
return std::find_if(sources_.begin(), sources_.end(), Source::hasName(namesource));
}
SourceList::iterator Session::find(Node *node)
{
return std::find_if(sources_.begin(), sources_.end(), Source::hasNode(node));
}
uint Session::numSource() const
{
return sources_.size();
}
std::list<uint64_t> Session::getIdList() const
{
std::list<uint64_t> idlist;
for( auto sit = sources_.begin(); sit != sources_.end(); sit++)
idlist.push_back( (*sit)->id() );
// make sure no duplicate
idlist.unique();
return idlist;
}
bool Session::empty() const
{
return sources_.empty();
}
SourceList::iterator Session::at(int index)
{
if (index<0)
return sources_.end();
int i = 0;
SourceList::iterator it = sources_.begin();
while ( i < index && it != sources_.end() ){
i++;
it++;
}
return it;
}
int Session::index(SourceList::iterator it) const
{
int index = -1;
int count = 0;
for(auto i = sources_.begin(); i != sources_.end(); i++, count++) {
if ( i == it ) {
index = count;
break;
}
}
return index;
}
//void Session::addFrameGrabber(FrameGrabber *rec)
//{
// if (rec != nullptr)
// grabbers_.push_back(rec);
//}
//FrameGrabber *Session::frontFrameGrabber()
//{
// if (grabbers_.empty())
// return nullptr;
// else
// return grabbers_.front();
//}
//FrameGrabber *Session::getFrameGrabber(uint64_t id)
//{
// if (id > 0 && grabbers_.size() > 0 )
// {
// std::list<FrameGrabber *>::iterator iter = std::find_if(grabbers_.begin(), grabbers_.end(), FrameGrabber::hasId(id));
// if (iter != grabbers_.end())
// return (*iter);
// }
// return nullptr;
//}
//void Session::stopAllFrameGrabbers()
//{
// std::list<FrameGrabber *>::iterator iter;
// for (iter=grabbers_.begin(); iter != grabbers_.end(); )
// (*iter)->stop();
//}
//void Session::clearAllFrameGrabbers()
//{
// std::list<FrameGrabber *>::iterator iter;
// for (iter=grabbers_.begin(); iter != grabbers_.end(); )
// {
// FrameGrabber *rec = *iter;
// rec->stop();
// iter = grabbers_.erase(iter);
// delete rec;
// }
//}
//void Session::transferFrameGrabber(Session *dest)
//{
// if (dest == nullptr)
// return;
// std::list<FrameGrabber *>::iterator iter;
// for (iter=grabbers_.begin(); iter != grabbers_.end(); )
// {
// dest->grabbers_.push_back(*iter);
// iter = grabbers_.erase(iter);
// }
//}
void Session::lock()
{
access_.lock();
}
void Session::unlock()
{
access_.unlock();
}
Session *Session::load(const std::string& filename)
{
SessionCreator creator;
creator.load(filename);
return creator.session();
}