cmdline.mm

extern "C"
{
#import <getopt.h>
}


#import <ApplicationServices/ApplicationServices.h>
#import <Foundation/Foundation.h>

#import <dlfcn.h>

#import "utils.h"

int cmdline_main(int argc, char * const*argv)
{
	//exit(0);
	
	NSAutoreleasePool* pool = [NSAutoreleasePool new];
 	{
		int width = 0;
		int height = 0;
		CGFloat scale = 0.0f;
		int bitRes = 0;
		int displayNo = -1;
		double rotation = -1;
		
		bool listDisplays = 0;
		bool listModes = 0;

		static struct option longopts[] = {
			{"width", required_argument, NULL, 'w'},
			{"height", required_argument, NULL, 'h'},
			{"scale", required_argument, NULL, 's'},
			{"bits", required_argument, NULL, 'b'},
			{"display", required_argument, NULL, 'd'},
			{"displays",no_argument, NULL, 'l'},
			{"modes", no_argument, NULL, 'm'},
			{NULL, 0, NULL, 0},
		};

		int ch;
		while ((ch = getopt_long(argc, argv, "w:h:s:b:d:lmr:", longopts, NULL)) != -1) {
			switch (ch) {
			case 'w':
				width = atoi(optarg);
				break;
			case 'h':
				height = atoi(optarg);
				break;
			case 's':
				scale = atof(optarg);
				break;
			case 'b':
				bitRes = atoi(optarg);
				break;
			case 'd':
				displayNo = atoi(optarg);
				break;
			case 'l':
				listDisplays = 1;
				break;
			case 'm':
				listModes = 1;
				break;
			case 'r':
				rotation = atof(optarg);
				break;
			default:
				return -1;
			}
		}

		uint32_t nDisplays;
		CGDirectDisplayID displays[0x10];
		CGGetOnlineDisplayList(0x10, displays, &nDisplays);
		
		//displays[0] = CGMainDisplayID();
		
		
		CGDirectDisplayID display;
	    
		if(displayNo > 0)
		{
			if(displayNo > nDisplays -1)
			{
				fprintf(stderr, "Error: display index %d exceeds display count %d\n", displayNo, nDisplays);
				exit(1);
			}
			display = displays[displayNo];
		}
		else
		{
		    display = CGMainDisplayID();
		}
		
		if(listDisplays)
		{
			for(int i=0; i<nDisplays; i++)
			{
				int modeNum;
				CGSGetCurrentDisplayMode(display, &modeNum);
				modes_D4 mode;
				CGSGetDisplayModeDescriptionOfLength(display, modeNum, &mode, 0xD4);
				
				int mBitres = (mode.derived.depth == 4) ? 32 : 16;
				
				fprintf(stdout, "Display %d: {resolution=%dx%d, scale = %.1f, freq = %d, bits/pixel = %d}\n", i, mode.derived.width, mode.derived.height, mode.derived.density, mode.derived.freq, mBitres);
				
			}
			
			return 0;
		}
		if(listModes)
		{
			int nModes;
			modes_D4* modes;
			CopyAllDisplayModes(display, &modes, &nModes);
			
			for(int i=0; i<nModes; i++)
			{
				modes_D4 mode = modes[i];
				if(width && mode.derived.width != width)
					continue;
				if(height && mode.derived.height != height)
					continue;
				int mBitres = (mode.derived.depth == 4) ? 32 : 16;
				if(bitRes && mBitres != bitRes)
					continue;
				if(scale && mode.derived.density != scale)
					continue;
				
				fprintf(stdout, "mode: {resolution=%dx%d, scale = %.1f, freq = %d, bits/pixel = %d}\n", mode.derived.width, mode.derived.height, mode.derived.density, mode.derived.freq, mBitres);
				
			}
			
			free(modes);
			
			return 0;
			/*
			CFArrayRef modes = CGDisplayCopyAllDisplayModes(display, NULL);
		    for (int i = 0; i < CFArrayGetCount(modes); i++) {
		        CGDisplayModeRef mode = (CGDisplayModeRef)CFArrayGetValueAtIndex(modes, i);
		        CFDictionaryRef infoDict = CGDisplayModeGetDictionary(mode);
				
		        CFNumberRef resolution = (CFNumberRef) CFDictionaryGetValue(infoDict, CFSTR("kCGDisplayResolution"));
		        CFNumberRef bits = (CFNumberRef) CFDictionaryGetValue(infoDict, CFSTR("BitsPerPixel"));
		        float modeScale = 1.0f;
		        int modeBitres;
		        if(resolution)
					CFNumberGetValue(resolution, kCFNumberFloatType, &modeScale);
		        CFNumberGetValue(bits, kCFNumberIntType, &modeBitres);
				
				int modeWidth = (int)CGDisplayModeGetWidth(mode);
				int modeHeight = (int)CGDisplayModeGetHeight(mode);
				
				//NSLog(@"%@", [mode description]);
				//NSLog(@"%@", [(NSDictionary*)infoDict description]);
				
				if(width && modeWidth != width)
					continue;
				if(height && modeHeight != height)
					continue;
				if(scale && modeScale != scale)
					continue;
				if(bitRes && modeBitres != bitRes)
					continue;
				
				
				
		        printf("mode: {resolution=%dx%d, scale = %.1f, bits/pixel = %d}\n", modeWidth,
		               modeHeight, modeScale, modeBitres);
		    }
		    CFRelease(modes);
			return 0;
			*/
		}
		
		if(rotation != -1.0f)
		{
			fprintf(stderr, "Sorry, cannot adjust rotation at this time!\n");
			exit(1);
		}
		
		// fill in missing details
		{
			int modeNum;
			CGSGetCurrentDisplayMode(display, &modeNum);
			modes_D4 mode;
			CGSGetDisplayModeDescriptionOfLength(display, modeNum, &mode, 0xD4);
			
			if(!width && !height)
			{
				width = mode.derived.width;
				height = mode.derived.height;
			}
			if(!scale)
			{
				scale = mode.derived.density;
			}
			int mBitres = (mode.derived.depth == 4) ? 32 : 16;
			if(!bitRes)
			{
				bitRes = mBitres;
			}
		}
		
		
		{
			int nModes;
			modes_D4* modes;
			CopyAllDisplayModes(display, &modes, &nModes);
			
			int iMode = -1;
			
			for(int i=0; i<nModes; i++)
			{
				modes_D4 mode = modes[i];
				if(width && mode.derived.width != width)
					continue;
				if(height && mode.derived.height != height)
					continue;
				int mBitres = (mode.derived.depth == 4) ? 32 : 16;
				if(bitRes && mBitres != bitRes)
					continue;
				if(scale && mode.derived.density != scale)
					continue;
				
				iMode = i;
				break;
				//fprintf(stdout, "mode: {resolution=%dx%d, scale = %.1f, freq = %d, bits/pixel = %d}\n", mode.derived.width, mode.derived.height, mode.derived.density, mode.derived.freq, mode.derived.depth);
			}
			
			if(iMode != -1)
			{
				SetDisplayModeNum(display, iMode);
			}
			else
			{
				fprintf(stderr, "Error: could not select a new mode\n");
			}
			
			free(modes);
		}
		
		
    }
	[pool release];
    return 0;
}