src/video_output/snapshot.c

   1 /*****************************************************************************
   2  * snapshot.c : vout internal snapshot
   3  *****************************************************************************
   4  * Copyright (C) 2009 Laurent Aimar
   5  * $Id$
   6  *
   7  * Authors: Gildas Bazin <gbazin _AT_ videolan _DOT_ org>
   8  *          Laurent Aimar <fenrir _AT_ videolan _DOT_ org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 #ifdef HAVE_CONFIG_H
  26 # include "config.h"
  27 #endif
  28
  29 #include <vlc_common.h>
  30 #include <vlc_charset.h>
  31 #include <vlc_strings.h>
  32 #include <vlc_block.h>
  33
  34 #include "snapshot.h"
  35
  36 #include <sys/stat.h>
  37 #include <sys/types.h>
  38 #include <dirent.h>
  39 #include <time.h>
  40
  41 /* */
  42 void vout_snapshot_Init(vout_snapshot_t *snap)
  43 {
  44     vlc_mutex_init(&snap->lock);
  45     vlc_cond_init(&snap->wait);
  46
  47     snap->is_available = true;
  48     snap->request_count = 0;
  49     snap->picture = NULL;
  50 }
  51 void vout_snapshot_Clean(vout_snapshot_t *snap)
  52 {
  53     picture_t *picture = snap->picture;
  54     while (picture) {
  55         picture_t *next = picture->p_next;
  56         picture_Release(picture);
  57         picture = next;
  58     }
  59
  60     vlc_cond_destroy(&snap->wait);
  61     vlc_mutex_destroy(&snap->lock);
  62 }
  63
  64 void vout_snapshot_End(vout_snapshot_t *snap)
  65 {
  66     vlc_mutex_lock(&snap->lock);
  67
  68     snap->is_available = false;
  69
  70     vlc_cond_broadcast(&snap->wait);
  71     vlc_mutex_unlock(&snap->lock);
  72 }
  73
  74 /* */
  75 picture_t *vout_snapshot_Get(vout_snapshot_t *snap, mtime_t timeout)
  76 {
  77     vlc_mutex_lock(&snap->lock);
  78
  79     /* */
  80     snap->request_count++;
  81
  82     /* */
  83     const mtime_t deadline = mdate() + timeout;
  84     while (snap->is_available && !snap->picture && mdate() < deadline)
  85         vlc_cond_timedwait(&snap->wait, &snap->lock, deadline);
  86
  87     /* */
  88     picture_t *picture = snap->picture;
  89     if (picture)
  90         snap->picture = picture->p_next;
  91     else if (snap->request_count > 0)
  92         snap->request_count--;
  93
  94     vlc_mutex_unlock(&snap->lock);
  95
  96     return picture;
  97 }
  98
  99 /* */
 100 bool vout_snapshot_IsRequested(vout_snapshot_t *snap)
 101 {
 102     bool has_request = false;
 103     if (!vlc_mutex_trylock(&snap->lock)) {
 104         has_request = snap->request_count > 0;
 105         vlc_mutex_unlock(&snap->lock);
 106     }
 107     return has_request;
 108 }
 109 void vout_snapshot_Set(vout_snapshot_t *snap,
 110                        const video_format_t *fmt,
 111                        const picture_t *picture)
 112 {
 113     if (!fmt)
 114         fmt = &picture->format;
 115
 116     vlc_mutex_lock(&snap->lock);
 117     while (snap->request_count > 0) {
 118         picture_t *dup = picture_NewFromFormat(fmt);
 119         if (!dup)
 120             break;
 121
 122         picture_Copy(dup, picture);
 123
 124         dup->p_next = snap->picture;
 125         snap->picture = dup;
 126         snap->request_count--;
 127     }
 128     vlc_cond_broadcast(&snap->wait);
 129     vlc_mutex_unlock(&snap->lock);
 130 }
 131 /* */
 132 char *vout_snapshot_GetDirectory(void)
 133 {
 134     return config_GetUserDir(VLC_PICTURES_DIR);
 135 }
 136 /* */
 137 int vout_snapshot_SaveImage(char **name, int *sequential,
 138                              const block_t *image,
 139                              vlc_object_t *object,
 140                              const vout_snapshot_save_cfg_t *cfg)
 141 {
 142     /* */
 143     char *filename;
 144     DIR *pathdir = utf8_opendir(cfg->path);
 145     if (pathdir != NULL) {
 146         /* The use specified a directory path */
 147         closedir(pathdir);
 148
 149         /* */
 150         char *prefix = NULL;
 151         if (cfg->prefix_fmt)
 152             prefix = str_format(object, cfg->prefix_fmt);
 153         if (!prefix) {
 154             prefix = strdup("vlcsnap-");
 155             if (!prefix)
 156                 goto error;
 157         }
 158
 159         if (cfg->is_sequential) {
 160             for (int num = cfg->sequence; ; num++) {
 161                 struct stat st;
 162
 163                 if (asprintf(&filename, "%s" DIR_SEP "%s%05d.%s",
 164                              cfg->path, prefix, num, cfg->format) < 0) {
 165                     free(prefix);
 166                     goto error;
 167                 }
 168                 if (utf8_stat(filename, &st)) {
 169                     *sequential = num;
 170                     break;
 171                 }
 172                 free(filename);
 173             }
 174         } else {
 175             struct tm    curtime;
 176             time_t       lcurtime = time(NULL) ;
 177
 178             if (!localtime_r(&lcurtime, &curtime)) {
 179                 const unsigned int id = (image->i_pts / 100000) & 0xFFFFFF;
 180
 181                 msg_Warn(object, "failed to get current time. Falling back to legacy snapshot naming");
 182
 183                 if (asprintf(&filename, "%s" DIR_SEP "%s%u.%s",
 184                              cfg->path, prefix, id, cfg->format) < 0)
 185                     filename = NULL;
 186             } else {
 187                 /* suffix with the last decimal digit in 10s of seconds resolution
 188                  * FIXME gni ? */
 189                 const int id = (image->i_pts / (100*1000)) & 0xFF;
 190                 char buffer[128];
 191
 192                 if (!strftime(buffer, sizeof(buffer), "%Y-%m-%d-%Hh%Mm%Ss", &curtime))
 193                     strcpy(buffer, "error");
 194
 195                 if (asprintf(&filename, "%s" DIR_SEP "%s%s%1u.%s",
 196                              cfg->path, prefix, buffer, id, cfg->format) < 0)
 197                     filename = NULL;
 198             }
 199         }
 200         free(prefix);
 201     } else {
 202         /* The user specified a full path name (including file name) */
 203         filename = str_format(object, cfg->path);
 204         path_sanitize(filename);
 205     }
 206
 207     if (!filename)
 208         goto error;
 209
 210     /* Save the snapshot */
 211     FILE *file = utf8_fopen(filename, "wb");
 212     if (!file) {
 213         msg_Err(object, "Failed to open '%s'", filename);
 214         free(filename);
 215         goto error;
 216     }
 217     if (fwrite(image->p_buffer, image->i_buffer, 1, file) != 1) {
 218         msg_Err(object, "Failed to write to '%s'", filename);
 219         fclose(file);
 220         free(filename);
 221         goto error;
 222     }
 223     fclose(file);
 224
 225     /* */
 226     if (name)
 227         *name = filename;
 228     else
 229         free(filename);
 230
 231     return VLC_SUCCESS;
 232
 233 error:
 234     msg_Err(object, "could not save snapshot");
 235     return VLC_EGENERIC;
 236 }
 237