11 #include <sys/types.h>
17 #include "img_format.h"
22 #include "libswscale/swscale.h"
23 #include "libavcodec/avcodec.h"
28 /// shot stores current screenshot mode:
29 /// 0: don't take screenshots
30 /// 1: take single screenshot, reset to 0 afterwards
31 /// 2: take screenshots of each frame
32 int shot
, store_slices
;
35 struct SwsContext
*ctx
;
36 AVCodecContext
*avctx
;
41 //===========================================================================//
43 static int config(struct vf_instance
* vf
,
44 int width
, int height
, int d_width
, int d_height
,
45 unsigned int flags
, unsigned int outfmt
)
47 vf
->priv
->ctx
=sws_getContextFromCmdLine(width
, height
, outfmt
,
48 d_width
, d_height
, IMGFMT_RGB24
);
50 vf
->priv
->outbuffer_size
= d_width
* d_height
* 3 * 2;
51 vf
->priv
->outbuffer
= realloc(vf
->priv
->outbuffer
, vf
->priv
->outbuffer_size
);
52 vf
->priv
->avctx
->width
= d_width
;
53 vf
->priv
->avctx
->height
= d_height
;
54 vf
->priv
->avctx
->pix_fmt
= PIX_FMT_RGB24
;
55 vf
->priv
->avctx
->compression_level
= 0;
56 vf
->priv
->dw
= d_width
;
57 vf
->priv
->dh
= d_height
;
58 vf
->priv
->stride
= (3*vf
->priv
->dw
+15)&~15;
60 if (vf
->priv
->buffer
) free(vf
->priv
->buffer
); // probably reconfigured
61 vf
->priv
->buffer
= NULL
;
63 return vf_next_config(vf
,width
,height
,d_width
,d_height
,flags
,outfmt
);
66 static void write_png(struct vf_priv_s
*priv
)
68 char *fname
= priv
->fname
;
73 fp
= fopen (fname
, "wb");
75 mp_msg(MSGT_VFILTER
,MSGL_ERR
,"\nPNG Error opening %s for writing!\n", fname
);
79 pic
.data
[0] = priv
->buffer
;
80 pic
.linesize
[0] = priv
->stride
;
81 size
= avcodec_encode_video(priv
->avctx
, priv
->outbuffer
, priv
->outbuffer_size
, &pic
);
83 fwrite(priv
->outbuffer
, size
, 1, fp
);
88 static int fexists(char *fname
)
91 if (stat(fname
, &dummy
) == 0) return 1;
95 static void gen_fname(struct vf_priv_s
* priv
)
98 snprintf (priv
->fname
, 100, "shot%04d.png", ++priv
->frameno
);
99 } while (fexists(priv
->fname
) && priv
->frameno
< 100000);
100 if (fexists(priv
->fname
)) {
101 priv
->fname
[0] = '\0';
105 mp_msg(MSGT_VFILTER
,MSGL_INFO
,"*** screenshot '%s' ***\n",priv
->fname
);
109 static void scale_image(struct vf_priv_s
* priv
, mp_image_t
*mpi
)
114 dst_stride
[0] = priv
->stride
;
115 dst_stride
[1] = dst_stride
[2] = 0;
117 priv
->buffer
= memalign(16, dst_stride
[0]*priv
->dh
);
119 dst
[0] = priv
->buffer
;
121 sws_scale_ordered(priv
->ctx
, mpi
->planes
, mpi
->stride
, 0, priv
->dh
, dst
, dst_stride
);
124 static void start_slice(struct vf_instance
* vf
, mp_image_t
*mpi
){
125 vf
->dmpi
=vf_get_image(vf
->next
,mpi
->imgfmt
,
126 mpi
->type
, mpi
->flags
, mpi
->width
, mpi
->height
);
127 if (vf
->priv
->shot
) {
128 vf
->priv
->store_slices
= 1;
129 if (!vf
->priv
->buffer
)
130 vf
->priv
->buffer
= memalign(16, vf
->priv
->stride
*vf
->priv
->dh
);
135 static void draw_slice(struct vf_instance
* vf
,
136 unsigned char** src
, int* stride
, int w
,int h
, int x
, int y
){
137 if (vf
->priv
->store_slices
) {
140 dst_stride
[0] = vf
->priv
->stride
;
141 dst_stride
[1] = dst_stride
[2] = 0;
142 dst
[0] = vf
->priv
->buffer
;
144 sws_scale_ordered(vf
->priv
->ctx
, src
, stride
, y
, h
, dst
, dst_stride
);
146 vf_next_draw_slice(vf
,src
,stride
,w
,h
,x
,y
);
149 static void get_image(struct vf_instance
* vf
, mp_image_t
*mpi
){
150 // FIXME: should vf.c really call get_image when using slices??
151 if (mpi
->flags
& MP_IMGFLAG_DRAW_CALLBACK
)
153 vf
->dmpi
= vf_get_image(vf
->next
, mpi
->imgfmt
,
154 mpi
->type
, mpi
->flags
/* | MP_IMGFLAG_READABLE*/, mpi
->width
, mpi
->height
);
156 mpi
->planes
[0]=vf
->dmpi
->planes
[0];
157 mpi
->stride
[0]=vf
->dmpi
->stride
[0];
158 if(mpi
->flags
&MP_IMGFLAG_PLANAR
){
159 mpi
->planes
[1]=vf
->dmpi
->planes
[1];
160 mpi
->planes
[2]=vf
->dmpi
->planes
[2];
161 mpi
->stride
[1]=vf
->dmpi
->stride
[1];
162 mpi
->stride
[2]=vf
->dmpi
->stride
[2];
164 mpi
->width
=vf
->dmpi
->width
;
166 mpi
->flags
|=MP_IMGFLAG_DIRECT
;
168 mpi
->priv
=(void*)vf
->dmpi
;
171 static int put_image(struct vf_instance
* vf
, mp_image_t
*mpi
, double pts
)
173 mp_image_t
*dmpi
= (mp_image_t
*)mpi
->priv
;
175 if (mpi
->flags
& MP_IMGFLAG_DRAW_CALLBACK
)
178 if(!(mpi
->flags
&MP_IMGFLAG_DIRECT
)){
179 dmpi
=vf_get_image(vf
->next
,mpi
->imgfmt
,
180 MP_IMGTYPE_EXPORT
, 0,
181 mpi
->width
, mpi
->height
);
182 vf_clone_mpi_attributes(dmpi
, mpi
);
183 dmpi
->planes
[0]=mpi
->planes
[0];
184 dmpi
->planes
[1]=mpi
->planes
[1];
185 dmpi
->planes
[2]=mpi
->planes
[2];
186 dmpi
->stride
[0]=mpi
->stride
[0];
187 dmpi
->stride
[1]=mpi
->stride
[1];
188 dmpi
->stride
[2]=mpi
->stride
[2];
189 dmpi
->width
=mpi
->width
;
190 dmpi
->height
=mpi
->height
;
194 if (vf
->priv
->shot
==1)
197 if (vf
->priv
->fname
[0]) {
198 if (!vf
->priv
->store_slices
)
199 scale_image(vf
->priv
, dmpi
);
202 vf
->priv
->store_slices
= 0;
205 return vf_next_put_image(vf
, dmpi
, pts
);
208 static int control (vf_instance_t
*vf
, int request
, void *data
)
210 /** data contains an integer argument
211 * 0: take screenshot with the next frame
212 * 1: take screenshots with each frame until the same command is given once again
214 if(request
==VFCTRL_SCREENSHOT
) {
215 if (data
&& *(int*)data
) { // repeated screenshot mode
216 if (vf
->priv
->shot
==2)
220 } else { // single screenshot
226 return vf_next_control (vf
, request
, data
);
230 //===========================================================================//
232 static int query_format(struct vf_instance
* vf
, unsigned int fmt
)
253 return vf_next_query_format(vf
, fmt
);
258 static void uninit(vf_instance_t
*vf
);
259 // open conflicts with stdio.h at least under MinGW
260 static int screenshot_open(vf_instance_t
*vf
, char* args
)
264 vf
->put_image
=put_image
;
265 vf
->query_format
=query_format
;
266 vf
->start_slice
=start_slice
;
267 vf
->draw_slice
=draw_slice
;
268 vf
->get_image
=get_image
;
270 vf
->priv
=malloc(sizeof(struct vf_priv_s
));
273 vf
->priv
->store_slices
=0;
275 vf
->priv
->outbuffer
=0;
277 vf
->priv
->avctx
= avcodec_alloc_context();
278 avcodec_register_all();
279 if (avcodec_open(vf
->priv
->avctx
, avcodec_find_encoder(CODEC_ID_PNG
))) {
280 mp_msg(MSGT_VFILTER
, MSGL_FATAL
, "Could not open libavcodec PNG encoder\n");
286 static void uninit(vf_instance_t
*vf
)
288 av_freep(&vf
->priv
->avctx
);
289 if(vf
->priv
->ctx
) sws_freeContext(vf
->priv
->ctx
);
290 if (vf
->priv
->buffer
) free(vf
->priv
->buffer
);
291 free(vf
->priv
->outbuffer
);
296 const vf_info_t vf_info_screenshot
= {
297 "screenshot to file",
299 "A'rpi, Jindrich Makovicka",
305 //===========================================================================//