comparison trunk/src/Modules/Output/Graphics/Devices/GraphicsOutputDeviceMovie.cc @ 397:7a573750b186

- First add of a lot of graphics code from the old version. Not working yet, not even compiling yet.

+// Copyright 2006, Willem van Engen
+//
+// AIM-C: A C++ implementation of the Auditory Image Model
+// http://www.acousticscale.org/AIMC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+/*!
+ * \file
+ * \brief Output device for output to a movie
+ *
+ * \author Willem van Engen <[e-mail]>
+ * \date created 2006/10/16
+ * \version \$Id: GraphicsOutputDeviceMovie.cpp 633 2008-09-11 04:20:16Z tom $
+ */
+
+/*! \todo
+ *  A recent experiment showed that the video was about an audioframe
+ *  (something like 30 ms) behind the audio in rdct.wav. It seems odd
+ *  to me, since I already output a frame at the beginning to compensate
+ *  for the missed buffer.
+ *  A solution that would solve this and be a broader improvement, is to
+ *  include the source's time when Fire()ing. The outputdevice can then
+ *  do audio/video synchronization. In the case of the movie, the very
+ *  first gGrab() looks at the time and emits as much empty output frames
+ *  as needed until the correct signal time is reached.
+ */
+#include "Support/Common.h"
+
+#ifdef _WINDOWS
+#	include <direct.h> // for _mkdir&_rmdir
+#else
+#	include <sys/types.h>
+#	include <dirent.h> // for opendir&friends
+#endif
+#include <stdio.h>
+#include <string.h>
+
+
+#include "Output/GraphicsOutputDeviceMovie.h"
+
+GraphicsOutputDeviceMovie::GraphicsOutputDeviceMovie(Parameters *pParam)
+	: GraphicsOutputDeviceCairo(pParam) {  // or GraphicsOutputDevicePlotutils
+	m_sMovieFile[0] = '\0';
+	m_sSoundFile[0] = '\0';
+}
+
+bool GraphicsOutputDeviceMovie::Initialize(const char *sSoundFile,
+                                           const char *sMovieFile) {
+	FILE *f;
+	AIM_ASSERT(sSoundFile);
+	AIM_ASSERT(sMovieFile);
+
+	// Check sound file exists
+	if ((f = fopen(sSoundFile, "r")) == NULL) {
+		AIM_ERROR(_T("Couldn't open sound file '%s' for movie creation."),
+             sSoundFile);
+		return false;
+	}
+	fclose(f);
+	strcpy(m_sSoundFile, sSoundFile);
+
+	// Check movie output file can be made
+	if ( (f=fopen(sMovieFile, "w"))==NULL ) {
+		aimERROR(_T("Couldn't open movie file '%s' to write to."),
+             sMovieFile);
+		return false;
+	}
+	fclose(f);
+	strcpy(m_sMovieFile, sMovieFile);
+
+	// Get a temporary image output directory
+	//! \warning Not really safe ... but windows has no mkdtemp()
+	//! \todo Make build system check for mkdtemp() to use it when available. See TODO.txt.
+	char *sTmpDir = NULL;
+#ifdef _WINDOWS
+	if ((sTmpDir = _tempnam(NULL, AIM_NAME))
+      && _mkdir(sTmpDir) >= 0) {
+		strcpy(m_sDir, sTmpDir);
+		strcat(m_sDir, "\\"); // Make sure to end with trailing slash
+	} else
+#else
+	strcpy(m_sDir, "/tmp/"AIM_NAME"-movie.XXXXXX");
+	if (mkdtemp(m_sDir)) {
+		strcat(m_sDir, "/"); // Make sure to end with trailing slash
+	} else
+#endif
+	{
+		AIM_ERROR(_T("Couldn't create a temporary directory for movie output."));
+		if (sTmpDir) free(sTmpDir);
+		return false;
+	}
+	if (sTmpDir) {
+    free(sTmpDir);
+  }
+
+	// We want png for movie conversion
+	//! \bug This may change the user preference in GUI, hmm what to do? See TODO.txt
+	m_pParam->SetString("output.img.format", "png");
+	//if ( !GraphicsOutputDevicePlotutils::Initialize(m_sDir) ) {
+	if ( !GraphicsOutputDeviceCairo::Initialize(m_sDir) ) {
+		return false;
+  }
+
+	return true;
+}
+
+void GraphicsOutputDeviceMovie::Start() {
+	//GraphicsOutputDevicePlotutils::Start();
+	GraphicsOutputDeviceCairo::Start();
+	// Just output a single frame to get audio/video in sync, put params in there
+	gGrab();
+	PlotParameterScreen();
+	gRelease();
+}
+
+void GraphicsOutputDeviceMovie::Stop() {
+	// Make sure Plotutils is really done writing.
+	//GraphicsOutputDevicePlotutils::Stop();
+	GraphicsOutputDeviceCairo::Stop();
+	CloseFile();
+
+#ifdef __WX__
+	// GUI only: popup dialog
+#else
+	printf("Generating movie ... \n");
+#endif
+	AIM_ASSERT(m_pParam);
+	// Convert images and sound file to a movie
+	//! \warning Movie files are overwritten without warning
+	//! \bug ffmpeg only works with colour images, not with bw. So make sure to not use bw only in drawing..
+	// Always convert to audio stream of 44.1kHz or problems may occur in playing or conversio.
+	float fFps = 1000.0 / m_pParam->GetFloat("output.frameperiod");
+	char sffmpegPath[1024];
+	if (!m_pParam->IsSet("output.ffmpeg_path")) {
+  strcpy(sffmpegPath,"ffmpeg");
+  } else {
+  	strcpy(sffmpegPath, m_pParam->GetString("output.ffmpeg_path"));
+  }
+  char sCodecOptions[1024];
+	if (!m_pParam->IsSet("output.ffmpeg_codec_options")) {
+    strcpy(sCodecOptions,"");
+  } else {
+	  strcpy(sCodecOptions, m_pParam->GetString("output.ffmpeg_codec_options"));
+  }
+
+	char sCmdLine[1024]; //!\todo check that snprintf does not want a larger buffer
+	snprintf(sCmdLine, sizeof(sCmdLine)/sizeof(sCmdLine[0]),
+		"%s -r %.2f -y -i \"%s\" -i \"%s%%06d.png\" "
+		"-title \"%s\" -comment \"Generated by "AIM_NAME" "AIM_VERSION_STRING"\" "
+		"-sameq -r %.2f -ar 44100 -acodec pcm_s16le %s \"%s\"",
+		sffmpegPath, fFps, m_sSoundFile, m_sDir,
+		m_pParam->GetString("output.movie.title"),
+		fFps, sCodecOptions, m_sMovieFile);
+    printf(sCmdLine);
+    printf("\n");
+	if (system(sCmdLine)) {
+		AIM_ERROR(_T("Couldn't create movie output."));
+	}
+
+#ifdef __WX__
+	// GUI only: close dialog again
+#endif
+	// Remove files in temporary directory and the dir itself
+	//! \todo make portable function, possibly decided on by build system
+#ifdef _WINDOWS
+	HANDLE hList;
+	WIN32_FIND_DATA FileData;
+	snprintf(sCmdLine, sizeof(sCmdLine)/sizeof(sCmdLine[0]), "%s/*.*", m_sDir);
+	if ((hList = FindFirstFile(sCmdLine, &FileData)) == INVALID_HANDLE_VALUE) {
+		AIM_ERROR(_T("Couldn't remove files from temporary directory."));
+		return;
+	}
+	bool bRMfinished = false;
+	while (!bRMfinished) {
+	  snprintf(sCmdLine,
+             sizeof(sCmdLine)/sizeof(sCmdLine[0]),
+			       "%s%s",
+             m_sDir,
+             FileData.cFileName);
+		remove(sCmdLine);
+		if (!FindNextFile(hList, &FileData) && GetLastError() == ERROR_NO_MORE_FILES) {
+			bRMfinished = true;
+    }
+	}
+	FindClose(hList);
+	_rmdir(m_sDir);
+#else
+	DIR *dir;
+	struct dirent *dirent;
+	if (!(dir = opendir(m_sDir))) {
+		AIM_ERROR(_T("Couldn't remove files in temporary directory."));
+		return;
+	}
+	while (dirent = readdir(dir)) {
+		snprintf(sCmdLine,
+             sizeof(sCmdLine)/sizeof(sCmdLine[0]),
+			       "%s%s",
+             m_sDir,
+             dirent->d_name);
+		unlink(sCmdLine);
+	}
+	closedir(dir);
+	rmdir(m_sDir);
+#endif
+}
+
+void GraphicsOutputDeviceMovie::PlotParameterScreen() {
+	AIM_ASSERT(m_pParam);
+	char sStr[50];
+	int lineno = 1;
+
+	float fMarL = m_pParam->GetFloat(_S("graph.margin.left"));
+	float fMarT = m_pParam->GetFloat(_S("graph.margin.top"));
+	float fTextHeight = 1.0f / 50.0f * 1.2; // change this when fontsizing is there!
+
+	gText2f(fMarL, 1-(fMarT+fTextHeight*lineno++),
+          _S("AIM-C"));
+	gText2f(fMarL,
+          1-(fMarT+fTextHeight*lineno++),
+          _S("(c) 2006-2010, Thomas Walters, Willem van Engen"));
+	gText2f(fMarL,
+          1-(fMarT+fTextHeight*lineno++),
+          _S("http://aimc.acousticscale.org/"));
+	lineno++;
+
+	static const char *pPlotParams[] = {
+		_S("input.buffersize"),
+		_S("input.samplerate"),
+		_S("bmm.freqstart"),
+		_S("bmm.freqend"),
+		_S("bmm.numchannels"),
+		_S("preset.name"),
+		_S("preset.title"),
+		NULL
+	};
+	for (int i = 0; pPlotParams[i]; i++) {
+		snprintf(sStr,
+             sizeof(sStr)/sizeof(sStr[0]), _S("%s=%s"),
+			       pPlotParams[i],
+             m_pParam->GetString(pPlotParams[i]));
+		gText2f(fMarL,
+            1-(fMarT+fTextHeight*lineno++),
+            sStr);
+	}
+}