2 // Copyright (c) 2010-2017 Intel Corporation
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
8 // http://www.apache.org/licenses/LICENSE-2.0
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
19 #include "memreader.hpp"
20 #include "mappedfile.hpp"
21 #include "stream3.hpp"
23 MemReader::MemReader(MappedFile *file, const vector<size_t> &offsets)
25 initRanges(file->getMapBeg(), file->getMapEnd(), offsets);
28 bool MemReader::read(Stream3 *stream)
33 readStream(stream, getLowestID());
38 uint32_t MemReader::getLowestID() const
40 uint32_t lowestID = UINT32_MAX;
43 for (size_t i = 0; i < ranges.size(); ++i) {
44 rangeID = Stream3::getIDFromMem(ranges[i].first);
45 if (rangeID < lowestID)
51 void MemReader::readStream(Stream3 *stream, uint32_t id)
53 stream->removeAllPackets();
57 for (size_t i = 0; i < ranges.size(); ++i) {
58 if (Stream3::getIDFromMem(ranges[i].first) == id) {
59 stream->addFromMemory(ranges[i].first, &len);
60 ranges[i].first += len;
65 void MemReader::removeEmptyRanges()
67 vector<pair <uint8_t *, uint8_t *> > original = ranges;
68 size_t destinationIdx = 0;
70 for (size_t i = 0; i < original.size(); ++i) {
71 if (original[i].first < original[i].second)
72 ranges[destinationIdx++] = original[i];
74 ranges.resize(destinationIdx);
77 void MemReader::initRanges(uint8_t *begin, uint8_t *end, const vector<size_t> &offsets)
79 ranges.resize(offsets.size());
82 for (size_t i = 0; i < offsets.size(); ++i) {
83 ranges[i].first = begin + offsets[i];
84 if (i != offsets.size() - 1)
85 ranges[i].second = begin + offsets[i + 1];
87 ranges[i].second = end;
88 totalLength += ranges[i].second - ranges[i].first;
93 size_t MemReader::getRangeLengths() const
97 for (size_t i = 0; i < ranges.size(); ++i) {
98 total += ranges[i].second - ranges[i].first;
103 size_t MemReader::consumed() const
105 return totalLength - getRangeLengths();