annotate libmpcodecs/vf_screenshot.c @ 16457:5ab14fa7a019

do nothing if no free filenames are available
author henry
date Sun, 11 Sep 2005 15:41:08 +0000
parents 34eb986b7e4a
children 0a828abf1f0f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
1 #include "../config.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
2 #ifdef HAVE_PNG
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
3
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
4 #include <stdio.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
5 #include <stdlib.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
6 #include <string.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
7 #include <inttypes.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
8
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
9 #include <sys/types.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
10 #include <sys/stat.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
11 #include <unistd.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
12
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
13 #include <png.h>
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
14
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
15 #include "../mp_msg.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
16
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
17 #include "img_format.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
18 #include "mp_image.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
19 #include "vf.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
20 #include "vf_scale.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
21
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
22 #include "../libvo/fastmemcpy.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
23 #include "../postproc/swscale.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
24 #include "../postproc/rgb2rgb.h"
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
25
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
26 struct vf_priv_s {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
27 int frameno;
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
28 char fname[102];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
29 int shot, store_slices;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
30 int dw, dh, stride;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
31 uint8_t *buffer;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
32 struct SwsContext *ctx;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
33 mp_image_t *dmpi;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
34 };
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
35
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
36 //===========================================================================//
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
37
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
38 static int config(struct vf_instance_s* vf,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
39 int width, int height, int d_width, int d_height,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
40 unsigned int flags, unsigned int outfmt)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
41 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
42 int int_sws_flags=0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
43 SwsFilter *srcFilter, *dstFilter;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
44
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
45 sws_getFlagsAndFilterFromCmdLine(&int_sws_flags, &srcFilter, &dstFilter);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
46
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
47 vf->priv->ctx=sws_getContext(width, height, outfmt,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
48 d_width, d_height, IMGFMT_BGR24,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
49 int_sws_flags | get_sws_cpuflags(), srcFilter, dstFilter, NULL);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
50
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
51 vf->priv->dw = d_width;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
52 vf->priv->dh = d_height;
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
53 vf->priv->stride = (3*vf->priv->dw+15)&~15;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
54
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
55 if (vf->priv->buffer) free(vf->priv->buffer); // probably reconfigured
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
56 vf->priv->dmpi = NULL;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
57
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
58 return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
59 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
60
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
61 static int write_png(char *fname, unsigned char *buffer, int width, int height, int stride)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
62 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
63 FILE * fp;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
64 png_structp png_ptr;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
65 png_infop info_ptr;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
66 png_byte **row_pointers;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
67 int k;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
68
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
69 png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
70 info_ptr = png_create_info_struct(png_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
71 fp = NULL;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
72
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
73 if (setjmp(png_ptr->jmpbuf)) {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
74 png_destroy_write_struct(&png_ptr, &info_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
75 fclose(fp);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
76 return 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
77 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
78
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
79 fp = fopen (fname, "wb");
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
80 if (fp == NULL) {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
81 mp_msg(MSGT_VFILTER,MSGL_ERR,"\nPNG Error opening %s for writing!\n", fname);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
82 return 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
83 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
84
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
85 png_init_io(png_ptr, fp);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
86 png_set_compression_level(png_ptr, 0);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
87
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
88 png_set_IHDR(png_ptr, info_ptr, width, height,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
89 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
90 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
91
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
92 png_write_info(png_ptr, info_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
93
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
94 png_set_bgr(png_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
95
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
96 row_pointers = (png_byte**)malloc(height*sizeof(png_byte*));
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
97 for (k = 0; k < height; k++) {
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
98 unsigned char* s=buffer + stride*k;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
99 row_pointers[k] = s;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
100 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
101
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
102 png_write_image(png_ptr, row_pointers);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
103 png_write_end(png_ptr, info_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
104 png_destroy_write_struct(&png_ptr, &info_ptr);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
105
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
106 free(row_pointers);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
107
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
108 fclose (fp);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
109 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
110
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
111 static int fexists(char *fname)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
112 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
113 struct stat dummy;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
114 if (stat(fname, &dummy) == 0) return 1;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
115 else return 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
116 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
117
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
118 static void gen_fname(struct vf_priv_s* priv)
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
119 {
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
120 do {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
121 snprintf (priv->fname, 100, "shot%04d.png", ++priv->frameno);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
122 } while (fexists(priv->fname) && priv->frameno < 100000);
16457
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
123 if (fexists(priv->fname)) {
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
124 priv->fname[0] = '\0';
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
125 return;
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
126 }
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
127
40222952440f DR and slice support
henry
parents: 16446
diff changeset
128 mp_msg(MSGT_VFILTER,MSGL_INFO,"*** screenshot '%s' ***\n",priv->fname);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
129
40222952440f DR and slice support
henry
parents: 16446
diff changeset
130 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
131
40222952440f DR and slice support
henry
parents: 16446
diff changeset
132 static void scale_image(struct vf_priv_s* priv)
40222952440f DR and slice support
henry
parents: 16446
diff changeset
133 {
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
134 uint8_t *dst[3];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
135 int dst_stride[3];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
136
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
137 dst_stride[0] = priv->stride;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
138 dst_stride[1] = dst_stride[2] = 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
139 if (!priv->buffer)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
140 priv->buffer = (uint8_t*)memalign(16, dst_stride[0]*priv->dh);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
141
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
142 dst[0] = priv->buffer;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
143 dst[1] = dst[2] = 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
144 sws_scale_ordered(priv->ctx, priv->dmpi->planes, priv->dmpi->stride, 0, priv->dmpi->height, dst, dst_stride);
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
145 }
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
146
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
147 static void start_slice(struct vf_instance_s* vf, mp_image_t *mpi){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
148 if(!vf->next->draw_slice) {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
149 mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
150 return;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
151 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
152 if(!(mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)) return; // shouldn't happen
40222952440f DR and slice support
henry
parents: 16446
diff changeset
153 mpi->priv=vf->dmpi=vf_get_image(vf->next,mpi->imgfmt,
40222952440f DR and slice support
henry
parents: 16446
diff changeset
154 mpi->type, mpi->flags, mpi->width, mpi->height);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
155 if (vf->priv->shot) {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
156 vf->priv->store_slices = 1;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
157 vf->priv->shot = 0;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
158 if (!vf->priv->buffer)
40222952440f DR and slice support
henry
parents: 16446
diff changeset
159 vf->priv->buffer = (uint8_t*)memalign(16, vf->priv->stride*vf->priv->dh);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
160 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
161
40222952440f DR and slice support
henry
parents: 16446
diff changeset
162 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
163
40222952440f DR and slice support
henry
parents: 16446
diff changeset
164 static void draw_slice(struct vf_instance_s* vf,
40222952440f DR and slice support
henry
parents: 16446
diff changeset
165 unsigned char** src, int* stride, int w,int h, int x, int y){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
166 mp_image_t *dmpi=vf->dmpi;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
167 if(!dmpi){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
168 mp_msg(MSGT_VFILTER,MSGL_FATAL,"vf_screenshot: draw_slice() called with dmpi=NULL (no get_image??)\n");
40222952440f DR and slice support
henry
parents: 16446
diff changeset
169 return;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
170 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
171 if (vf->priv->store_slices) {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
172 uint8_t *dst[3];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
173 int dst_stride[3];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
174 dst_stride[0] = vf->priv->stride;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
175 dst_stride[1] = dst_stride[2] = 0;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
176 dst[0] = vf->priv->buffer;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
177 dst[1] = dst[2] = 0;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
178 sws_scale_ordered(vf->priv->ctx, src, stride, y, h, dst, dst_stride);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
179 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
180 vf_next_draw_slice(vf,src,stride,w,h,x,y);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
181 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
182
40222952440f DR and slice support
henry
parents: 16446
diff changeset
183 static void get_image(struct vf_instance_s* vf, mp_image_t *mpi){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
184 vf->dmpi= vf_get_image(vf->next, mpi->imgfmt,
16452
f5cc3e7e7d10 forgotten MP_IMGFLAG_READABLE
henry
parents: 16450
diff changeset
185 mpi->type, mpi->flags | MP_IMGFLAG_READABLE, mpi->width, mpi->height);
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
186
16456
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
187 if((vf->dmpi->flags & MP_IMGFLAG_DRAW_CALLBACK) &&
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
188 !(vf->dmpi->flags & MP_IMGFLAG_DIRECT)){
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
189 // use slices if DR isn't possible
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
190 return;
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
191 }
34eb986b7e4a use slices if DR isn't available
henry
parents: 16452
diff changeset
192
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
193 mpi->planes[0]=vf->dmpi->planes[0];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
194 mpi->stride[0]=vf->dmpi->stride[0];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
195 if(mpi->flags&MP_IMGFLAG_PLANAR){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
196 mpi->planes[1]=vf->dmpi->planes[1];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
197 mpi->planes[2]=vf->dmpi->planes[2];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
198 mpi->stride[1]=vf->dmpi->stride[1];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
199 mpi->stride[2]=vf->dmpi->stride[2];
40222952440f DR and slice support
henry
parents: 16446
diff changeset
200 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
201 mpi->width=vf->dmpi->width;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
202
40222952440f DR and slice support
henry
parents: 16446
diff changeset
203 mpi->flags|=MP_IMGFLAG_DIRECT;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
204 mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
205
40222952440f DR and slice support
henry
parents: 16446
diff changeset
206 mpi->priv=(void*)vf->dmpi;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
207 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
208
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
209 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
210 {
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
211 if (mpi->flags&MP_IMGFLAG_DRAW_CALLBACK) {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
212 if(vf->priv->store_slices) {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
213 vf->priv->store_slices = 0;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
214 gen_fname(vf->priv);
16457
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
215 if (vf->priv->fname[0])
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
216 write_png(vf->priv->fname, vf->priv->buffer, vf->priv->dw, vf->priv->dh, vf->priv->stride);
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
217 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
218 return vf_next_put_image(vf,vf->dmpi);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
219 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
220
40222952440f DR and slice support
henry
parents: 16446
diff changeset
221 if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
40222952440f DR and slice support
henry
parents: 16446
diff changeset
222 vf->priv->dmpi=vf_get_image(vf->next,mpi->imgfmt,
40222952440f DR and slice support
henry
parents: 16446
diff changeset
223 MP_IMGTYPE_EXPORT, 0,
40222952440f DR and slice support
henry
parents: 16446
diff changeset
224 mpi->width, mpi->height);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
225 vf_clone_mpi_attributes(vf->priv->dmpi, mpi);
40222952440f DR and slice support
henry
parents: 16446
diff changeset
226 } else {
40222952440f DR and slice support
henry
parents: 16446
diff changeset
227 vf->priv->dmpi=vf->dmpi;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
228 }
40222952440f DR and slice support
henry
parents: 16446
diff changeset
229
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
230 vf->priv->dmpi->planes[0]=mpi->planes[0];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
231 vf->priv->dmpi->planes[1]=mpi->planes[1];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
232 vf->priv->dmpi->planes[2]=mpi->planes[2];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
233 vf->priv->dmpi->stride[0]=mpi->stride[0];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
234 vf->priv->dmpi->stride[1]=mpi->stride[1];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
235 vf->priv->dmpi->stride[2]=mpi->stride[2];
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
236 vf->priv->dmpi->width=mpi->width;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
237 vf->priv->dmpi->height=mpi->height;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
238
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
239 if(vf->priv->shot) {
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
240 vf->priv->shot=0;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
241 gen_fname(vf->priv);
16457
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
242 if (vf->priv->fname[0]) {
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
243 scale_image(vf->priv);
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
244 write_png(vf->priv->fname, vf->priv->buffer, vf->priv->dw, vf->priv->dh, vf->priv->stride);
5ab14fa7a019 do nothing if no free filenames are available
henry
parents: 16456
diff changeset
245 }
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
246 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
247
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
248 return vf_next_put_image(vf, vf->priv->dmpi);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
249 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
250
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
251 int control (vf_instance_t *vf, int request, void *data)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
252 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
253 if(request==VFCTRL_SCREENSHOT) {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
254 vf->priv->shot=1;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
255 return CONTROL_TRUE;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
256 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
257 return vf_next_control (vf, request, data);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
258 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
259
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
260
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
261 //===========================================================================//
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
262
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
263 static int query_format(struct vf_instance_s* vf, unsigned int fmt)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
264 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
265 switch(fmt){
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
266 case IMGFMT_YV12:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
267 case IMGFMT_I420:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
268 case IMGFMT_IYUV:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
269 case IMGFMT_UYVY:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
270 case IMGFMT_YUY2:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
271 case IMGFMT_BGR32:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
272 case IMGFMT_BGR24:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
273 case IMGFMT_BGR16:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
274 case IMGFMT_BGR15:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
275 case IMGFMT_RGB32:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
276 case IMGFMT_RGB24:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
277 case IMGFMT_Y800:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
278 case IMGFMT_Y8:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
279 case IMGFMT_YVU9:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
280 case IMGFMT_IF09:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
281 case IMGFMT_444P:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
282 case IMGFMT_422P:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
283 case IMGFMT_411P:
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
284 return vf_next_query_format(vf, fmt);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
285 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
286 return 0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
287 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
288
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
289 static int open(vf_instance_t *vf, char* args)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
290 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
291 vf->config=config;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
292 vf->control=control;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
293 vf->put_image=put_image;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
294 vf->query_format=query_format;
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
295 vf->start_slice=start_slice;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
296 vf->draw_slice=draw_slice;
40222952440f DR and slice support
henry
parents: 16446
diff changeset
297 vf->get_image=get_image;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
298 vf->priv=malloc(sizeof(struct vf_priv_s));
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
299 vf->priv->frameno=0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
300 vf->priv->shot=0;
16450
40222952440f DR and slice support
henry
parents: 16446
diff changeset
301 vf->priv->store_slices=0;
16446
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
302 vf->priv->buffer=0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
303 vf->priv->ctx=0;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
304 return 1;
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
305 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
306
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
307 static void uninit(vf_instance_t *vf)
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
308 {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
309 if(vf->priv->ctx) sws_freeContext(vf->priv->ctx);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
310 if (vf->priv->buffer) free(vf->priv->buffer);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
311 free(vf->priv);
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
312 }
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
313
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
314
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
315 vf_info_t vf_info_screenshot = {
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
316 "screenshot to file",
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
317 "screenshot",
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
318 "A'rpi, Jindrich Makovicka",
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
319 "",
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
320 open,
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
321 NULL
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
322 };
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
323
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
324 //===========================================================================//
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
325
c82c0d407ec9 screenshot filter
henry
parents:
diff changeset
326 #endif /* HAVE_PNG */