1d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver/*
2d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * Copyright 2012, Google Inc.
3d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * All rights reserved.
4d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *
5d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * Redistribution and use in source and binary forms, with or without
6d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * modification, are permitted provided that the following conditions are
7d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * met:
8d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *
9d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *     * Redistributions of source code must retain the above copyright
10d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * notice, this list of conditions and the following disclaimer.
11d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *     * Redistributions in binary form must reproduce the above
12d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * copyright notice, this list of conditions and the following disclaimer
13d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * in the documentation and/or other materials provided with the
14d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * distribution.
15d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *     * Neither the name of Google Inc. nor the names of its
16d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * contributors may be used to endorse or promote products derived from
17d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * this software without specific prior written permission.
18d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver *
19d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver */
31d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
32d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruverpackage org.jf.dexlib2.dexbacked.util;
33d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
3484c1762a62d7fc6638432c6c56e0422aa8cc6939Ben Gruverimport org.jf.dexlib2.dexbacked.DexBackedDexFile;
35d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruverimport org.jf.dexlib2.dexbacked.DexReader;
36d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
37d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruverimport javax.annotation.Nonnull;
38d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruverimport javax.annotation.Nullable;
39d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruverimport java.util.Iterator;
4065d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruverimport java.util.NoSuchElementException;
41d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
4265d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruverpublic abstract class VariableSizeLookaheadIterator<T> implements Iterator<T> {
4365d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver    @Nonnull private final DexReader reader;
44d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
45d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    private T cachedItem = null;
46d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
4784c1762a62d7fc6638432c6c56e0422aa8cc6939Ben Gruver    protected VariableSizeLookaheadIterator(@Nonnull DexBackedDexFile dexFile, int offset) {
4884c1762a62d7fc6638432c6c56e0422aa8cc6939Ben Gruver        this.reader = dexFile.readerAt(offset);
4965d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver        cachedItem = readNextItem(reader);
50d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    }
51d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
52d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    /**
53d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver     * Reads the next item from reader. If the end of the list has been reached, it should return null.
54d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver     *
55d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver     * @return The item that was read, or null if the end of the list has been reached.
56d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver     */
5765d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver    @Nullable protected abstract T readNextItem(@Nonnull DexReader reader);
58d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
59d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    @Override
60d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    public boolean hasNext() {
61d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver        return cachedItem != null;
62d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    }
63d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
64d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    @Override
6565d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver    @Nonnull
66d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    public T next() {
6765d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver        if (cachedItem == null) {
6865d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver            throw new NoSuchElementException();
6965d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver        }
70d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver        T ret = cachedItem;
7165d969c23b5ef14f8a51c88fdb7c5d5319a1b555Ben Gruver        cachedItem = readNextItem(reader);
72d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver        return ret;
73d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    }
74d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver
75d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver    @Override public void remove() { throw new UnsupportedOperationException(); }
76d1662b67fecaf835227aff3a136949a2358ccd4eBen Gruver}
77