ds_draw_item.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2013-2014 Jean-Pierre Charras, jp.charras at wanadoo.fr
 * Copyright (C) 1992-2021 KiCad Developers, see AUTHORS.txt for contributors.
 *
 * 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 2
 * 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, you may find one here:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * or you may search the http://www.gnu.org website for the version 2 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
 */

#ifndef DS_DRAW_ITEM_H
#define DS_DRAW_ITEM_H

#include <core/typeinfo.h>
#include <math/vector2d.h>
#include <eda_text.h>
#include "widgets/msgpanel.h"
#include <geometry/shape_poly_set.h>
#include <eda_item.h>
#include <eda_units.h>

#include <algorithm>
#include <vector>

class DS_DATA_ITEM;
class TITLE_BLOCK;
class PAGE_INFO;
class EDA_ITEM;
class EDA_DRAW_FRAME;
class PROJECT;

class DS_DRAW_ITEM_BASE : public EDA_ITEM
{
public:
 virtual ~DS_DRAW_ITEM_BASE() {}

 DS_DATA_ITEM* GetPeer() const { return m_peer; }
 int GetIndexInPeer() const { return m_index; }

 void ViewGetLayers( int aLayers[], int& aCount ) const override;

 virtual void SetEnd( const wxPoint& aPos ) { /* not all types will need this */ }

 virtual int GetPenWidth() const
 {
 if( m_penWidth > 0 )
 return m_penWidth;
 else
 return 1;
 }

 // The function to print a WS_DRAW_ITEM
 virtual void PrintWsItem( const RENDER_SETTINGS* aSettings )
 {
 PrintWsItem( aSettings, wxPoint( 0, 0 ) );
 }

 // More advanced version of DrawWsItem. This is what must be defined in the derived type.
 virtual void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) = 0;

 // Derived types must define GetBoundingBox() as a minimum, and can then override the
 // two HitTest() functions if they need something more specific.
 const EDA_RECT GetBoundingBox() const override = 0;

 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override
 {
 // This is just here to prevent annoying compiler warnings about hidden overloaded
 // virtual functions
 return EDA_ITEM::HitTest( aPosition, aAccuracy );
 }

 bool HitTest( const EDA_RECT& aRect, bool aContained, int aAccuracy = 0 ) const override;

 void GetMsgPanelInfo( EDA_DRAW_FRAME* aFrame, std::vector<MSG_PANEL_ITEM>& aList ) override;

protected:
 DS_DRAW_ITEM_BASE( DS_DATA_ITEM* aPeer, int aIndex, KICAD_T aType ) :
 EDA_ITEM( aType )
 {
 m_peer = aPeer;
 m_index = aIndex;
 m_penWidth = 0;
 m_flags = 0;
 }

 DS_DATA_ITEM* m_peer; // the parent DS_DATA_ITEM item in the DS_DATA_MODEL
 int m_index; // the index in the parent's repeat count
 int m_penWidth;
};


// This class draws a thick segment
class DS_DRAW_ITEM_LINE : public DS_DRAW_ITEM_BASE
{
public:
 DS_DRAW_ITEM_LINE( DS_DATA_ITEM* aPeer, int aIndex, wxPoint aStart, wxPoint aEnd,
 int aPenWidth ) :
 DS_DRAW_ITEM_BASE( aPeer, aIndex, WSG_LINE_T )
 {
 m_start = aStart;
 m_end = aEnd;
 m_penWidth = aPenWidth;
 }

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_LINE" ); }

 const wxPoint& GetStart() const { return m_start; }
 void SetStart( const wxPoint& aPos ) { m_start = aPos; }
 const wxPoint& GetEnd() const { return m_end; }
 void SetEnd( const wxPoint& aPos ) override { m_end = aPos; }

 wxPoint GetPosition() const override { return GetStart(); }
 void SetPosition( const wxPoint& aPos ) override { SetStart( aPos ); }

 const EDA_RECT GetBoundingBox() const override;
 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override;

 void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) override;

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif

private:
 wxPoint m_start; // start point of line/rect
 wxPoint m_end; // end point
};


class DS_DRAW_ITEM_POLYPOLYGONS : public DS_DRAW_ITEM_BASE
{
public:
 DS_DRAW_ITEM_POLYPOLYGONS( DS_DATA_ITEM* aPeer, int aIndex, wxPoint aPos, int aPenWidth ) :
 DS_DRAW_ITEM_BASE( aPeer, aIndex, WSG_POLY_T )
 {
 m_penWidth = aPenWidth;
 m_pos = aPos;
 }

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_POLYPOLYGONS" ); }

 SHAPE_POLY_SET& GetPolygons() { return m_Polygons; }
 wxPoint GetPosition() const override { return m_pos; }
 void SetPosition( const wxPoint& aPos ) override;

