kicad/pcbnew/router/router_preview_item.cpp

/*
 * KiRouter - a push-and-(sometimes-)shove PCB router
 *
 * Copyright (C) 2013  CERN
 * Author: Tomasz Wlostowski <tomasz.wlostowski@cern.ch>
 * 
 * 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.or/licenses/>.
 */
 
#include <gal/color4d.h>

#include "class_track.h"
#include <pcb_painter.h>

#include "router_preview_item.h"

#include "pns_line.h"
#include "pns_segment.h"
#include "pns_via.h"

using namespace KiGfx;

ROUTER_PREVIEW_ITEM::ROUTER_PREVIEW_ITEM( const PNS_ITEM *aItem, VIEW_GROUP *aParent ) 
	: EDA_ITEM( NOT_USED )
		{
			m_Flags = 0;
			m_parent = aParent;
			if(aItem)
				Update(aItem);
		}
	
ROUTER_PREVIEW_ITEM::~ROUTER_PREVIEW_ITEM()
{
	
}
			
void ROUTER_PREVIEW_ITEM::Update(const PNS_ITEM *aItem)
{
	m_layer = aItem->GetLayers().Start();


	m_color = getLayerColor( m_layer );
	m_color.a = 0.8;
		
	switch(aItem->GetKind())
	{
		case PNS_ITEM::LINE:
			m_type = PR_LINE;
			m_width = static_cast<const PNS_LINE *>(aItem)->GetWidth();
			m_line = * static_cast<const SHAPE_LINE_CHAIN *>(aItem->GetShape());
			break;

		case PNS_ITEM::SEGMENT:
			m_type = PR_LINE;
			m_width = static_cast<const PNS_SEGMENT *>(aItem)->GetWidth();
			m_line = * static_cast<const SHAPE_LINE_CHAIN *>(aItem->GetShape());
			break;

		case PNS_ITEM::VIA:
			m_type = PR_VIA;
			m_color = COLOR4D(0.7, 0.7, 0.7, 0.8);
			m_width = static_cast<const PNS_VIA *>(aItem)->GetDiameter();
			m_viaCenter = static_cast<const PNS_VIA *>(aItem)->GetPos();
			break;

		default:
			break;	
	}
	
	ViewSetVisible(true);
	ViewUpdate(GEOMETRY | APPEARANCE);
}

void ROUTER_PREVIEW_ITEM::MarkAsHead( )
{
	if(m_type != PR_VIA)
		m_color.Saturate(1.0);
}

const BOX2I ROUTER_PREVIEW_ITEM::ViewBBox() const 
{
	BOX2I bbox;

	switch(m_type)
	{
		case PR_LINE:
			bbox = m_line.BBox();
			bbox.Inflate( m_width / 2);
			return bbox;
		case PR_VIA:
			bbox = BOX2I( m_viaCenter, VECTOR2I(0, 0));
			bbox.Inflate( m_width / 2);
			return bbox;
		default:
			break;
	}
	return bbox;	
}

void ROUTER_PREVIEW_ITEM::ViewDraw( int aLayer, KiGfx::GAL* aGal ) const
{
	switch( m_type )
	{
		case PR_LINE:

			aGal->SetLayerDepth(-100.0);
			aGal->SetLineWidth(m_width);
	    	aGal->SetStrokeColor(m_color);
	    	aGal->SetIsStroke(true);
	    	aGal->SetIsFill(false);
	    	for(int s= 0 ; s < m_line.SegmentCount(); s++)
	    		aGal->DrawLine(m_line.CSegment(s).a, m_line.CSegment(s).b);
	    	if(m_line.IsClosed())
	    		aGal->DrawLine(m_line.CSegment(-1).b, m_line.CSegment(0).a);
	    	break;
	    case PR_VIA:
    		
    		aGal->SetLayerDepth(-101.0);
			aGal->SetIsStroke(false);
    		aGal->SetIsFill(true);
    		aGal->SetFillColor(m_color);
    		aGal->DrawCircle(m_viaCenter, m_width / 2);
			break;
		default:
			break;
	}
}


void ROUTER_PREVIEW_ITEM::DebugLine ( const SHAPE_LINE_CHAIN& aLine, int aWidth , int aStyle  )
{
#if 0
	m_line = aLine;
	m_width = aWidth;
	m_color = assignColor(aStyle);
	

	m_type = PR_LINE;
	ViewUpdate(GEOMETRY | APPEARANCE);
#endif
}

void ROUTER_PREVIEW_ITEM::DebugBox ( const BOX2I& aBox, int aStyle )
{
#if 0
	assert(false);

	m_line.Clear();
	m_line.Append( aBox.GetX(), aBox.GetY() );
	m_line.Append( aBox.GetX() + aBox.GetWidth(), aBox.GetY() + aBox.GetHeight());
	m_line.Append( aBox.GetX() + aBox.GetWidth(), aBox.GetY() + aBox.GetHeight());
	m_line.Append( aBox.GetX(), aBox.GetY() + aBox.GetHeight());
	m_line.SetClosed(true);
	m_width = 20000;
	m_color = assignColor(aStyle);
	m_type = PR_LINE;
	ViewUpdate(GEOMETRY | APPEARANCE);
#endif
}
		
const COLOR4D ROUTER_PREVIEW_ITEM::getLayerColor (int layer ) const
{
	//assert (m_view != NULL);

	PCB_RENDER_SETTINGS *settings = static_cast <PCB_RENDER_SETTINGS*> (m_parent -> GetView() -> GetPainter() -> GetSettings());
	return settings->GetLayerColor(layer);
}


const COLOR4D ROUTER_PREVIEW_ITEM::assignColor ( int style ) const
{
	COLOR4D color;
	switch(style)
	{
    case 0: color =COLOR4D(0, 1, 0, 1);break;
    case 1: color =COLOR4D(1, 0, 0, 0.3);break;
    case 2: color =COLOR4D(1, 0.5, 0.5, 1);break;
    case 3: color =COLOR4D(0, 0, 1, 1);break;
    case 4: color =COLOR4D(1, 1, 1, 1); break;
    case 5: color =COLOR4D(1, 1, 0, 1); break;
    case 6: color =COLOR4D(0, 1, 1, 1); break;
    case 32: color =COLOR4D(0, 0, 1, 0.5); break;
    default: break;
	}
	return color;
}