component_class_manager.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 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 3 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 <http://www.gnu.org/licenses/>.
 */
 
#ifndef PCBNEW_COMPONENT_CLASS_MANAGER_H
#define PCBNEW_COMPONENT_CLASS_MANAGER_H
 
#include <memory>
#include <unordered_map>
#include <unordered_set>
#include <wx/string.h>
 
#include <board_item.h>
 
/*
 * A lightweight representation of a component class. The membership within
 * m_consituentClasses allows determination of the type of class this is:
 * 
 * m_constituentClasses.size() == 0: This is a null class (no assigment).
 *     m_name is empty.
 * m_constituentClasses.size() == 1: This is an atomic class. The constituent class
 *     pointer refers to itself. m_name contains the name of the atomic class
 * m_constituentClasses.size() > 1: This is a composite class. The constituent class
 *     pointers refer to all atomic members. m_name contains a comma-delimited list of
 *     all atomic member class names.
 */
class COMPONENT_CLASS
{
public:
    COMPONENT_CLASS( const wxString& name ) : m_name( name ) {}
 
    wxString GetName() const;
 
    const wxString& GetFullName() const { return m_name; }
 
    void AddConstituentClass( COMPONENT_CLASS* componentClass );
 
    bool ContainsClassName( const wxString& className ) const;
 
    bool IsEmpty() const;
 
    const std::vector<COMPONENT_CLASS*>& GetConstituentClasses() const
    {
        return m_constituentClasses;
    }
 
private:
    wxString m_name;
 
    std::vector<COMPONENT_CLASS*> m_constituentClasses;
};
 
/*
 * A class to manage Component Classes in a board context
 *
 * This manager owns generated COMPONENT_CLASS objects, and guarantees that pointers to managed
 * objects are valid for the duration of the board lifetime. Note that, in order to maintain this
 * guarantee, there are two methods that must be called when updating the board from the netlist
 * (InitNetlistUpdate and FinishNetlistUpdate).
 */
class COMPONENT_CLASS_MANAGER
{
public:
    COMPONENT_CLASS_MANAGER();
 
    static wxString GetFullClassNameForConstituents( std::unordered_set<wxString>& classNames );
 
    static wxString GetFullClassNameForConstituents( std::vector<wxString>& classNames );
 
    COMPONENT_CLASS* GetEffectiveComponentClass( std::unordered_set<wxString>& classNames );
 
    const COMPONENT_CLASS* GetNoneComponentClass() const { return m_noneClass.get(); }
 
    void InitNetlistUpdate();
 
    void FinishNetlistUpdate();
 
    // All pointers to COMPONENT_CLASS objects will being invalid
    void Reset();
 
    const std::unordered_map<wxString, std::unique_ptr<COMPONENT_CLASS>>& GetClasses() const
    {
        return m_classes;
    }
 
protected:
    std::unordered_map<wxString, std::unique_ptr<COMPONENT_CLASS>> m_classes;
 
    std::unordered_map<wxString, std::unique_ptr<COMPONENT_CLASS>> m_effectiveClasses;
 
    std::unordered_map<wxString, std::unique_ptr<COMPONENT_CLASS>> m_classesCache;
 
    std::unordered_map<wxString, std::unique_ptr<COMPONENT_CLASS>> m_effectiveClassesCache;
 
    std::unique_ptr<COMPONENT_CLASS> m_noneClass;
};
 
#endif