X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=synfig-core%2Fsrc%2Fmodules%2Fmod_imagemagick%2Fmptr_imagemagick.cpp;fp=synfig-core%2Fsrc%2Fmodules%2Fmod_imagemagick%2Fmptr_imagemagick.cpp;h=2860c1f97434413f6e4689fce916294cff52aa4f;hb=a095981e18cc37a8ecc7cd237cc22b9c10329264;hp=0000000000000000000000000000000000000000;hpb=9459638ad6797b8139f1e9f0715c96076dbf0890;p=synfig.git

diff --git a/synfig-core/src/modules/mod_imagemagick/mptr_imagemagick.cpp b/synfig-core/src/modules/mod_imagemagick/mptr_imagemagick.cpp
new file mode 100644
index 0000000..2860c1f
--- /dev/null
+++ b/synfig-core/src/modules/mod_imagemagick/mptr_imagemagick.cpp
@@ -0,0 +1,299 @@
+/* === S Y N F I G ========================================================= */
+/*!	\file mptr_imagemagick.cpp
+**	\brief ppm Target Module
+**
+**	$Id$
+**
+**	\legal
+**	Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
+**	Copyright (c) 2007 Chris Moore
+**
+**	This package is free software; you can redistribute it and/or
+**	modify it under the terms of the GNU General Public License as
+**	published by the Free Software Foundation; either version 2 of
+**	the License, or (at your option) any later version.
+**
+**	This package 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
+**	General Public License for more details.
+**	\endlegal
+**
+** === N O T E S ===========================================================
+**
+** ========================================================================= */
+
+/* === H E A D E R S ======================================================= */
+
+#ifdef USING_PCH
+#	include "pch.h"
+#else
+#ifdef HAVE_CONFIG_H
+#	include <config.h>
+#endif
+
+#include <ETL/stringf>
+#include "mptr_imagemagick.h"
+#include <stdio.h>
+#include <sys/types.h>
+#if HAVE_SYS_WAIT_H
+ #include <sys/wait.h>
+#endif
+#if HAVE_IO_H
+ #include <io.h>
+#endif
+#if HAVE_PROCESS_H
+ #include <process.h>
+#endif
+#if HAVE_FCNTL_H
+ #include <fcntl.h>
+#endif
+#include <unistd.h>
+#include <algorithm>
+#include <functional>
+#include <ETL/stringf>
+#include <synfig/general.h>
+
+#endif
+
+/* === M A C R O S ========================================================= */
+
+using namespace synfig;
+using namespace std;
+using namespace etl;
+
+#if defined(HAVE_FORK) && defined(HAVE_PIPE) && defined(HAVE_WAITPID)
+ #define UNIX_PIPE_TO_PROCESSES
+#else
+ #define WIN32_PIPE_TO_PROCESSES
+#endif
+
+/* === G L O B A L S ======================================================= */
+
+SYNFIG_IMPORTER_INIT(imagemagick_mptr);
+SYNFIG_IMPORTER_SET_NAME(imagemagick_mptr,"imagemagick");
+SYNFIG_IMPORTER_SET_EXT(imagemagick_mptr,"miff");
+SYNFIG_IMPORTER_SET_VERSION(imagemagick_mptr,"0.1");
+SYNFIG_IMPORTER_SET_CVS_ID(imagemagick_mptr,"$Id$");
+
+/* === M E T H O D S ======================================================= */
+
+
+imagemagick_mptr::imagemagick_mptr(const char *f)
+{
+
+	filename=f;
+	file=NULL;
+}
+
+imagemagick_mptr::~imagemagick_mptr()
+{
+	if(file)
+		pclose(file);
+}
+
+bool
+imagemagick_mptr::get_frame(synfig::Surface &surface,Time /*time*/, synfig::ProgressCallback *cb)
+{
+//#define HAS_LIBPNG 1
+
+#if 1
+	if(filename.empty())
+	{
+		if(cb)cb->error(_("No file to load"));
+		else synfig::error(_("No file to load"));
+		return false;
+	}
+	string temp_file="/tmp/deleteme.png";
+
+#if defined(WIN32_PIPE_TO_PROCESSES)
+
+	if(file)
+		pclose(file);
+
+	string command;
+
+	if(filename.find("psd")!=String::npos)
+		command=strprintf("convert \"%s\" -flatten \"png32:%s\"\n",filename.c_str(),temp_file.c_str());
+	else
+		command=strprintf("convert \"%s\" \"png32:%s\"\n",filename.c_str(),temp_file.c_str());
+
+	if(system(command.c_str())!=0)
+		return false;
+
+#elif defined(UNIX_PIPE_TO_PROCESSES)
+
+	string output="png32:"+temp_file;
+
+	pid_t pid = fork();
+
+	if (pid == -1) {
+		return false;
+	}
+
+	if (pid == 0){
+		// Child process
+		if(filename.find("psd")!=String::npos)
+			execlp("convert", "convert", filename.c_str(), "-flatten", output.c_str(), (const char *)NULL);
+		else
+			execlp("convert", "convert", filename.c_str(), output.c_str(), (const char *)NULL);
+		// We should never reach here unless the exec failed
+		return false;
+	}
+
+	int status;
+	waitpid(pid, &status, 0);
+	if( (WIFEXITED(status) && WEXITSTATUS(status) != 0) || !WIFEXITED(status) )
+		return false;
+
+#else
+	#error There are no known APIs for creating child processes
+#endif
+
+	Importer::Handle importer(Importer::open(temp_file));
+
+	if(!importer)
+	{
+		if(cb)cb->error(_("Unable to open ")+temp_file);
+		else synfig::error(_("Unable to open ")+temp_file);
+		return false;
+	}
+
+	if(!importer->get_frame(surface,0,cb))
+	{
+		if(cb)cb->error(_("Unable to get frame from ")+temp_file);
+		else synfig::error(_("Unable to get frame from ")+temp_file);
+		return false;
+	}
+
+	if(!surface)
+	{
+		if(cb)cb->error(_("Bad surface from ")+temp_file);
+		else synfig::error(_("Bad surface from ")+temp_file);
+		return false;
+	}
+
+	if(1)
+	{
+		// remove odd premultiplication
+		for(int i=0;i<surface.get_w()*surface.get_h();i++)
+		{
+			Color c(surface[0][i]);
+
+			if(c.get_a())
+			{
+				surface[0][i].set_r(c.get_r()/c.get_a()/c.get_a());
+				surface[0][i].set_g(c.get_g()/c.get_a()/c.get_a());
+				surface[0][i].set_b(c.get_b()/c.get_a()/c.get_a());
+			}
+			else
+			{
+				surface[0][i].set_r(0);
+				surface[0][i].set_g(0);
+				surface[0][i].set_b(0);
+			}
+			surface[0][i].set_a(c.get_a());
+		}
+	}
+
+	Surface bleh(surface);
+	surface=bleh;
+
+	//remove(temp_file.c_str());
+	return true;
+
+#else
+
+#error This code contains tempfile and arbitrary shell command execution vulnerabilities
+
+	if(file)
+		pclose(file);
+
+	string command;
+
+	if(filename.empty())
+	{
+		if(cb)cb->error(_("No file to load"));
+		else synfig::error(_("No file to load"));
+		return false;
+	}
+
+	command=strprintf("convert \"%s\" -flatten ppm:-\n",filename.c_str());
+
+	file=popen(command.c_str(),POPEN_BINARY_READ_TYPE);
+
+	if(!file)
+	{
+		if(cb)cb->error(_("Unable to open pipe to imagemagick"));
+		else synfig::error(_("Unable to open pipe to imagemagick"));
+		return false;
+	}
+	int w,h;
+	float divisor;
+	char cookie[2];
+
+	while((cookie[0]=fgetc(file))!='P' && !feof(file));
+
+	if(feof(file))
+	{
+		if(cb)cb->error(_("Reached end of stream without finding PPM header"));
+		else synfig::error(_("Reached end of stream without finding PPM header"));
+		return false;
+	}
+
+	cookie[1]=fgetc(file);
+
+	if(cookie[0]!='P' || cookie[1]!='6')
+	{
+		if(cb)cb->error(string(_("stream not in PPM format"))+" \""+cookie[0]+cookie[1]+'"');
+		else synfig::error(string(_("stream not in PPM format"))+" \""+cookie[0]+cookie[1]+'"');
+		return false;
+	}
+
+	fgetc(file);
+	fscanf(file,"%d %d\n",&w,&h);
+	fscanf(file,"%f",&divisor);
+	fgetc(file);
+
+	if(feof(file))
+	{
+		if(cb)cb->error(_("Premature end of file (after header)"));
+		else synfig::error(_("Premature end of file (after header)"));
+		return false;
+	}
+
+	int x;
+	int y;
+	frame.set_wh(w,h);
+	for(y=0;y<frame.get_h();y++)
+		for(x=0;x<frame.get_w();x++)
+		{
+			if(feof(file))
+			{
+				if(cb)cb->error(_("Premature end of file"));
+				else synfig::error(_("Premature end of file"));
+				return false;
+			}
+			float b=gamma().r_U8_to_F32((unsigned char)fgetc(file));
+			float g=gamma().g_U8_to_F32((unsigned char)fgetc(file));
+			float r=gamma().b_U8_to_F32((unsigned char)fgetc(file));
+/*
+			float b=(float)(unsigned char)fgetc(file)/divisor;
+			float g=(float)(unsigned char)fgetc(file)/divisor;
+			float r=(float)(unsigned char)fgetc(file)/divisor;
+*/
+			frame[y][x]=Color(
+				b,
+				g,
+				r,
+				1.0
+			);
+		}
+
+	surface=frame;
+
+	return true;
+#endif
+
+
+}