mirror of
https://gitlab.com/kicad/code/kicad.git
synced 2025-09-14 10:13:19 +02:00
a310c0a05a
The new behavior is to only expand to the first child of the root sheet level. On very complex hierarchies, this makes the navigator far more useful. Do not update schematic hierarchy navigator on every edit. Now only sheet changes will trigger a rebuild of the tree. Fixes https://gitlab.com/kicad/code/kicad/-/issues/16371
329 lines
10 KiB
C++
329 lines
10 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2004 Jean-Pierre Charras, jp.charras at wanadoo.fr
|
|
* Copyright (C) 2008 Wayne Stambaugh <stambaughw@gmail.com>
|
|
* Copyright (C) 2004-2024 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
|
|
*/
|
|
|
|
#include <bitmaps.h>
|
|
#include <sch_edit_frame.h>
|
|
#include <sch_sheet.h>
|
|
#include <sch_sheet_path.h>
|
|
#include <schematic.h>
|
|
#include <tool/tool_manager.h>
|
|
#include <tools/ee_actions.h>
|
|
|
|
#include <hierarchy_pane.h>
|
|
#include <kiface_base.h>
|
|
#include <eeschema_settings.h>
|
|
|
|
#include <wx/object.h>
|
|
#include <wx/generic/textdlgg.h>
|
|
#include <wx/menu.h>
|
|
|
|
/**
|
|
* Store an SCH_SHEET_PATH of each sheet in hierarchy.
|
|
*/
|
|
class TREE_ITEM_DATA : public wxTreeItemData
|
|
{
|
|
public:
|
|
SCH_SHEET_PATH m_SheetPath;
|
|
|
|
TREE_ITEM_DATA( SCH_SHEET_PATH& sheet ) :
|
|
wxTreeItemData()
|
|
{
|
|
m_SheetPath = sheet;
|
|
}
|
|
};
|
|
|
|
|
|
// Need to use wxRTTI macros in order for OnCompareItems to work properly
|
|
// See: https://docs.wxwidgets.org/3.1/classwx_tree_ctrl.html#ab90a465793c291ca7aa827a576b7d146
|
|
wxIMPLEMENT_ABSTRACT_CLASS( HIERARCHY_TREE, wxTreeCtrl );
|
|
|
|
|
|
int HIERARCHY_TREE::OnCompareItems( const wxTreeItemId& item1, const wxTreeItemId& item2 )
|
|
{
|
|
SCH_SHEET_PATH* item1Path = &static_cast<TREE_ITEM_DATA*>( GetItemData( item1 ) )->m_SheetPath;
|
|
SCH_SHEET_PATH* item2Path = &static_cast<TREE_ITEM_DATA*>( GetItemData( item2 ) )->m_SheetPath;
|
|
|
|
return item1Path->ComparePageNum( *item2Path );
|
|
}
|
|
|
|
|
|
HIERARCHY_PANE::HIERARCHY_PANE( SCH_EDIT_FRAME* aParent ) :
|
|
WX_PANEL( aParent )
|
|
{
|
|
wxASSERT( dynamic_cast<SCH_EDIT_FRAME*>( aParent ) );
|
|
|
|
m_frame = aParent;
|
|
|
|
wxBoxSizer* sizer = new wxBoxSizer( wxVERTICAL );
|
|
SetSizer( sizer );
|
|
m_tree = new HIERARCHY_TREE( this );
|
|
|
|
// Make an image list containing small icons
|
|
// All icons are expected having the same size.
|
|
wxBitmap tree_nosel_bm( KiBitmap( BITMAPS::tree_nosel ) );
|
|
wxImageList* imageList = new wxImageList( tree_nosel_bm.GetWidth(), tree_nosel_bm.GetHeight(),
|
|
true, 2 );
|
|
|
|
imageList->Add( tree_nosel_bm );
|
|
imageList->Add( KiBitmap( BITMAPS::tree_sel ) );
|
|
|
|
m_tree->AssignImageList( imageList );
|
|
|
|
sizer->Add( m_tree, 1, wxEXPAND, wxBORDER_NONE, 0 );
|
|
|
|
m_events_bound = false;
|
|
|
|
UpdateHierarchyTree();
|
|
|
|
// Enable selection events
|
|
Bind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
|
|
m_events_bound = true;
|
|
}
|
|
|
|
|
|
HIERARCHY_PANE::~HIERARCHY_PANE()
|
|
{
|
|
Unbind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
}
|
|
|
|
|
|
void HIERARCHY_PANE::buildHierarchyTree( SCH_SHEET_PATH* aList, const wxTreeItemId& aParent )
|
|
{
|
|
std::vector<SCH_ITEM*> sheetChildren;
|
|
aList->LastScreen()->GetSheets( &sheetChildren );
|
|
|
|
for( SCH_ITEM* aItem : sheetChildren )
|
|
{
|
|
SCH_SHEET* sheet = static_cast<SCH_SHEET*>( aItem );
|
|
aList->push_back( sheet );
|
|
|
|
wxString sheetName = formatPageString( sheet->GetFields()[SHEETNAME].GetShownText( false ),
|
|
aList->GetPageNumber() );
|
|
wxTreeItemId child = m_tree->AppendItem( aParent, sheetName, 0, 1 );
|
|
m_tree->SetItemData( child, new TREE_ITEM_DATA( *aList ) );
|
|
|
|
buildHierarchyTree( aList, child );
|
|
aList->pop_back();
|
|
}
|
|
|
|
m_tree->SortChildren( aParent );
|
|
}
|
|
|
|
|
|
void HIERARCHY_PANE::UpdateHierarchySelection()
|
|
{
|
|
bool eventsWereBound = m_events_bound;
|
|
|
|
if( eventsWereBound )
|
|
{
|
|
// Disable selection events
|
|
Unbind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
|
|
m_events_bound = false;
|
|
}
|
|
|
|
bool sheetSelected = false;
|
|
|
|
std::function<void( const wxTreeItemId& )> recursiveDescent =
|
|
[&]( const wxTreeItemId& id )
|
|
{
|
|
wxCHECK_RET( id.IsOk(), wxT( "Invalid tree item" ) );
|
|
|
|
TREE_ITEM_DATA* itemData = static_cast<TREE_ITEM_DATA*>( m_tree->GetItemData( id ) );
|
|
|
|
if( itemData->m_SheetPath == m_frame->GetCurrentSheet() )
|
|
{
|
|
m_tree->EnsureVisible( id );
|
|
m_tree->SetItemBold( id, true );
|
|
}
|
|
else
|
|
{
|
|
m_tree->SetItemBold( id, false );
|
|
}
|
|
|
|
if( itemData->m_SheetPath.Last()->IsSelected() )
|
|
{
|
|
m_tree->EnsureVisible( id );
|
|
m_tree->SelectItem( id );
|
|
sheetSelected = true;
|
|
}
|
|
|
|
wxTreeItemIdValue cookie;
|
|
wxTreeItemId child = m_tree->GetFirstChild( id, cookie );
|
|
|
|
while( child.IsOk() )
|
|
{
|
|
recursiveDescent( child );
|
|
child = m_tree->GetNextChild( id, cookie );
|
|
}
|
|
};
|
|
|
|
recursiveDescent( m_tree->GetRootItem() );
|
|
|
|
if( !sheetSelected && m_tree->GetSelection() )
|
|
m_tree->SelectItem( m_tree->GetSelection(), false );
|
|
|
|
if( eventsWereBound )
|
|
{
|
|
// Enable selection events
|
|
Bind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
|
|
m_events_bound = true;
|
|
}
|
|
}
|
|
|
|
|
|
void HIERARCHY_PANE::UpdateHierarchyTree()
|
|
{
|
|
Freeze();
|
|
|
|
bool eventsWereBound = m_events_bound;
|
|
|
|
if( eventsWereBound )
|
|
{
|
|
// Disable selection events
|
|
Unbind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Unbind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
|
|
m_events_bound = false;
|
|
}
|
|
|
|
m_list.clear();
|
|
m_list.push_back( &m_frame->Schematic().Root() );
|
|
|
|
m_tree->DeleteAllItems();
|
|
|
|
wxTreeItemId root = m_tree->AddRoot( getRootString(), 0, 1 );
|
|
m_tree->SetItemData( root, new TREE_ITEM_DATA( m_list ) );
|
|
|
|
buildHierarchyTree( &m_list, root );
|
|
UpdateHierarchySelection();
|
|
|
|
if( m_tree->ItemHasChildren( root ) )
|
|
m_tree->Expand( root );
|
|
|
|
if( eventsWereBound )
|
|
{
|
|
// Enable selection events
|
|
Bind( wxEVT_TREE_ITEM_ACTIVATED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_SEL_CHANGED, &HIERARCHY_PANE::onSelectSheetPath, this );
|
|
Bind( wxEVT_TREE_ITEM_RIGHT_CLICK, &HIERARCHY_PANE::onRightClick, this );
|
|
|
|
m_events_bound = true;
|
|
}
|
|
|
|
Thaw();
|
|
}
|
|
|
|
|
|
void HIERARCHY_PANE::onSelectSheetPath( wxTreeEvent& aEvent )
|
|
{
|
|
wxTreeItemId itemSel = m_tree->GetSelection();
|
|
TREE_ITEM_DATA* itemData = static_cast<TREE_ITEM_DATA*>( m_tree->GetItemData( itemSel ) );
|
|
|
|
SetCursor( wxCURSOR_ARROWWAIT );
|
|
m_frame->GetToolManager()->RunAction<SCH_SHEET_PATH*>( EE_ACTIONS::changeSheet,
|
|
&itemData->m_SheetPath );
|
|
SetCursor( wxCURSOR_ARROW );
|
|
}
|
|
|
|
|
|
void HIERARCHY_PANE::onRightClick( wxTreeEvent& aEvent )
|
|
{
|
|
wxTreeItemId itemSel = m_tree->GetFocusedItem();
|
|
|
|
if( !itemSel.IsOk() )
|
|
return;
|
|
|
|
TREE_ITEM_DATA* itemData = static_cast<TREE_ITEM_DATA*>( m_tree->GetItemData( itemSel ) );
|
|
|
|
if( !itemData )
|
|
return;
|
|
|
|
wxMenu ctxMenu;
|
|
|
|
ctxMenu.Append( 1, _( "Edit Page Number" ) );
|
|
|
|
if( GetPopupMenuSelectionFromUser( ctxMenu ) == 1 )
|
|
{
|
|
wxString msg;
|
|
wxString sheetPath = itemData->m_SheetPath.PathHumanReadable( false, true );
|
|
wxString pageNumber = itemData->m_SheetPath.GetPageNumber();
|
|
|
|
msg.Printf( _( "Enter page number for sheet path %s" ),
|
|
( sheetPath.Length() > 20 ) ? wxS( " \n" ) + sheetPath + wxT( ": " )
|
|
: sheetPath + wxT( ": " ) );
|
|
|
|
wxTextEntryDialog dlg( m_frame, msg, _( "Edit Sheet Page Number" ), pageNumber );
|
|
|
|
dlg.SetTextValidator( wxFILTER_ALPHANUMERIC ); // No white space.
|
|
|
|
if( dlg.ShowModal() == wxID_OK && dlg.GetValue() != itemData->m_SheetPath.GetPageNumber() )
|
|
{
|
|
SCH_SHEET_PATH parentPath = itemData->m_SheetPath;
|
|
parentPath.pop_back();
|
|
|
|
m_frame->SaveCopyInUndoList( parentPath.LastScreen(), itemData->m_SheetPath.Last(),
|
|
UNDO_REDO::CHANGED, false );
|
|
|
|
itemData->m_SheetPath.SetPageNumber( dlg.GetValue() );
|
|
|
|
if( itemData->m_SheetPath == m_frame->GetCurrentSheet() )
|
|
{
|
|
m_frame->GetScreen()->SetPageNumber( dlg.GetValue() );
|
|
m_frame->OnPageSettingsChange();
|
|
}
|
|
|
|
m_frame->OnModify();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
wxString HIERARCHY_PANE::getRootString()
|
|
{
|
|
SCH_SHEET* rootSheet = &m_frame->Schematic().Root();
|
|
SCH_SHEET_PATH rootPath;
|
|
rootPath.push_back( rootSheet );
|
|
|
|
return formatPageString( rootSheet->GetShownName( false ), rootPath.GetPageNumber() );
|
|
}
|
|
|
|
|
|
wxString HIERARCHY_PANE::formatPageString( const wxString& aName, const wxString& aPage )
|
|
{
|
|
return aName + wxT( " " ) + wxString::Format( _( "(page %s)" ), aPage );
|
|
}
|