]> git.sesse.net Git - casparcg/blobdiff - core/producer/framerate/framerate_producer.cpp
[ffmpeg] Ported 2.0.7 ffmpeg producer to 2.1.0 while still keeping the usage of the...
[casparcg] / core / producer / framerate / framerate_producer.cpp
index 956ca5e49050091285886db5ecb39a504bca7472..d84b1f1d39e4c38482039a1679dae7c7ff8bf68f 100644 (file)
@@ -30,6 +30,7 @@
 #include "../../frame/frame_transform.h"
 #include "../../frame/pixel_format.h"
 #include "../../monitor/monitor.h"
+#include "../../help/help_sink.h"
 
 #include <common/future.h>
 #include <common/tweener.h>
@@ -37,6 +38,7 @@
 #include <functional>
 #include <queue>
 #include <future>
+#include <stack>
 
 namespace caspar { namespace core {
 
@@ -54,7 +56,7 @@ draw_frame blend(const draw_frame& source, const draw_frame& destination, const
 
        auto under                                      = source;
        auto over                                       = destination;
-       double float_distance           = static_cast<double>(distance.numerator()) / static_cast<double>(distance.denominator());
+       double float_distance           = boost::rational_cast<double>(distance);
 
        under.transform().image_transform.is_mix        = true;
        under.transform().image_transform.opacity       = 1 - float_distance;
@@ -99,7 +101,7 @@ struct blend_all
                middle.transform().image_transform.is_mix                       = true;
                next_frame.transform().image_transform.is_mix           = true;
 
-               double float_distance                                                           = static_cast<double>(distance.numerator()) / static_cast<double>(distance.denominator());
+               double float_distance                                                           = boost::rational_cast<double>(distance);
                previous_frame.transform().image_transform.opacity      = std::max(0.0, 0.5 - float_distance * 0.5);
                middle.transform().image_transform.opacity                      = 0.5;
                next_frame.transform().image_transform.opacity          = 1.0 - previous_frame.transform().image_transform.opacity - middle.transform().image_transform.opacity;
@@ -110,20 +112,30 @@ struct blend_all
        }
 };
 
-struct audio_extractor : public frame_visitor
+class audio_extractor : public frame_visitor
 {
-       std::function<void(const const_frame& frame)> on_frame_;
-
+       std::stack<core::audio_transform>                               transform_stack_;
+       std::function<void(const const_frame& frame)>   on_frame_;
+public:
        audio_extractor(std::function<void(const const_frame& frame)> on_frame)
                : on_frame_(std::move(on_frame))
        {
+               transform_stack_.push(audio_transform());
+       }
+
+       void push(const frame_transform& transform) override
+       {
+               transform_stack_.push(transform_stack_.top() * transform.audio_transform);
+       }
+
+       void pop() override
+       {
+               transform_stack_.pop();
        }
 
-       void push(const frame_transform& transform) override    { }
-       void pop() override                                                                             { }
        void visit(const const_frame& frame) override
        {
-               if (!frame.audio_data().empty())
+               if (!frame.audio_data().empty() && !transform_stack_.top().is_still)
                        on_frame_(frame);
        }
 };
@@ -160,9 +172,10 @@ public:
        {
                if (time_ == duration_)
                        return dest_;
-               double source   = static_cast<double>(source_.numerator()) / static_cast<double>(source_.denominator());
-               double delta    = static_cast<double>(dest_.numerator()) / static_cast<double>(dest_.denominator()) - source;
-               double result = tweener_(time_, source, delta, duration_);
+
+               double source   = boost::rational_cast<double>(source_);
+               double delta    = boost::rational_cast<double>(dest_) - source;
+               double result   = tweener_(time_, source, delta, duration_);
                
                return boost::rational<int64_t>(static_cast<int64_t>(result * 1000000.0), 1000000);
        }
@@ -177,8 +190,11 @@ public:
 class framerate_producer : public frame_producer_base
 {
        spl::shared_ptr<frame_producer>                                         source_;
-       boost::rational<int>                                                            source_framerate_;
-       audio_channel_layout                                                            source_channel_layout_          = audio_channel_layout::invalid();
+       std::function<boost::rational<int>()>                           get_source_framerate_;
+       boost::rational<int>                                                            source_framerate_                               = -1;
+       audio_channel_layout                                                            source_channel_layout_                  = audio_channel_layout::invalid();
+       boost::rational<int>                                                            original_destination_framerate_;
+       field_mode                                                                                      original_destination_fieldmode_;
        boost::rational<int>                                                            destination_framerate_;
        field_mode                                                                                      destination_fieldmode_;
        std::vector<int>                                                                        destination_audio_cadence_;
@@ -187,69 +203,28 @@ class framerate_producer : public frame_producer_base
        std::function<draw_frame (
                        const draw_frame& source,
                        const draw_frame& destination,
-                       const boost::rational<int64_t>& distance)>      interpolator_                           = drop_and_skip;
+                       const boost::rational<int64_t>& distance)>      interpolator_                                   = drop_and_skip;
        
-       boost::rational<std::int64_t>                                           current_frame_number_           = 0;
-       draw_frame                                                                                      previous_frame_                         = draw_frame::empty();
-       draw_frame                                                                                      next_frame_                                     = draw_frame::empty();
+       boost::rational<std::int64_t>                                           current_frame_number_                   = 0;
+       draw_frame                                                                                      previous_frame_                                 = draw_frame::empty();
+       draw_frame                                                                                      next_frame_                                             = draw_frame::empty();
        mutable_audio_buffer                                                            audio_samples_;
 
-       unsigned int                                                                            output_repeat_                          = 0;
-       unsigned int                                                                            output_frame_                           = 0;
+       unsigned int                                                                            output_repeat_                                  = 0;
+       unsigned int                                                                            output_frame_                                   = 0;
 public:
        framerate_producer(
                        spl::shared_ptr<frame_producer> source,
-                       boost::rational<int> source_framerate,
+                       std::function<boost::rational<int> ()> get_source_framerate,
                        boost::rational<int> destination_framerate,
                        field_mode destination_fieldmode,
                        std::vector<int> destination_audio_cadence)
                : source_(std::move(source))
-               , source_framerate_(std::move(source_framerate))
-               , destination_framerate_(std::move(destination_framerate))
-               , destination_fieldmode_(destination_fieldmode)
+               , get_source_framerate_(std::move(get_source_framerate))
+               , original_destination_framerate_(std::move(destination_framerate))
+               , original_destination_fieldmode_(destination_fieldmode)
                , destination_audio_cadence_(std::move(destination_audio_cadence))
        {
-               // Coarse adjustment to correct fps family (23.98 - 30 vs 47.95 - 60)
-               if (destination_fieldmode_ != field_mode::progressive)  // Interlaced output
-               {
-                       auto diff_double        = boost::abs(source_framerate_ - destination_framerate_ * 2);
-                       auto diff_keep          = boost::abs(source_framerate_ - destination_framerate_);
-
-                       if (diff_double < diff_keep)                                            // Double rate interlaced
-                       {
-                               destination_framerate_ *= 2;
-                       }
-                       else                                                                                            // Progressive non interlaced
-                       {
-                               destination_fieldmode_ = field_mode::progressive;
-                       }
-               }
-               else                                                                                                    // Progressive
-               {
-                       auto diff_halve = boost::abs(source_framerate_ * 2      - destination_framerate_);
-                       auto diff_keep  = boost::abs(source_framerate_          - destination_framerate_);
-
-                       if (diff_halve < diff_keep)                                                     // Repeat every frame two times
-                       {
-                               destination_framerate_  /= 2;
-                               output_repeat_                  = 2;
-                       }
-               }
-
-               speed_ = boost::rational<int64_t>(source_framerate_ / destination_framerate_);
-
-               // drop_and_skip will only be used by default for exact framerate multiples (half, same and double)
-               // for all other framerates a frame interpolator will be chosen.
-               if (speed_ != 1 && speed_ * 2 != 1 && speed_ != 2)
-               {
-                       if (source_framerate_ > 47)             // The bluriness of blend_all is acceptable on high framerates.
-                               interpolator_ = blend_all();
-                       else                                                    // blend_all is mostly too blurry on low framerates. blend provides a compromise.
-                               interpolator_ = &blend;
-
-                       CASPAR_LOG(warning) << source_->print() << L" Frame blending frame rate conversion required to conform to channel frame rate.";
-               }
-
                // Note: Uses 1 step rotated cadence for 1001 modes (1602, 1602, 1601, 1602, 1601)
                // This cadence fills the audio mixer most optimally.
                boost::range::rotate(destination_audio_cadence_, std::end(destination_audio_cadence_) - 1);
@@ -319,7 +294,33 @@ public:
 
        boost::property_tree::wptree info() const override
        {
-               return source_->info();
+               auto info = source_->info();
+
+               auto incorrect_frame_number = info.get_child_optional(L"frame-number");
+               if (incorrect_frame_number)
+                       incorrect_frame_number->put_value(frame_number());
+
+               auto incorrect_nb_frames = info.get_child_optional(L"nb-frames");
+               if (incorrect_nb_frames)
+                       incorrect_nb_frames->put_value(nb_frames());
+
+               return info;
+       }
+
+       uint32_t nb_frames() const override
+       {
+               auto source_nb_frames = source_->nb_frames();
+               auto multiple = boost::rational_cast<double>(1 / get_speed() * (output_repeat_ != 0 ? 2 : 1));
+
+               return static_cast<uint32_t>(source_nb_frames * multiple);
+       }
+
+       uint32_t frame_number() const override
+       {
+               auto source_frame_number = source_->frame_number() - 1; // next frame already received
+               auto multiple = boost::rational_cast<double>(1 / get_speed() * (output_repeat_ != 0 ? 2 : 1));
+
+               return static_cast<uint32_t>(source_frame_number * multiple);
        }
 
        constraints& pixel_constraints() override
@@ -331,7 +332,7 @@ private:
        {
                user_speed_.fetch_and_tick();
 
-               if (output_repeat_ && ++output_frame_ % output_repeat_)
+               if (output_repeat_ && output_frame_++ % output_repeat_)
                {
                        auto frame = draw_frame::still(last_frame());
 
@@ -371,9 +372,14 @@ private:
 
                for (std::int64_t i = 0; i < num_frames; ++i)
                {
-                       previous_frame_ = std::move(next_frame_);
+                       if (next_frame_ == draw_frame::empty())
+                               previous_frame_ = pop_frame_from_source();
+                       else
+                       {
+                               previous_frame_ = std::move(next_frame_);
 
-                       next_frame_ = pop_frame_from_source();
+                               next_frame_ = pop_frame_from_source();
+                       }
                }
        }
 
@@ -385,6 +391,7 @@ private:
        draw_frame pop_frame_from_source()
        {
                auto frame = source_->receive();
+               update_source_framerate();
 
                if (user_speed_.fetch() == 1)
                {
@@ -441,7 +448,8 @@ private:
                {
                        auto needed = destination_audio_cadence_.front();
                        auto got = audio_samples_.size() / source_channel_layout_.num_channels;
-                       CASPAR_LOG(debug) << print() << L" Too few audio samples. Needed " << needed << L" but got " << got;
+                       if (got != 0) // If at end of stream we don't care
+                               CASPAR_LOG(debug) << print() << L" Too few audio samples. Needed " << needed << L" but got " << got;
                        buffer.swap(audio_samples_);
                        buffer.resize(needed * source_channel_layout_.num_channels, 0);
                }
@@ -463,18 +471,87 @@ private:
                                || user_speed_.fetch() != 1
                                || audio_samples_.size() / source_channel_layout_.num_channels >= destination_audio_cadence_.at(0);
        }
+
+       void update_source_framerate()
+       {
+               auto source_framerate = get_source_framerate_();
+
+               if (source_framerate_ == source_framerate)
+                       return;
+
+               source_framerate_               = source_framerate;
+               destination_framerate_  = original_destination_framerate_;
+               destination_fieldmode_  = original_destination_fieldmode_;
+
+               // Coarse adjustment to correct fps family (23.98 - 30 vs 47.95 - 60)
+               if (destination_fieldmode_ != field_mode::progressive)  // Interlaced output
+               {
+                       auto diff_double        = boost::abs(source_framerate_ - destination_framerate_ * 2);
+                       auto diff_keep          = boost::abs(source_framerate_ - destination_framerate_);
+
+                       if (diff_double < diff_keep)                                            // Double rate interlaced
+                       {
+                               destination_framerate_ *= 2;
+                       }
+                       else                                                                                            // Progressive non interlaced
+                       {
+                               destination_fieldmode_ = field_mode::progressive;
+                       }
+               }
+               else                                                                                                    // Progressive
+               {
+                       auto diff_halve = boost::abs(source_framerate_ * 2      - destination_framerate_);
+                       auto diff_keep  = boost::abs(source_framerate_          - destination_framerate_);
+
+                       if (diff_halve < diff_keep)                                                     // Repeat every frame two times
+                       {
+                               destination_framerate_  /= 2;
+                               output_repeat_                  = 2;
+                       }
+               }
+
+               speed_ = boost::rational<int64_t>(source_framerate_ / destination_framerate_);
+
+               // drop_and_skip will only be used by default for exact framerate multiples (half, same and double)
+               // for all other framerates a frame interpolator will be chosen.
+               if (speed_ != 1 && speed_ * 2 != 1 && speed_ != 2)
+               {
+                       auto high_source_framerate              = source_framerate_ > 47;
+                       auto high_destination_framerate = destination_framerate_ > 47
+                                       || destination_fieldmode_ != field_mode::progressive;
+
+                       if (high_source_framerate && high_destination_framerate)        // The bluriness of blend_all is acceptable on high framerates.
+                               interpolator_ = blend_all();
+                       else                                                                                                            // blend_all is mostly too blurry on low framerates. blend provides a compromise.
+                               interpolator_ = &blend;
+
+                       CASPAR_LOG(warning) << source_->print() << L" Frame blending frame rate conversion required to conform to channel frame rate.";
+               }
+       }
 };
 
+void describe_framerate_producer(help_sink& sink)
+{
+       sink.para()->text(L"Framerate conversion control / Slow motion examples:");
+       sink.example(L">> CALL 1-10 FRAMERATE INTERPOLATION BLEND", L"enables 2 frame blend interpolation.");
+       sink.example(L">> CALL 1-10 FRAMERATE INTERPOLATION BLEND_ALL", L"enables 3 frame blend interpolation.");
+       sink.example(L">> CALL 1-10 FRAMERATE INTERPOLATION DROP_AND_SKIP", L"disables frame interpolation.");
+       sink.example(L">> CALL 1-10 FRAMERATE SPEED 0.25", L"immediately changes the speed to 25%. Sound will be disabled.");
+       sink.example(L">> CALL 1-10 FRAMERATE SPEED 0.25 50", L"changes the speed to 25% linearly over 50 frames. Sound will be disabled.");
+       sink.example(L">> CALL 1-10 FRAMERATE SPEED 0.25 50 easeinoutsine", L"changes the speed to 25% over 50 frames using specified easing curve. Sound will be disabled.");
+       sink.example(L">> CALL 1-10 FRAMERATE SPEED 1 50", L"changes the speed to 100% linearly over 50 frames. Sound will be enabled when the destination speed of 100% has been reached.");
+}
+
 spl::shared_ptr<frame_producer> create_framerate_producer(
                spl::shared_ptr<frame_producer> source,
-               boost::rational<int> source_framerate,
+               std::function<boost::rational<int> ()> get_source_framerate,
                boost::rational<int> destination_framerate,
                field_mode destination_fieldmode,
                std::vector<int> destination_audio_cadence)
 {
        return spl::make_shared<framerate_producer>(
                        std::move(source),
-                       std::move(source_framerate),
+                       std::move(get_source_framerate),
                        std::move(destination_framerate),
                        destination_fieldmode,
                        std::move(destination_audio_cadence));