zhao
2021-07-19 8347f2fbddbd25369359dcb2da1233ac48a19fdc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
* 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.
*/
/*
 * Created on May 15, 2005
 *
 */
namespace HH.WMS.Utils.NPOI.SS.Formula.Functions
{
    using System;
    using System.Text;
    using HH.WMS.Utils.NPOI.SS.Formula.Eval;
 
    /**
     * An implementation of the SUBSTITUTE function:
     * Substitutes text in a text string with new text, some number of times.
     * @author Manda Wilson < wilson at c bio dot msk cc dot org >
     */
    public class Substitute : TextFunction
    {
        // fix warning CS0414 "never used": private static int Replace_ALL = -1;
 
        /**
         *Substitutes text in a text string with new text, some number of times.
         * 
         * @see org.apache.poi.hssf.record.formula.eval.Eval
         */
        public override ValueEval EvaluateFunc(ValueEval[] args, int srcCellRow, int srcCellCol)
        {
            if (args.Length < 3 || args.Length > 4)
            {
                return ErrorEval.VALUE_INVALID;
            }
 
            String oldStr = EvaluateStringArg(args[0], srcCellRow, srcCellCol);
            String searchStr = EvaluateStringArg(args[1], srcCellRow, srcCellCol);
            String newStr = EvaluateStringArg(args[2], srcCellRow, srcCellCol);
            
 
            String result;
 
            switch (args.Length)
            {
                case 4:
                    int instanceNumber = EvaluateIntArg(args[3], srcCellRow, srcCellCol);
                    if (instanceNumber < 1)
                    {
                        return ErrorEval.VALUE_INVALID;
                    }
                    result = ReplaceOneOccurrence(oldStr, searchStr, newStr, instanceNumber);
                    break;
                case 3:
                    result = ReplaceAllOccurrences(oldStr, searchStr, newStr);
                    break;
                default:
                    throw new InvalidOperationException("Cannot happen");
 
            }
 
            return new StringEval(result);
        }
 
        private static String ReplaceAllOccurrences(String oldStr, String searchStr, String newStr)
        {
            StringBuilder sb = new StringBuilder();
            int startIndex = 0;
            int nextMatch = -1;
            while (true)
            {
                nextMatch = oldStr.IndexOf(searchStr, startIndex, StringComparison.CurrentCulture);
                if (nextMatch < 0)
                {
                    // store everything from end of last match to end of string
                    sb.Append(oldStr.Substring(startIndex));
                    return sb.ToString();
                }
                // store everything from end of last match to start of this match
                sb.Append(oldStr.Substring(startIndex, nextMatch));
                sb.Append(newStr);
                startIndex = nextMatch + searchStr.Length;
            }
        }
 
        private static String ReplaceOneOccurrence(String oldStr, String searchStr, String newStr, int instanceNumber)
        {
            if (searchStr.Length < 1)
            {
                return oldStr;
            }
            int startIndex = 0;
            int nextMatch = -1;
            int count = 0;
            while (true)
            {
                nextMatch = oldStr.IndexOf(searchStr, startIndex, StringComparison.CurrentCulture);
                if (nextMatch < 0)
                {
                    // not enough occurrences found - leave unchanged
                    return oldStr;
                }
                count++;
                if (count == instanceNumber)
                {
                    StringBuilder sb = new StringBuilder(oldStr.Length + newStr.Length);
                    sb.Append(oldStr.Substring(0, nextMatch));
                    sb.Append(newStr);
                    sb.Append(oldStr.Substring(nextMatch + searchStr.Length));
                    return sb.ToString();
                }
                startIndex = nextMatch + searchStr.Length;
            }
        }
    }
}