symbol_library_manager.h

Go to the documentation of this file.

/*
 * This program source code file is symbol of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2017 CERN
 * Copyright The KiCad Developers, see AUTHORS.txt for contributors.
 *
 * @author Maciej Suminski <[email protected]>
 *
 * 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 SYMBOLICULAR 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/>.
 */
 
#ifndef SYMBOL_LIBRARY_MANAGER_H
#define SYMBOL_LIBRARY_MANAGER_H
 
#include <map>
#include <list>
#include <deque>
#include <set>
#include <memory>
#include <wx/arrstr.h>
#include <sch_io/sch_io_mgr.h>
#include <sch_screen.h>
#include <libraries/library_table.h>
 
class LIB_SYMBOL;
class LEGACY_SYMBOL_LIB;
class PROGRESS_REPORTER;
class SCH_IO;
class SCH_BASE_FRAME;
class LIB_LOGGER;
 
 
enum class SYMBOL_NAME_FILTER
{
    ALL,
    ROOT_ONLY,
    DERIVED_ONLY
};
 
 
class SYMBOL_BUFFER
{
public:
    SYMBOL_BUFFER( std::unique_ptr<LIB_SYMBOL> aSymbol = nullptr,
                   std::unique_ptr<SCH_SCREEN> aScreen = nullptr );
    ~SYMBOL_BUFFER();
 
    LIB_SYMBOL& GetSymbol() const { return *m_symbol; }
    void        SetSymbol( std::unique_ptr<LIB_SYMBOL> aSymbol );
 
    LIB_SYMBOL& GetOriginal() const { return *m_original; }
    void        SetOriginal( std::unique_ptr<LIB_SYMBOL> aSymbol );
 
    bool        IsModified() const;
    SCH_SCREEN* GetScreen() const { return m_screen.get(); }
 
private:
    std::unique_ptr<SCH_SCREEN> m_screen;
    std::unique_ptr<LIB_SYMBOL> m_symbol;   // Working copy
    std::unique_ptr<LIB_SYMBOL> m_original; // Initial state of the symbol
};
 

class LIB_BUFFER
{
public:
    LIB_BUFFER( const wxString& aLibrary ) :
            m_libName( aLibrary ),
            m_hash( 1 )
    {}
 
    bool IsModified() const
    {
        if( !m_deleted.empty() )
            return true;
 
        for( const std::shared_ptr<SYMBOL_BUFFER>& symbolBuf : m_symbols )
        {
            if( symbolBuf->IsModified() )
                return true;
        }
 
        return false;
    }
 
    int GetHash() const { return m_hash; }

    LIB_SYMBOL* GetSymbol( const wxString& aAlias ) const;

    bool CreateBuffer( std::unique_ptr<LIB_SYMBOL> aCopy, std::unique_ptr<SCH_SCREEN> aScreen );

    bool UpdateBuffer( SYMBOL_BUFFER& aSymbolBuf, const LIB_SYMBOL& aCopy );

    bool DeleteBuffer( const SYMBOL_BUFFER& aSymbolBuf );

    const std::deque<std::shared_ptr<SYMBOL_BUFFER>>& GetDeletedBuffers() const { return m_deleted; }
 
    void ClearDeletedBuffer() { m_deleted.clear(); }

    bool SaveBuffer( SYMBOL_BUFFER& aSymbolBuf, const wxString& aFileName, SCH_IO* aPlugin,
                     bool aBuffer );

    std::shared_ptr<SYMBOL_BUFFER> GetBuffer( const wxString& aAlias ) const;

    const std::deque<std::shared_ptr<SYMBOL_BUFFER>>& GetBuffers() const { return m_symbols; }

    bool HasDerivedSymbols( const wxString& aParentName ) const;

    void GetSymbolNames( wxArrayString&     aSymbolNames,
                         SYMBOL_NAME_FILTER aFilter = SYMBOL_NAME_FILTER::ALL );

    size_t GetDerivedSymbolNames( const wxString& aSymbolName, wxArrayString& aList );
 
private:
    int removeChildSymbols( const SYMBOL_BUFFER& aSymbolBuf );
 
private:
    std::deque<std::shared_ptr<SYMBOL_BUFFER>> m_symbols;
    std::deque<std::shared_ptr<SYMBOL_BUFFER>> m_deleted;   
    const wxString                             m_libName;   
    int                                        m_hash;
};
 

