Mercurial > mplayer.hg
annotate libmpdemux/aviwrite.c @ 2622:fb7220c79b69
verbose fixed
author | arpi |
---|---|
date | Fri, 02 Nov 2001 02:08:00 +0000 |
parents | 66837325b929 |
children | c1e24e01601b |
rev | line source |
---|---|
587
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
1 |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
2 #include <stdio.h> |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
3 #include <stdlib.h> |
2529 | 4 #include <string.h> |
5 | |
2555
66837325b929
config.h cleanup, few things added to steram/demuxer headers
arpi
parents:
2529
diff
changeset
|
6 #include "config.h" |
66837325b929
config.h cleanup, few things added to steram/demuxer headers
arpi
parents:
2529
diff
changeset
|
7 |
2529 | 8 //#include "stream.h" |
9 //#include "demuxer.h" | |
10 //#include "stheader.h" | |
587
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
11 |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
12 #include "wine/mmreg.h" |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
13 #include "wine/avifmt.h" |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
14 #include "wine/vfw.h" |
8511095c5283
stage#1 completed: c files no more included from mplayer.c
arpi_esp
parents:
1
diff
changeset
|
15 |
2529 | 16 #include "aviwrite.h" |
1 | 17 |
2529 | 18 aviwrite_stream_t* aviwrite_new_stream(aviwrite_t *muxer,int type){ |
19 aviwrite_stream_t* s; | |
20 if(muxer->avih.dwStreams>=AVIWRITE_MAX_STREAMS){ | |
21 printf("Too many streams! increase AVIWRITE_MAX_STREAMS !\n"); | |
22 return NULL; | |
23 } | |
24 s=malloc(sizeof(aviwrite_stream_t)); | |
25 memset(s,0,sizeof(aviwrite_stream_t)); | |
26 if(!s) return NULL; // no mem!? | |
27 muxer->streams[muxer->avih.dwStreams]=s; | |
28 s->type=type; | |
29 s->id=muxer->avih.dwStreams; | |
30 s->timer=0.0; | |
31 switch(type){ | |
32 case AVIWRITE_TYPE_VIDEO: | |
33 s->ckid=mmioFOURCC(('0'+s->id/10),('0'+(s->id%10)),'d','c'); | |
34 s->h.fccType=streamtypeVIDEO; | |
35 if(!muxer->def_v) muxer->def_v=s; | |
36 break; | |
37 case AVIWRITE_TYPE_AUDIO: | |
38 s->ckid=mmioFOURCC(('0'+s->id/10),('0'+(s->id%10)),'w','b'); | |
39 s->h.fccType=streamtypeAUDIO; | |
40 break; | |
41 default: | |
42 printf("WarninG! unknown stream type: %d\n",type); | |
43 return NULL; | |
44 } | |
45 muxer->avih.dwStreams++; | |
46 return s; | |
47 } | |
1 | 48 |
2529 | 49 aviwrite_t* aviwrite_new_muxer(){ |
50 aviwrite_t* muxer=malloc(sizeof(aviwrite_t)); | |
51 memset(muxer,0,sizeof(aviwrite_t)); | |
52 return muxer; | |
53 } | |
54 | |
55 static void write_avi_chunk(FILE *f,unsigned int id,int len,void* data){ | |
1 | 56 fwrite(&id,4,1,f); |
57 fwrite(&len,4,1,f); | |
58 if(len>0){ | |
59 if(data){ | |
60 // DATA | |
61 fwrite(data,len,1,f); | |
62 if(len&1){ // padding | |
63 unsigned char zerobyte=0; | |
64 fwrite(&zerobyte,1,1,f); | |
65 } | |
66 } else { | |
67 // JUNK | |
68 char *avi_junk_data="[= MPlayer junk data! =]"; | |
69 if(len&1) ++len; // padding | |
70 while(len>0){ | |
71 int l=strlen(avi_junk_data); | |
72 if(l>len) l=len; | |
73 fwrite(avi_junk_data,l,1,f); | |
74 len-=l; | |
75 } | |
76 } | |
77 } | |
2529 | 78 } |
79 | |
80 void aviwrite_write_chunk(aviwrite_t *muxer,aviwrite_stream_t *s, FILE *f,int len,unsigned int flags){ | |
81 | |
82 // add to the index: | |
83 if(muxer->idx_pos>=muxer->idx_size){ | |
84 muxer->idx_size+=256; // 4kB | |
85 muxer->idx=realloc(muxer->idx,16*muxer->idx_size); | |
86 } | |
87 muxer->idx[muxer->idx_pos].ckid=s->ckid; | |
88 muxer->idx[muxer->idx_pos].dwFlags=flags; // keyframe? | |
89 muxer->idx[muxer->idx_pos].dwChunkOffset=ftell(f)-(muxer->movi_start-4); | |
90 muxer->idx[muxer->idx_pos].dwChunkLength=len; | |
91 ++muxer->idx_pos; | |
92 | |
93 // write out the chunk: | |
94 write_avi_chunk(f,s->ckid,len,s->buffer); | |
95 | |
96 // alter counters: | |
97 if(s->h.dwSampleSize){ | |
98 // CBR | |
99 s->h.dwLength+=len/s->h.dwSampleSize; | |
100 if(len%s->h.dwSampleSize) printf("Warning! len isn't divisable by samplesize!\n"); | |
101 } else { | |
102 // VBR | |
103 s->h.dwLength++; | |
104 } | |
105 s->timer=(double)s->h.dwLength*s->h.dwScale/s->h.dwRate; | |
106 if(len>s->h.dwSuggestedBufferSize) s->h.dwSuggestedBufferSize=len; | |
1 | 107 |
108 } | |
109 | |
2529 | 110 static void write_avi_list(FILE *f,unsigned int id,int len){ |
1 | 111 unsigned int list_id=FOURCC_LIST; |
112 len+=4; // list fix | |
113 fwrite(&list_id,4,1,f); | |
114 fwrite(&len,4,1,f); | |
115 fwrite(&id,4,1,f); | |
116 } | |
117 | |
2529 | 118 void aviwrite_write_header(aviwrite_t *muxer,FILE *f){ |
1 | 119 unsigned int riff[3]; |
2529 | 120 int i; |
121 unsigned int hdrsize; | |
1 | 122 // RIFF header: |
123 riff[0]=mmioFOURCC('R','I','F','F'); | |
2529 | 124 riff[1]=muxer->file_end; // filesize |
1 | 125 riff[2]=formtypeAVI; // 'AVI ' |
126 fwrite(&riff,12,1,f); | |
2529 | 127 |
128 // update AVI header: | |
129 if(muxer->def_v){ | |
130 muxer->avih.dwMicroSecPerFrame=1000000.0*muxer->def_v->h.dwScale/muxer->def_v->h.dwRate; | |
131 muxer->avih.dwMaxBytesPerSec=1000000; // dummy!!!!! FIXME | |
132 muxer->avih.dwPaddingGranularity=2; // ??? | |
133 muxer->avih.dwFlags|=AVIF_ISINTERLEAVED|AVIF_TRUSTCKTYPE; | |
134 muxer->avih.dwTotalFrames=muxer->def_v->h.dwLength; | |
135 muxer->avih.dwSuggestedBufferSize=muxer->def_v->h.dwSuggestedBufferSize; | |
136 muxer->avih.dwWidth=muxer->def_v->bih->biWidth; | |
137 muxer->avih.dwHeight=muxer->def_v->bih->biHeight; | |
138 } | |
139 | |
1 | 140 // AVI header: |
2529 | 141 hdrsize=sizeof(muxer->avih)+8; |
142 // calc total header size: | |
143 for(i=0;i<muxer->avih.dwStreams;i++){ | |
144 hdrsize+=12; // LIST | |
145 hdrsize+=sizeof(muxer->streams[i]->h)+8; // strh | |
146 switch(muxer->streams[i]->type){ | |
147 case AVIWRITE_TYPE_VIDEO: | |
148 hdrsize+=muxer->streams[i]->bih->biSize+8; // strf | |
149 break; | |
150 case AVIWRITE_TYPE_AUDIO: | |
151 hdrsize+=sizeof(WAVEFORMATEX)+muxer->streams[i]->wf->cbSize+8; // strf | |
152 break; | |
153 } | |
154 } | |
155 write_avi_list(f,listtypeAVIHEADER,hdrsize); | |
156 write_avi_chunk(f,ckidAVIMAINHDR,sizeof(muxer->avih),&muxer->avih); | |
157 | |
158 // stream headers: | |
159 for(i=0;i<muxer->avih.dwStreams;i++){ | |
160 hdrsize=sizeof(muxer->streams[i]->h)+8; // strh | |
161 switch(muxer->streams[i]->type){ | |
162 case AVIWRITE_TYPE_VIDEO: | |
163 hdrsize+=muxer->streams[i]->bih->biSize+8; // strf | |
164 break; | |
165 case AVIWRITE_TYPE_AUDIO: | |
166 hdrsize+=sizeof(WAVEFORMATEX)+muxer->streams[i]->wf->cbSize+8; // strf | |
167 break; | |
168 } | |
169 write_avi_list(f,listtypeSTREAMHEADER,hdrsize); | |
170 write_avi_chunk(f,ckidSTREAMHEADER,sizeof(muxer->streams[i]->h),&muxer->streams[i]->h); // strh | |
171 switch(muxer->streams[i]->type){ | |
172 case AVIWRITE_TYPE_VIDEO: | |
173 write_avi_chunk(f,ckidSTREAMFORMAT,muxer->streams[i]->bih->biSize,muxer->streams[i]->bih); | |
174 break; | |
175 case AVIWRITE_TYPE_AUDIO: | |
176 write_avi_chunk(f,ckidSTREAMFORMAT,sizeof(WAVEFORMATEX)+muxer->streams[i]->wf->cbSize,muxer->streams[i]->wf); | |
177 break; | |
178 } | |
179 } | |
180 | |
181 // JUNK: | |
1 | 182 write_avi_chunk(f,ckidAVIPADDING,2048-(ftell(f)&2047)-8,NULL); |
183 // 'movi' header: | |
2529 | 184 write_avi_list(f,listtypeAVIMOVIE,muxer->movi_end-ftell(f)-12); |
185 muxer->movi_start=ftell(f); | |
1 | 186 } |
187 | |
2529 | 188 void aviwrite_write_index(aviwrite_t *muxer,FILE *f){ |
189 muxer->movi_end=ftell(f); | |
190 if(muxer->idx && muxer->idx_pos>0){ | |
191 // fixup index entries: | |
192 // int i; | |
193 // for(i=0;i<muxer->idx_pos;i++) muxer->idx[i].dwChunkOffset-=muxer->movi_start-4; | |
194 // write index chunk: | |
195 write_avi_chunk(f,ckidAVINEWINDEX,16*muxer->idx_pos,muxer->idx); | |
196 muxer->avih.dwFlags|=AVIF_HASINDEX; | |
197 } | |
198 muxer->file_end=ftell(f); | |
1 | 199 } |
200 |