cpu.c 4.54 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

19 20
#include "config.h"

21 22
#ifdef __APPLE__
#include <sys/sysctl.h>
23 24 25 26
#elif defined(__linux__)
#include <asm/cputable.h>
#include <linux/auxvec.h>
#include <fcntl.h>
27
#if HAVE_UNISTD_H
28
#include <unistd.h>
29
#endif
30
#elif defined(__OpenBSD__)
31 32 33
#include <sys/param.h>
#include <sys/sysctl.h>
#include <machine/cpu.h>
34
#elif defined(__AMIGAOS4__)
35 36 37 38 39
#include <exec/exec.h>
#include <interfaces/exec.h>
#include <proto/exec.h>
#endif /* __APPLE__ */

40
#include "libavutil/avassert.h"
41
#include "libavutil/cpu.h"
42
#include "libavutil/cpu_internal.h"
43

44
/**
Diego Biurrun's avatar
Diego Biurrun committed
45 46
 * This function MAY rely on signal() or fork() in order to make sure AltiVec
 * is present.
47
 */
48
int ff_get_cpu_flags_ppc(void)
49
{
50
#if HAVE_ALTIVEC
51 52 53 54 55
#ifdef __AMIGAOS4__
    ULONG result = 0;
    extern struct ExecIFace *IExec;

    IExec->GetCPUInfoTags(GCIT_VectorUnit, &result, TAG_DONE);
56 57
    if (result == VECTORTYPE_ALTIVEC)
        return AV_CPU_FLAG_ALTIVEC;
58
    return 0;
59 60 61 62
#elif defined(__APPLE__) || defined(__OpenBSD__)
#ifdef __OpenBSD__
    int sels[2] = {CTL_MACHDEP, CPU_ALTIVEC};
#else
63
    int sels[2] = {CTL_HW, HW_VECTORUNIT};
64
#endif
65 66 67 68 69 70
    int has_vu = 0;
    size_t len = sizeof(has_vu);
    int err;

    err = sysctl(sels, 2, &has_vu, &len, NULL, 0);

71 72
    if (err == 0)
        return has_vu ? AV_CPU_FLAG_ALTIVEC : 0;
73
    return 0;
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
#elif defined(__linux__)
    // The linux kernel could have the altivec support disabled
    // even if the cpu has it.
    int i, ret = 0;
    int fd = open("/proc/self/auxv", O_RDONLY);
    unsigned long buf[64] = { 0 };
    ssize_t count;

    if (fd < 0)
        return 0;

    while ((count = read(fd, buf, sizeof(buf))) > 0) {
        for (i = 0; i < count / sizeof(*buf); i += 2) {
            if (buf[i] == AT_NULL)
                goto out;
            if (buf[i] == AT_HWCAP) {
                if (buf[i + 1] & PPC_FEATURE_HAS_ALTIVEC)
                    ret = AV_CPU_FLAG_ALTIVEC;
92 93 94 95
#ifdef PPC_FEATURE_HAS_VSX
                if (buf[i + 1] & PPC_FEATURE_HAS_VSX)
                    ret |= AV_CPU_FLAG_VSX;
#endif
96 97
                if (ret & AV_CPU_FLAG_VSX)
                    av_assert0(ret & AV_CPU_FLAG_ALTIVEC);
98 99 100 101 102
            } else if (buf[i] == AT_HWCAP2) {
#ifdef PPC_FEATURE2_ARCH_2_07
                if (buf[i + 1] & PPC_FEATURE2_ARCH_2_07)
                    ret |= AV_CPU_FLAG_POWER8;
#endif
103 104 105 106 107 108 109
            }
        }
    }

out:
    close(fd);
    return ret;
110
#elif CONFIG_RUNTIME_CPUDETECT && defined(__linux__)
111 112 113 114 115 116 117 118 119
#define PVR_G4_7400  0x000C
#define PVR_G5_970   0x0039
#define PVR_G5_970FX 0x003C
#define PVR_G5_970MP 0x0044
#define PVR_G5_970GX 0x0045
#define PVR_POWER6   0x003E
#define PVR_POWER7   0x003F
#define PVR_POWER8   0x004B
#define PVR_CELL_PPU 0x0070
120
    int ret = 0;
121
    int proc_ver;
Diego Biurrun's avatar
Diego Biurrun committed
122
    // Support of mfspr PVR emulation added in Linux 2.6.17.
123
    __asm__ volatile("mfspr %0, 287" : "=r" (proc_ver));
124 125
    proc_ver >>= 16;
    if (proc_ver  & 0x8000 ||
126 127 128 129 130 131 132 133 134
        proc_ver == PVR_G4_7400  ||
        proc_ver == PVR_G5_970   ||
        proc_ver == PVR_G5_970FX ||
        proc_ver == PVR_G5_970MP ||
        proc_ver == PVR_G5_970GX ||
        proc_ver == PVR_POWER6   ||
        proc_ver == PVR_POWER7   ||
        proc_ver == PVR_POWER8   ||
        proc_ver == PVR_CELL_PPU)
135 136 137 138 139 140 141 142
        ret = AV_CPU_FLAG_ALTIVEC;
    if (proc_ver == PVR_POWER7 ||
        proc_ver == PVR_POWER8)
        ret |= AV_CPU_FLAG_VSX;
    if (proc_ver == PVR_POWER8)
        ret |= AV_CPU_FLAG_POWER8;

    return ret;
143
#else
Diego Biurrun's avatar
Diego Biurrun committed
144
    // Since we were compiled for AltiVec, just assume we have it
145
    // until someone comes up with a proper way (not involving signal hacks).
146
    return AV_CPU_FLAG_ALTIVEC;
147
#endif /* __AMIGAOS4__ */
148 149
#endif /* HAVE_ALTIVEC */
    return 0;
150
}
151 152 153 154 155 156 157 158 159 160 161 162

size_t ff_get_cpu_max_align_ppc(void)
{
    int flags = av_get_cpu_flags();

    if (flags & (AV_CPU_FLAG_ALTIVEC   |
                 AV_CPU_FLAG_VSX       |
                 AV_CPU_FLAG_POWER8))
        return 16;

    return 8;
}