nut.c 6.75 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*
 * nut
 * Copyright (c) 2004-2007 Michael Niedermayer
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * 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.
 *
 * FFmpeg 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

22
#include "libavutil/mathematics.h"
23
#include "libavutil/tree.h"
24
#include "nut.h"
25
#include "internal.h"
26

27 28 29 30 31 32 33 34
const AVCodecTag ff_nut_subtitle_tags[] = {
    { CODEC_ID_TEXT        , MKTAG('U', 'T', 'F', '8') },
    { CODEC_ID_SSA         , MKTAG('S', 'S', 'A',  0 ) },
    { CODEC_ID_DVD_SUBTITLE, MKTAG('D', 'V', 'D', 'S') },
    { CODEC_ID_DVB_SUBTITLE, MKTAG('D', 'V', 'B', 'S') },
    { CODEC_ID_NONE        , 0                         }
};

35 36 37 38 39 40 41 42 43
const AVCodecTag ff_nut_video_tags[] = {
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 15 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 15 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(15 , 'B', 'G', 'R') },
    { CODEC_ID_RAWVIDEO, MKTAG(15 , 'R', 'G', 'B') },
    { CODEC_ID_RAWVIDEO, MKTAG(16 , 'B', 'G', 'R') },
    { CODEC_ID_RAWVIDEO, MKTAG(16 , 'R', 'G', 'B') },
Michael Niedermayer's avatar
Michael Niedermayer committed
44 45 46 47
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 12 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 12 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(12 , 'B', 'G', 'R') },
    { CODEC_ID_RAWVIDEO, MKTAG(12 , 'R', 'G', 'B') },
48
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 'A') },
49
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B',  0 ) },
50
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 'A') },
51
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R',  0 ) },
52
    { CODEC_ID_RAWVIDEO, MKTAG('A', 'B', 'G', 'R') },
53
    { CODEC_ID_RAWVIDEO, MKTAG( 0 , 'B', 'G', 'R') },
54
    { CODEC_ID_RAWVIDEO, MKTAG('A', 'R', 'G', 'B') },
55
    { CODEC_ID_RAWVIDEO, MKTAG( 0 , 'R', 'G', 'B') },
56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 24 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 24 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '1', '1', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
    { CODEC_ID_RAWVIDEO, MKTAG('B', '1', 'W', '0') },
    { CODEC_ID_RAWVIDEO, MKTAG('B', '0', 'W', '1') },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R',  8 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B',  8 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R',  4 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B',  4 ) },
71 72
    { CODEC_ID_RAWVIDEO, MKTAG('B', '4', 'B', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('R', '4', 'B', 'Y') },
73
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 48 ) },
74
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 48 ) },
75
    { CODEC_ID_RAWVIDEO, MKTAG(48 , 'B', 'G', 'R') },
76
    { CODEC_ID_RAWVIDEO, MKTAG(48 , 'R', 'G', 'B') },
77 78 79 80
    { CODEC_ID_RAWVIDEO, MKTAG('B', 'R', 'A', 64 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('R', 'B', 'A', 64 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(64 , 'B', 'R', 'A') },
    { CODEC_ID_RAWVIDEO, MKTAG(64 , 'R', 'B', 'A') },
Luca Barbato's avatar
Luca Barbato committed
81 82 83 84 85 86
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 10 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(10 , 11 , '3', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 10 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(10 , 10 , '3', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3',  0 , 10 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(10 ,  0 , '3', 'Y') },
87 88 89 90 91 92 93 94 95 96
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '1',  0 , 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(16 ,  0 , '1', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(16 , 11 , '3', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(16 , 10 , '3', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '3',  0 , 16 ) },
    { CODEC_ID_RAWVIDEO, MKTAG(16 ,  0 , '3', 'Y') },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11 ,  8 ) },
    { CODEC_ID_RAWVIDEO, MKTAG('Y', '2',  0 ,  8 ) },
97 98 99
    { CODEC_ID_NONE    , 0                         }
};

100 101 102 103
void ff_nut_reset_ts(NUTContext *nut, AVRational time_base, int64_t val){
    int i;
    for(i=0; i<nut->avf->nb_streams; i++){
        nut->stream[i].last_pts= av_rescale_rnd(
104
            val,
105 106 107 108 109
            time_base.num * (int64_t)nut->stream[i].time_base->den,
            time_base.den * (int64_t)nut->stream[i].time_base->num,
            AV_ROUND_DOWN);
    }
}
110 111 112 113 114 115 116

int64_t ff_lsb2full(StreamContext *stream, int64_t lsb){
    int64_t mask = (1<<stream->msb_pts_shift)-1;
    int64_t delta= stream->last_pts - mask/2;
    return  ((lsb - delta)&mask) + delta;
}

117
int ff_nut_sp_pos_cmp(const Syncpoint *a, const Syncpoint *b){
118
    return ((a->pos - b->pos) >> 32) - ((b->pos - a->pos) >> 32);
119 120
}

121
int ff_nut_sp_pts_cmp(const Syncpoint *a, const Syncpoint *b){
122
    return ((a->ts - b->ts) >> 32) - ((b->ts - a->ts) >> 32);
123 124 125
}

void ff_nut_add_sp(NUTContext *nut, int64_t pos, int64_t back_ptr, int64_t ts){
126
    Syncpoint *sp= av_mallocz(sizeof(Syncpoint));
127
    struct AVTreeNode *node= av_mallocz(av_tree_node_size);
128 129 130 131

    sp->pos= pos;
    sp->back_ptr= back_ptr;
    sp->ts= ts;
132
    av_tree_insert(&nut->syncpoints, sp, (void *) ff_nut_sp_pos_cmp, &node);
133
    if(node){
134
        av_free(sp);
135 136
        av_free(node);
    }
137
}
138

Vitor Sessak's avatar
Vitor Sessak committed
139
static int enu_free(void *opaque, void *elem)
140 141
{
    av_free(elem);
Vitor Sessak's avatar
Vitor Sessak committed
142
    return 0;
143 144 145 146 147 148 149 150
}

void ff_nut_free_sp(NUTContext *nut)
{
    av_tree_enumerate(nut->syncpoints, NULL, NULL, enu_free);
    av_tree_destroy(nut->syncpoints);
}

151 152 153 154 155 156 157 158 159 160
const Dispositions ff_nut_dispositions[] = {
    {"default"     , AV_DISPOSITION_DEFAULT},
    {"dub"         , AV_DISPOSITION_DUB},
    {"original"    , AV_DISPOSITION_ORIGINAL},
    {"comment"     , AV_DISPOSITION_COMMENT},
    {"lyrics"      , AV_DISPOSITION_LYRICS},
    {"karaoke"     , AV_DISPOSITION_KARAOKE},
    {""            , 0}
};

161 162 163 164 165 166 167 168 169 170 171 172 173
const AVMetadataConv ff_nut_metadata_conv[] = {
    { "Author",         "artist"      },
    { "X-CreationTime", "date"        },
    { "CreationTime",   "date"        },
    { "SourceFilename", "filename"    },
    { "X-Language",     "language"    },
    { "X-Disposition",  "disposition" },
    { "X-Replaces",     "replaces"    },
    { "X-Depends",      "depends"     },
    { "X-Uses",         "uses"        },
    { "X-UsesFont",     "usesfont"    },
    { 0 },
};