1/* 2 * This file is part of FFmpeg. 3 * 4 * FFmpeg is free software; you can redistribute it and/or 5 * modify it under the terms of the GNU Lesser General Public 6 * License as published by the Free Software Foundation; either 7 * version 2.1 of the License, or (at your option) any later version. 8 * 9 * FFmpeg is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12 * Lesser General Public License for more details. 13 * 14 * You should have received a copy of the GNU Lesser General Public 15 * License along with FFmpeg; if not, write to the Free Software 16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 17 */ 18 19#include <stdio.h> 20 21#include "config.h" 22 23#include "libavutil/cpu.h" 24#include "libavutil/avstring.h" 25 26#if HAVE_UNISTD_H 27#include <unistd.h> 28#endif 29#if !HAVE_GETOPT 30#include "compat/getopt.c" 31#endif 32 33static const struct { 34 int flag; 35 const char *name; 36} cpu_flag_tab[] = { 37#if ARCH_AARCH64 38 { AV_CPU_FLAG_ARMV8, "armv8" }, 39 { AV_CPU_FLAG_NEON, "neon" }, 40 { AV_CPU_FLAG_VFP, "vfp" }, 41#elif ARCH_ARM 42 { AV_CPU_FLAG_ARMV5TE, "armv5te" }, 43 { AV_CPU_FLAG_ARMV6, "armv6" }, 44 { AV_CPU_FLAG_ARMV6T2, "armv6t2" }, 45 { AV_CPU_FLAG_VFP, "vfp" }, 46 { AV_CPU_FLAG_VFP_VM, "vfp_vm" }, 47 { AV_CPU_FLAG_VFPV3, "vfpv3" }, 48 { AV_CPU_FLAG_NEON, "neon" }, 49 { AV_CPU_FLAG_SETEND, "setend" }, 50#elif ARCH_PPC 51 { AV_CPU_FLAG_ALTIVEC, "altivec" }, 52#elif ARCH_MIPS 53 { AV_CPU_FLAG_MMI, "mmi" }, 54 { AV_CPU_FLAG_MSA, "msa" }, 55#elif ARCH_X86 56 { AV_CPU_FLAG_MMX, "mmx" }, 57 { AV_CPU_FLAG_MMXEXT, "mmxext" }, 58 { AV_CPU_FLAG_SSE, "sse" }, 59 { AV_CPU_FLAG_SSE2, "sse2" }, 60 { AV_CPU_FLAG_SSE2SLOW, "sse2slow" }, 61 { AV_CPU_FLAG_SSE3, "sse3" }, 62 { AV_CPU_FLAG_SSE3SLOW, "sse3slow" }, 63 { AV_CPU_FLAG_SSSE3, "ssse3" }, 64 { AV_CPU_FLAG_ATOM, "atom" }, 65 { AV_CPU_FLAG_SSE4, "sse4.1" }, 66 { AV_CPU_FLAG_SSE42, "sse4.2" }, 67 { AV_CPU_FLAG_AVX, "avx" }, 68 { AV_CPU_FLAG_AVXSLOW, "avxslow" }, 69 { AV_CPU_FLAG_XOP, "xop" }, 70 { AV_CPU_FLAG_FMA3, "fma3" }, 71 { AV_CPU_FLAG_FMA4, "fma4" }, 72 { AV_CPU_FLAG_3DNOW, "3dnow" }, 73 { AV_CPU_FLAG_3DNOWEXT, "3dnowext" }, 74 { AV_CPU_FLAG_CMOV, "cmov" }, 75 { AV_CPU_FLAG_AVX2, "avx2" }, 76 { AV_CPU_FLAG_BMI1, "bmi1" }, 77 { AV_CPU_FLAG_BMI2, "bmi2" }, 78 { AV_CPU_FLAG_AESNI, "aesni" }, 79 { AV_CPU_FLAG_AVX512, "avx512" }, 80 { AV_CPU_FLAG_SLOW_GATHER, "slowgather" }, 81#elif ARCH_LOONGARCH 82 { AV_CPU_FLAG_LSX, "lsx" }, 83 { AV_CPU_FLAG_LASX, "lasx" }, 84#endif 85 { 0 } 86}; 87 88static void print_cpu_flags(int cpu_flags, const char *type) 89{ 90 int i; 91 92 printf("cpu_flags(%s) = 0x%08X\n", type, cpu_flags); 93 printf("cpu_flags_str(%s) =", type); 94 for (i = 0; cpu_flag_tab[i].flag; i++) 95 if (cpu_flags & cpu_flag_tab[i].flag) 96 printf(" %s", cpu_flag_tab[i].name); 97 printf("\n"); 98} 99 100 101int main(int argc, char **argv) 102{ 103 int cpu_flags_raw = av_get_cpu_flags(); 104 int cpu_flags_eff; 105 int cpu_count = av_cpu_count(); 106 char threads[5] = "auto"; 107 int i; 108 109 for(i = 0; cpu_flag_tab[i].flag; i++) { 110 unsigned tmp = 0; 111 if (av_parse_cpu_caps(&tmp, cpu_flag_tab[i].name) < 0) { 112 fprintf(stderr, "Table missing %s\n", cpu_flag_tab[i].name); 113 return 4; 114 } 115 } 116 117 if (cpu_flags_raw < 0) 118 return 1; 119 120 for (;;) { 121 int c = getopt(argc, argv, "c:t:"); 122 if (c == -1) 123 break; 124 switch (c) { 125 case 'c': 126 { 127 unsigned flags = av_get_cpu_flags(); 128 if (av_parse_cpu_caps(&flags, optarg) < 0) 129 return 2; 130 131 av_force_cpu_flags(flags); 132 break; 133 } 134 case 't': 135 { 136 int len = av_strlcpy(threads, optarg, sizeof(threads)); 137 if (len >= sizeof(threads)) { 138 fprintf(stderr, "Invalid thread count '%s'\n", optarg); 139 return 2; 140 } 141 } 142 } 143 } 144 145 cpu_flags_eff = av_get_cpu_flags(); 146 147 if (cpu_flags_eff < 0) 148 return 3; 149 150 print_cpu_flags(cpu_flags_raw, "raw"); 151 print_cpu_flags(cpu_flags_eff, "effective"); 152 printf("threads = %s (cpu_count = %d)\n", threads, cpu_count); 153 154 return 0; 155} 156