src/colors.cpp

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 Santeri Piippo
 *  
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "common.h"
#include "colors.h"
#include "file.h"
#include "misc.h"
#include <qcolor.h>

static color* g_LDColors[MAX_COLORS];

void initColors () {
	printf ("%s: initializing color information.\n", __func__);
	
	color* col;
	
	// Always make sure there's 16 and 24 available. They're special like that.
	col = new color;
	col->hexcode = "#AAAAAA";
	col->faceColor = col->hexcode.chars ();
	col->edgeColor = Qt::black;
	g_LDColors[maincolor] = col;
	
	col = new color;
	col->hexcode = "#000000";
	col->edgeColor = col->faceColor = Qt::black;
	g_LDColors[edgecolor] = col;
	
	parseLDConfig ();
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
color* getColor (short dColorNum) {
	// Check bounds
	if (dColorNum < 0 || dColorNum >= MAX_COLORS)
		return null;
	
	return g_LDColors[dColorNum];
}

// =============================================================================
uchar luma (QColor& col) {
	return (0.2126f * col.red ()) +
		(0.7152f * col.green ()) +
		(0.0722f * col.blue ());
}

// =============================================================================
// Helper function for parseLDConfig
static bool parseLDConfigTag (StringParser& pars, char const* tag, str& val) {
	short pos;
	
	// Try find the token and get its position
	if (!pars.findToken (pos, tag, 1))
		return false;
	
	// Get the token after it and store it into val
	return pars.getToken (val, pos + 1);
}

// =============================================================================
void parseLDConfig () {
	FILE* fp = openLDrawFile ("LDConfig.ldr", false);
	
	if (!fp)
		return;
	
	// Read in the lines
	char buf[1024];
	while (fgets (buf, sizeof buf, fp)) {
		if (strlen (buf) == 0 || buf[0] != '0')
			continue; // empty or illogical
		
		// Use StringParser to parse the LDConfig.ldr file.
		str line = str (buf).strip ({'\n', '\r'});
		StringParser pars (line, ' ');
		
		short code = 0, alpha = 255;
		str name, facename, edgename, valuestr;
		
		// Check 0 !COLOUR, parse the name
		if (!pars.tokenCompare (0, "0") || !pars.tokenCompare (1, "!COLOUR") || !pars.getToken (name, 2))
			continue;
		
		// Replace underscores in the name with spaces for readability
		name.replace ("_", " ");
		
		// Get the CODE tag
		if (!parseLDConfigTag (pars, "CODE", valuestr))
			continue;
		
		if (!isNumber (valuestr))
			continue; // not a number
		
		// Ensure that the code is within [0 - 511]
		code = atoi (valuestr);
		if (code < 0 || code >= 512)
			continue;
		
		// VALUE and EDGE tags
		if (!parseLDConfigTag (pars, "VALUE", facename) || !parseLDConfigTag (pars, "EDGE", edgename))
			continue;
		
		// Ensure that our colors are correct
		QColor faceColor (facename.chars()),
			edgeColor (edgename.chars());
		
		if (!faceColor.isValid () || !edgeColor.isValid ())
			continue;
		
		// Parse alpha if given.
		if (parseLDConfigTag (pars, "ALPHA", valuestr))
			alpha = clamp<short> (atoi (valuestr), 0, 255);
		
		color* col = new color;
		col->name = name;
		col->faceColor = faceColor;
		col->edgeColor = edgeColor;
		col->hexcode = facename;
		col->faceColor.setAlpha (alpha);
		col->index = code;
		
		g_LDColors[code] = col;
	}
	
	fclose (fp);
}