mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-07 18:06:55 -06:00
85 lines
2.2 KiB
C++
Executable File
85 lines
2.2 KiB
C++
Executable File
// BZip2Update.cpp
|
|
|
|
#include "StdAfx.h"
|
|
|
|
#include "../../Common/ProgressUtils.h"
|
|
#include "Windows/PropVariant.h"
|
|
|
|
#include "BZip2Update.h"
|
|
|
|
#ifdef COMPRESS_BZIP2
|
|
#include "../../Compress/BZip2/BZip2Encoder.h"
|
|
#else
|
|
// {23170F69-40C1-278B-0402-020000000100}
|
|
DEFINE_GUID(CLSID_CCompressBZip2Encoder,
|
|
0x23170F69, 0x40C1, 0x278B, 0x04, 0x02, 0x02, 0x00, 0x00, 0x00, 0x01, 0x00);
|
|
#include "../Common/CoderLoader.h"
|
|
extern CSysString GetBZip2CodecPath();
|
|
#endif
|
|
|
|
namespace NArchive {
|
|
namespace NBZip2 {
|
|
|
|
HRESULT UpdateArchive(UInt64 unpackSize,
|
|
ISequentialOutStream *outStream,
|
|
int indexInClient,
|
|
UInt32 dictionary,
|
|
UInt32 numPasses,
|
|
#ifdef COMPRESS_MT
|
|
UInt32 numThreads,
|
|
#endif
|
|
IArchiveUpdateCallback *updateCallback)
|
|
{
|
|
RINOK(updateCallback->SetTotal(unpackSize));
|
|
UInt64 complexity = 0;
|
|
RINOK(updateCallback->SetCompleted(&complexity));
|
|
|
|
CMyComPtr<ISequentialInStream> fileInStream;
|
|
|
|
RINOK(updateCallback->GetStream(indexInClient, &fileInStream));
|
|
|
|
CLocalProgress *localProgressSpec = new CLocalProgress;
|
|
CMyComPtr<ICompressProgressInfo> localProgress = localProgressSpec;
|
|
localProgressSpec->Init(updateCallback, true);
|
|
|
|
#ifndef COMPRESS_BZIP2
|
|
CCoderLibrary lib;
|
|
#endif
|
|
CMyComPtr<ICompressCoder> encoder;
|
|
#ifdef COMPRESS_BZIP2
|
|
encoder = new NCompress::NBZip2::CEncoder;
|
|
#else
|
|
RINOK(lib.LoadAndCreateCoder(GetBZip2CodecPath(),
|
|
CLSID_CCompressBZip2Encoder, &encoder));
|
|
#endif
|
|
|
|
CMyComPtr<ICompressSetCoderProperties> setCoderProperties;
|
|
encoder.QueryInterface(IID_ICompressSetCoderProperties, &setCoderProperties);
|
|
if (setCoderProperties)
|
|
{
|
|
NWindows::NCOM::CPropVariant properties[] =
|
|
{
|
|
dictionary,
|
|
numPasses
|
|
#ifdef COMPRESS_MT
|
|
, numThreads
|
|
#endif
|
|
};
|
|
PROPID propIDs[] =
|
|
{
|
|
NCoderPropID::kDictionarySize,
|
|
NCoderPropID::kNumPasses
|
|
#ifdef COMPRESS_MT
|
|
, NCoderPropID::kNumThreads
|
|
#endif
|
|
};
|
|
RINOK(setCoderProperties->SetCoderProperties(propIDs, properties, sizeof(propIDs) / sizeof(propIDs[0])));
|
|
}
|
|
|
|
RINOK(encoder->Code(fileInStream, outStream, NULL, NULL, localProgress));
|
|
|
|
return updateCallback->SetOperationResult(NArchive::NUpdate::NOperationResult::kOK);
|
|
}
|
|
|
|
}}
|