ardour/gtk2_ardour/utils_videotl.cc

377 lines
9.5 KiB
C++

/*
Copyright (C) 2010-2013 Paul Davis
Author: Robin Gareus <robin@gareus.org>
This program 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 program 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.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include <cstdio>
#include <string>
#include <cerrno>
#include <gtkmm.h>
#include <curl/curl.h>
#include "pbd/error.h"
#include "ardour/ardour.h"
#include "ardour/session_directory.h"
#include "video_image_frame.h"
#include "utils_videotl.h"
#ifdef WAF_BUILD
#include "gtk2ardour-version.h"
#endif
#ifndef ARDOUR_CURL_TIMEOUT
#define ARDOUR_CURL_TIMEOUT (60)
#endif
#include "i18n.h"
using namespace Gtk;
using namespace std;
using namespace PBD;
using namespace ARDOUR;
using namespace VideoUtils;
bool
VideoUtils::confirm_video_outfn (std::string outfn, std::string docroot)
{
/* replace docroot's '/' to G_DIR_SEPARATOR for the comparison */
size_t look_here = 0;
size_t found_here;
const char ds = G_DIR_SEPARATOR;
while((found_here = docroot.find('/', look_here)) != string::npos) {
docroot.replace(found_here, 1, std::string(&ds, 1));
look_here = found_here + 1;
}
if (!docroot.empty() && docroot.compare(0, docroot.length(), outfn, 0, docroot.length())) {
ArdourDialog confirm (_("Destination is outside Video Server's docroot. "), true);
Label m (_("The destination file path is outside of the Video Server's docroot. The file will not be readable by the Video Server. Do you still want to continue?"));
confirm.get_vbox()->pack_start (m, true, true);
confirm.add_button (Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
confirm.add_button (_("Continue"), Gtk::RESPONSE_ACCEPT);
confirm.show_all ();
if (confirm.run() == RESPONSE_CANCEL) { return false; }
}
if (Glib::file_test(outfn, Glib::FILE_TEST_EXISTS)) {
ArdourDialog confirm (_("Confirm Overwrite"), true);
Label m (_("A file with the same name already exists. Do you want to overwrite it?"));
confirm.get_vbox()->pack_start (m, true, true);
confirm.add_button (Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
confirm.add_button (_("Overwrite"), Gtk::RESPONSE_ACCEPT);
confirm.show_all ();
if (confirm.run() == RESPONSE_CANCEL) { return false; }
}
std::string dir = Glib::path_get_dirname (outfn);
if (g_mkdir_with_parents (dir.c_str(), 0755) < 0) {
error << string_compose(_("Cannot create video folder \"%1\" (%2)"), dir, strerror (errno)) << endmsg;
return false;
}
return true;
}
std::string
VideoUtils::video_dest_dir (const std::string sessiondir, const std::string docroot)
{
std::string dir = docroot;
if (dir.empty() || !dir.compare(0, dir.length(), sessiondir, 0, dir.length())) {
dir=sessiondir;
}
if ((dir.empty() || dir.at(dir.length()-1) != G_DIR_SEPARATOR)) { dir += G_DIR_SEPARATOR; }
if (g_mkdir_with_parents (dir.c_str(), 0755) < 0) {
error << string_compose(_("Cannot create video folder \"%1\" (%2)"), dir, strerror (errno)) << endmsg;
}
return dir;
}
std::string
VideoUtils::video_get_docroot (ARDOUR::RCConfiguration* config)
{
if (config->get_video_advanced_setup()) {
return config->get_video_server_docroot();
}
#ifndef PLATFORM_WINDOWS
return X_("/");
#else
return X_("C:\\");
#endif
}
std::string
VideoUtils::video_get_server_url (ARDOUR::RCConfiguration* config)
{
if (config->get_video_advanced_setup()) {
return config->get_video_server_url();
}
return X_("http://localhost:1554");
}
std::string
VideoUtils::strip_file_extension (const std::string infile)
{
std::string rv;
char *ext, *bn = strdup(infile.c_str());
if ((ext=strrchr(bn, '.'))) {
if (!strchr(ext, G_DIR_SEPARATOR)) {
*ext = 0;
}
}
rv = std::string(bn);
free(bn);
return rv;
}
std::string
VideoUtils::get_file_extension (const std::string infile)
{
std::string rv = "";
char *ext, *bn = strdup(infile.c_str());
if ((ext=strrchr(bn, '.'))) {
if (!strchr(ext, G_DIR_SEPARATOR)) {
rv=std::string(ext+1);
}
}
free(bn);
return rv;
}
std::string
VideoUtils::video_dest_file (const std::string dir, const std::string infile)
{
return Glib::build_filename(dir, strip_file_extension(Glib::path_get_basename(infile)) + ".avi");
}
std::string
VideoUtils::video_map_path (std::string server_docroot, std::string filepath)
{
std::string rv = filepath;
/* strip docroot */
if (server_docroot.length() > 0) {
if (rv.compare(0, server_docroot.length(), server_docroot) == 0 ) {
rv = rv.substr(server_docroot.length());
}
}
/* replace all G_DIR_SEPARATOR with '/' */
size_t look_here = 0;
size_t found_here;
while((found_here = rv.find(G_DIR_SEPARATOR, look_here)) != string::npos) {
rv.replace(found_here, 1, "/");
look_here = found_here + 1;
}
CURL *curl;
char *ue;
curl = curl_easy_init();
ue = curl_easy_escape(curl, rv.c_str(),rv.length());
if (ue) {
rv = std::string(ue);
curl_free(ue);
}
curl_easy_cleanup(curl);
return rv;
}
void
VideoUtils::ParseCSV (const std::string &csv, std::vector<std::vector<std::string> > &lines)
{
bool inQuote(false);
bool newLine(false);
std::string field;
lines.clear();
std::vector<std::string> line;
std::string::const_iterator aChar = csv.begin();
while (aChar != csv.end()) {
switch (*aChar) {
case '"':
newLine = false;
inQuote = !inQuote;
break;
case ',':
newLine = false;
if (inQuote == true) {
field += *aChar;
} else {
line.push_back(field);
field.clear();
}
break;
case '\n':
case '\r':
if (inQuote == true) {
field += *aChar;
} else {
if (newLine == false) {
line.push_back(field);
lines.push_back(line);
field.clear();
line.clear();
newLine = true;
}
}
break;
default:
newLine = false;
field.push_back(*aChar);
break;
}
aChar++;
}
if (field.size())
line.push_back(field);
if (line.size())
lines.push_back(line);
}
bool
VideoUtils::video_query_info (
std::string video_server_url,
std::string filepath,
double &video_file_fps,
long long int &video_duration,
double &video_start_offset,
double &video_aspect_ratio
)
{
char url[2048];
snprintf(url, sizeof(url), "%s%sinfo/?file=%s&format=csv"
, video_server_url.c_str()
, (video_server_url.length()>0 && video_server_url.at(video_server_url.length()-1) == '/')?"":"/"
, filepath.c_str());
char *res = a3_curl_http_get(url, NULL);
if (!res) {
return false;
}
std::vector<std::vector<std::string> > lines;
ParseCSV(std::string(res), lines);
free(res);
if (lines.empty() || lines.at(0).empty() || lines.at(0).size() != 6) {
return false;
}
if (atoi(lines.at(0).at(0)) != 1) return false; // version
video_start_offset = 0.0;
video_aspect_ratio = atof (lines.at(0).at(3));
video_file_fps = atof (lines.at(0).at(4));
video_duration = atoll(lines.at(0).at(5));
return true;
}
void
VideoUtils::video_draw_cross (Glib::RefPtr<Gdk::Pixbuf> img)
{
int rowstride = img->get_rowstride();
int n_channels = img->get_n_channels();
guchar *pixels, *p;
pixels = img->get_pixels();
int x,y;
int clip_width = img->get_width();
int clip_height = img->get_height();
for (x=0;x<clip_width;x++) {
y = clip_height * x / clip_width;
p = pixels + y * rowstride + x * n_channels;
p[0] = 192; p[1] = 192; p[2] = 192;
if (n_channels>3) p[3] = 255;
p = pixels + y * rowstride + (clip_width-x-1) * n_channels;
p[0] = 192; p[1] = 192; p[2] = 192;
if (n_channels>3) p[3] = 255;
}
}
extern "C" {
#include <curl/curl.h>
struct A3MemoryStruct {
char *data;
size_t size;
};
static size_t
WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data) {
size_t realsize = size * nmemb;
struct A3MemoryStruct *mem = (struct A3MemoryStruct *)data;
mem->data = (char *)realloc(mem->data, mem->size + realsize + 1);
if (mem->data) {
memcpy(&(mem->data[mem->size]), ptr, realsize);
mem->size += realsize;
mem->data[mem->size] = 0;
}
return realsize;
}
char *a3_curl_http_get (const char *u, int *status) {
CURL *curl;
CURLcode res;
struct A3MemoryStruct chunk;
long int httpstatus;
if (status) *status = 0;
//Glib::usleep(500000); return NULL; // TEST & DEBUG
if (strncmp("http://", u, 7)) return NULL;
chunk.data=NULL;
chunk.size=0;
curl = curl_easy_init();
if(!curl) return NULL;
curl_easy_setopt(curl, CURLOPT_URL, u);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
curl_easy_setopt(curl, CURLOPT_USERAGENT, PROGRAM_NAME VERSIONSTRING);
curl_easy_setopt(curl, CURLOPT_TIMEOUT, ARDOUR_CURL_TIMEOUT);
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
#ifdef CURLERRORDEBUG
char curlerror[CURL_ERROR_SIZE] = "";
curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, curlerror);
#endif
res = curl_easy_perform(curl);
curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &httpstatus);
curl_easy_cleanup(curl);
if (status) *status = httpstatus;
if (res) {
#ifdef CURLERRORDEBUG
printf("a3_curl_http_get() failed: %s\n", curlerror);
#endif
return NULL;
}
if (httpstatus != 200) {
free (chunk.data);
chunk.data = NULL;
}
return (chunk.data);
}
} /* end extern "C" */