690e2a438b3c3f9a32fe3e24fea30255a91caeab
[yangtools.git] / yang / yang-data-api / src / main / java / org / opendaylight / yangtools / yang / data / api / StackedPathArguments.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.yang.data.api;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.base.Verify;
12 import com.google.common.collect.UnmodifiableIterator;
13 import java.util.Iterator;
14 import java.util.List;
15 import javax.annotation.Nonnull;
16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
17
18 final class StackedPathArguments extends PathArgumentList {
19     private final List<PathArgument> base;
20     private final List<PathArgument> stack;
21
22     public StackedPathArguments(@Nonnull final YangInstanceIdentifier base, @Nonnull final List<PathArgument> stack) {
23         Verify.verify(!stack.isEmpty());
24         this.base = base.getPathArguments();
25         this.stack = stack;
26     }
27
28     @Override
29     public int size() {
30         return stack.size() + base.size();
31     }
32
33     @Override
34     public boolean contains(final Object o) {
35         final PathArgument srch = (PathArgument) Preconditions.checkNotNull(o);
36         return stack.contains(srch) || base.contains(srch);
37     }
38
39     @Override
40     public PathArgument get(final int index) {
41         if (index < base.size()) {
42             return base.get(index);
43         }
44         return stack.get(index - base.size());
45     }
46
47     @Override
48     public int indexOf(final Object o) {
49         final PathArgument srch = (PathArgument) Preconditions.checkNotNull(o);
50
51         int ret = base.indexOf(srch);
52         if (ret == -1) {
53             ret = stack.indexOf(srch);
54             if (ret != -1) {
55                 return base.size() + ret;
56             }
57         }
58         return ret;
59     }
60
61     @Override
62     public int lastIndexOf(final Object o) {
63         final PathArgument srch = (PathArgument) Preconditions.checkNotNull(o);
64
65         final int ret = stack.lastIndexOf(srch);
66         if (ret != -1) {
67             return base.size() + ret;
68         }
69
70         return base.lastIndexOf(srch);
71     }
72
73     @Nonnull
74     @Override
75     public UnmodifiableIterator<PathArgument> iterator() {
76         return new IteratorImpl(base, stack);
77     }
78
79     private static final class IteratorImpl extends UnmodifiableIterator<PathArgument> {
80         private final Iterator<PathArgument> stack;
81         private final Iterator<PathArgument> base;
82
83         IteratorImpl(final Iterable<PathArgument> base, final Iterable<PathArgument> stack) {
84             this.base = base.iterator();
85             this.stack = stack.iterator();
86         }
87
88         @Override
89         public boolean hasNext() {
90             return stack.hasNext();
91         }
92
93         @Override
94         public PathArgument next() {
95             if (base.hasNext()) {
96                 return base.next();
97             }
98             return stack.next();
99         }
100     }
101 }