23#include <unordered_set>
25#include <wx/datetime.h>
27#include <wx/tokenzr.h>
29#include <boost/algorithm/string.hpp>
45 SCH_IO( wxS(
"Database library" ) ),
62 const wxString& aLibraryPath,
63 const std::map<std::string, UTF8>* aProperties )
65 std::vector<LIB_SYMBOL*> symbols;
69 aSymbolNameList.Add( symbol->GetName() );
74 const wxString& aLibraryPath,
75 const std::map<std::string, UTF8>* aProperties )
77 wxCHECK_RET(
m_adapter,
"Database plugin missing library manager adapter handle!" );
91 if( !powerSymbolsOnly || symbol->
IsPower() )
92 aSymbolList.emplace_back( symbol );
98 const wxString& aAliasName,
99 const std::map<std::string, UTF8>* aProperties )
101 wxCHECK_MSG(
m_adapter,
nullptr,
"Database plugin missing library manager adapter handle!" );
119 std::string tableName;
120 std::string symbolName( aAliasName.ToUTF8() );
126 tableName = sanitizedIt->second.first;
127 symbolName = sanitizedIt->second.second;
133 if( aAliasName.Contains(
'/' ) )
135 tableName = std::string( aAliasName.BeforeFirst(
'/' ).ToUTF8() );
136 symbolName = std::string( aAliasName.AfterFirst(
'/' ).ToUTF8() );
140 std::vector<const DATABASE_LIB_TABLE*> tablesToTry;
145 if( tableName.empty() || tableIter.
name == tableName )
146 tablesToTry.emplace_back( &tableIter );
149 if( tablesToTry.empty() )
151 wxLogTrace(
traceDatabase, wxT(
"LoadSymbol: table '%s' not found in config" ), tableName );
160 if(
m_conn->SelectOne(
table->table, std::make_pair(
table->key_col, symbolName ),
164 wxLogTrace(
traceDatabase, wxT(
"LoadSymbol: SelectOne (%s, %s) found in %s" ),
169 wxLogTrace(
traceDatabase, wxT(
"LoadSymbol: SelectOne (%s, %s) failed for table %s" ),
187 std::set<wxString> tableNames;
191 if( tableNames.count( tableIter.
name ) )
194 aNames.emplace_back( tableIter.
name );
195 tableNames.insert( tableIter.
name );
209 std::back_inserter( aNames ) );
220 if( aErrorMsg && ( !
m_conn || !
m_conn->IsConnected() ) )
235 long long currentTimestampSeconds = wxDateTime::Now().GetValue().GetValue() / 1000;
251 struct CACHE_LIB_GUARD
254 ~CACHE_LIB_GUARD() { *
flag =
false; }
260 std::vector<std::pair<const DATABASE_LIB_TABLE*, std::vector<DATABASE_CONNECTION::ROW>>> tableResults;
261 size_t signature = 0;
265 std::vector<DATABASE_CONNECTION::ROW> results;
269 if( !
m_conn->GetLastError().empty() )
271 wxString msg = wxString::Format(
_(
"Error reading database table %s: %s" ),
283 for(
const auto& [column, value] :
result )
287 if(
const std::string* str = std::any_cast<std::string>( &value ) )
298 if(
const std::string* str = std::any_cast<std::string>( &it->second ) )
301 symbolId.
Parse( *str );
308 if( std::optional<int> libHash =
m_adapter->GetLibraryModifyHash( nickname ) )
315 tableResults.emplace_back( &
table, std::move( results ) );
325 std::map<wxString, std::unique_ptr<LIB_SYMBOL>> newSymbolCache;
326 std::map<wxString, std::pair<std::string, std::string>> newSanitizedNameMap;
328 for(
const auto& [
table, results] : tableResults )
335 std::string rawName = std::any_cast<std::string>(
result.at(
table->key_col ) );
337 std::string sanitizedKey = sanitizedName.
c_str();
339 (
m_settings->m_GloballyUniqueKeys ||
table->name.empty() ) ?
"" : fmt::format(
"{}/",
table->name );
340 std::string sanitizedDisplayName = fmt::format(
"{}{}", prefix, sanitizedKey );
341 wxString
name( sanitizedDisplayName );
343 newSanitizedNameMap[
name] = std::make_pair(
table->name, rawName );
348 newSymbolCache[symbol->GetName()] = std::move( symbol );
367 wxString msg = wxString::Format(
368 _(
"Could not load database library: settings file %s missing or invalid" ),
377 std::string
path( aSettingsPath.ToUTF8() );
390 wxASSERT_MSG( aSettingsPath ==
m_settings->GetFilename(),
391 "Path changed for database library without re-initializing plugin!" );
395 wxLogTrace(
traceDatabase, wxT(
"ensureSettings: no settings but no valid path!" ) );
402 wxCHECK_RET(
m_settings,
"Call ensureSettings before ensureConnection!" );
408 wxString msg = wxString::Format(
409 _(
"Could not load database library: could not connect to database %s (%s)" ),
419 wxCHECK_RET(
m_settings,
"Call ensureSettings before connect()!" );
426 if(
m_settings->m_Source.connection_string.empty() )
435 std::string cs =
m_settings->m_Source.connection_string;
436 std::string basePath( wxFileName(
m_settings->GetFilename() ).GetPath().ToUTF8() );
440 boost::replace_all( cs,
"${CWD}", basePath );
442 m_conn = std::make_unique<DATABASE_CONNECTION>( cs,
m_settings->m_Source.timeout );
445 if( !
m_conn->IsConnected() )
454 std::set<std::string> columns;
456 columns.insert( boost::to_lower_copy( tableIter.
key_col ) );
457 columns.insert( boost::to_lower_copy( tableIter.
footprints_col ) );
458 columns.insert( boost::to_lower_copy( tableIter.
symbols_col ) );
468 columns.insert( boost::to_lower_copy( field.
column ) );
470 m_conn->CacheTableInfo( tableIter.
table, columns );
482 bool val = std::any_cast<bool>( aVal );
485 catch(
const std::bad_any_cast& )
491 int val = std::any_cast<int>( aVal );
492 return static_cast<bool>( val );
494 catch(
const std::bad_any_cast& )
500 wxString strval( std::any_cast<std::string>( aVal ).c_str(), wxConvUTF8 );
502 if( strval.IsEmpty() )
507 for(
const auto& trueVal : { wxS(
"true" ), wxS(
"yes" ), wxS(
"y" ), wxS(
"1" ) } )
509 if( strval.Matches( trueVal ) )
513 for(
const auto& falseVal : { wxS(
"false" ), wxS(
"no" ), wxS(
"n" ), wxS(
"0" ) } )
515 if( strval.Matches( falseVal ) )
519 catch(
const std::bad_any_cast& )
531 std::unique_ptr<LIB_SYMBOL> symbol =
nullptr;
538 std::string symbolIdStr = std::any_cast<std::string>( aRow.at( aTable.
symbols_col ) );
540 symbolId.
Parse( std::any_cast<std::string>( aRow.at( aTable.
symbols_col ) ) );
548 std::unordered_set<wxString>* set;
570 wxT(
"loadSymbolFromRow: cycle detected resolving '%s' "
571 "(row '%s' in table '%s'); skipping recursive load" ),
572 symbolIdStr, aSymbolName, aTable.
name );
576 originalSymbol =
m_adapter->LoadSymbol( symbolId );
582 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: found original symbol '%s'" ),
584 symbol.reset( originalSymbol->
Duplicate() );
585 symbol->SetSourceLibId( symbolId );
589 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: source symbol '%s' is a "
590 "self-reference, will create empty symbol" ),
595 wxLogTrace(
traceDatabase, wxT(
"loadSymboFromRow: source symbol id '%s' is invalid, "
596 "will create empty symbol" ), symbolIdStr );
600 wxLogTrace(
traceDatabase, wxT(
"loadSymboFromRow: source symbol '%s' not found, "
601 "will create empty symbol" ), symbolIdStr );
609 symbol.reset(
new LIB_SYMBOL( aSymbolName ) );
613 symbol->SetName( aSymbolName );
616 LIB_ID libId = symbol->GetLibId();
618 symbol->SetLibId( libId );
619 wxArrayString footprintsList;
623 std::string footprints = std::any_cast<std::string>( aRow.at( aTable.
footprints_col ) );
625 wxString footprintsStr = wxString( footprints.c_str(), wxConvUTF8 );
626 wxStringTokenizer tokenizer( footprintsStr,
";\t\r\n", wxTOKEN_STRTOK );
628 while( tokenizer.HasMoreTokens() )
629 footprintsList.Add( tokenizer.GetNextToken() );
631 if( footprintsList.size() > 0 )
632 symbol->GetFootprintField().SetText( footprintsList[0] );
636 wxLogTrace(
traceDatabase, wxT(
"loadSymboFromRow: footprint field %s not found." ),
645 symbol->SetDescription( value );
650 wxString value( std::any_cast<std::string>( aRow.at( aTable.
properties.
keywords ) ).c_str(),
652 symbol->SetKeyWords( value );
661 footprintsList.push_back( value );
664 symbol->SetFPFilters( footprintsList );
673 symbol->SetExcludedFromSim( *val );
677 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: exclude_from_sim value for %s "
678 "could not be cast to a boolean" ), aSymbolName );
689 symbol->SetExcludedFromBoard( *val );
693 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: exclude_from_board value for %s "
694 "could not be cast to a boolean" ), aSymbolName );
705 symbol->SetExcludedFromBOM( *val );
709 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: exclude_from_bom value for %s "
710 "could not be cast to a boolean" ), aSymbolName );
714 std::vector<SCH_FIELD*> fields;
715 symbol->GetFields( fields );
717 std::unordered_map<wxString, SCH_FIELD*> fieldsMap;
720 fieldsMap[field->GetName()] = field;
722 static const wxString c_valueFieldName( wxS(
"Value" ) );
723 static const wxString c_datasheetFieldName( wxS(
"Datasheet" ) );
724 static const wxString c_footprintFieldName( wxS(
"Footprint" ) );
730 int dbFieldOrdinal = symbol->GetNextFieldOrdinal();
734 if( !aRow.count( mapping.
column ) )
736 wxLogTrace(
traceDatabase, wxT(
"loadSymbolFromRow: field %s not found in result" ),
750 strValue = std::any_cast<std::string>( aRow.at( mapping.
column ) );
752 catch( std::bad_any_cast& )
756 wxString value(
strValue.c_str(), wxConvUTF8 );
758 if( mapping.
name_wx == c_valueFieldName )
760 SCH_FIELD& field = symbol->GetValueField();
770 else if( mapping.
name_wx == c_datasheetFieldName )
772 SCH_FIELD& field = symbol->GetDatasheetField();
790 if( fieldsMap.count( mapping.
name_wx ) )
792 field = fieldsMap[mapping.
name_wx];
799 fieldsMap[mapping.
name_wx] = field;
821 symbol->AddDrawItem( field,
false );
829 symbol->GetDrawItems().sort();
834 symbol->RefreshLibraryTreeCaches();
std::map< std::string, std::any > ROW
Dialog helper object to sit in the inheritance tree between wxDialog and any class written by wxFormB...
virtual void SetVisible(bool aVisible)
A logical library item identifier and consists of various portions much like a URI.
int Parse(const UTF8 &aId, bool aFix=false)
Parse LIB_ID with the information from aId.
bool IsValid() const
Check if this LID_ID is valid.
void SetSubLibraryName(const UTF8 &aName)
const UTF8 & GetLibNickname() const
Return the logical library name portion of a LIB_ID.
static UTF8 FixIllegalChars(const UTF8 &aLibItemName, bool aLib)
Replace illegal LIB_ID item name characters with underscores '_'.
Define a library symbol object.
bool IsPower() const override
virtual LIB_SYMBOL * Duplicate() const
Create a copy of a LIB_SYMBOL and assigns unique KIIDs to the copy and its children.
void SetOrdinal(int aOrdinal)
void SetAutoAdded(bool aAutoAdded)
void SetName(const wxString &aName)
void SetText(const wxString &aText) override
void SetNameShown(bool aShown=true)
void EnumerateSymbolLib(wxArrayString &aSymbolNameList, const wxString &aLibraryPath, const std::map< std::string, UTF8 > *aProperties=nullptr) override
Populate a list of LIB_SYMBOL alias names contained within the library aLibraryPath.
std::unordered_set< wxString > m_inProgressLoads
LIB_IDs whose resolution is in flight, used to break self-referential cycles where a row's Symbols co...
std::unique_ptr< DATABASE_CONNECTION > m_conn
Generally will be null if no valid connection is established.
void GetDefaultSymbolFields(std::vector< wxString > &aNames) override
Retrieves a list of (custom) field names that should be shown by default for this library in the symb...
void ensureSettings(const wxString &aSettingsPath)
bool m_cachePopulated
True once the LIB_SYMBOL cache has been materialized at least once.
bool TestConnection(wxString *aErrorMsg=nullptr)
bool m_inCacheLib
Re-entrancy guard for cacheLib(), tripped when a self-referential load routes back through the adapte...
std::map< wxString, std::unique_ptr< LIB_SYMBOL > > m_nameToSymbolcache
std::map< wxString, std::pair< std::string, std::string > > m_sanitizedNameMap
std::set< wxString > m_defaultShownFields
virtual ~SCH_IO_DATABASE()
size_t m_cacheSignature
Signature of the raw database rows at last materialization; used to skip rebuilding the LIB_SYMBOL ca...
void GetSubLibraryNames(std::vector< wxString > &aNames) override
Retrieves a list of sub-libraries in this library.
std::unique_ptr< DATABASE_LIB_SETTINGS > m_settings
static std::optional< bool > boolFromAny(const std::any &aVal)
SYMBOL_LIBRARY_ADAPTER * m_adapter
long long m_cacheTimestamp
DIALOG_SHIM * CreateConfigurationDialog(wxWindow *aParent) override
void GetAvailableSymbolFields(std::vector< wxString > &aNames) override
Retrieves a list of (custom) field names that are present on symbols in this library.
std::unique_ptr< LIB_SYMBOL > loadSymbolFromRow(const wxString &aSymbolName, const DATABASE_LIB_TABLE &aTable, const DATABASE_CONNECTION::ROW &aRow)
std::set< wxString > m_customFields
LIB_SYMBOL * LoadSymbol(const wxString &aLibraryPath, const wxString &aAliasName, const std::map< std::string, UTF8 > *aProperties=nullptr) override
Load a LIB_SYMBOL object having aPartName from the aLibraryPath containing a library format that this...
SCH_IO(const wxString &aName)
static const char * PropPowerSymsOnly
An 8 bit string that is assuredly encoded in UTF8, and supplies special conversion support to and fro...
const char * c_str() const
const char *const traceDatabase
static constexpr void hash_combine(std::size_t &seed)
This is a dummy function to take the final case of hash_combine below.
#define THROW_IO_ERROR(msg)
macro which captures the "call site" values of FILE_, __FUNCTION & LINE
static std::string strValue(double aValue)
bool visible_in_chooser
Whether the column is shown by default in the chooser.
std::string column
Database column name.
bool inherit_properties
Whether or not to inherit properties from symbol field.
bool visible_on_add
Whether to show the field when placing the symbol.
bool show_name
Whether or not to show the field name as well as its value.
wxString name_wx
KiCad field name (converted)
A database library table will be mapped to a sub-library provided by the database library entry in th...
std::string key_col
Unique key column name (will form part of the LIB_ID)
std::string name
KiCad library nickname (will form part of the LIB_ID)
std::string symbols_col
Column name containing KiCad symbol refs.
std::string footprints_col
Column name containing KiCad footprint refs.
std::vector< DATABASE_FIELD_MAPPING > fields
std::string table
Database table to pull content from.
MAPPABLE_SYMBOL_PROPERTIES properties
std::string footprint_filters
std::string exclude_from_sim
std::string exclude_from_board
std::string exclude_from_bom
@ USER
The field ID hasn't been set yet; field is invalid.
std::vector< std::vector< std::string > > table
wxString result
Test unit parsing edge cases and error handling.