• Craig Watson's avatar
    Podcast recording on OSX fixed. Details: · 427cdbb6
    Craig Watson authored
    The threading logic was changed somewhat. UBQuickTimeFile's run()
    function no longer handles enqueuing the video/audio samples in a while
    loop. Instead, it runs once, and uses Apple's Dispatch Queues to handle
    enqueuing samples.
    One dispatch queue was thus added for each input to the AssetWriter.
    Each input is associated to one queue, and the requestMediaDataWhenReady
    function insures that the inputs go and fetch any available samples when
    they are able to write them.
    
    As tested (for short podcasts, repeatedly), this solves all the problems
    encountered earlier, such as the program hanging due to one input not
    being ready, or corrupt files due (presumably) to missing samples.
    427cdbb6
UBQuickTimeFile.h 3.91 KB
/*
 * Copyright (C) 2013 Open Education Foundation
 *
 * Copyright (C) 2010-2013 Groupement d'Intérêt Public pour
 * l'Education Numérique en Afrique (GIP ENA)
 *
 * This file is part of OpenBoard.
 *
 * OpenBoard is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3 of the License,
 * with a specific linking exception for the OpenSSL project's
 * "OpenSSL" library (or with modified versions of it that use the
 * same license as the "OpenSSL" library).
 *
 * OpenBoard is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with OpenBoard. If not, see <http://www.gnu.org/licenses/>.
 */




#ifndef UBQUICKTIMEFILE_H_
#define UBQUICKTIMEFILE_H_

#include <QtCore>

#include <CoreVideo/CoreVideo.h>
#include <CoreMedia/CoreMedia.h>

#include "UBAudioQueueRecorder.h"



// Trick to get around the fact that the C++ compiler doesn't
// like Objective C code.

#ifdef __OBJC__ // defined by the Objective C compiler
    @class AVAssetWriter;
    @class AVAssetWriterInput;
    @class AVAssetWriterInputPixelBufferAdaptor;

    typedef AVAssetWriter* AssetWriterPTR;
    typedef AVAssetWriterInput* AssetWriterInputPTR;
    typedef AVAssetWriterInputPixelBufferAdaptor* AssetWriterInputAdaptorPTR;
#else
    typedef void* AssetWriterPTR;
    typedef void* AssetWriterInputPTR;
    typedef void* AssetWriterInputAdaptorPTR;
#endif

class UBQuickTimeFile : public QThread
{
    Q_OBJECT;

    public:
        UBQuickTimeFile(QObject * pParent = 0);
        virtual ~UBQuickTimeFile();

        bool init(const QString& videoFileName, const QString& profileData
                , int pFramesPerSecond, const QSize& pFrameSize
                , bool recordAudio = true, const QString& audioDeviceName = QString("Default"));

        void stop();

        CVPixelBufferRef newPixelBuffer();

        bool isCompressionSessionRunning() { return mCompressionSessionRunning; }

        QString lastErrorMessage() const { return mLastErrorMessage; }

        void endSession();

        struct VideoFrame
        {
            CVPixelBufferRef buffer;
            long timestamp;
        };

        static QQueue<VideoFrame> frameQueue;
        static QMutex frameQueueMutex;
        static QWaitCondition frameBufferNotEmpty;

    signals:
        void audioLevelChanged(quint8 level);
        void compressionSessionStarted();
        void compressionFinished();

    protected:
        void run();

    private slots:
        void enqueueAudioBuffer(void* pBuffer, long pLength);

    private:

        bool beginSession();
        void setLastErrorMessage(const QString& error);

        void appendVideoFrame(CVPixelBufferRef pixelBuffer, long msTimeStamp);
        bool appendSampleBuffer(CMSampleBufferRef sampleBuffer);
        
        QSize mFrameSize;
        QString mVideoFileName;

        AssetWriterPTR mVideoWriter;

        AssetWriterInputPTR mVideoWriterInput;
        AssetWriterInputAdaptorPTR mAdaptor;

        AssetWriterInputPTR mAudioWriterInput;

        QPointer<UBAudioQueueRecorder> mWaveRecorder;
        CFAbsoluteTime mStartTime;

        CMAudioFormatDescriptionRef mAudioFormatDescription;
        
        long mTimeScale;
        bool mRecordAudio;
        
        volatile bool mShouldStopCompression;
        volatile bool mCompressionSessionRunning;

        QString mLastErrorMessage;
        QString mAudioRecordingDeviceName;

        dispatch_queue_t mVideoDispatchQueue;
        dispatch_queue_t mAudioDispatchQueue;

        static QQueue<CMSampleBufferRef> audioQueue;
        static QMutex audioQueueMutex;

        static QMutex audioWriterMutex;
};

#endif /* UBQUICKTIMEFILE_H_ */