pcbnew_action_plugins.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2017-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
 */

#include "pcbnew_action_plugins.h"
#include <bitmaps.h>
#include <board.h>
#include <board_commit.h>
#include <footprint.h>
#include <pcb_track.h>
#include <zone.h>
#include <menus_helpers.h>
#include <pcbnew_settings.h>
#include <tool/action_menu.h>
#include <tool/action_toolbar.h>
#include <wx/msgdlg.h>
#include "../../scripting/python_scripting.h"

PYTHON_ACTION_PLUGIN::PYTHON_ACTION_PLUGIN( PyObject* aAction )
{
 PyLOCK lock;

 m_PyAction = aAction;
 Py_XINCREF( aAction );
}


PYTHON_ACTION_PLUGIN::~PYTHON_ACTION_PLUGIN()
{
 PyLOCK lock;

 Py_XDECREF( m_PyAction );
}


PyObject* PYTHON_ACTION_PLUGIN::CallMethod( const char* aMethod, PyObject* aArglist )
{
 PyLOCK lock;

 PyErr_Clear();

 // pFunc is a new reference to the desired method
 PyObject* pFunc = PyObject_GetAttrString( m_PyAction, aMethod );

 if( pFunc && PyCallable_Check( pFunc ) )
 {
 PyObject* result = PyObject_CallObject( pFunc, aArglist );

 if( PyErr_Occurred() )
 {
 wxMessageBox( PyErrStringWithTraceback(),
 _( "Exception on python action plugin code" ),
 wxICON_ERROR | wxOK );
 }

 if( result )
 {
 Py_XDECREF( pFunc );
 return result;
 }
 }
 else
 {
 wxString msg = wxString::Format( _( "Method '%s' not found, or not callable" ), aMethod );
 wxMessageBox( msg, _( "Unknown Method" ), wxICON_ERROR | wxOK );
 }

 if( pFunc )
 {
 Py_XDECREF( pFunc );
 }

 return nullptr;
}


wxString PYTHON_ACTION_PLUGIN::CallRetStrMethod( const char* aMethod, PyObject* aArglist )
{
 wxString ret;
 PyLOCK lock;

 PyObject* result = CallMethod( aMethod, aArglist );

 ret = PyStringToWx( result );
 Py_XDECREF( result );

 return ret;
}


wxString PYTHON_ACTION_PLUGIN::GetCategoryName()
{
 PyLOCK lock;

 return CallRetStrMethod( "GetCategoryName" );
}


wxString PYTHON_ACTION_PLUGIN::GetName()
{
 PyLOCK lock;

 return CallRetStrMethod( "GetName" );
}


wxString PYTHON_ACTION_PLUGIN::GetDescription()
{
 PyLOCK lock;

 return CallRetStrMethod( "GetDescription" );
}


bool PYTHON_ACTION_PLUGIN::GetShowToolbarButton()
{
 PyLOCK lock;

 PyObject* result = CallMethod( "GetShowToolbarButton");

 return PyObject_IsTrue(result);
}


wxString PYTHON_ACTION_PLUGIN::GetIconFileName( bool aDark )
{
 PyLOCK lock;

 PyObject* arglist = Py_BuildValue( "(i)", static_cast<int>( aDark ) );

 wxString result = CallRetStrMethod( "GetIconFileName", arglist );

 Py_DECREF( arglist );

 return result;
}


wxString PYTHON_ACTION_PLUGIN::GetPluginPath()
{
 PyLOCK lock;

 return CallRetStrMethod( "GetPluginPath" );
}


void PYTHON_ACTION_PLUGIN::Run()
{
 PyLOCK lock;

 CallMethod( "Run" );
}


void* PYTHON_ACTION_PLUGIN::GetObject()
{
 return (void*) m_PyAction;
}


void PYTHON_ACTION_PLUGINS::register_action( PyObject* aPyAction )
{
 PYTHON_ACTION_PLUGIN* fw = new PYTHON_ACTION_PLUGIN( aPyAction );

 fw->register_action();
}


void PYTHON_ACTION_PLUGINS::deregister_action( PyObject* aPyAction )
{
 // deregister also destroys the previously created "PYTHON_ACTION_PLUGIN object"
 ACTION_PLUGINS::deregister_object( (void*) aPyAction );
}


