bc482f1dc523f13dc99e74a5af15482db43ae76d
[vxquery.git] / vxquery-core / src / main / java / org / apache / vxquery / runtime / functions / strings / SubstringBeforeCharacterIterator.java
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements.  See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License.  You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 package org.apache.vxquery.runtime.functions.strings;
18
19 public class SubstringBeforeCharacterIterator implements ICharacterIterator {
20     private ICharacterIterator searchIterator;
21     private final UTF8StringCharacterIterator stringIterator;
22     private int currentByteOffset;
23
24     public SubstringBeforeCharacterIterator(UTF8StringCharacterIterator stringIterator) {
25         this.stringIterator = stringIterator;
26     }
27
28     public void setSearch(ICharacterIterator searchIterator) {
29         this.searchIterator = searchIterator;
30     }
31
32     @Override
33     public char next() {
34         // Default - no character exists.
35         int c = ICharacterIterator.EOS_CHAR;
36         boolean firstPass = true;
37         searchIterator.reset();
38
39         while (true) {
40             int c1 = stringIterator.next();
41             int c2 = searchIterator.next();
42             if (firstPass) {
43                 // Save character and location for next call.
44                 currentByteOffset = stringIterator.getByteOffset();
45                 c = c1;
46                 firstPass = false;
47             }
48             if (c2 == ICharacterIterator.EOS_CHAR) {
49                 // End of string.
50                 c = ICharacterIterator.EOS_CHAR;
51                 break;
52             }
53             if (c1 != c2) {
54                 // No match found.
55                 break;
56             }
57         }
58
59         stringIterator.setByteOffset(currentByteOffset);
60         return (char) c;
61     }
62
63     @Override
64     public void reset() {
65         stringIterator.reset();
66     }
67 }