kicad/qa/unittests/pcbnew/group_saveload.cpp

270 lines
8.7 KiB
C++

/*
* This program source code file is part of KiCad, a free EDA CAD application.
*
* Copyright (C) 2020 Joshua Redstone redstone at gmail.com
* Copyright (C) 1992-2020 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 <bitset>
#include <string>
#include <boost/filesystem.hpp>
#include <board.h>
#include <footprint.h>
#include <pcb_group.h>
#include <pcb_text.h>
#include <common.h>
#include <pcbnew_utils/board_construction_utils.h>
#include <pcbnew_utils/board_file_utils.h>
#include <qa_utils/wx_utils/unit_test_utils.h>
BOOST_AUTO_TEST_SUITE( GroupSaveLoad )
// The tests below set up a test case with a spec for the set of groups to create.
// A group can contain members from this list of candidates.
enum ItemType
{
TEXT0,
TEXT1,
TEXT2,
TEXT3,
TEXT4,
TEXT5,
TEXT6,
TEXT7,
TEXT8,
REMOVED_TEXT, // Text not added to board
GROUP0,
GROUP1,
GROUP2,
NAME_GROUP3,
NAME_GROUP4,
NAME_GROUP3_DUP, // Group with name identical to NAME_GROUP3
REMOVED_GROUP, // Group not added to board
NUM_ITEMS
};
// The objects associated with item REMOVED_TEXT and REMOVED_GROUP are not added to the board,
// so they are not cleaned up when the board is deleted. These pointers stores the objects
// so they can be deleted once they are done being used.
static PCB_TEXT* s_removedText = nullptr;
static PCB_GROUP* s_removedGroup = nullptr;
/*
* Takes a vector of group specifications for groups to create.
* Each group is a vector of which ItemTypes to put in the group.
* The first group corresponds to GROUP0, the second to GROUP1, and os on.
*/
std::unique_ptr<BOARD> createBoard( const std::vector<std::vector<ItemType>>& spec )
{
std::unique_ptr<BOARD> board = std::make_unique<BOARD>();
std::vector<BOARD_ITEM*> items;
// Create text items and add to board.
for( int idx = 0; idx <= REMOVED_TEXT; idx++ )
{
PCB_TEXT* textItem = new PCB_TEXT( board.get() );
textItem->SetText( wxString::Format( _( "some text-%d" ), idx ) );
// Don't add REMOVED_TEXT to the board
if( idx < REMOVED_TEXT )
board->Add( textItem );
items.push_back( textItem );
}
// Create groups
for( int idx = 0; idx < ( NUM_ITEMS - GROUP0 ); idx++ )
{
PCB_GROUP* gr = new PCB_GROUP( board.get() );
if( idx >= ( NAME_GROUP3 - GROUP0 ) )
{
wxString name = wxString::Format( _( "group-%d" ),
( idx == ( NAME_GROUP3_DUP - GROUP0 ) ) ? 3 : idx );
gr->SetName( name );
BOOST_CHECK_EQUAL( gr->GetName(), name );
}
items.push_back( gr );
}
std::bitset<NUM_ITEMS> used;
// Populate groups based on spec
for( int offset = 0; offset < ( NUM_ITEMS - GROUP0 ); offset++ )
{
int groupIdx = GROUP0 + offset;
PCB_GROUP* group = static_cast<PCB_GROUP*>( items[groupIdx] );
if( offset < spec.size() )
{
const std::vector<ItemType>& groupSpec = spec[offset];
for( ItemType item : groupSpec )
{
used.set( static_cast<size_t>( item ) );
group->AddItem( items[item] );
}
BOOST_CHECK_EQUAL( group->GetItems().size(), groupSpec.size() );
board->Add( group );
}
else if( groupIdx != REMOVED_GROUP && used.test( groupIdx ) )
{
// This group is used in another group, so it must be on the board
board->Add( group );
}
else if( groupIdx != REMOVED_GROUP )
{
// If the group isn't used, delete it
delete group;
}
}
// Delete the removed text item if it isn't used
if( used.test( REMOVED_TEXT ) )
s_removedText = static_cast<PCB_TEXT*>( items[REMOVED_TEXT] );
else
delete items[REMOVED_TEXT];
// Delete the removed group item if it isn't used
if( used.test( REMOVED_GROUP ) )
s_removedGroup = static_cast<PCB_GROUP*>( items[REMOVED_GROUP] );
else
delete items[REMOVED_GROUP];
BOOST_TEST_CHECKPOINT( "Returning fresh board" );
return board;
}
// Check if two groups are identical by comparing the fields (by Uuid).
void testGroupEqual( const PCB_GROUP& group1, const PCB_GROUP& group2 )
{
BOOST_CHECK_EQUAL( group1.m_Uuid.AsString(), group2.m_Uuid.AsString() );
BOOST_CHECK_EQUAL( group1.GetName(), group2.GetName() );
const std::unordered_set<BOARD_ITEM*>& items1 = group1.GetItems();
const std::unordered_set<BOARD_ITEM*>& items2 = group2.GetItems();
BOOST_CHECK_EQUAL( items1.size(), items2.size() );
// Test that the sets items1 and items2 are identical, by checking m_Uuid
for( BOARD_ITEM* item1 : items1 )
{
auto cmp = [&]( BOARD_ITEM* elem )
{
return elem->m_Uuid.AsString() == item1->m_Uuid.AsString();
};
auto item2 = std::find_if( items2.begin(), items2.end(), cmp );
BOOST_CHECK( item2 != items2.end() );
}
}
// Check if two GROUPS are identical by comparing the groups in each of them.
void testGroupsEqual( const GROUPS& groups1, const GROUPS& groups2 )
{
BOOST_CHECK_EQUAL( groups1.size(), groups2.size() );
for( PCB_GROUP* group1 : groups1 )
{
auto cmp = [&]( BOARD_ITEM* elem )
{
return elem->m_Uuid.AsString() == group1->m_Uuid.AsString();
};
auto group2 = std::find_if( groups2.begin(), groups2.end(), cmp );
BOOST_CHECK( group2 != groups2.end() );
testGroupEqual( *group1, **group2 );
}
}
/*
* Create board based on spec, save it to a file, load it, and make sure the
* groups in the resulting board are the same as the groups we started with.
*/
void testSaveLoad( const std::vector<std::vector<ItemType>>& spec )
{
std::unique_ptr<BOARD> board1 = createBoard( spec );
auto path = boost::filesystem::temp_directory_path() / "group_saveload_tst.kicad_pcb";
::KI_TEST::DumpBoardToFile( *board1, path.string() );
std::unique_ptr<BOARD> board2 = ::KI_TEST::ReadBoardFromFileOrStream( path.string() );
testGroupsEqual( board1->Groups(), board2->Groups() );
}
// Test saving & loading of a few configurations
BOOST_AUTO_TEST_CASE( HealthyGroups )
{
// Test board with no groups
testSaveLoad( {} );
// Single group
testSaveLoad( { { TEXT0 } } );
testSaveLoad( { { TEXT0, TEXT1 } } );
// Two groups
testSaveLoad( { { TEXT0, TEXT1 }, { TEXT2, TEXT3 } } );
testSaveLoad( { { TEXT0, TEXT1 }, { TEXT2, GROUP0 } } );
// Subgroups by no cycle
testSaveLoad( { { TEXT0, GROUP1 }, { TEXT2 }, { TEXT3, GROUP0 } } );
testSaveLoad( { { TEXT0 }, { TEXT2 }, { GROUP1, GROUP0 } } );
testSaveLoad( { { TEXT0 }, { TEXT1 }, { TEXT2, NAME_GROUP3 }, { TEXT3 } } );
testSaveLoad( { { TEXT0 }, { TEXT1 }, { TEXT2, NAME_GROUP3 }, { TEXT3, GROUP0 } } );
testSaveLoad( { { TEXT0 }, { TEXT1 }, { TEXT2 }, { TEXT3 }, { NAME_GROUP3, GROUP0 } } );
}
BOOST_AUTO_TEST_CASE( InvalidGroups )
{
// A cycle
std::unique_ptr<BOARD> board1 = createBoard( { { TEXT0, GROUP1 }, { TEXT2, GROUP0 } } );
BOOST_CHECK_EQUAL( board1->GroupsSanityCheck(), "Cycle detected in group membership" );
// More complex cycle
board1 = createBoard( { { TEXT0, GROUP1 }, { TEXT1 }, { TEXT2, NAME_GROUP4 },
{ TEXT3, GROUP2 }, { TEXT4, NAME_GROUP3 } } );
BOOST_CHECK_EQUAL( board1->GroupsSanityCheck(), "Cycle detected in group membership" );
// Delete the removed group since the test is over
board1.reset( nullptr );
delete s_removedGroup;
s_removedGroup = nullptr;
// Delete the removed text since the test is over
board1.reset( nullptr );
delete s_removedText;
s_removedText = nullptr;
}
BOOST_AUTO_TEST_SUITE_END()