Mercurial > ldforge / file revision

/*
 *  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 <QGraphicsScene>
#include <QGraphicsItem>
#include <QMouseEvent>
#include <QScrollBar>

#include "common.h"
#include "gui.h"
#include "colorSelectDialog.h"
#include "colors.h"
#include "config.h"
#include "misc.h"
#include "ui_colorsel.h"

static const int g_numColumns = 16;
static const short g_squareSize = 32;

extern_cfg( str, gl_maincolor );
extern_cfg( float, gl_maincolor_alpha );

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
ColorSelector::ColorSelector( short defval, QWidget* parent ) : QDialog( parent )
{
	// Remove the default color if it's invalid
	if( !::getColor( defval ))
		defval = -1;
	
	m_firstResize = true;
	ui = new Ui_ColorSelUI;
	ui->setupUi( this );
	
	m_scene = new QGraphicsScene;
	ui->viewport->setScene( m_scene );
	setSelection( ::getColor( defval ));
	
	// not really an icon but eh
	m_scene->setBackgroundBrush( getIcon( "checkerboard" ));
	
	drawScene();
	
	int width = viewportWidth();
	ui->viewport->setMinimumWidth( width );
	ui->viewport->setMaximumWidth( width );
	
	drawColorInfo();
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
ColorSelector::~ColorSelector()
{
	delete ui;
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
void ColorSelector::drawScene()
{
	const int numCols = g_numColumns;
	const int square = g_squareSize;
	const int g_maxHeight = ( numRows() * square );
	QRect sceneRect( 0, 0, viewportWidth(), g_maxHeight );
	
	m_scene->setSceneRect( sceneRect );
	ui->viewport->setSceneRect( sceneRect );
	
	const double penWidth = 1.0f;
	QPen pen( Qt::black, penWidth, Qt::SolidLine, Qt::FlatCap, Qt::MiterJoin );
	
	// Draw the color rectangles.
	m_scene->clear();
	
	for( short i = 0; i < MAX_COLORS; ++i )
	{
		color* info = ::getColor( i );
		
		if( !info )
			continue;
		
		const double x = ( i % numCols ) * square;
		const double y = ( i / numCols ) * square;
		const double w = square - ( penWidth / 2 );
		
		QColor col = info->faceColor;
		
		if( i == maincolor )
		{
			// Use the user preferences for main color here
			col = QColor( gl_maincolor );
			col.setAlpha( gl_maincolor_alpha * 255.0f );
		}
		
		m_scene->addRect( x, y, w, w, pen, col );
		QGraphicsTextItem* numtext = m_scene->addText( fmt( "%1", i ));
		numtext->setDefaultTextColor(( luma( col ) < 80 ) ? Qt::white : Qt::black );
		numtext->setPos( x, y );
		
		if( i == sel()->index )
		{
			auto curspic = m_scene->addPixmap( getIcon( "colorcursor" ));
			curspic->setPos( x, y );
		}
	}
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
int ColorSelector::numRows() const
{
	return ( MAX_COLORS / g_numColumns );
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
int ColorSelector::viewportWidth() const
{
	return g_numColumns * g_squareSize + 21;
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
void ColorSelector::drawColorInfo()
{
	if( !sel() )
	{
		ui->colorLabel->setText( "---" );
		return;
	}
	
	ui->colorLabel->setText( fmt( "%1 - %2", sel()->index, sel()->name ));
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
void ColorSelector::resizeEvent( QResizeEvent* ev )
{
	// If this is the first resize, check if we need to scroll down to see the
	// currently selected color. We cannot do this in the constructor because the
	// height is not set properly there.
	if( m_firstResize )
	{
		int visibleColors = ( ui->viewport->height() / g_squareSize ) * g_numColumns;
		
		if( sel() && sel()->index >= visibleColors )
		{
			int y = ( sel()->index / g_numColumns ) * g_squareSize;
			ui->viewport->verticalScrollBar()->setValue( y );
		}
		
		m_firstResize = false;
	}
	
	drawScene();
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
void ColorSelector::mousePressEvent( QMouseEvent* event )
{
	QPointF scenepos = ui->viewport->mapToScene( event->pos() );
	
	ulong x = (( ulong ) scenepos.x() - ( g_squareSize / 2 )) / g_squareSize;
	ulong y = (( ulong ) scenepos.y() - ( g_squareSize / 2 )) / g_squareSize;
	ulong idx = ( y * g_numColumns ) + x;
	
	color* col = ::getColor( idx );
	
	if( !col )
		return;
	
	setSelection( col );
	drawScene();
	drawColorInfo();
}

// =============================================================================
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// =============================================================================
bool ColorSelector::getColor( short& val, short int defval, QWidget* parent )
{
	ColorSelector dlg( defval, parent );
	
	if( dlg.exec() && dlg.sel() != null )
	{
		val = dlg.sel()->index;
		return true;
	}
	
	return false;
}