 const EDA_RECT GetBoundingBox() const override;
 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override;
 bool HitTest( const EDA_RECT& aRect, bool aContained, int aAccuracy = 0 ) const override;

 void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) override;

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif

public:
 SHAPE_POLY_SET m_Polygons;


private:
 wxPoint m_pos; // position of reference point, from the DS_DATA_ITEM_POLYGONS parent
 // (used only in drawing sheet editor to draw anchors)
};


class DS_DRAW_ITEM_RECT : public DS_DRAW_ITEM_BASE
{
public:
 DS_DRAW_ITEM_RECT( DS_DATA_ITEM* aPeer, int aIndex, wxPoint aStart, wxPoint aEnd,
 int aPenWidth ) :
 DS_DRAW_ITEM_BASE( aPeer, aIndex, WSG_RECT_T )
 {
 m_start = aStart;
 m_end = aEnd;
 m_penWidth = aPenWidth;
 }

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_RECT" ); }

 const wxPoint& GetStart() const { return m_start; }
 void SetStart( const wxPoint& aPos ) { m_start = aPos; }
 const wxPoint& GetEnd() const { return m_end; }
 void SetEnd( const wxPoint& aPos ) override { m_end = aPos; }

 wxPoint GetPosition() const override { return GetStart(); }
 void SetPosition( const wxPoint& aPos ) override { SetStart( aPos ); }

 void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) override;

 const EDA_RECT GetBoundingBox() const override;
 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override;
 bool HitTest( const EDA_RECT& aRect, bool aContained, int aAccuracy = 0 ) const override;

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif

private:
 wxPoint m_start; // start point of line/rect
 wxPoint m_end; // end point
};


class DS_DRAW_ITEM_PAGE : public DS_DRAW_ITEM_BASE
{
public:
 DS_DRAW_ITEM_PAGE( int aPenWidth, double aMarkerSize ) :
 DS_DRAW_ITEM_BASE( nullptr, 0, WSG_PAGE_T )
 {
 m_penWidth = aPenWidth;
 m_markerSize = aMarkerSize;
 }

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_PAGE" ); }

 void SetPageSize( const wxSize& aSize ) { m_pageSize = aSize; }
 wxSize GetPageSize() const { return m_pageSize; }
 const wxPoint& GetMarkerPos() const { return m_markerPos; }
 void SetMarkerPos( const wxPoint& aPos ) { m_markerPos = aPos; }
 double GetMarkerSize() const { return m_markerSize; }

 wxPoint GetPosition() const override { return wxPoint( 0, 0 ); }
 void SetPosition( const wxPoint& aPos ) override { /* do nothing */ }

 void PrintWsItem( const RENDER_SETTINGS* , const wxPoint& ) override { /* do nothing */ }

 const EDA_RECT GetBoundingBox() const override;
 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override { return false; }

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif

private:
 wxPoint m_markerPos; // position of the marker
 wxSize m_pageSize; // full size of the page
 double m_markerSize;
};


class DS_DRAW_ITEM_TEXT : public DS_DRAW_ITEM_BASE, public EDA_TEXT
{
public:
 DS_DRAW_ITEM_TEXT( DS_DATA_ITEM* aPeer, int aIndex, const wxString& aText, const wxPoint& aPos,
 const wxSize& aSize, int aPenWidth, bool aItalic = false,
 bool aBold = false ) :
 DS_DRAW_ITEM_BASE( aPeer, aIndex, WSG_TEXT_T),
 EDA_TEXT( aText )
 {
 SetTextPos( aPos );
 SetTextSize( aSize );
 SetTextThickness( aPenWidth );
 SetItalic( aItalic );
 SetBold( aBold );
 }

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_TEXT" ); }

 void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) override;

 void SetTextAngle( double aAngle ) override;

 wxPoint GetPosition() const override { return GetTextPos(); }
 void SetPosition( const wxPoint& aPos ) override { SetTextPos( aPos ); }

 const EDA_RECT GetBoundingBox() const override;
 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override;
 bool HitTest( const EDA_RECT& aRect, bool aContained, int aAccuracy = 0 ) const override;

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif
};


class DS_DRAW_ITEM_BITMAP : public DS_DRAW_ITEM_BASE
{
public:
 DS_DRAW_ITEM_BITMAP( DS_DATA_ITEM* aPeer, int aIndex, wxPoint aPos ) :
 DS_DRAW_ITEM_BASE( aPeer, aIndex, WSG_BITMAP_T )
 {
 m_pos = aPos;
 }

 ~DS_DRAW_ITEM_BITMAP() {}

 virtual wxString GetClass() const override { return wxT( "DS_DRAW_ITEM_BITMAP" ); }

 wxPoint GetPosition() const override { return m_pos; }
 void SetPosition( const wxPoint& aPos ) override { m_pos = aPos; }

