Statistics
| Branch: | Revision:

ffmpeg / libavutil / log.c @ 0a51362b

History | View | Annotate | Download (4.95 KB)

1
/*
2
 * log functions
3
 * Copyright (c) 2003 Michel Bardiaux
4
 *
5
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg is free software; you can redistribute it and/or
8
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21

    
22
/**
23
 * @file
24
 * logging functions
25
 */
26

    
27
#include <unistd.h>
28
#include <stdlib.h>
29
#include "avutil.h"
30
#include "log.h"
31

    
32
#if LIBAVUTIL_VERSION_MAJOR > 50
33
static
34
#endif
35
int av_log_level = AV_LOG_INFO;
36
static int flags;
37

    
38
#if defined(_WIN32) && !defined(__MINGW32CE__)
39
#include <windows.h>
40
static const uint8_t color[] = {12,12,12,14,7,7,7};
41
static int16_t background, attr_orig;
42
static HANDLE con;
43
#define set_color(x)  SetConsoleTextAttribute(con, background | color[x])
44
#define reset_color() SetConsoleTextAttribute(con, attr_orig)
45
#else
46
static const uint8_t color[]={0x41,0x41,0x11,0x03,9,9,9};
47
#define set_color(x)  fprintf(stderr, "\033[%d;3%dm", color[x]>>4, color[x]&15)
48
#define reset_color() fprintf(stderr, "\033[0m")
49
#endif
50
static int use_color=-1;
51

    
52
#undef fprintf
53
static void colored_fputs(int level, const char *str){
54
    if(use_color<0){
55
#if defined(_WIN32) && !defined(__MINGW32CE__)
56
        CONSOLE_SCREEN_BUFFER_INFO con_info;
57
        con = GetStdHandle(STD_ERROR_HANDLE);
58
        use_color = (con != INVALID_HANDLE_VALUE) && !getenv("NO_COLOR") && !getenv("FFMPEG_FORCE_NOCOLOR");
59
        if (use_color) {
60
            GetConsoleScreenBufferInfo(con, &con_info);
61
            attr_orig  = con_info.wAttributes;
62
            background = attr_orig & 0xF0;
63
        }
64
#elif HAVE_ISATTY
65
        use_color= !getenv("NO_COLOR") && !getenv("FFMPEG_FORCE_NOCOLOR") &&
66
            (getenv("TERM") && isatty(2) || getenv("FFMPEG_FORCE_COLOR"));
67
#else
68
        use_color= getenv("FFMPEG_FORCE_COLOR") && !getenv("NO_COLOR") && !getenv("FFMPEG_FORCE_NOCOLOR");
69
#endif
70
    }
71

    
72
    if(use_color){
73
        set_color(level);
74
    }
75
    fputs(str, stderr);
76
    if(use_color){
77
        reset_color();
78
    }
79
}
80

    
81
const char* av_default_item_name(void* ptr){
82
    return (*(AVClass**)ptr)->class_name;
83
}
84

    
85
static void sanitize(uint8_t *line){
86
    while(*line){
87
        if(*line < 0x08 || (*line > 0x0D && *line < 0x20))
88
            *line='?';
89
        line++;
90
    }
91
}
92

    
93
void av_log_default_callback(void* ptr, int level, const char* fmt, va_list vl)
94
{
95
    static int print_prefix=1;
96
    static int count;
97
    static char line[1024], prev[1024];
98
    static int is_atty;
99
    AVClass* avc= ptr ? *(AVClass**)ptr : NULL;
100
    if(level>av_log_level)
101
        return;
102
    line[0]=0;
103
#undef fprintf
104
    if(print_prefix && avc) {
105
        if(avc->version >= (50<<16 | 15<<8 | 3) && avc->parent_log_context_offset){
106
            AVClass** parent= *(AVClass***)(((uint8_t*)ptr) + avc->parent_log_context_offset);
107
            if(parent && *parent){
108
                snprintf(line, sizeof(line), "[%s @ %p] ", (*parent)->item_name(parent), parent);
109
            }
110
        }
111
        snprintf(line + strlen(line), sizeof(line) - strlen(line), "[%s @ %p] ", avc->item_name(ptr), ptr);
112
    }
113

    
114
    vsnprintf(line + strlen(line), sizeof(line) - strlen(line), fmt, vl);
115

    
116
    print_prefix= line[strlen(line)-1] == '\n';
117

    
118
#if HAVE_ISATTY
119
    if(!is_atty) is_atty= isatty(2) ? 1 : -1;
120
#endif
121

    
122
    if(print_prefix && (flags & AV_LOG_SKIP_REPEATED) && !strcmp(line, prev)){
123
        count++;
124
        if(is_atty==1)
125
            fprintf(stderr, "    Last message repeated %d times\r", count);
126
        return;
127
    }
128
    if(count>0){
129
        fprintf(stderr, "    Last message repeated %d times\n", count);
130
        count=0;
131
    }
132
    strcpy(prev, line);
133
    sanitize(line);
134
    colored_fputs(av_clip(level>>3, 0, 6), line);
135
}
136

    
137
static void (*av_log_callback)(void*, int, const char*, va_list) = av_log_default_callback;
138

    
139
void av_log(void* avcl, int level, const char *fmt, ...)
140
{
141
    AVClass* avc= avcl ? *(AVClass**)avcl : NULL;
142
    va_list vl;
143
    va_start(vl, fmt);
144
    if(avc && avc->version >= (50<<16 | 15<<8 | 2) && avc->log_level_offset_offset && level>=AV_LOG_FATAL)
145
        level += *(int*)(((uint8_t*)avcl) + avc->log_level_offset_offset);
146
    av_vlog(avcl, level, fmt, vl);
147
    va_end(vl);
148
}
149

    
150
void av_vlog(void* avcl, int level, const char *fmt, va_list vl)
151
{
152
    av_log_callback(avcl, level, fmt, vl);
153
}
154

    
155
int av_log_get_level(void)
156
{
157
    return av_log_level;
158
}
159

    
160
void av_log_set_level(int level)
161
{
162
    av_log_level = level;
163
}
164

    
165
void av_log_set_flags(int arg)
166
{
167
    flags= arg;
168
}
169

    
170
void av_log_set_callback(void (*callback)(void*, int, const char*, va_list))
171
{
172
    av_log_callback = callback;
173
}