asset_archive.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2021 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/>.
 */
 
#include <wx/tarstrm.h>
#include <wx/wfstream.h>
#include <wx/zstream.h>
 
#include <asset_archive.h>
 
 
ASSET_ARCHIVE::ASSET_ARCHIVE( const wxString& aFilePath, bool aLoadNow ) :
        m_filePath( aFilePath )
{
    if( aLoadNow )
        Load();
}
 
 
bool ASSET_ARCHIVE::Load()
{
    // We don't support hot-reloading yet
    if( !m_fileInfoCache.empty() )
        return false;
 
    wxFFileInputStream zipFile( m_filePath );
 
    if( !zipFile.IsOk() )
        return false;
 
    wxZlibInputStream stream( zipFile, wxZLIB_GZIP );
    wxTarInputStream tarStream( stream );
    wxTarEntry* entry;
 
    // Avoid realloc while reading: we're not going to get better than 2:1 compression
    m_cache.resize( 2 * zipFile.GetLength() );
 
    size_t offset = 0;
 
    while( ( entry = tarStream.GetNextEntry() ) != nullptr )
    {
        if( entry->IsDir() )
        {
            delete entry;
            continue;
        }
 
        size_t length = entry->GetSize();
 
        FILE_INFO fi;
        fi.offset = offset;
        fi.length = length;
 
        if( offset + length > m_cache.size() )
            m_cache.resize( m_cache.size() * 2 );
 
        tarStream.Read( &m_cache[offset], length );
 
        m_fileInfoCache[entry->GetName()] = fi;
 
        offset += length;
 
        delete entry;
    }
 
    m_cache.resize( offset );
 
    return true;
}
 
 
long ASSET_ARCHIVE::GetFileContents( const wxString& aFilePath, const unsigned char* aDest,
                                     size_t aMaxLen )
{
    wxFAIL_MSG( wxS( "Unimplemented" ) );
    return 0;
}
 
 
long ASSET_ARCHIVE::GetFilePointer( const wxString& aFilePath, const unsigned char** aDest )
{
    if( aFilePath.IsEmpty() )
        return -1;
 
    wxASSERT( aDest );
 
    if( !m_fileInfoCache.count( aFilePath ) )
        return -1;
 
    const FILE_INFO& fi = m_fileInfoCache.at( aFilePath );
 
    *aDest = &m_cache[fi.offset];
 
    return fi.length;
}