void PCB_EDIT_FRAME::OnActionPluginMenu( wxCommandEvent& aEvent )
{
 ACTION_PLUGIN* actionPlugin = ACTION_PLUGINS::GetActionByMenu( aEvent.GetId() );

 if( actionPlugin )
 RunActionPlugin( actionPlugin );
}


void PCB_EDIT_FRAME::OnActionPluginButton( wxCommandEvent& aEvent )
{
 ACTION_PLUGIN* actionPlugin = ACTION_PLUGINS::GetActionByButton( aEvent.GetId() );

 if( actionPlugin )
 RunActionPlugin( actionPlugin );
}

void PCB_EDIT_FRAME::RunActionPlugin( ACTION_PLUGIN* aActionPlugin )
{

 PICKED_ITEMS_LIST itemsList;
 BOARD* currentPcb = GetBoard();
 bool fromEmpty = false;

 // Append tracks:
 for( PCB_TRACK* item : currentPcb->Tracks() )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::CHANGED );
 itemsList.PushItem( picker );
 }

 // Append footprints:
 for( FOOTPRINT* item : currentPcb->Footprints() )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::CHANGED );
 itemsList.PushItem( picker );
 }

 // Append drawings
 for( BOARD_ITEM* item : currentPcb->Drawings() )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::CHANGED );
 itemsList.PushItem( picker );
 }

 // Append zones outlines
 for( ZONE* zone : currentPcb->Zones() )
 {
 ITEM_PICKER picker( nullptr, zone, UNDO_REDO::CHANGED );
 itemsList.PushItem( picker );
 }

 if( itemsList.GetCount() > 0 )
 SaveCopyInUndoList( itemsList, UNDO_REDO::CHANGED );
 else
 fromEmpty = true;

 itemsList.ClearItemsList();

 BOARD_COMMIT commit( this );

 // Execute plugin itself...
 ACTION_PLUGINS::SetActionRunning( true );
 aActionPlugin->Run();
 ACTION_PLUGINS::SetActionRunning( false );

 // Get back the undo buffer to fix some modifications
 PICKED_ITEMS_LIST* oldBuffer = nullptr;

 if( fromEmpty )
 {
 oldBuffer = new PICKED_ITEMS_LIST();
 }
 else
 {
 oldBuffer = PopCommandFromUndoList();
 wxASSERT( oldBuffer );
 }

 // Try do discover what was modified
 PICKED_ITEMS_LIST deletedItemsList;

 // The list of existing items after running the action script
 std::set<BOARD_ITEM*> currItemList;

 // Append tracks:
 for( PCB_TRACK* item : currentPcb->Tracks() )
 currItemList.insert( item );

 // Append footprints:
 for( FOOTPRINT* item : currentPcb->Footprints() )
 currItemList.insert( item );

 // Append drawings
 for( BOARD_ITEM* item : currentPcb->Drawings() )
 currItemList.insert( item );

 // Append zones outlines
 for( ZONE* zone : currentPcb->Zones() )
 currItemList.insert( zone );

 // Found deleted items
 for( unsigned int i = 0; i < oldBuffer->GetCount(); i++ )
 {
 BOARD_ITEM* item = (BOARD_ITEM*) oldBuffer->GetPickedItem( i );
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::DELETED );

 wxASSERT( item );

 if( currItemList.find( item ) == currItemList.end() )
 {
 deletedItemsList.PushItem( picker );
 commit.Removed( item );
 }
 }

 // Mark deleted elements in undolist

 for( unsigned int i = 0; i < deletedItemsList.GetCount(); i++ )
 {
 oldBuffer->PushItem( deletedItemsList.GetItemWrapper( i ) );
 }

 // Find new footprints
 for( FOOTPRINT* item : currentPcb->Footprints() )
 {
 if( !oldBuffer->ContainsItem( item ) )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::NEWITEM );
 oldBuffer->PushItem( picker );
 commit.Added( item );
 }
 }

 for( PCB_TRACK* item : currentPcb->Tracks() )
 {
 if( !oldBuffer->ContainsItem( item ) )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::NEWITEM );
 oldBuffer->PushItem( picker );
 commit.Added( item );
 }
 }

 for( BOARD_ITEM* item : currentPcb->Drawings() )
 {
 if( !oldBuffer->ContainsItem( item ) )
 {
 ITEM_PICKER picker( nullptr, item, UNDO_REDO::NEWITEM );
 oldBuffer->PushItem( picker );
 commit.Added( item );
 }
 }

 for( ZONE* zone : currentPcb->Zones() )
 {
 if( !oldBuffer->ContainsItem( zone ) )
 {
 ITEM_PICKER picker( nullptr, zone, UNDO_REDO::NEWITEM );
 oldBuffer->PushItem( picker );
 commit.Added( zone );
 }
 }


 if( oldBuffer->GetCount() )
 {
 OnModify();
 PushCommandToUndoList( oldBuffer );
 }
 else
 {
 delete oldBuffer;
 }

 // Apply changes, UndoList already handled
 commit.Push( _( "Apply action script" ), false );
 ActivateGalCanvas();
}


