AviRecorder.cc

Go to the documentation of this file.

#include "AviRecorder.hh"
#include "AviWriter.hh"
#include "WavWriter.hh"
#include "Reactor.hh"
#include "MSXMotherBoard.hh"
#include "FileContext.hh"
#include "CommandException.hh"
#include "Display.hh"
#include "PostProcessor.hh"
#include "MSXMixer.hh"
#include "Filename.hh"
#include "CliComm.hh"
#include "FileOperations.hh"
#include "TclObject.hh"
#include "memory.hh"
#include "outer.hh"
#include "vla.hh"
#include <cassert>

using std::string;
using std::vector;

namespace openmsx {

AviRecorder::AviRecorder(Reactor& reactor_)
        : reactor(reactor_)
        , recordCommand(reactor.getCommandController())
        , mixer(nullptr)
        , duration(EmuDuration::infinity)
        , prevTime(EmuTime::infinity)
        , frameHeight(0)
{
}

AviRecorder::~AviRecorder()
{
        assert(!aviWriter);
        assert(!wavWriter);
}

void AviRecorder::start(bool recordAudio, bool recordVideo, bool recordMono,
                        bool recordStereo, const Filename& filename)
{
        stop();
        MSXMotherBoard* motherBoard = reactor.getMotherBoard();
        if (!motherBoard) {
                throw CommandException("No active MSX machine.");
        }
        if (recordAudio) {
                mixer = &motherBoard->getMSXMixer();
                warnedStereo = false;
                if (recordStereo) {
                        stereo = true;
                } else if (recordMono) {
                        stereo = false;
                        warnedStereo = true; // no warning if data is actually stereo
                } else {
                        stereo = mixer->needStereoRecording();
                }
                sampleRate = mixer->getSampleRate();
                warnedSampleRate = false;
        }
        if (recordVideo) {
                // Set V99x8, V9990, Laserdisc, ... in record mode (when
                // present). Only the active one will actually send frames to
                // the video. This also works for Video9000.
                postProcessors.clear();
                for (auto* l : reactor.getDisplay().getAllLayers()) {
                        if (auto* pp = dynamic_cast<PostProcessor*>(l)) {
                                postProcessors.push_back(pp);
                        }
                }
                if (postProcessors.empty()) {
                        throw CommandException(
                                "Current renderer doesn't support video recording.");
                }
                // any source is fine because they all have the same bpp
                unsigned bpp = postProcessors.front()->getBpp();
                warnedFps = false;
                duration = EmuDuration::infinity;
                prevTime = EmuTime::infinity;

                try {
                        aviWriter = make_unique<AviWriter>(
                                filename, frameWidth, frameHeight, bpp,
                                (recordAudio && stereo) ? 2 : 1, sampleRate);
                } catch (MSXException& e) {
                        throw CommandException("Can't start recording: " +
                                               e.getMessage());
                }
        } else {
                assert(recordAudio);
                wavWriter = make_unique<Wav16Writer>(
                        filename, stereo ? 2 : 1, sampleRate);
        }
        // only set recorders when all errors are checked for
        for (auto* pp : postProcessors) {
                pp->setRecorder(this);
        }
        if (mixer) mixer->setRecorder(this);
}

void AviRecorder::stop()
{
        for (auto* pp : postProcessors) {
                pp->setRecorder(nullptr);
        }
        postProcessors.clear();
        if (mixer) {
                mixer->setRecorder(nullptr);
                mixer = nullptr;
        }
        sampleRate = 0;
        aviWriter.reset();
        wavWriter.reset();
}

void AviRecorder::addWave(unsigned num, int16_t* data)
{
        if (!warnedSampleRate && (mixer->getSampleRate() != sampleRate)) {
                warnedSampleRate = true;
                reactor.getCliComm().printWarning(
                        "Detected audio sample frequency change during "
                        "avi recording. Audio/video might get out of sync "
                        "because of this.");
        }
        if (stereo) {
                if (wavWriter) {
                        wavWriter->write(data, 2, num);
                } else {
                        assert(aviWriter);
                        audioBuf.insert(end(audioBuf), data, data + 2 * num);
                }
        } else {
                VLA(int16_t, buf, num);
                unsigned i = 0;
                for (; !warnedStereo && i < num; ++i) {
                        if (data[2 * i + 0] != data[2 * i + 1]) {
                                reactor.getCliComm().printWarning(
                                    "Detected stereo sound during mono recording. "
                                    "Channels will be mixed down to mono. To "
                                    "avoid this warning you can explicity pass the "
                                    "-mono or -stereo flag to the record command.");
                                warnedStereo = true;
                                break;
                        }
                        buf[i] = data[2 * i];
                }
                for (; i < num; ++i) {
                        buf[i] = (int(data[2 * i + 0]) + int(data[2 * i + 1])) / 2;
                }

                if (wavWriter) {
                        wavWriter->write(buf, 1, num);
                } else {
                        assert(aviWriter);
                        audioBuf.insert(end(audioBuf), buf, buf + num);
                }
        }
}

void AviRecorder::addImage(FrameSource* frame, EmuTime::param time)
{
        assert(!wavWriter);
        if (duration != EmuDuration::infinity) {
                if (!warnedFps && ((time - prevTime) != duration)) {
                        warnedFps = true;
                        reactor.getCliComm().printWarning(
                                "Detected frame rate change (PAL/NTSC or frameskip) "
                                "during avi recording. Audio/video might get out of "
                                "sync because of this.");
                }
        } else if (prevTime != EmuTime::infinity) {
                duration = time - prevTime;
                aviWriter->setFps(1.0 / duration.toDouble());
        }
        prevTime = time;

        if (mixer) {
                mixer->updateStream(time);
        }
        aviWriter->addFrame(frame, unsigned(audioBuf.size()), audioBuf.data());
        audioBuf.clear();
}

// TODO: Can this be dropped?
unsigned AviRecorder::getFrameHeight() const {
        assert (frameHeight != 0); // someone uses the getter too early?
        return frameHeight;
}

void AviRecorder::processStart(array_ref<TclObject> tokens, TclObject& result)
{
        string filename;
        string prefix = "openmsx";
        bool recordAudio = true;
        bool recordVideo = true;
        bool recordMono = false;
        bool recordStereo = false;
        frameWidth = 320;
        frameHeight = 240;

        vector<string> arguments;
        for (unsigned i = 2; i < tokens.size(); ++i) {
                string_ref token = tokens[i].getString();
                if (token.starts_with('-')) {
                        if (token == "--") {
                                for (auto it = std::begin(tokens) + i + 1;
                                     it != std::end(tokens); ++it) {
                                        arguments.push_back(it->getString().str());
                                }
                                break;
                        }
                        if (token == "-prefix") {
                                if (++i == tokens.size()) {
                                        throw CommandException("Missing argument");
                                }
                                prefix = tokens[i].getString().str();
                        } else if (token == "-audioonly") {
                                recordVideo = false;
                        } else if (token == "-mono") {
                                recordMono = true;
                        } else if (token == "-stereo") {
                                recordStereo = true;
                        } else if (token == "-videoonly") {
                                recordAudio = false;
                        } else if (token == "-doublesize") {
                                frameWidth = 640;
                                frameHeight = 480;
                        } else if (token == "-triplesize") {
                                frameWidth = 960;
                                frameHeight = 720;
                        } else {
                                throw CommandException("Invalid option: " + token);
                        }
                } else {
                        arguments.push_back(token.str());
                }
        }
        if (!recordAudio && !recordVideo) {
                throw CommandException("Can't have both -videoonly and -audioonly.");
        }
        if (recordStereo && recordMono) {
                throw CommandException("Can't have both -mono and -stereo.");
        }
        if (!recordAudio && (recordStereo || recordMono)) {
                throw CommandException("Can't have both -videoonly and -stereo or -mono.");
        }
        switch (arguments.size()) {
        case 0:
                // nothing
                break;
        case 1:
                filename = arguments[0];
                break;
        default:
                throw SyntaxError();
        }

        string directory = recordVideo ? "videos" : "soundlogs";
        string extension = recordVideo ? ".avi"   : ".wav";
        filename = FileOperations::parseCommandFileArgument(
                filename, directory, prefix, extension);

        if (aviWriter || wavWriter) {
                result.setString("Already recording.");
        } else {
                start(recordAudio, recordVideo, recordMono, recordStereo,
                                Filename(filename));
                result.setString("Recording to " + filename);
        }
}

void AviRecorder::processStop(array_ref<TclObject> tokens)
{
        if (tokens.size() != 2) {
                throw SyntaxError();
        }
        stop();
}

void AviRecorder::processToggle(array_ref<TclObject> tokens, TclObject& result)
{
        if (aviWriter || wavWriter) {
                // drop extra tokens
                processStop(make_array_ref(tokens.data(), 2));
        } else {
                processStart(tokens, result);
        }
}

void AviRecorder::status(array_ref<TclObject> tokens, TclObject& result) const
{
        if (tokens.size() != 2) {
                throw SyntaxError();
        }
        result.addListElement("status");
        if (aviWriter || wavWriter) {
                result.addListElement("recording");
        } else {
                result.addListElement("idle");
        }

}

// class AviRecorder::Cmd

AviRecorder::Cmd::Cmd(CommandController& commandController_)
        : Command(commandController_, "record")
{
}

void AviRecorder::Cmd::execute(array_ref<TclObject> tokens, TclObject& result)
{
        if (tokens.size() < 2) {
                throw CommandException("Missing argument");
        }
        auto& recorder = OUTER(AviRecorder, recordCommand);
        const string_ref subcommand = tokens[1].getString();
        if (subcommand == "start") {
                recorder.processStart(tokens, result);
        } else if (subcommand == "stop") {
                recorder.processStop(tokens);
        } else if (subcommand == "toggle") {
                recorder.processToggle(tokens, result);
        } else if (subcommand == "status") {
                recorder.status(tokens, result);
        } else {
                throw SyntaxError();
        }
}

string AviRecorder::Cmd::help(const vector<string>& /*tokens*/) const
{
        return "Controls video recording: Write openMSX audio/video to a .avi file.\n"
               "record start              Record to file 'openmsxNNNN.avi'\n"
               "record start <filename>   Record to given file\n"
               "record start -prefix foo  Record to file 'fooNNNN.avi'\n"
               "record stop               Stop recording\n"
               "record toggle             Toggle recording (useful as keybinding)\n"
               "record status             Query recording state\n"
               "\n"
               "The start subcommand also accepts an optional -audioonly, -videoonly, "
               " -mono, -stereo, -doublesize flag.\n"
               "Videos are recorded in a 320x240 size by default, at 640x480 when the "
               "-doublesize flag is used and at 960x720 when the -triplesize flag is used.";
}

void AviRecorder::Cmd::tabCompletion(vector<string>& tokens) const
{
        if (tokens.size() == 2) {
                static const char* const cmds[] = {
                        "start", "stop", "toggle", "status",
                };
                completeString(tokens, cmds);
        } else if ((tokens.size() >= 3) && (tokens[1] == "start")) {
                static const char* const options[] = {
                        "-prefix", "-videoonly", "-audioonly", "-doublesize", "-triplesize",
                        "-mono", "-stereo",
                };
                completeFileName(tokens, userFileContext(), options);
        }
}

} // namespace openmsx