summaryrefslogtreecommitdiff
path: root/src/lib
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib')
-rw-r--r--src/lib/cross.cc76
-rw-r--r--src/lib/cross.h4
-rw-r--r--src/lib/film.cc18
3 files changed, 80 insertions, 18 deletions
diff --git a/src/lib/cross.cc b/src/lib/cross.cc
index 7fbba86f9..cdb7f2819 100644
--- a/src/lib/cross.cc
+++ b/src/lib/cross.cc
@@ -20,11 +20,14 @@
#include <fstream>
#include <boost/algorithm/string.hpp>
#include "cross.h"
+#include "log.h"
#ifdef DVDOMATIC_POSIX
#include <unistd.h>
#endif
#ifdef DVDOMATIC_WINDOWS
-#include "windows.h"
+#include <windows.h>
+#undef DATADIR
+#include <shlwapi.h>
#endif
#ifdef DVDOMATIC_OSX
#include <sys/sysctl.h>
@@ -33,6 +36,7 @@
using std::pair;
using std::ifstream;
using std::string;
+using boost::shared_ptr;
void
dvdomatic_sleep (int s)
@@ -81,3 +85,73 @@ cpu_info ()
return info;
}
+void
+run_ffprobe (boost::filesystem::path content, boost::filesystem::path out, shared_ptr<Log> log)
+{
+#ifdef DVDOMATIC_WINDOWS
+ SECURITY_ATTRIBUTES security;
+ security.nLength = sizeof (security);
+ security.bInheritHandle = TRUE;
+ security.lpSecurityDescriptor = 0;
+
+ HANDLE child_stderr_read;
+ HANDLE child_stderr_write;
+ if (!CreatePipe (&child_stderr_read, &child_stderr_write, &security, 0)) {
+ log->log ("ffprobe call failed (could not CreatePipe)");
+ return;
+ }
+
+ wchar_t dir[512];
+ GetModuleFileName (GetModuleHandle (0), dir, sizeof (dir));
+ PathRemoveFileSpec (dir);
+ SetCurrentDirectory (dir);
+
+ STARTUPINFO startup_info;
+ ZeroMemory (&startup_info, sizeof (startup_info));
+ startup_info.cb = sizeof (startup_info);
+ startup_info.hStdError = child_stderr_write;
+ startup_info.dwFlags |= STARTF_USESTDHANDLES;
+
+ wchar_t command[512];
+ wcscpy (command, L"ffprobe.exe ");
+
+ wchar_t file[512];
+ MultiByteToWideChar (CP_UTF8, 0, content.string().c_str(), -1, file, sizeof(file));
+ wcscat (command, file);
+
+ PROCESS_INFORMATION process_info;
+ ZeroMemory (&process_info, sizeof (process_info));
+ if (!CreateProcess (0, command, 0, 0, TRUE, CREATE_NO_WINDOW, 0, 0, &startup_info, &process_info)) {
+ log->log ("ffprobe call failed (could not CreateProcess)");
+ return;
+ }
+
+ FILE* o = fopen (out.string().c_str(), "w");
+ if (!o) {
+ log->log ("ffprobe call failed (could not create output file)");
+ return;
+ }
+
+ CloseHandle (child_stderr_write);
+
+ while (1) {
+ char buffer[512];
+ DWORD read;
+ if (!ReadFile(child_stderr_read, buffer, sizeof(buffer), &read, 0) || read == 0) {
+ break;
+ }
+ fwrite (buffer, read, 1, o);
+ }
+
+ fclose (o);
+
+ WaitForSingleObject (process_info.hProcess, INFINITE);
+ CloseHandle (process_info.hProcess);
+ CloseHandle (process_info.hThread);
+ CloseHandle (child_stderr_read);
+#else
+ string ffprobe = "ffprobe 2> \"" + c + "\" 2> \"" + file ("ffprobe.log");
+ log->log (String::compose ("Probing with %1", ffprobe));
+ system (ffprobe.c_str ());
+#endif
+}
diff --git a/src/lib/cross.h b/src/lib/cross.h
index 970bf3e9d..1a7a8cb4d 100644
--- a/src/lib/cross.h
+++ b/src/lib/cross.h
@@ -18,6 +18,9 @@
*/
#include <string>
+#include <boost/filesystem.hpp>
+
+class Log;
#ifdef DVDOMATIC_WINDOWS
#define WEXITSTATUS(w) (w)
@@ -25,3 +28,4 @@
void dvdomatic_sleep (int);
extern std::pair<std::string, int> cpu_info ();
+extern void run_ffprobe (boost::filesystem::path, boost::filesystem::path, boost::shared_ptr<Log>);
diff --git a/src/lib/film.cc b/src/lib/film.cc
index cb66e2783..3d9b3eeb4 100644
--- a/src/lib/film.cc
+++ b/src/lib/film.cc
@@ -25,10 +25,6 @@
#include <sstream>
#include <iomanip>
#include <unistd.h>
-#ifdef DVDOMATIC_WINDOWS
-#undef DATADIR
-#include <shlwapi.h>
-#endif
#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>
#include <boost/lexical_cast.hpp>
@@ -1003,19 +999,7 @@ Film::set_content (string c)
set_use_content_audio (false);
}
-#ifdef DVDOMATIC_WINDOWS
- wchar_t dir[512];
- GetModuleFileName (GetModuleHandle (0), dir, sizeof (dir));
- PathRemoveFileSpec (dir);
- SetCurrentDirectory (dir);
- string ffprobe = "ffprobe.exe ";
-#else
- string ffprobe = "ffprobe ";
-#endif
- ffprobe += "\"" + c + "\"";
- ffprobe += " 2> \"" + file ("ffprobe.log") + "\"";
- log()->log (String::compose ("Probing with %1", ffprobe));
- system (ffprobe.c_str ());
+ run_ffprobe (c, file ("ffprobe.log"), _log);
}
void