1 // ProgressUtils.h 2 3 #include "StdAfx.h" 4 5 #include "ProgressUtils.h" 6 7 CLocalProgress::CLocalProgress() 8 { 9 ProgressOffset = InSize = OutSize = 0; 10 SendRatio = SendProgress = true; 11 } 12 13 void CLocalProgress::Init(IProgress *progress, bool inSizeIsMain) 14 { 15 _ratioProgress.Release(); 16 _progress = progress; 17 _progress.QueryInterface(IID_ICompressProgressInfo, &_ratioProgress); 18 _inSizeIsMain = inSizeIsMain; 19 } 20 21 STDMETHODIMP CLocalProgress::SetRatioInfo(const UInt64 *inSize, const UInt64 *outSize) 22 { 23 UInt64 inSizeNew = InSize, outSizeNew = OutSize; 24 if (inSize) 25 inSizeNew += (*inSize); 26 if (outSize) 27 outSizeNew += (*outSize); 28 if (SendRatio && _ratioProgress) 29 { 30 RINOK(_ratioProgress->SetRatioInfo(&inSizeNew, &outSizeNew)); 31 } 32 inSizeNew += ProgressOffset; 33 outSizeNew += ProgressOffset; 34 if (SendProgress) 35 return _progress->SetCompleted(_inSizeIsMain ? &inSizeNew : &outSizeNew); 36 return S_OK; 37 } 38 39 HRESULT CLocalProgress::SetCur() 40 { 41 return SetRatioInfo(NULL, NULL); 42 } 43
