301 lines
8.4 KiB
C
301 lines
8.4 KiB
C
/*
|
|
* Copyright (C) 2009-2010 Freescale Semiconductor, Inc. All rights reserved.
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include <gst/gst.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include "gst_snapshot.h"
|
|
|
|
|
|
static void
|
|
feed_fakesrc (GstElement * src, GstBuffer * buf, GstPad * pad, gpointer data)
|
|
{
|
|
GstBuffer *in_buf = GST_BUFFER (data);
|
|
|
|
g_assert (GST_BUFFER_SIZE (buf) >= GST_BUFFER_SIZE (in_buf));
|
|
g_assert (!GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_READONLY));
|
|
|
|
gst_buffer_set_caps (buf, GST_BUFFER_CAPS (in_buf));
|
|
|
|
memcpy (GST_BUFFER_DATA (buf), GST_BUFFER_DATA (in_buf),
|
|
GST_BUFFER_SIZE (in_buf));
|
|
|
|
GST_BUFFER_SIZE (buf) = GST_BUFFER_SIZE (in_buf);
|
|
|
|
GST_DEBUG ("feeding buffer %p, size %u, caps %" GST_PTR_FORMAT,
|
|
buf, GST_BUFFER_SIZE (buf), GST_BUFFER_CAPS (buf));
|
|
}
|
|
|
|
static void
|
|
save_result (GstElement * sink, GstBuffer * buf, GstPad * pad, gpointer data)
|
|
{
|
|
GstBuffer **p_buf = (GstBuffer **) data;
|
|
|
|
*p_buf = gst_buffer_ref (buf);
|
|
|
|
GST_DEBUG ("received converted buffer %p with caps %" GST_PTR_FORMAT,
|
|
*p_buf, GST_BUFFER_CAPS (*p_buf));
|
|
}
|
|
|
|
static gboolean
|
|
create_element (const gchar *factory_name, GstElement **element, GError **err)
|
|
{
|
|
*element = gst_element_factory_make (factory_name, NULL);
|
|
if (*element)
|
|
return TRUE;
|
|
|
|
if (err && *err == NULL) {
|
|
*err = g_error_new (GST_CORE_ERROR, GST_CORE_ERROR_MISSING_PLUGIN,
|
|
"cannot create element '%s' - please check your GStreamer installation",
|
|
factory_name);
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/* takes ownership of the input buffer */
|
|
GstBuffer* gst_frame_convert(GstBuffer* buf, GstCaps* to_caps)
|
|
{
|
|
GstElement *src, *csp, *filter1, *vscale, *filter2, *sink, *pipeline;
|
|
GstMessage *msg;
|
|
GstBuffer *result = NULL;
|
|
GError *error = NULL;
|
|
GstBus *bus;
|
|
GstCaps *to_caps_no_par;
|
|
|
|
g_return_val_if_fail (GST_BUFFER_CAPS (buf) != NULL, NULL);
|
|
|
|
/* videoscale is here to correct for the pixel-aspect-ratio for us */
|
|
GST_DEBUG ("creating elements");
|
|
if (!create_element ("fakesrc", &src, &error) ||
|
|
!create_element ("ffmpegcolorspace", &csp, &error) ||
|
|
!create_element ("videoscale", &vscale, &error) ||
|
|
!create_element ("capsfilter", &filter1, &error) ||
|
|
!create_element ("capsfilter", &filter2, &error) ||
|
|
!create_element ("fakesink", &sink, &error)) {
|
|
g_warning ("Could not take screenshot: %s", error->message);
|
|
g_error_free (error);
|
|
return NULL;
|
|
}
|
|
|
|
pipeline = gst_pipeline_new ("screenshot-pipeline");
|
|
if (pipeline == NULL) {
|
|
g_warning ("Could not take screenshot: %s", "no pipeline (unknown error)");
|
|
return NULL;
|
|
}
|
|
|
|
GST_DEBUG ("adding elements");
|
|
gst_bin_add_many (GST_BIN (pipeline), src, csp, filter1, vscale, filter2,
|
|
sink, NULL);
|
|
|
|
g_signal_connect (src, "handoff", G_CALLBACK (feed_fakesrc), buf);
|
|
|
|
/* set to 'fixed' sizetype */
|
|
g_object_set (src, "sizemax", GST_BUFFER_SIZE (buf), "sizetype", 2,
|
|
"num-buffers", 1, "signal-handoffs", TRUE, NULL);
|
|
|
|
/* adding this superfluous capsfilter makes linking cheaper */
|
|
to_caps_no_par = gst_caps_copy (to_caps);
|
|
gst_structure_remove_field (gst_caps_get_structure (to_caps_no_par, 0),
|
|
"pixel-aspect-ratio");
|
|
g_object_set (filter1, "caps", to_caps_no_par, NULL);
|
|
gst_caps_unref (to_caps_no_par);
|
|
|
|
g_object_set (filter2, "caps", to_caps, NULL);
|
|
|
|
g_signal_connect (sink, "handoff", G_CALLBACK (save_result), &result);
|
|
|
|
g_object_set (sink, "preroll-queue-len", 1, "signal-handoffs", TRUE, NULL);
|
|
|
|
/* FIXME: linking is still way too expensive, profile this properly */
|
|
GST_DEBUG ("linking src->csp");
|
|
if (!gst_element_link_pads (src, "src", csp, "sink"))
|
|
return NULL;
|
|
|
|
GST_DEBUG ("linking csp->filter1");
|
|
if (!gst_element_link_pads (csp, "src", filter1, "sink"))
|
|
return NULL;
|
|
|
|
GST_DEBUG ("linking filter1->vscale");
|
|
if (!gst_element_link_pads (filter1, "src", vscale, "sink"))
|
|
return NULL;
|
|
|
|
GST_DEBUG ("linking vscale->capsfilter");
|
|
if (!gst_element_link_pads (vscale, "src", filter2, "sink"))
|
|
return NULL;
|
|
|
|
GST_DEBUG ("linking capsfilter->sink");
|
|
if (!gst_element_link_pads (filter2, "src", sink, "sink"))
|
|
return NULL;
|
|
|
|
GST_DEBUG ("running conversion pipeline");
|
|
gst_element_set_state (pipeline, GST_STATE_PLAYING);
|
|
|
|
bus = gst_element_get_bus (pipeline);
|
|
msg = gst_bus_poll (bus, GST_MESSAGE_ERROR | GST_MESSAGE_EOS, 25*GST_SECOND);
|
|
|
|
if (msg) {
|
|
switch (GST_MESSAGE_TYPE (msg)) {
|
|
case GST_MESSAGE_EOS: {
|
|
if (result) {
|
|
GST_DEBUG ("conversion successful: result = %p", result);
|
|
} else {
|
|
GST_WARNING ("EOS but no result frame?!");
|
|
}
|
|
break;
|
|
}
|
|
case GST_MESSAGE_ERROR: {
|
|
gchar *dbg = NULL;
|
|
|
|
gst_message_parse_error (msg, &error, &dbg);
|
|
if (error) {
|
|
g_warning ("Could not take screenshot: %s", error->message);
|
|
GST_DEBUG ("%s [debug: %s]", error->message, GST_STR_NULL (dbg));
|
|
g_error_free (error);
|
|
} else {
|
|
g_warning ("Could not take screenshot (and NULL error!)");
|
|
}
|
|
g_free (dbg);
|
|
result = NULL;
|
|
break;
|
|
}
|
|
default: {
|
|
g_return_val_if_reached (NULL);
|
|
}
|
|
}
|
|
} else {
|
|
g_warning ("Could not take screenshot: %s", "timeout during conversion");
|
|
result = NULL;
|
|
}
|
|
|
|
gst_element_set_state (pipeline, GST_STATE_NULL);
|
|
gst_object_unref (pipeline);
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
typedef long BOOL;
|
|
typedef long LONG;
|
|
typedef unsigned char BYTE;
|
|
typedef unsigned int DWORD;
|
|
typedef unsigned short WORD;
|
|
|
|
#pragma pack(2)
|
|
typedef struct {
|
|
WORD bfType;
|
|
DWORD bfSize;
|
|
WORD bfReserved1;
|
|
WORD bfReserved2;
|
|
DWORD bfOffBits;
|
|
} BMPFILEHEADER_T;
|
|
|
|
struct BMPFILEHEADER_S{
|
|
WORD bfType;
|
|
DWORD bfSize;
|
|
WORD bfReserved1;
|
|
WORD bfReserved2;
|
|
DWORD bfOffBits;
|
|
};
|
|
|
|
typedef struct{
|
|
DWORD biSize;
|
|
LONG biWidth;
|
|
LONG biHeight;
|
|
WORD biPlanes;
|
|
WORD biBitCount;
|
|
DWORD biCompression;
|
|
DWORD biSizeImage;
|
|
LONG biXPelsPerMeter;
|
|
LONG biYPelsPerMeter;
|
|
DWORD biClrUsed;
|
|
DWORD biClrImportant;
|
|
} BMPINFOHEADER_T;
|
|
#pragma pack()
|
|
|
|
gboolean gst_save_bmp( char* gst_buffer, int width, int height, int red_mask, int green_mask, int blue_mask, char* filename )
|
|
{
|
|
int i = 0;
|
|
int j = 0;
|
|
int size = ((((width*24)+31)&(~31))/8)*height; // 24 bit per pixel
|
|
char stuff[3] = {0,0,0};
|
|
int stuff_number = width % 4;
|
|
|
|
FILE * fp = fopen( filename,"wb" );
|
|
if( !fp ) return FALSE;
|
|
|
|
// First part of bitmap, file header
|
|
BMPFILEHEADER_T bfh;
|
|
bfh.bfType = 0x4d42; //bm
|
|
bfh.bfSize = size // data size
|
|
+ sizeof( BMPFILEHEADER_T ) // first section size
|
|
+ sizeof( BMPINFOHEADER_T ) // second section size
|
|
;
|
|
bfh.bfReserved1 = 0; // reserved
|
|
bfh.bfReserved2 = 0; // reserved
|
|
bfh.bfOffBits = bfh.bfSize - size;
|
|
|
|
// Second part of bitmap, bitmap information header
|
|
BMPINFOHEADER_T bih;
|
|
bih.biSize = sizeof(BMPINFOHEADER_T);
|
|
bih.biWidth = width;
|
|
bih.biHeight = height;
|
|
bih.biPlanes = 1;
|
|
bih.biBitCount = 24;
|
|
bih.biCompression = 0;
|
|
bih.biSizeImage = size;
|
|
bih.biXPelsPerMeter = 0;
|
|
bih.biYPelsPerMeter = 0;
|
|
bih.biClrUsed = 0;
|
|
bih.biClrImportant = 0;
|
|
|
|
fwrite( &bfh, 1, sizeof(BMPFILEHEADER_T), fp );
|
|
fwrite( &bih, 1, sizeof(BMPINFOHEADER_T), fp );
|
|
|
|
for( j=height-1; j>=0; j-- )
|
|
{
|
|
for( i=0; i<width; i++ )
|
|
{
|
|
fwrite(gst_buffer+(j*width+i)*3+stuff_number*j+2,1,1,fp);
|
|
fwrite(gst_buffer+(j*width+i)*3+stuff_number*j+1,1,1,fp);
|
|
fwrite(gst_buffer+(j*width+i)*3+stuff_number*j+0,1,1,fp);
|
|
}
|
|
if( 0 != stuff_number )
|
|
{
|
|
fwrite(gst_buffer+(j*width+i)*3+stuff_number*j+3, 1, stuff_number, fp);
|
|
}
|
|
}
|
|
/*for( i=width*height-1; i>=0; i-- )
|
|
{
|
|
fwrite(gst_buffer+i*4,3,1,fp);
|
|
}
|
|
for( i=0; i<width*height; i++ )
|
|
{
|
|
fwrite(gst_buffer+i*4,3,1,fp);
|
|
}*/
|
|
|
|
fclose( fp );
|
|
return TRUE;
|
|
}
|
|
|
|
|