Browse Source
network.js file moved to main thread and `PDFNetworkStream` implemented at worker thread, that is used to ask for data whenever worker needs.
Mukul Mishra
8 years ago
9 changed files with 481 additions and 238 deletions
@ -0,0 +1,243 @@
@@ -0,0 +1,243 @@
|
||||
/* Copyright 2012 Mozilla Foundation |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
import { assert, createPromiseCapability } from '../shared/util'; |
||||
|
||||
/** @implements {IPDFStream} */ |
||||
var PDFDataTransportStream = (function PDFDataTransportStreamClosure() { |
||||
function PDFDataTransportStream(params, pdfDataRangeTransport) { |
||||
assert(pdfDataRangeTransport); |
||||
|
||||
this._queuedChunks = []; |
||||
var initialData = params.initialData; |
||||
if (initialData && initialData.length > 0) { |
||||
let buffer = new Uint8Array(initialData).buffer; |
||||
this._queuedChunks.push(buffer); |
||||
} |
||||
|
||||
this._pdfDataRangeTransport = pdfDataRangeTransport; |
||||
this._isRangeSupported = !(params.disableRange); |
||||
this._isStreamingSupported = !(params.disableStream); |
||||
this._contentLength = params.length; |
||||
|
||||
this._fullRequestReader = null; |
||||
this._rangeReaders = []; |
||||
|
||||
this._pdfDataRangeTransport.addRangeListener((begin, chunk) => { |
||||
this._onReceiveData({ begin, chunk, }); |
||||
}); |
||||
|
||||
this._pdfDataRangeTransport.addProgressListener((loaded) => { |
||||
this._onProgress({ loaded, }); |
||||
}); |
||||
|
||||
this._pdfDataRangeTransport.addProgressiveReadListener((chunk) => { |
||||
this._onReceiveData({ chunk, }); |
||||
}); |
||||
|
||||
this._pdfDataRangeTransport.transportReady(); |
||||
} |
||||
PDFDataTransportStream.prototype = { |
||||
_onReceiveData: function PDFDataTransportStream_onReceiveData(args) { |
||||
let buffer = new Uint8Array(args.chunk).buffer; |
||||
if (args.begin === undefined) { |
||||
if (this._fullRequestReader) { |
||||
this._fullRequestReader._enqueue(buffer); |
||||
} else { |
||||
this._queuedChunks.push(buffer); |
||||
} |
||||
} else { |
||||
var found = this._rangeReaders.some(function (rangeReader) { |
||||
if (rangeReader._begin !== args.begin) { |
||||
return false; |
||||
} |
||||
rangeReader._enqueue(buffer); |
||||
return true; |
||||
}); |
||||
assert(found); |
||||
} |
||||
}, |
||||
|
||||
_onProgress: function PDFDataTransportStream_onDataProgress(evt) { |
||||
if (this._rangeReaders.length > 0) { |
||||
// Reporting to first range reader.
|
||||
var firstReader = this._rangeReaders[0]; |
||||
if (firstReader.onProgress) { |
||||
firstReader.onProgress({ loaded: evt.loaded, }); |
||||
} |
||||
} |
||||
}, |
||||
|
||||
_removeRangeReader: |
||||
function PDFDataTransportStream_removeRangeReader(reader) { |
||||
var i = this._rangeReaders.indexOf(reader); |
||||
if (i >= 0) { |
||||
this._rangeReaders.splice(i, 1); |
||||
} |
||||
}, |
||||
|
||||
getFullReader: function PDFDataTransportStream_getFullReader() { |
||||
assert(!this._fullRequestReader); |
||||
var queuedChunks = this._queuedChunks; |
||||
this._queuedChunks = null; |
||||
return new PDFDataTransportStreamReader(this, queuedChunks); |
||||
}, |
||||
|
||||
getRangeReader: function PDFDataTransportStream_getRangeReader(begin, end) { |
||||
var reader = new PDFDataTransportStreamRangeReader(this, begin, end); |
||||
this._pdfDataRangeTransport.requestDataRange(begin, end); |
||||
this._rangeReaders.push(reader); |
||||
return reader; |
||||
}, |
||||
|
||||
cancelAllRequests: |
||||
function PDFDataTransportStream_cancelAllRequests(reason) { |
||||
if (this._fullRequestReader) { |
||||
this._fullRequestReader.cancel(reason); |
||||
} |
||||
var readers = this._rangeReaders.slice(0); |
||||
readers.forEach(function (rangeReader) { |
||||
rangeReader.cancel(reason); |
||||
}); |
||||
this._pdfDataRangeTransport.abort(); |
||||
}, |
||||
}; |
||||
|
||||
/** @implements {IPDFStreamReader} */ |
||||
function PDFDataTransportStreamReader(stream, queuedChunks) { |
||||
this._stream = stream; |
||||
this._done = false; |
||||
this._queuedChunks = queuedChunks || []; |
||||
this._requests = []; |
||||
this._headersReady = Promise.resolve(); |
||||
stream._fullRequestReader = this; |
||||
|
||||
this.onProgress = null; // not used
|
||||
} |
||||
PDFDataTransportStreamReader.prototype = { |
||||
_enqueue: function PDFDataTransportStreamReader_enqueue(chunk) { |
||||
if (this._done) { |
||||
return; // ignore new data
|
||||
} |
||||
if (this._requests.length > 0) { |
||||
var requestCapability = this._requests.shift(); |
||||
requestCapability.resolve({ value: chunk, done: false, }); |
||||
return; |
||||
} |
||||
this._queuedChunks.push(chunk); |
||||
}, |
||||
|
||||
get headersReady() { |
||||
return this._headersReady; |
||||
}, |
||||
|
||||
get isRangeSupported() { |
||||
return this._stream._isRangeSupported; |
||||
}, |
||||
|
||||
get isStreamingSupported() { |
||||
return this._stream._isStreamingSupported; |
||||
}, |
||||
|
||||
get contentLength() { |
||||
return this._stream._contentLength; |
||||
}, |
||||
|
||||
read: function PDFDataTransportStreamReader_read() { |
||||
if (this._queuedChunks.length > 0) { |
||||
var chunk = this._queuedChunks.shift(); |
||||
return Promise.resolve({ value: chunk, done: false, }); |
||||
} |
||||
if (this._done) { |
||||
return Promise.resolve({ value: undefined, done: true, }); |
||||
} |
||||
var requestCapability = createPromiseCapability(); |
||||
this._requests.push(requestCapability); |
||||
return requestCapability.promise; |
||||
}, |
||||
|
||||
cancel: function PDFDataTransportStreamReader_cancel(reason) { |
||||
this._done = true; |
||||
this._requests.forEach(function (requestCapability) { |
||||
requestCapability.resolve({ value: undefined, done: true, }); |
||||
}); |
||||
this._requests = []; |
||||
}, |
||||
}; |
||||
|
||||
/** @implements {IPDFStreamRangeReader} */ |
||||
function PDFDataTransportStreamRangeReader(stream, begin, end) { |
||||
this._stream = stream; |
||||
this._begin = begin; |
||||
this._end = end; |
||||
this._queuedChunk = null; |
||||
this._requests = []; |
||||
this._done = false; |
||||
|
||||
this.onProgress = null; |
||||
} |
||||
PDFDataTransportStreamRangeReader.prototype = { |
||||
_enqueue: function PDFDataTransportStreamRangeReader_enqueue(chunk) { |
||||
if (this._done) { |
||||
return; // ignore new data
|
||||
} |
||||
if (this._requests.length === 0) { |
||||
this._queuedChunk = chunk; |
||||
} else { |
||||
var requestsCapability = this._requests.shift(); |
||||
requestsCapability.resolve({ value: chunk, done: false, }); |
||||
this._requests.forEach(function (requestCapability) { |
||||
requestCapability.resolve({ value: undefined, done: true, }); |
||||
}); |
||||
this._requests = []; |
||||
} |
||||
this._done = true; |
||||
this._stream._removeRangeReader(this); |
||||
}, |
||||
|
||||
get isStreamingSupported() { |
||||
return false; |
||||
}, |
||||
|
||||
read: function PDFDataTransportStreamRangeReader_read() { |
||||
if (this._queuedChunk) { |
||||
let chunk = this._queuedChunk; |
||||
this._queuedChunk = null; |
||||
return Promise.resolve({ value: chunk, done: false, }); |
||||
} |
||||
if (this._done) { |
||||
return Promise.resolve({ value: undefined, done: true, }); |
||||
} |
||||
var requestCapability = createPromiseCapability(); |
||||
this._requests.push(requestCapability); |
||||
return requestCapability.promise; |
||||
}, |
||||
|
||||
cancel: function PDFDataTransportStreamRangeReader_cancel(reason) { |
||||
this._done = true; |
||||
this._requests.forEach(function (requestCapability) { |
||||
requestCapability.resolve({ value: undefined, done: true, }); |
||||
}); |
||||
this._requests = []; |
||||
this._stream._removeRangeReader(this); |
||||
}, |
||||
}; |
||||
|
||||
return PDFDataTransportStream; |
||||
})(); |
||||
|
||||
export { |
||||
PDFDataTransportStream, |
||||
}; |
||||
Loading…
Reference in new issue