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() ) )
 {
 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( wxT( "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;
}