librobotcontrol-sys 0.4.0

/**
 * @file rc_cpu.c
 * @example    rc_cpu
 *
 * This cycles the CPU through each available frequency and leaves the frequency
 * set to the default "ondemand" governor.
 */

#include <stdio.h>
#include <getopt.h>
#include <string.h>
#include <rc/cpu.h>

typedef enum mode_t{
	NONE,
	SET,
	READ,
	TEST
}mode_t;


// printed if some invalid argument was given
static void __print_usage(void)
{
	printf("\n");
	printf(" Options\n");
	printf(" -g {gov}   set the governor\n");
	printf(" -r         read the frequency in mhz\n");
	printf(" -t         run a test through all governors\n");
	printf(" -h         print this help message\n\n");
	printf("available governors:\n");
	printf("conservative ondemand powersave performance schedutil\n\n");
	return;
}


static int __run_test(void)
{
	printf("setting governor to POWERSAVE\n");
	if(rc_cpu_set_governor(RC_GOV_POWERSAVE)<0){
		fprintf(stderr,"Failed to set governor to POWERSAVE\n");
		return -1;
	}
	printf("setting governor to PERFORMANCE\n");
	if(rc_cpu_set_governor(RC_GOV_PERFORMANCE)<0){
		fprintf(stderr,"Failed to set governor to PERFORMANCE\n");
		return -1;
	}
	printf("setting governor to SCHEDUTIL\n");
	if(rc_cpu_set_governor(RC_GOV_SCHEDUTIL)<0){
		fprintf(stderr,"Failed to set governor to SCHEDUTIL\n");
		return -1;
	}
	printf("setting governor to CONSERVATIVE\n");
	if(rc_cpu_set_governor(RC_GOV_CONSERVATIVE)<0){
		fprintf(stderr,"Failed to set governor to CONSERVATIVE\n");
		return -1;
	}
	printf("setting governor to ONDEMAND\n");
	if(rc_cpu_set_governor(RC_GOV_ONDEMAND)<0){
		fprintf(stderr,"Failed to set governor to ONDEMAND\n");
		return -1;
	}
	return 0;
}



int main(int argc, char *argv[])
{
	int c;
	mode_t mode = NONE;
	rc_governor_t gov;

	if(argc>3){
		fprintf(stderr,"ERROR: too many arguments\n");
		__print_usage();
		return -1;
	}
	if(argc<2){
		fprintf(stderr,"ERROR: not enough arguments\n");
		__print_usage();
		return -1;
	}

	// parse arguments
	opterr = 0;
	while ((c = getopt(argc, argv, "g:rth")) != -1){
		switch (c){
		case 'g':
			if(mode!=NONE){
				fprintf(stderr,"Invalid Argument\n");
				__print_usage();
				return -1;
			}
			mode = SET;
			if(!strcmp(optarg, "powersave")) gov=RC_GOV_POWERSAVE;
			else if(!strcmp(optarg, "performance")) gov=RC_GOV_PERFORMANCE;
			else if(!strcmp(optarg, "ondemand")) gov=RC_GOV_ONDEMAND;
			else if(!strcmp(optarg, "schedutil")) gov=RC_GOV_SCHEDUTIL;
			else if(!strcmp(optarg, "conservative")) gov=RC_GOV_CONSERVATIVE;
			else{
				fprintf(stderr,"ERROR invalid governor\n");
				__print_usage();
				return -1;
			}
			break;
		case 'r':
			if(mode!=NONE){
				fprintf(stderr,"Invalid Argument\n");
				__print_usage();
				return -1;
			}
			mode = READ;
			break;

		case 't':
			if(mode!=NONE){
				fprintf(stderr,"Invalid Argument\n");
				__print_usage();
				return -1;
			}
			mode = TEST;
			break;

		case 'h':
			__print_usage();
			return 0;

		default:
			fprintf(stderr,"Invalid Argument\n");
			__print_usage();
			return -1;
		}
	}

	switch(mode){
	case NONE:
		__print_usage();
		return 0;
	case READ:
		printf("Current Frequency: ");
		rc_cpu_print_freq();
		printf("\n");
		return 0;
	case TEST:
		return __run_test();
	case SET:
		return rc_cpu_set_governor(gov);
	default:
		fprintf(stderr,"ERROR: Invalid mode\n");
		return -1;
	}

	return 0;
}