macos-stats/Modules/CPU/CPU.c

//
//  CPU.c
//  CPU
//
//  Created by Serhiy Mytrovtsiy on 13/10/2020.
//  Using Swift 5.0.
//  Running on macOS 10.15.
//
//  Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
//

#include "CPU.h"

#include <unistd.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <sched.h>

#include <IntelPowerGadget/PowerGadgetLib.h>

pthread_t thread_id;
double CPUFrequency;
bool stop;
double min;
double max;
PGSampleID sampleID_1;
PGSampleID sampleID_2;

extern bool PG_Initialize(void) __attribute__((weak_import));

static inline bool intelPowerGadgetIsAvailable() {
    return (PG_Initialize != NULL);
}

void *listener () {
    while(stop == false) {
        PG_ReadSample(0, &sampleID_1);
        sleep(1);
        PG_ReadSample(0, &sampleID_2);

        PGSample_GetIAFrequency(sampleID_1, sampleID_2, &CPUFrequency, &min, &max);

        PGSample_Release(sampleID_2);
        PGSample_Release(sampleID_1);
    }

    pthread_exit(NULL);
}

void PG_start() {
    if (intelPowerGadgetIsAvailable() == false) {
        return;
    }

    stop = false;
    PG_Initialize();
    pthread_create(&thread_id, NULL, listener, NULL);
}

void PG_stop() {
    if (intelPowerGadgetIsAvailable() == false) {
        return;
    }

    stop = true;
    PG_Shutdown();
}

double* PG_getCPUFrequency() {
    if (intelPowerGadgetIsAvailable() == false) {
        return NULL;
    }

    if (stop == true || CPUFrequency == 0) {
        return NULL;
    } else {
        return &CPUFrequency;
    }
    return NULL;
}