sch_design_block_utils.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright The 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, see <https://www.gnu.org/licenses/>.
 */
 
#include <pgm_base.h>
#include <kiway.h>
#include <design_block.h>
#include <design_block_library_adapter.h>
#include <sch_design_block_pane.h>
#include <sch_edit_frame.h>
#include <sch_group.h>
#include <wx/choicdlg.h>
#include <wx/msgdlg.h>
#include <wx/textdlg.h>
#include <wildcards_and_files_ext.h>
#include <paths.h>
#include <env_paths.h>
#include <common.h>
#include <kidialog.h>
#include <confirm.h>
#include <tool/actions.h>
#include <tool/tool_manager.h>
#include <sch_selection_tool.h>
#include <dialogs/dialog_design_block_properties.h>
#include <json_common.h>
 
bool checkOverwriteDb( wxWindow* aFrame, wxString& libname, wxString& newName )
{
    wxString msg = wxString::Format( _( "Design block '%s' already exists in library '%s'." ),
                                     newName.GetData(),
                                     libname.GetData() );
 
    if( OKOrCancelDialog( aFrame, _( "Confirmation" ), msg, _( "Overwrite existing design block?" ), _( "Overwrite" ) )
        != wxID_OK )
    {
        return false;
    }
 
    return true;
}
 
 
bool checkOverwriteDbSchematic( wxWindow* aFrame, const LIB_ID& aLibId )
{
    wxString msg = wxString::Format( _( "Design block '%s' already has a schematic." ),
                                     aLibId.GetUniStringLibItemName() );
 
    if( OKOrCancelDialog( aFrame, _( "Confirmation" ), msg, _( "Overwrite existing schematic?" ), _( "Overwrite" ) )
        != wxID_OK )
    {
        return false;
    }
 
    return true;
}
 
 
bool SCH_EDIT_FRAME::SaveSheetAsDesignBlock( const wxString& aLibraryName, SCH_SHEET_PATH& aSheetPath )
{
    // Make sure the user has selected a library to save into
    if( m_designBlocksPane->GetSelectedLibId().GetLibNickname().empty() )
    {
        DisplayErrorMessage( this, _( "Please select a library to save the design block to." ) );
        return false;
    }
 
    // Just block all attempts to create design blocks with nested sheets at this point
    std::vector<SCH_ITEM*> sheets;
    aSheetPath.LastScreen()->GetSheets( &sheets );
 
    if( !sheets.empty() )
    {
        DisplayErrorMessage( this, _( "Design blocks with nested sheets are not supported." ) );
        return false;
    }
 
    DESIGN_BLOCK blk;
    wxFileName   fn = wxFileNameFromPath( aSheetPath.Last()->GetName() );
 
    blk.SetLibId( LIB_ID( aLibraryName, fn.GetName() ) );
 
    // Copy all fields from the sheet to the design block
    for( SCH_FIELD& field : aSheetPath.Last()->GetFields() )
    {
        if( field.GetId() == FIELD_T::SHEET_NAME || field.GetId() == FIELD_T::SHEET_FILENAME )
            continue;
 
        blk.GetFields()[field.GetUntranslatedName()] = field.GetText();
    }
 
    DIALOG_DESIGN_BLOCK_PROPERTIES dlg( this, &blk );
 
    if( dlg.ShowModal() != wxID_OK )
        return false;
 
    wxString libName = blk.GetLibId().GetLibNickname();
    wxString newName = blk.GetLibId().GetLibItemName();
 
    if( Prj().DesignBlockLibs()->DesignBlockExists( libName, newName ) && !checkOverwriteDb( this, libName, newName ) )
        return false;
 
    // Save a temporary copy of the schematic file, as the plugin is just going to move it
    wxString tempFile = wxFileName::CreateTempFileName( "design_block" );
 
    if( !saveSchematicFile( aSheetPath.Last(), tempFile, true ) )
    {
        DisplayErrorMessage( this, _( "Error saving temporary schematic file to create design block." ) );
        wxRemoveFile( tempFile );
        return false;
    }
 
    blk.SetSchematicFile( tempFile );
 
    bool success = false;
 
    try
    {
        success = Prj().DesignBlockLibs()->SaveDesignBlock( aLibraryName, &blk )
                  == DESIGN_BLOCK_LIBRARY_ADAPTER::SAVE_OK;
    }
    catch( const IO_ERROR& ioe )
    {
        DisplayError( this, ioe.What() );
    }
 
    // Clean up the temporary file
    wxRemoveFile( tempFile );
 
    m_designBlocksPane->RefreshLibs();
    m_designBlocksPane->SelectLibId( blk.GetLibId() );
 
    return success;
}
 
 
bool SCH_EDIT_FRAME::UpdateDesignBlockFromSheet( const LIB_ID& aLibId, SCH_SHEET_PATH& aSheetPath )
{
    // Make sure the user has selected a library to save into
    if( !Prj().DesignBlockLibs()->DesignBlockExists( aLibId.GetLibNickname(), aLibId.GetLibItemName() ) )
    {
        DisplayErrorMessage( this, _( "Please select a design block to save the schematic to." ) );
        return false;
    }
 
    // Just block all attempts to create design blocks with nested sheets at this point
    std::vector<SCH_ITEM*> sheets;
    aSheetPath.LastScreen()->GetSheets( &sheets );
 
    if( !sheets.empty() )
    {
        DisplayErrorMessage( this, _( "Design blocks with nested sheets are not supported." ) );
        return false;
    }
 
    wxString                      error;
    std::unique_ptr<DESIGN_BLOCK> blk;
 
    blk.reset( Prj().DesignBlockLibs()->LoadDesignBlock( aLibId.GetLibNickname(), aLibId.GetLibItemName(), false,
                                                         &error ) );
 
    if( !blk )
    {
        DisplayErrorMessage( this, error );
        return false;
    }
 
    if( !blk->GetSchematicFile().IsEmpty() && !checkOverwriteDbSchematic( this, aLibId ) )
        return false;
 
    // Copy all fields from the sheet to the design block.
    // Note: this will overwrite any existing fields in the design block, but
    // will leave extra fields not in this source sheet alone.
    for( SCH_FIELD& field : aSheetPath.Last()->GetFields() )
    {
        if( field.GetId() == FIELD_T::SHEET_NAME || field.GetId() == FIELD_T::SHEET_FILENAME )
            continue;
 
        blk->GetFields()[field.GetUntranslatedName()] = field.GetText();
    }
 
    DIALOG_DESIGN_BLOCK_PROPERTIES dlg( this, blk.get(), true );
 
    if( dlg.ShowModal() != wxID_OK )
        return false;
 
    // Save a temporary copy of the schematic file, as the plugin is just going to move it
    wxString tempFile = wxFileName::CreateTempFileName( "design_block" );
 
    if( !saveSchematicFile( aSheetPath.Last(), tempFile, true ) )
    {
        DisplayErrorMessage( this, _( "Error saving temporary schematic file to create design block." ) );
        wxRemoveFile( tempFile );
        return false;
    }
 
    blk->SetSchematicFile( tempFile );
 
    bool success = false;
 
    try
    {
        success = Prj().DesignBlockLibs()->SaveDesignBlock( aLibId.GetLibNickname(), blk.get() )
                  == DESIGN_BLOCK_LIBRARY_ADAPTER::SAVE_OK;
    }
    catch( const IO_ERROR& ioe )
    {
        DisplayError( this, ioe.What() );
    }
 
    // Clean up the temporary file
    wxRemoveFile( tempFile );
 
    m_designBlocksPane->RefreshLibs();
    m_designBlocksPane->SelectLibId( blk->GetLibId() );
 
    return success;
}
 
 
bool SCH_EDIT_FRAME::SaveSelectionAsDesignBlock( const wxString& aLibraryName )
{
    // Get all selected items
    SCH_SELECTION selection = m_toolManager->GetTool<SCH_SELECTION_TOOL>()->GetSelection();
 
    if( selection.Empty() )
    {
        DisplayErrorMessage( this, _( "Please select some items to save as a design block." ) );
        return false;
    }
 
    // Make sure the user has selected a library to save into
    if( m_designBlocksPane->GetSelectedLibId().GetLibNickname().empty() )
    {
        DisplayErrorMessage( this, _( "Please select a library to save the design block to." ) );
        return false;
    }
 
    // Just block all attempts to create design blocks with nested sheets at this point
    if( selection.HasType( SCH_SHEET_T ) )
    {
        if( selection.Size() == 1 )
        {
            SCH_SHEET*     sheet = static_cast<SCH_SHEET*>( selection.Front() );
            SCH_SHEET_PATH curPath = GetCurrentSheet();
 
            curPath.push_back( sheet );
            return SaveSheetAsDesignBlock( aLibraryName, curPath );
        }
        else
        {
            DisplayErrorMessage( this, _( "Design blocks with nested sheets are not supported." ) );
            return false;
        }
    }
 
    DESIGN_BLOCK blk;
    SCH_GROUP*   group = nullptr;
 
    if( selection.Size() == 1 && selection.HasType( SCH_GROUP_T ) )
        group = static_cast<SCH_GROUP*>( selection.Front() );
 
    if( group && !group->GetName().IsEmpty() )
        // If the user has selected a single group, they probably want the design block named after the group
        blk.SetLibId( LIB_ID( aLibraryName, group->GetName() ) );
    else
    {
        // Otherwise, use the current screen name
        wxFileName fn = wxFileNameFromPath( GetScreen()->GetFileName() );
        blk.SetLibId( LIB_ID( aLibraryName, fn.GetName() ) );
    }
 
    DIALOG_DESIGN_BLOCK_PROPERTIES dlg( this, &blk );
 
    if( dlg.ShowModal() != wxID_OK )
        return false;
 
    wxString libName = blk.GetLibId().GetLibNickname();
    wxString newName = blk.GetLibId().GetLibItemName();
 
    if( Prj().DesignBlockLibs()->DesignBlockExists( libName, newName ) && !checkOverwriteDb( this, libName, newName ) )
        return false;
 
    // Create a temporary screen
    SCH_SCREEN* tempScreen = new SCH_SCREEN( m_schematic );
 
    // If we have a single group, we want to strip the group and select the children
    if( group )
    {
        selection.Remove( group );
 
        // Don't recurse; if we have a group of groups the user probably intends the inner groups to be saved
        group->RunOnChildren(
                [&]( EDA_ITEM* aItem )
                {
                    selection.Add( aItem );
                },
                RECURSE_MODE::NO_RECURSE );
    }
 
    // Copy the selected items to the temporary screen
    for( EDA_ITEM* item : selection )
    {
        // We need to deep copy since selections of groups will not have the children
        if( item->Type() == SCH_GROUP_T )
        {
            SCH_GROUP* clonedGroup = static_cast<SCH_GROUP*>( item )->DeepClone();
 
            tempScreen->Append( clonedGroup );
 
            clonedGroup->RunOnChildren(
                                        [&]( EDA_ITEM* aItem )
                                        {
                                            tempScreen->Append( static_cast<SCH_ITEM*>( aItem ) );
                                        },
                                        RECURSE_MODE::RECURSE );
        }
        else if( item->Type() == SCH_SYMBOL_T )
        {
            SCH_SYMBOL* clonedSymbol = static_cast<SCH_SYMBOL*>( item->Clone() );
            tempScreen->Append( clonedSymbol );
        }
        else if( item->Type() == SCH_PIN_T || item->Type() == SCH_FIELD_T )
        {
            // Handled as symbol children
            continue;
        }
        else
        {
            EDA_ITEM* copy = item->Clone();
            tempScreen->Append( static_cast<SCH_ITEM*>( copy ) );
        }
    }
 
    // Create a sheet for the temporary screen
    std::unique_ptr<SCH_SHEET> tempSheet = std::make_unique<SCH_SHEET>( m_schematic );
    tempSheet->SetScreen( tempScreen );
 
    // Save a temporary copy of the schematic file, as the plugin is just going to move it
    wxString tempFile = wxFileName::CreateTempFileName( "design_block" );
 
    if( !saveSchematicFile( tempSheet.get(), tempFile, true ) )
    {
        DisplayErrorMessage( this, _( "Error saving temporary schematic file to create design block." ) );
        wxRemoveFile( tempFile );
        return false;
    }
 
    blk.SetSchematicFile( tempFile );
 
    bool success = false;
 
    try
    {
        success = Prj().DesignBlockLibs()->SaveDesignBlock( aLibraryName, &blk )
                  == DESIGN_BLOCK_LIBRARY_ADAPTER::SAVE_OK;
    }
    catch( const IO_ERROR& ioe )
    {
        DisplayError( this, ioe.What() );
    }
 
    if( success && !group )
    {
        SCH_COMMIT  commit( m_toolManager );
        SCH_SCREEN* screen = GetScreen();
 
        SCH_GROUP* newGroup = new SCH_GROUP;
        newGroup->SetParent( screen );
        newGroup->SetName( blk.GetLibId().GetUniStringLibItemName() );
        newGroup->SetDesignBlockLibId( blk.GetLibId() );
 
        int addedCount = 0;
 
        for( EDA_ITEM* edaItem : selection )
        {
            if( !edaItem->IsSCH_ITEM() )
                continue;
 
            SCH_ITEM* item = static_cast<SCH_ITEM*>( edaItem );
 
            if( item->GetParentSymbol() )
                continue;
 
            if( !item->IsGroupableType() )
                continue;
 
            if( EDA_GROUP* existingGroup = item->GetParentGroup() )
                commit.Modify( existingGroup->AsEdaItem(), screen, RECURSE_MODE::NO_RECURSE );
 
            commit.Modify( item, screen, RECURSE_MODE::NO_RECURSE );
            newGroup->AddItem( item );
            addedCount++;
        }
 
        if( addedCount > 0 )
        {
            commit.Add( newGroup, screen );
            commit.Push( _( "Group Items" ) );
 
            m_toolManager->RunAction( ACTIONS::selectionClear );
            m_toolManager->RunAction( ACTIONS::selectItem, newGroup->AsEdaItem() );
        }
        else
        {
            newGroup->RemoveAll();
            delete newGroup;
        }
    }
    else if( success && group && !group->HasDesignBlockLink() )
    {
        SCH_COMMIT commit( m_toolManager );
 
        commit.Modify( group, GetScreen() );
        group->SetDesignBlockLibId( blk.GetLibId() );
 
        commit.Push( _( "Set Group Design Block Link" ) );
    }
 
    // Clean up the temporaries
    wxRemoveFile( tempFile );
 
    m_designBlocksPane->RefreshLibs();
    m_designBlocksPane->SelectLibId( blk.GetLibId() );
 
    return success;
}
 
 
bool SCH_EDIT_FRAME::UpdateDesignBlockFromSelection( const LIB_ID& aLibId )
{
    // Get all selected items
    SCH_SELECTION selection = m_toolManager->GetTool<SCH_SELECTION_TOOL>()->GetSelection();
 
    if( selection.Empty() )
    {
        DisplayErrorMessage( this, _( "Please select some items to save as a design block." ) );
        return false;
    }
 
    // Make sure the user has selected a library to save into
    if( !Prj().DesignBlockLibs()->DesignBlockExists( aLibId.GetLibNickname(), aLibId.GetLibItemName() ) )
    {
        DisplayErrorMessage( this, _( "Please select a design block to save the schematic to." ) );
        return false;
    }
 
    // Just block all attempts to create design blocks with nested sheets at this point
    if( selection.HasType( SCH_SHEET_T ) )
    {
        if( selection.Size() == 1 )
        {
            SCH_SHEET*     sheet = static_cast<SCH_SHEET*>( selection.Front() );
            SCH_SHEET_PATH curPath = GetCurrentSheet();
 
            curPath.push_back( sheet );
            return UpdateDesignBlockFromSheet( aLibId, curPath );
        }
        else
        {
            DisplayErrorMessage( this, _( "Design blocks with nested sheets are not supported." ) );
            return false;
        }
    }
 
    // If the selection is a single group, or contains this block's linked group plus extra items,
    // we want to strip the group and select the children
    SCH_GROUP*             group = nullptr;
    std::vector<SCH_ITEM*> extraItems;
 
    if( selection.Size() == 1 )
    {
        EDA_ITEM* item = selection.Front();
 
        if( item->Type() == SCH_GROUP_T )
            group = static_cast<SCH_GROUP*>( item );
    }
    else
    {
        for( EDA_ITEM* item : selection )
        {
            if( item->Type() != SCH_GROUP_T )
                continue;
 
            if( static_cast<SCH_GROUP*>( item )->GetDesignBlockLibId() != aLibId )
                continue;
 
            if( group )
            {
                DisplayErrorMessage( this, _( "The selection contains more than one group linked to "
                                              "this design block." ) );
                return false;
            }
 
            group = static_cast<SCH_GROUP*>( item );
        }
    }
 
    if( group )
    {
        for( EDA_ITEM* item : selection )
        {
            if( item != group && item->IsSCH_ITEM() )
                extraItems.push_back( static_cast<SCH_ITEM*>( item ) );
        }
 
        selection.Remove( group );
 
        // Don't recurse; if we have a group of groups the user probably intends the inner groups to be saved
        group->RunOnChildren(
                [&]( EDA_ITEM* aItem )
                {
                    selection.Add( aItem );
                },
                RECURSE_MODE::NO_RECURSE );
    }
 
    wxString                      error;
    std::unique_ptr<DESIGN_BLOCK> blk;
 
    blk.reset( Prj().DesignBlockLibs()->LoadDesignBlock( aLibId.GetLibNickname(), aLibId.GetLibItemName(), false,
                                                         &error ) );
 
    if( !blk )
    {
        DisplayErrorMessage( this, error );
        return false;
    }
 
    if( !blk->GetSchematicFile().IsEmpty() && !checkOverwriteDbSchematic( this, aLibId ) )
        return false;
 
    // Create a temporary screen
    SCH_SCREEN* tempScreen = new SCH_SCREEN( m_schematic );
 
    auto cloneAndAdd =
            [&] ( EDA_ITEM* aItem ) -> SCH_ITEM*
            {
                if( !aItem->IsSCH_ITEM() )
                    return nullptr;
 
                SCH_ITEM* copy = static_cast<SCH_ITEM*>( aItem->Clone() );
                tempScreen->Append( static_cast<SCH_ITEM*>( copy ) );
                return copy;
            };
 
    // Copy the selected items to the temporary board
    for( EDA_ITEM* item : selection )
    {
        // Remove parent group membership since we strip the first group layer
        if( SCH_ITEM* copy = cloneAndAdd( item ) )
            copy->SetParentGroup( nullptr );
 
        if( item->Type() == SCH_GROUP_T )
        {
            SCH_GROUP* innerGroup = static_cast<SCH_GROUP*>( item );
 
            // Groups also need their children copied
            innerGroup->RunOnChildren( cloneAndAdd, RECURSE_MODE::RECURSE );
        }
    }
 
    // Create a sheet for the temporary screen
    std::unique_ptr<SCH_SHEET> tempSheet = std::make_unique<SCH_SHEET>( m_schematic );
    tempSheet->SetScreen( tempScreen );
 
    // Save a temporary copy of the schematic file, as the plugin is just going to move it
    wxString tempFile = wxFileName::CreateTempFileName( "design_block" );
 
    if( !saveSchematicFile( tempSheet.get(), tempFile, true ) )
    {
        DisplayErrorMessage( this, _( "Error saving temporary schematic file to create design block." ) );
        wxRemoveFile( tempFile );
        return false;
    }
 
    blk->SetSchematicFile( tempFile );
 
    bool success = false;
 
    try
    {
        success = Prj().DesignBlockLibs()->SaveDesignBlock( aLibId.GetLibNickname(), blk.get() )
                  == DESIGN_BLOCK_LIBRARY_ADAPTER::SAVE_OK;
 
        // If we had a group, extend it with the extra items and reselect it
        if( group )
        {
            selection.Clear();
            selection.Add( group );
 
            SCH_COMMIT  commit( m_toolManager );
            SCH_SCREEN* screen = GetScreen();
            bool        changed = false;
 
            for( SCH_ITEM* item : extraItems )
            {
                if( item->GetParentSymbol() || !item->IsGroupableType() )
                    continue;
 
                if( item->GetParentGroup() == group )
                    continue;
 
                if( EDA_GROUP* existingGroup = item->GetParentGroup() )
                    commit.Modify( existingGroup->AsEdaItem(), screen, RECURSE_MODE::NO_RECURSE );
 
                if( !changed )
                    commit.Modify( group, screen, RECURSE_MODE::NO_RECURSE );
 
                commit.Modify( item, screen, RECURSE_MODE::NO_RECURSE );
                group->AddItem( item );
                changed = true;
            }
 
            // If we didn't have a design block link before, add one for convenience
            if( !group->HasDesignBlockLink() )
            {
                commit.Modify( group, screen, RECURSE_MODE::NO_RECURSE );
                group->SetDesignBlockLibId( aLibId );
                changed = true;
            }
 
            if( changed )
                commit.Push( _( "Update Design Block Group" ) );
        }
    }
    catch( const IO_ERROR& ioe )
    {
        DisplayError( this, ioe.What() );
    }
 
    if( success && !group )
    {
        SCH_COMMIT  commit( m_toolManager );
        SCH_SCREEN* screen = GetScreen();
 
        SCH_GROUP* newGroup = new SCH_GROUP;
        newGroup->SetParent( screen );
        newGroup->SetName( aLibId.GetUniStringLibItemName() );
        newGroup->SetDesignBlockLibId( aLibId );
 
        int addedCount = 0;
 
        for( EDA_ITEM* edaItem : selection )
        {
            if( !edaItem->IsSCH_ITEM() )
                continue;
 
            SCH_ITEM* item = static_cast<SCH_ITEM*>( edaItem );
 
            if( item->GetParentSymbol() )
                continue;
 
            if( !item->IsGroupableType() )
                continue;
 
            if( EDA_GROUP* existingGroup = item->GetParentGroup() )
                commit.Modify( existingGroup->AsEdaItem(), screen, RECURSE_MODE::NO_RECURSE );
 
            commit.Modify( item, screen, RECURSE_MODE::NO_RECURSE );
            newGroup->AddItem( item );
            addedCount++;
        }
 
        if( addedCount > 0 )
        {
            commit.Add( newGroup, screen );
            commit.Push( _( "Group Items" ) );
 
            m_toolManager->RunAction( ACTIONS::selectionClear );
            m_toolManager->RunAction( ACTIONS::selectItem, newGroup->AsEdaItem() );
        }
        else
        {
            newGroup->RemoveAll();
            delete newGroup;
        }
    }
 
    // Clean up the temporaries
    wxRemoveFile( tempFile );
 
    m_designBlocksPane->RefreshLibs();
    m_designBlocksPane->SelectLibId( blk->GetLibId() );
 
    return success;
}