 void PrintWsItem( const RENDER_SETTINGS* aSettings, const wxPoint& aOffset ) override;

 bool HitTest( const wxPoint& aPosition, int aAccuracy = 0 ) const override;
 bool HitTest( const EDA_RECT& aRect, bool aContained, int aAccuracy = 0 ) const override;
 const EDA_RECT GetBoundingBox() const override;

 wxString GetSelectMenuText( EDA_UNITS aUnits ) const override;

#if defined(DEBUG)
 void Show( int nestLevel, std::ostream& os ) const override { ShowDummy( os ); }
#endif

private:
 wxPoint m_pos; // position of reference point
};


class DS_DRAW_ITEM_LIST
{
public:
 DS_DRAW_ITEM_LIST()
 {
 m_idx = 0;
 m_milsToIu = 1.0;
 m_penSize = 1;
 m_pageNumber = "1";
 m_sheetCount = 1;
 m_sheetLayer = nullptr;
 m_titleBlock = nullptr;
 m_paperFormat = nullptr;
 m_project = nullptr;
 m_isFirstPage = true;
 }

 ~DS_DRAW_ITEM_LIST()
 {
 // Items in the m_graphicList are owned by their respective DS_DATA_ITEMs.
 // for( DS_DRAW_ITEM_BASE* item : m_graphicList )
 // delete item;
 }

 void SetProject( const PROJECT* aProject ) { m_project = aProject; }

 void SetTitleBlock( const TITLE_BLOCK* aTblock ) { m_titleBlock = aTblock; }

 void SetPaperFormat( const wxString* aFormatName ) { m_paperFormat = aFormatName; }

 void SetFileName( const wxString& aFileName )
 {
 m_fileName = aFileName;
 }

 void SetSheetName( const wxString& aSheetName )
 {
 m_sheetFullName = aSheetName;
 }

 void SetSheetLayer( const wxString& aSheetLayer )
 {
 m_sheetLayer = &aSheetLayer;
 }

 void SetDefaultPenSize( int aPenSize ) { m_penSize = aPenSize; }
 int GetDefaultPenSize() const { return m_penSize; }

 void SetMilsToIUfactor( double aMils2Iu )
 {
 m_milsToIu = aMils2Iu;
 }

 double GetMilsToIUfactor() { return m_milsToIu; }

 void SetPageNumber( const wxString& aPageNumber )
 {
 m_pageNumber = aPageNumber;
 }

 void SetIsFirstPage( bool aIsFirstPage ) { m_isFirstPage = aIsFirstPage; }

 void SetSheetCount( int aSheetCount )
 {
 m_sheetCount = aSheetCount;
 }

 void Append( DS_DRAW_ITEM_BASE* aItem )
 {
 m_graphicList.push_back( aItem );
 }

 void Remove( DS_DRAW_ITEM_BASE* aItem )
 {
 auto newEnd = std::remove( m_graphicList.begin(), m_graphicList.end(), aItem );
 m_graphicList.erase( newEnd, m_graphicList.end() );
 }

 DS_DRAW_ITEM_BASE* GetFirst()
 {
 m_idx = 0;

 if( m_graphicList.size() )
 return m_graphicList[0];
 else
 return nullptr;
 }

 DS_DRAW_ITEM_BASE* GetNext()
 {
 m_idx++;

 if( m_graphicList.size() > m_idx )
 return m_graphicList[m_idx];
 else
 return nullptr;
 }

 void GetAllItems( std::vector<DS_DRAW_ITEM_BASE*>* aList )
 {
 *aList = m_graphicList;
 }

 void Print( const RENDER_SETTINGS* aSettings );

 void BuildDrawItemsList( const PAGE_INFO& aPageInfo, const TITLE_BLOCK& aTitleBlock );

 static void GetTextVars( wxArrayString* aVars );

 wxString BuildFullText( const wxString& aTextbase );

protected:
 std::vector <DS_DRAW_ITEM_BASE*> m_graphicList; // Items to draw/plot
 unsigned m_idx; // for GetFirst, GetNext functions
 double m_milsToIu; // the scalar to convert pages units ( mils)
 // to draw/plot units.
 int m_penSize; // The default line width for drawings.
 // used when an item has a pen size = 0
 bool m_isFirstPage; 
 int m_sheetCount; 
 // for basic inscriptions, in schematic
 const TITLE_BLOCK* m_titleBlock; // for basic inscriptions
 const wxString* m_paperFormat; // for basic inscriptions
 wxString m_fileName; // for basic inscriptions
 wxString m_sheetFullName; // for basic inscriptions
 wxString m_pageNumber; 
 const wxString* m_sheetLayer; // for basic inscriptions
 const PROJECT* m_project; // for project-based variable substitutions
};


#endif // DS_DRAW_ITEM_H