Blender Frameserver

/* * ***** BEGIN GPL LICENSE BLOCK ***** * * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * * Copyright (c) 2006 Peter Schlaile * * Contributor(s): * * ***** END GPL LICENSE BLOCK ***** */ /** \file blender/blenkernel/intern/writeframeserver.c * \ingroup bke * * Frameserver * Makes Blender accessible from TMPGenc directly using VFAPI (you can * use firefox too ;-) */ #ifdef WITH_FRAMESERVER #include #include #if defined(_WIN32) #include #include #include #include #else #include #include #include #include #include #include #include #include #include #include #include #include #endif #include #include "DNA_userdef_types.h" #include "BLI_utildefines.h" #include "BKE_writeframeserver.h" #include "BKE_global.h" #include "BKE_report.h" #include "DNA_scene_types.h" #include "MEM_guardedalloc.h" typedef struct FrameserverContext { int sock; int connsock; int write_ppm; int render_width; int render_height; } FrameserverContext; #if defined(_WIN32) static int startup_socket_system(void) { WSADATA wsa; return (WSAStartup(MAKEWORD(2, 0), &wsa) == 0); } static void shutdown_socket_system(void) { WSACleanup(); } static int select_was_interrupted_by_signal(void) { return (WSAGetLastError() == WSAEINTR); } #else static int startup_socket_system(void) { return 1; } static void shutdown_socket_system(void) { } static int select_was_interrupted_by_signal(void) { return (errno == EINTR); } static int closesocket(int fd) { return close(fd); } #endif int BKE_frameserver_start(void *context_v, struct Scene *scene, RenderData *UNUSED(rd), int rectx, int recty, ReportList *reports, bool UNUSED(preview), const char *UNUSED(suffix)) { struct sockaddr_in addr; int arg = 1; FrameserverContext *context = context_v; (void)scene; /* unused */ if (!startup_socket_system()) { BKE_report(reports, RPT_ERROR, "Cannot startup socket system"); return 0; } if ((context->sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) { shutdown_socket_system(); BKE_report(reports, RPT_ERROR, "Cannot open socket"); return 0; } setsockopt(context->sock, SOL_SOCKET, SO_REUSEADDR, (char *) &arg, sizeof(arg)); addr.sin_family = AF_INET; addr.sin_port = htons(U.frameserverport); addr.sin_addr.s_addr = INADDR_ANY; if (bind(context->sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) { shutdown_socket_system(); BKE_report(reports, RPT_ERROR, "Cannot bind to socket"); return 0; } if (listen(context->sock, SOMAXCONN) < 0) { shutdown_socket_system(); BKE_report(reports, RPT_ERROR, "Cannot establish listen backlog"); return 0; } context->connsock = -1; context->render_width = rectx; context->render_height = recty; return 1; } static char index_page[] = "HTTP/1.1 200 OK\r\n" "Content-Type: text/html\r\n" "\r\n" "Blender Frameserver\n" "

\n"
"Blender Frameserver\n"
"Render Info
\n"
"Stop Rendering
\n"
"\n"
"Images can be found here\n"
"\n"
"images/ppm/%d.ppm\n"
"\n"
"

\n"; static char good_bye[] = "HTTP/1.1 200 OK\r\n" "Content-Type: text/html\r\n" "\r\n" "Blender Frameserver\n" "

\n"
"Render stopped. Goodbye

