testpatternnode.h

Go to the documentation of this file.

 
#pragma once
 
#include <promeki/core/namespace.h>
#include <promeki/core/timecodegenerator.h>
#include <promeki/proav/medianode.h>
#include <promeki/proav/videodesc.h>
#include <promeki/proav/imagedesc.h>
#include <promeki/proav/audiodesc.h>
#include <promeki/proav/audiogen.h>
#include <promeki/proav/ltcencoder.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class TestPatternNode : public MediaNode {
        PROMEKI_OBJECT(TestPatternNode, MediaNode)
        public:
                enum Pattern {
                        ColorBars,      
                        ColorBars75,    
                        Ramp,           
                        Grid,           
                        Crosshatch,     
                        Checkerboard,   
                        SolidColor,     
                        White,          
                        Black,          
                        Noise,          
                        ZonePlate       
                };
 
                enum AudioMode {
                        Tone,           
                        Silence,        
                        LTC             
                };
 
                TestPatternNode(ObjectBase *parent = nullptr);
 
                virtual ~TestPatternNode();
 
                // ---- Video configuration ----
 
                void setPattern(Pattern p) { _pattern = p; return; }
 
                Pattern pattern() const { return _pattern; }
 
                void setVideoDesc(const VideoDesc &desc) { _videoDesc = desc; return; }
 
                const VideoDesc &videoDesc() const { return _videoDesc; }
 
                void setSolidColor(uint16_t r, uint16_t g, uint16_t b) {
                        _solidR = r; _solidG = g; _solidB = b;
                        return;
                }
 
                // ---- Motion ----
 
                void setMotion(double speed) { _motion = speed; return; }
 
                double motion() const { return _motion; }
 
                // ---- Timecode ----
 
                TimecodeGenerator &timecodeGenerator() { return _tcGen; }
 
                const TimecodeGenerator &timecodeGenerator() const { return _tcGen; }
 
                void setStartTimecode(const Timecode &tc) { _tcGen.setTimecode(tc); return; }
 
                void setDropFrame(bool df) { _tcGen.setDropFrame(df); return; }
 
                Timecode currentTimecode() const { return _tcGen.timecode(); }
 
                uint64_t frameCount() const { return _frameCount; }
 
                // ---- Audio ----
 
                void setAudioDesc(const AudioDesc &desc) { _audioDesc = desc; return; }
 
                const AudioDesc &audioDesc() const { return _audioDesc; }
 
                void setAudioEnabled(bool enable) { _audioEnabled = enable; return; }
 
                bool audioEnabled() const { return _audioEnabled; }
 
                void setAudioMode(AudioMode mode) { _audioMode = mode; return; }
 
                AudioMode audioMode() const { return _audioMode; }
 
                void setChannelConfig(size_t chan, AudioGen::Config config);
 
                void setToneFrequency(double hz) { _toneFreq = hz; return; }
 
                void setToneAmplitude(double amplitude) { _toneAmplitude = amplitude; return; }
 
                void setLtcLevel(float level) { _ltcLevel = level; return; }
 
                void setLtcChannel(int chan) { _ltcChannel = chan; return; }
 
                // ---- Lifecycle overrides ----
 
                Error configure() override;
                Error start() override;
                void process() override;
                void stop() override;
 
                // ---- Extended stats ----
 
                Map<String, Variant> extendedStats() const override;
 
        private:
                // Video config
                Pattern                 _pattern = ColorBars;
                VideoDesc               _videoDesc;
                uint16_t                _solidR = 0;
                uint16_t                _solidG = 0;
                uint16_t                _solidB = 0;
                double                  _motion = 0.0;
                double                  _motionOffset = 0.0;
 
                // Timecode
                TimecodeGenerator       _tcGen;
                uint64_t                _frameCount = 0;
 
                // Audio config
                AudioDesc               _audioDesc;
                bool                    _audioEnabled = true;
                AudioMode               _audioMode = Tone;
                double                  _toneFreq = 1000.0;
                double                  _toneAmplitude = 0.5;
                float                   _ltcLevel = 0.5f;
                int                     _ltcChannel = 0;
                List<AudioGen::Config>  _channelConfigs;
 
                // Runtime state
                AudioGen                *_audioGen = nullptr;
                LtcEncoder              *_ltcEncoder = nullptr;
                ImageDesc               _imageDesc;
                size_t                  _samplesPerFrame = 0;
 
                // Pattern rendering
                void renderPattern(Image &img, double motionOffset);
                void renderColorBars(Image &img, double offset, bool full);
                void renderRamp(Image &img, double offset);
                void renderGrid(Image &img, double offset);
                void renderCrosshatch(Image &img, double offset);
                void renderCheckerboard(Image &img, double offset);
                void renderZonePlate(Image &img, double phase);
                void renderNoise(Image &img);
                void renderSolid(Image &img, uint16_t r, uint16_t g, uint16_t b);
};
 
PROMEKI_NAMESPACE_END