a64.c 6.32 KB
Newer Older
1 2 3 4
/*
 * a64 muxer
 * Copyright (c) 2009 Tobias Bindhammer
 *
5
 * This file is part of Libav.
6
 *
7
 * Libav is free software; you can redistribute it and/or
8 9 10 11
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
12
 * Libav is distributed in the hope that it will be useful,
13 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with Libav; if not, write to the Free Software
19 20 21 22 23 24 25 26
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "libavcodec/avcodec.h"
#include "libavcodec/a64enc.h"
#include "libavcodec/bytestream.h"
#include "avformat.h"

27 28 29 30 31 32
typedef struct A64MuxerContext {
    int interleaved;
    AVPacket prev_pkt;
    int prev_frame_count;
} A64MuxerContext;

33 34
static int a64_write_header(struct AVFormatContext *s)
{
Tobias Bindhammer's avatar
Tobias Bindhammer committed
35
    AVCodecContext *avctx = s->streams[0]->codec;
36
    A64MuxerContext *c = s->priv_data;
37 38 39 40 41 42 43
    uint8_t header[5] = {
        0x00, //load
        0x40, //address
        0x00, //mode
        0x00, //charset_lifetime (multi only)
        0x00  //fps in 50/fps;
    };
44
    c->interleaved = 0;
45 46 47
    switch (avctx->codec->id) {
    case CODEC_ID_A64_MULTI:
        header[2] = 0x00;
48
        header[3] = AV_RB32(avctx->extradata+0);
49 50 51 52
        header[4] = 2;
        break;
    case CODEC_ID_A64_MULTI5:
        header[2] = 0x01;
53
        header[3] = AV_RB32(avctx->extradata+0);
54 55 56
        header[4] = 3;
        break;
    default:
57
        return AVERROR(EINVAL);
58 59
        break;
    }
60
    avio_write(s->pb, header, 2);
61 62
    c->prev_pkt.size = 0;
    c->prev_frame_count = 0;
63 64 65 66 67
    return 0;
}

static int a64_write_packet(struct AVFormatContext *s, AVPacket *pkt)
{
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
    AVCodecContext *avctx = s->streams[0]->codec;
    A64MuxerContext *c = s->priv_data;
    int i, j;
    int ch_chunksize;
    int lifetime;
    int frame_count;
    int charset_size;
    int frame_size;
    int num_frames;

    /* fetch values from extradata */
    switch (avctx->codec->id) {
    case CODEC_ID_A64_MULTI:
    case CODEC_ID_A64_MULTI5:
        if(c->interleaved) {
            /* Write interleaved, means we insert chunks of the future charset before each current frame.
             * Reason: if we load 1 charset + corresponding frames in one block on c64, we need to store
             * them first and then display frame by frame to keep in sync. Thus we would read and write
             * the data for colram from/to ram first and waste too much time. If we interleave and send the
             * charset beforehand, we assemble a new charset chunk by chunk, write current screen data to
             * screen-ram to be displayed and decode the colram directly to colram-location $d800 during
Tobias Bindhammer's avatar
Tobias Bindhammer committed
89
             * the overscan, while reading directly from source.
90 91 92 93 94 95 96 97 98
             * This is the only way so far, to achieve 25fps on c64 */
            if(avctx->extradata) {
                /* fetch values from extradata */
                lifetime     = AV_RB32(avctx->extradata + 0);
                frame_count  = AV_RB32(avctx->extradata + 4);
                charset_size = AV_RB32(avctx->extradata + 8);
                frame_size   = AV_RB32(avctx->extradata + 12);

                /* TODO: sanity checks? */
Tobias Bindhammer's avatar
Tobias Bindhammer committed
99
            } else {
100 101 102
                av_log(avctx, AV_LOG_ERROR, "extradata not set\n");
                return AVERROR(EINVAL);
            }
Tobias Bindhammer's avatar
Tobias Bindhammer committed
103

104 105
            ch_chunksize=charset_size/lifetime;
            /* TODO: check if charset/size is % lifetime, but maybe check in codec */
Tobias Bindhammer's avatar
Tobias Bindhammer committed
106

107 108
            if(pkt->data) num_frames = lifetime;
            else num_frames = c->prev_frame_count;
Tobias Bindhammer's avatar
Tobias Bindhammer committed
109

110 111 112
            for(i = 0; i < num_frames; i++) {
                if(pkt->data) {
                    /* if available, put newest charset chunk into buffer */
113
                    avio_write(s->pb, pkt->data + ch_chunksize * i, ch_chunksize);
Tobias Bindhammer's avatar
Tobias Bindhammer committed
114
                } else {
Tobias Bindhammer's avatar
Tobias Bindhammer committed
115
                    /* a bit ugly, but is there an alternative to put many zeros? */
116
                    for(j = 0; j < ch_chunksize; j++) avio_w8(s->pb, 0);
117
                }
Tobias Bindhammer's avatar
Tobias Bindhammer committed
118

119 120
                if(c->prev_pkt.data) {
                    /* put frame (screen + colram) from last packet into buffer */
121
                    avio_write(s->pb, c->prev_pkt.data + charset_size + frame_size * i, frame_size);
Tobias Bindhammer's avatar
Tobias Bindhammer committed
122
                } else {
Tobias Bindhammer's avatar
Tobias Bindhammer committed
123
                    /* a bit ugly, but is there an alternative to put many zeros? */
124
                    for(j = 0; j < frame_size; j++) avio_w8(s->pb, 0);
125 126
                }
            }
Tobias Bindhammer's avatar
Tobias Bindhammer committed
127

128 129
            /* backup current packet for next turn */
            if(pkt->data) {
130 131 132 133 134 135 136 137 138 139
                /* no backup packet yet? create one! */
                if(!c->prev_pkt.data) av_new_packet(&c->prev_pkt, pkt->size);
                /* we have a packet and data is big enough, reuse it */
                if(c->prev_pkt.data && c->prev_pkt.size >= pkt->size) {
                    memcpy(c->prev_pkt.data, pkt->data, pkt->size);
                    c->prev_pkt.size = pkt->size;
                } else {
                    av_log(avctx, AV_LOG_ERROR, "Too less memory for prev_pkt.\n");
                    return AVERROR(ENOMEM);
                }
140
            }
Tobias Bindhammer's avatar
Tobias Bindhammer committed
141

142 143 144 145 146 147
            c->prev_frame_count = frame_count;
            break;
        }
        default:
            /* Write things as is. Nice for self-contained frames from non-multicolor modes or if played
             * directly from ram and not from a streaming device (rrnet/mmc) */
148
            if(pkt) avio_write(s->pb, pkt->data, pkt->size);
149 150 151
        break;
    }

152
    avio_flush(s->pb);
153 154 155
    return 0;
}

156 157 158
static int a64_write_trailer(struct AVFormatContext *s)
{
    A64MuxerContext *c = s->priv_data;
159
    AVPacket pkt = {0};
160 161
    /* need to flush last packet? */
    if(c->interleaved) a64_write_packet(s, &pkt);
162 163
    /* discard backed up packet */
    if(c->prev_pkt.data) av_destruct_packet(&c->prev_pkt);
164 165 166
    return 0;
}

167
AVOutputFormat ff_a64_muxer = {
168 169 170 171 172 173 174 175
    .name = "a64",
    .long_name = NULL_IF_CONFIG_SMALL("a64 - video for Commodore 64"),
    .mime_type = NULL,
    .extensions = "a64, A64",
    .priv_data_size = sizeof (A64Context),
    .video_codec = CODEC_ID_A64_MULTI,
    a64_write_header,
    a64_write_packet,
176
    a64_write_trailer
177
};