class SYMBOL_LIBRARY_MANAGER
{
public:
    SYMBOL_LIBRARY_MANAGER( SCH_BASE_FRAME& aFrame );
    virtual ~SYMBOL_LIBRARY_MANAGER();
 
    int GetHash() const;
 
    bool HasModifications() const;

    int GetLibraryHash( const wxString& aLibrary ) const;

    wxArrayString GetLibraryNames() const;
 
    std::list<LIB_SYMBOL*> EnumerateSymbols( const wxString& aLibrary ) const;

    bool CreateLibrary( const wxString& aFilePath, LIBRARY_TABLE_SCOPE aScope )
    {
        return addLibrary( aFilePath, true, aScope );
    }

    bool AddLibrary( const wxString& aFilePath, LIBRARY_TABLE_SCOPE aScope )
    {
        return addLibrary( aFilePath, false, aScope );
    }

    bool UpdateSymbol( LIB_SYMBOL* aSymbol, const wxString& aLibrary );

    bool UpdateSymbolAfterRename( LIB_SYMBOL* aSymbol, const wxString& aOldSymbolName,
                                  const wxString& aLibrary );

    bool UpdateLibraryBuffer( const wxString& aLibrary );

    bool RemoveSymbol( const wxString& aSymbolName, const wxString& aLibrary );

    LIB_SYMBOL* GetSymbol( const wxString& aSymbolName, const wxString& aLibrary ) const;

    LIB_SYMBOL* GetBufferedSymbol( const wxString& aSymbolName, const wxString& aLibrary );

    SCH_SCREEN* GetScreen( const wxString& aSymbolName, const wxString& aLibrary );

    SYMBOL_BUFFER* GetBuffer( const wxString& aSymbolName, const wxString& aLibrary );

    bool SymbolExists( const wxString& aSymbolName, const wxString& aLibrary ) const;

    bool SymbolNameInUse( const wxString& aName, const wxString& aLibrary );

    bool LibraryExists( const wxString& aLibrary, bool aCheckEnabled = false ) const;

    bool IsLibraryLoaded( const wxString& aLibrary ) const;

    bool IsLibraryModified( const wxString& aLibrary ) const;

    bool IsSymbolModified( const wxString& aSymbolName, const wxString& aLibrary ) const;
 
    void SetSymbolModified( const wxString& aSymbolName, const wxString& aLibrary );

    bool ClearLibraryModified( const wxString& aLibrary ) const;

    bool ClearSymbolModified( const wxString& aSymbolName, const wxString& aLibrary ) const;

    bool IsLibraryReadOnly( const wxString& aLibrary ) const;

    bool SaveLibrary( const wxString& aLibrary, const wxString& aFileName,
                      SCH_IO_MGR::SCH_FILE_T aFileType = SCH_IO_MGR::SCH_FILE_T::SCH_LEGACY );

    LIB_ID RevertSymbol( const wxString& aSymbolName, const wxString& aLibrary );

    bool RevertLibrary( const wxString& aLibrary );

    bool RevertAll();

    wxString GetUniqueLibraryName() const;
 
    void GetSymbolNames( const wxString& aLibName, wxArrayString& aSymbolNames,
                         SYMBOL_NAME_FILTER aFilter = SYMBOL_NAME_FILTER::ALL );

    size_t GetDerivedSymbolNames( const wxString& aSymbolName, const wxString& aLibraryName, wxArrayString& aList );
 
    size_t GetLibraryCount() const;
 
protected:
    virtual void OnDataChanged() const {}

    static wxString getLibraryName( const wxString& aFilePath );

    bool addLibrary( const wxString& aFilePath, bool aCreate, LIBRARY_TABLE_SCOPE aScope );

    std::set<LIB_SYMBOL*> getOriginalSymbols( const wxString& aLibrary );

    LIB_BUFFER& getLibraryBuffer( const wxString& aLibrary );
 
protected:
    std::map<wxString, LIB_BUFFER> m_libs;       
    SCH_BASE_FRAME&                m_frame;      
    LIB_LOGGER*                    m_logger;
};
 
#endif /* SYMBOL_LIBRARY_MANAGER_H */