"; static int safe_write(const int connsock, char *s, int tosend) { int total = tosend; do { int got = send(connsock, s, tosend, 0); if (got < 0) { return got; } tosend -= got; s += got; } while (tosend > 0); return total; } static int safe_puts(const int connsock, char *s) { return safe_write(connsock, s, strlen(s)); } static int handle_request(FrameserverContext *context, RenderData *rd, char *req) { char *p; char *path; int pathlen; if (memcmp(req, "GET ", 4) != 0) { return -1; } p = req + 4; path = p; while (*p != ' ' && *p) p++; *p = 0; if (STREQ(path, "/index.html") || strcmp(path, "/")) { safe_puts(context->connsock, index_page); return -1; } context->write_ppm = 0; pathlen = strlen(path); if (pathlen > 12 && memcmp(path, "/images/ppm/", 12) == 0) { context->write_ppm = 1; return atoi(path + 12); } if (STREQ(path, "/info.txt")) { char buf[4096]; sprintf(buf, "HTTP/1.1 200 OK\r\n" "Content-Type: text/html\r\n" "\r\n" "start %d\n" "end %d\n" "width %d\n" "height %d\n" "rate %d\n" "ratescale %d\n", rd->sfra, rd->efra, context->render_width, context->render_height, rd->frs_sec, 1 ); safe_puts(context->connsock, buf); return -1; } if (STREQ(path, "/close.txt")) { safe_puts(context->connsock, good_bye); G.is_break = true; /* Abort render */ return -1; } return -1; } int BKE_frameserver_loop(void *context_v, RenderData *rd, ReportList *UNUSED(reports)) { fd_set readfds; struct timeval tv; struct sockaddr_in addr; int len, rval; #ifdef FREE_WINDOWS int socklen; #else unsigned int socklen; #endif char buf[4096]; FrameserverContext *context = context_v; if (context->connsock != -1) { closesocket(context->connsock); context->connsock = -1; } tv.tv_sec = 1; tv.tv_usec = 0; FD_ZERO(&readfds); FD_SET(context->sock, &readfds); rval = select(context->sock + 1, &readfds, NULL, NULL, &tv); if (rval < 0) { return -1; } if (rval == 0) { /* nothing to be done */ return -1; } socklen = sizeof(addr); if ((context->connsock = accept(context->sock, (struct sockaddr *)&addr, &socklen)) < 0) { return -1; } FD_ZERO(&readfds); FD_SET(context->connsock, &readfds); for (;;) { /* give 10 seconds for telnet testing... */ tv.tv_sec = 10; tv.tv_usec = 0; rval = select(context->connsock + 1, &readfds, NULL, NULL, &tv); if (rval > 0) { break; } else if (rval == 0) { return -1; } else if (rval < 0) { if (!select_was_interrupted_by_signal()) { return -1; } } } len = recv(context->connsock, buf, sizeof(buf) - 1, 0); if (len < 0) { return -1; } buf[len] = 0; return handle_request(context, rd, buf); } static void serve_ppm(FrameserverContext *context, int *pixels, int rectx, int recty) { unsigned char *rendered_frame; unsigned char *row = (unsigned char *) malloc(context->render_width * 3); int y; char header[1024]; sprintf(header, "HTTP/1.1 200 OK\r\n" "Content-Type: image/ppm\r\n" "Connection: close\r\n" "\r\n" "P6\n" "# Creator: blender frameserver v0.0.1\n" "%d %d\n" "255\n", rectx, recty); safe_puts(context->connsock, header); rendered_frame = (unsigned char *)pixels; for (y = recty - 1; y >= 0; y--) { unsigned char *target = row; unsigned char *src = rendered_frame + rectx * 4 * y; unsigned char *end = src + rectx * 4; while (src != end) { target[2] = src[2]; target[1] = src[1]; target[0] = src[0]; target += 3; src += 4; } safe_write(context->connsock, (char *)row, 3 * rectx); } free(row); closesocket(context->connsock); context->connsock = -1; } int BKE_frameserver_append(void *context_v, RenderData *UNUSED(rd), int UNUSED(start_frame), int frame, int *pixels, int rectx, int recty, const char *UNUSED(suffix), ReportList *UNUSED(reports)) { FrameserverContext *context = context_v; fprintf(stderr, "Serving frame: %d\n", frame); if (context->write_ppm) { serve_ppm(context, pixels, rectx, recty); } if (context->connsock != -1) { closesocket(context->connsock); context->connsock = -1; } return 1; } void BKE_frameserver_end(void *context_v) { FrameserverContext *context = context_v; if (context->connsock != -1) { closesocket(context->connsock); context->connsock = -1; } closesocket(context->sock); shutdown_socket_system(); } void *BKE_frameserver_context_create(void) { FrameserverContext *context = MEM_mallocN(sizeof(FrameserverContext), "Frameserver Context"); return context; } void BKE_frameserver_context_free(void *context_v) { FrameserverContext *context = context_v; if (context) { MEM_freeN(context); } } #endif /* WITH_FRAMESERVER */