MappingCharFilter.cs

Go to the documentation of this file.

/* 
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

using System.Collections.Generic;

namespace Lucene.Net.Analysis
{
    
    public class MappingCharFilter : BaseCharFilter
    {
        private readonly NormalizeCharMap normMap;
        private LinkedList<char> buffer;
        private System.String replacement;
        private int charPointer;
        private int nextCharCounter;
        
        public MappingCharFilter(NormalizeCharMap normMap, CharStream @in)
            : base(@in)
        {
            this.normMap = normMap;
        }
        
        public MappingCharFilter(NormalizeCharMap normMap, System.IO.TextReader @in)
            : base(CharReader.Get(@in))
        {
            this.normMap = normMap;
        }
        
        public  override int Read()
        {
            while (true)
            {
                if (replacement != null && charPointer < replacement.Length)
                {
                    return replacement[charPointer++];
                }
                
                int firstChar = NextChar();
                if (firstChar == - 1)
                    return - 1;
                NormalizeCharMap nm = normMap.submap != null
                                          ? normMap.submap[(char) firstChar]
                                          : null;
                if (nm == null)
                    return firstChar;
                NormalizeCharMap result = Match(nm);
                if (result == null)
                    return firstChar;
                replacement = result.normStr;
                charPointer = 0;
                if (result.diff != 0)
                {
                    int prevCumulativeDiff = LastCumulativeDiff;
                    if (result.diff < 0)
                    {
                        for (int i = 0; i < - result.diff; i++)
                            AddOffCorrectMap(nextCharCounter + i - prevCumulativeDiff, prevCumulativeDiff - 1 - i);
                    }
                    else
                    {
                        AddOffCorrectMap(nextCharCounter - result.diff - prevCumulativeDiff, prevCumulativeDiff + result.diff);
                    }
                }
            }
        }
        
        private int NextChar()
        {
            nextCharCounter++;
            if (buffer != null && buffer.Count != 0)
            {
                char tempObject = buffer.First.Value;
                buffer.RemoveFirst();
                return (tempObject);
            }
            return input.Read();
        }
        
        private void  PushChar(int c)
        {
            nextCharCounter--;
            if (buffer == null)
            {
                buffer = new LinkedList<char>();
            }
            buffer.AddFirst((char)c);
        }
        
        private void  PushLastChar(int c)
        {
            if (buffer == null)
            {
                buffer = new LinkedList<char>();
            }
            buffer.AddLast((char)c);
        }
        
        private NormalizeCharMap Match(NormalizeCharMap map)
        {
            NormalizeCharMap result = null;
            if (map.submap != null)
            {
                int chr = NextChar();
                if (chr != - 1)
                {
                    NormalizeCharMap subMap = map.submap[(char)chr];
                    if (subMap != null)
                    {
                        result = Match(subMap);
                    }
                    if (result == null)
                    {
                        PushChar(chr);
                    }
                }
            }
            if (result == null && map.normStr != null)
            {
                result = map;
            }
            return result;
        }
        
        public  override int Read(System.Char[] cbuf, int off, int len)
        {
            var tmp = new char[len];
            int l = input.Read(tmp, 0, len);
            if (l != 0)
            {
                for (int i = 0; i < l; i++)
                    PushLastChar(tmp[i]);
            }
            l = 0;
            for (int i = off; i < off + len; i++)
            {
                int c = Read();
                if (c == - 1)
                    break;
                cbuf[i] = (char) c;
                l++;
            }
            return l == 0?- 1:l;
        }
    }
}