Files
fastfetch/src/modules/cpucache/cpucache.c
T

261 lines
9.7 KiB
C

#include "common/printing.h"
#include "common/jsonconfig.h"
#include "common/size.h"
#include "common/strutil.h"
#include "detection/cpucache/cpucache.h"
#include "modules/cpucache/cpucache.h"
static void printCPUCacheNormal(const FFCPUCacheResult* result, FFCPUCacheOptions* options) {
FF_STRBUF_AUTO_DESTROY buffer = ffStrbufCreate();
FF_STRBUF_AUTO_DESTROY key = ffStrbufCreate();
char levelStr[4] = "L";
for (uint32_t i = 0; i < ARRAY_SIZE(result->caches) && result->caches[i].length > 0; i++) {
ffStrbufClear(&key);
levelStr[1] = (char) ('1' + i);
if (options->moduleArgs.key.length == 0) {
ffStrbufAppendF(&key, "%s (%s)", FF_MODULE_GET_DISPLAY_NAME(CPUCache), levelStr);
} else {
uint32_t index = i + 1;
FF_PARSE_FORMAT_STRING_CHECKED(&key, &options->moduleArgs.key, ((FFformatarg[]) {
FF_ARG(index, "index"),
FF_ARG(levelStr, "level"),
FF_ARG(options->moduleArgs.keyIcon, "icon"),
FF_ARG(FF_MODULE_GET_DISPLAY_NAME(CPUCache), "module-name"),
}));
}
ffStrbufClear(&buffer);
uint32_t sum = 0;
FF_LIST_FOR_EACH (FFCPUCache, src, result->caches[i]) {
char typeStr = '?';
switch (src->type) {
case FF_CPU_CACHE_TYPE_DATA:
typeStr = 'D';
break;
case FF_CPU_CACHE_TYPE_INSTRUCTION:
typeStr = 'I';
break;
case FF_CPU_CACHE_TYPE_UNIFIED:
typeStr = 'U';
break;
case FF_CPU_CACHE_TYPE_TRACE:
typeStr = 'T';
break;
}
if (buffer.length) {
ffStrbufAppendS(&buffer, ", ");
}
if (src->num > 1) {
ffStrbufAppendF(&buffer, "%ux", src->num);
}
ffSizeAppendNum(src->size, &buffer);
ffStrbufAppendF(&buffer, " (%c)", typeStr);
sum += src->size * src->num;
}
if (options->moduleArgs.outputFormat.length == 0) {
ffPrintLogoAndKey(key.chars, 0, &options->moduleArgs, FF_PRINT_TYPE_NO_CUSTOM_KEY);
ffStrbufPutTo(&buffer, stdout);
} else {
FF_STRBUF_AUTO_DESTROY buffer2 = ffStrbufCreate();
ffSizeAppendNum(sum, &buffer2);
FF_PRINT_FORMAT_CHECKED(key.chars, 0, &options->moduleArgs, FF_PRINT_TYPE_NO_CUSTOM_KEY, ((FFformatarg[]) {
FF_ARG(buffer, "result"),
FF_ARG(buffer2, "sum"),
}));
}
}
}
static void printCPUCacheCompact(const FFCPUCacheResult* result, FFCPUCacheOptions* options) {
FF_STRBUF_AUTO_DESTROY buffer = ffStrbufCreate();
uint64_t sum = 0;
for (uint32_t i = 0; i < ARRAY_SIZE(result->caches) && result->caches[i].length > 0; i++) {
if (buffer.length) {
ffStrbufAppendS(&buffer, ", ");
}
uint32_t value = 0;
FF_LIST_FOR_EACH (FFCPUCache, src, result->caches[i]) {
value += src->size * src->num;
}
ffSizeAppendNum(value, &buffer);
ffStrbufAppendF(&buffer, " (L%u)", i + 1);
sum += value;
}
if (options->moduleArgs.outputFormat.length == 0) {
ffPrintLogoAndKey(FF_MODULE_GET_DISPLAY_NAME(CPUCache), 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT);
ffStrbufPutTo(&buffer, stdout);
} else {
FF_STRBUF_AUTO_DESTROY buffer2 = ffStrbufCreate();
ffSizeAppendNum(sum, &buffer2);
FF_PRINT_FORMAT_CHECKED(FF_MODULE_GET_DISPLAY_NAME(CPUCache), 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, ((FFformatarg[]) {
FF_ARG(buffer, "result"),
FF_ARG(buffer2, "sum"),
}));
}
}
bool ffPrintCPUCache(FFCPUCacheOptions* options) {
bool success = false;
FFCPUCacheResult result = {
.caches = {
ffListCreate(),
ffListCreate(),
ffListCreate(),
ffListCreate(),
},
};
const char* error = ffDetectCPUCache(&result);
if (error) {
ffPrintError(FF_MODULE_GET_DISPLAY_NAME(CPUCache), 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "%s", error);
goto exit;
}
if (!options->compact) {
printCPUCacheNormal(&result, options);
} else {
printCPUCacheCompact(&result, options);
}
success = true;
exit:
ffListDestroy(&result.caches[0]);
ffListDestroy(&result.caches[1]);
ffListDestroy(&result.caches[2]);
ffListDestroy(&result.caches[3]);
return success;
}
void ffParseCPUCacheJsonObject(FFCPUCacheOptions* options, yyjson_val* module) {
yyjson_val *key, *val;
size_t idx, max;
yyjson_obj_foreach (module, idx, max, key, val) {
if (ffJsonConfigParseModuleArgs(key, val, &options->moduleArgs)) {
continue;
}
if (unsafe_yyjson_equals_str(key, "compact")) {
options->compact = yyjson_get_bool(val);
continue;
}
ffPrintError(FF_MODULE_GET_DISPLAY_NAME(CPUCache), 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "Unknown JSON key %s", unsafe_yyjson_get_str(key));
}
}
void ffGenerateCPUCacheJsonConfig(FFCPUCacheOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) {
ffJsonConfigGenerateModuleArgsConfig(doc, module, &options->moduleArgs);
yyjson_mut_obj_add_bool(doc, module, "compact", options->compact);
}
bool ffGenerateCPUCacheJsonResult([[maybe_unused]] FFCPUCacheOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) {
bool success = false;
FFCPUCacheResult result = {
.caches = {
ffListCreate(),
ffListCreate(),
ffListCreate(),
ffListCreate(),
},
};
const char* error = ffDetectCPUCache(&result);
if (error) {
yyjson_mut_obj_add_str(doc, module, "error", error);
goto exit;
}
yyjson_mut_val* caches = yyjson_mut_obj_add_obj(doc, module, "result");
for (uint32_t i = 0; i < ARRAY_SIZE(result.caches) && result.caches[i].length > 0; i++) {
yyjson_mut_val* level = yyjson_mut_obj_add_arr(doc, caches, &"l1\0l2\0l3\0l4\0"[i * 3]);
FF_LIST_FOR_EACH (FFCPUCache, src, result.caches[i]) {
yyjson_mut_val* item = yyjson_mut_arr_add_obj(doc, level);
yyjson_mut_obj_add_uint(doc, item, "size", src->size);
yyjson_mut_obj_add_uint(doc, item, "num", src->num);
const char* typeStr = "unknown";
switch (src->type) {
case FF_CPU_CACHE_TYPE_DATA:
typeStr = "data";
break;
case FF_CPU_CACHE_TYPE_INSTRUCTION:
typeStr = "instruction";
break;
case FF_CPU_CACHE_TYPE_UNIFIED:
typeStr = "unified";
break;
case FF_CPU_CACHE_TYPE_TRACE:
typeStr = "trace";
break;
}
yyjson_mut_obj_add_uint(doc, item, "lineSize", src->lineSize);
yyjson_mut_obj_add_str(doc, item, "type", typeStr);
}
}
success = true;
exit:
ffListDestroy(&result.caches[0]);
ffListDestroy(&result.caches[1]);
ffListDestroy(&result.caches[2]);
ffListDestroy(&result.caches[3]);
return success;
}
void ffInitCPUCacheOptions(FFCPUCacheOptions* options) {
ffOptionInitModuleArg(&options->moduleArgs, "");
options->compact = false;
}
void ffDestroyCPUCacheOptions(FFCPUCacheOptions* options) {
ffOptionDestroyModuleArg(&options->moduleArgs);
}
FFModuleBaseInfo ffCPUCacheModuleInfo = {
.name = "CPUCache",
.description = "Print CPU cache sizes",
.displayName = {
.en = "CPU Cache",
.ar = "ذاكرة المعالج المؤقتة",
.cs = "Mezipaměť CPU",
.de = "CPU-Cache",
.es = "Caché de la CPU",
.fr = "Cache CPU",
.gl = "Caché da CPU",
.he = "מטמון מעבד",
.id = "Cache CPU",
.it = "Cache CPU",
.ja = "CPUキャッシュ",
.ko = "CPU 캐시",
.pl = "Pamięć podręczna CPU",
.pt = "Cache da CPU",
.ru = "Кэш процессора",
.tr = "CPU Önbelleği",
.uk = "Кеш ЦП",
.vi = "Bộ nhớ đệm CPU",
.zh_CN = "CPU缓存",
.zh_TW = "CPU快取記憶體",
},
.initOptions = (void*) ffInitCPUCacheOptions,
.destroyOptions = (void*) ffDestroyCPUCacheOptions,
.parseJsonObject = (void*) ffParseCPUCacheJsonObject,
.printModule = (void*) ffPrintCPUCache,
.generateJsonResult = (void*) ffGenerateCPUCacheJsonResult,
.generateJsonConfig = (void*) ffGenerateCPUCacheJsonConfig,
.formatArgs = FF_FORMAT_ARG_LIST(((FFModuleFormatArg[]) {
{ "Separate result", "result" },
{ "Sum result", "sum" },
})),
.defaultOrder = 34,
};