7279b58f4b0c4253698791eba1e9880ceefa1ca2
[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     @Override
74     public UnmodifiableIterator<PathArgument> iterator() {
75         return new IteratorImpl(base, stack);
76     }
77
78     private static final class IteratorImpl extends UnmodifiableIterator<PathArgument> {
79         private final Iterator<PathArgument> stack;
80         private final Iterator<PathArgument> base;
81
82         IteratorImpl(final Iterable<PathArgument> base, final Iterable<PathArgument> stack) {
83             this.base = base.iterator();
84             this.stack = stack.iterator();
85         }
86
87         @Override
88         public boolean hasNext() {
89             return stack.hasNext();
90         }
91
92         @Override
93         public PathArgument next() {
94             if (base.hasNext()) {
95                 return base.next();
96             }
97             return stack.next();
98         }
99     }
100 }