void PCB_EDIT_FRAME::buildActionPluginMenus( ACTION_MENU* actionMenu )
{
 if( !actionMenu ) // Should not occur.
 return;

 for( int ii = 0; ii < ACTION_PLUGINS::GetActionsCount(); ii++ )
 {
 wxMenuItem* item;
 ACTION_PLUGIN* ap = ACTION_PLUGINS::GetAction( ii );
 const wxBitmap& bitmap = ap->iconBitmap.IsOk() ? ap->iconBitmap :
 KiBitmap( BITMAPS::puzzle_piece );

 item = AddMenuItem( actionMenu, wxID_ANY, ap->GetName(), ap->GetDescription(), bitmap );

 Connect( item->GetId(), wxEVT_COMMAND_MENU_SELECTED,
 wxCommandEventHandler( PCB_EDIT_FRAME::OnActionPluginMenu ) );

 ACTION_PLUGINS::SetActionMenu( ii, item->GetId() );
 }
}


void PCB_EDIT_FRAME::AddActionPluginTools()
{
 bool need_separator = true;
 const std::vector<ACTION_PLUGIN*>& orderedPlugins = GetOrderedActionPlugins();

 for( ACTION_PLUGIN* ap : orderedPlugins )
 {
 if( GetActionPluginButtonVisible( ap->GetPluginPath(), ap->GetShowToolbarButton() ) )
 {
 if( need_separator )
 {
 m_mainToolBar->AddScaledSeparator( this );
 need_separator = false;
 }

 // Add button
 wxBitmap bitmap;

 if ( ap->iconBitmap.IsOk() )
 bitmap = KiScaledBitmap( ap->iconBitmap, this );
 else
 bitmap = KiScaledBitmap( BITMAPS::puzzle_piece, this );

 wxAuiToolBarItem* button = m_mainToolBar->AddTool( wxID_ANY, wxEmptyString,
 bitmap, ap->GetName() );

 Connect( button->GetId(), wxEVT_COMMAND_MENU_SELECTED,
 wxCommandEventHandler( PCB_EDIT_FRAME::OnActionPluginButton ) );

 // Link action plugin to button
 ACTION_PLUGINS::SetActionButton( ap, button->GetId() );
 }
 }
}


std::vector<ACTION_PLUGIN*> PCB_EDIT_FRAME::GetOrderedActionPlugins()
{
 std::vector<ACTION_PLUGIN*> plugins;
 std::vector<ACTION_PLUGIN*> orderedPlugins;

 for( int i = 0; i < ACTION_PLUGINS::GetActionsCount(); i++ )
 plugins.push_back( ACTION_PLUGINS::GetAction( i ) );

 // First add plugins that have entries in settings
 for( const auto& pair : m_settings->m_VisibleActionPlugins )
 {
 auto loc = std::find_if( plugins.begin(), plugins.end(),
 [pair] ( ACTION_PLUGIN* plugin )
 {
 return plugin->GetPluginPath() == pair.first;
 } );

 if( loc != plugins.end() )
 {
 orderedPlugins.push_back( *loc );
 plugins.erase( loc );
 }
 }

 // Now append new plugins that have not been configured yet
 for( auto remaining_plugin : plugins )
 orderedPlugins.push_back( remaining_plugin );

 return orderedPlugins;
}


bool PCB_EDIT_FRAME::GetActionPluginButtonVisible( const wxString& aPluginPath,
 bool aPluginDefault )
{
 auto& settings = m_settings->m_VisibleActionPlugins;

 for( const auto& entry : settings )
 {
 if( entry.first == aPluginPath )
 return entry.second;
 }

 // Plugin is not in settings, return default.
 return aPluginDefault;
}