jef.tools.string.StringSpliterEx.java Source code

Introduction

Here is the source code for jef.tools.string.StringSpliterEx.java

Source

/*
 * JEF - Copyright 2009-2010 Jiyi (mr.jiyi@gmail.com)
 *
 * Licensed 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.
 */
package jef.tools.string;

import org.apache.commons.lang.ArrayUtils;

/**
 * ??Substringkeyword?
 * ?intString???String??
 */
public class StringSpliterEx {
    private Substring keyword1;
    private Substring keyword2;
    private Substring source;

    //------left-----| key1 | --middle- |key2|----right------  

    public static final int MODE_FROM_LEFT = 0;// ??
    public static final int MODE_FROM_ENDS = 1;// ?
    private int findMode = MODE_FROM_LEFT;

    public StringSpliterEx(Substring source) {
        this.source = source;
    }

    public StringSpliterEx(String source) {
        this.source = new Substring(source);
    }

    public Substring getKeyword1() {
        return keyword1;
    }

    public Substring getKeyword2() {
        return keyword2;
    }

    public Substring getSource() {
        return source;
    }

    public void setSource(Substring source) {
        this.source = source;
        this.keyword1 = null;
        this.keyword2 = null;
    }

    /**
     * ?
     */
    public Substring getLeft() {
        if (keyword1 == null && keyword2 == null) {
            return null;
        } else if (keyword1 == null) {
            return source.subAbsoultOffset(source.getStart(), keyword2.getStart());
        } else {
            return source.subAbsoultOffset(source.getStart(), keyword1.getStart());
        }
    }

    /**
     * ??
     */
    public Substring getRight() {
        if (keyword1 == null && keyword2 == null) {
            return null;
        } else if (keyword2 == null) {
            return source.subAbsoultOffset(keyword1.getEnd(), source.getEnd());
        } else {
            return source.subAbsoultOffset(keyword2.getEnd(), source.getEnd());
        }

    }

    /**
     * 
     * @return
     */
    public Substring getMiddle() {
        if (keyword1 == null && keyword2 == null) {
            return null;
        } else if (keyword1 == null) {
            return null;
        } else if (keyword2 == null) {
            return null;
        } else {
            return source.subAbsoultOffset(keyword1.getEnd(), keyword2.getStart());
        }
    }

    public static final int RESULT_NO_KEY = 3;
    public static final int RESULT_BOTH_KEY = 0;
    public static final int RESULT_KEY1_ONLY = 1;
    public static final int RESULT_KEY2_ONLY = 2;

    /**
     * ?keyStr?Substring.
     * @param key1, key2
     */
    public int setKeys(String key1, String key2) {
        StringSpliter sp = new StringSpliter(source);
        sp.setKey(key1);
        if (sp.getKeyword() != null) {
            this.keyword1 = sp.getKeyword();
            sp = new StringSpliter(sp.getRight());
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKey(key2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_BOTH_KEY;
            } else {
                return RESULT_KEY1_ONLY;
            }
        } else {
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKey(key2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_KEY2_ONLY;
            } else {
                return RESULT_NO_KEY;
            }
        }

    }

    public int setKeyOfAny(String[] key1, String[] key2) {
        StringSpliter sp = new StringSpliter(source);
        sp.setKeyOfAny(key1);
        if (sp.getKeyword() != null) {
            this.keyword1 = sp.getKeyword();
            sp = new StringSpliter(sp.getRight());
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKeyOfAny(key2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_BOTH_KEY;
            } else {
                return RESULT_KEY1_ONLY;
            }
        } else {
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKeyOfAny(key2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_KEY2_ONLY;
            } else {
                return RESULT_NO_KEY;
            }
        }
    }

    /**
     * setKeyOfAny?key???
     *  key1= "<","[" key2=">","]" <????>??
     * @param key1
     * @param key2
     * @return
     */
    public int setKeyPairOfAny(String[] key1, String[] key2) {
        if (key1.length < key2.length) {
            throw new IllegalArgumentException("the length between two input keys do not equals!");
        }
        StringSpliter sp = new StringSpliter(source);
        sp.setKeyOfAny(key1);
        if (sp.getKeyword() != null) {
            this.keyword1 = sp.getKeyword();
            int n = ArrayUtils.indexOf(key1, this.keyword1.toString());
            if (n > key2.length - 1) {//key1??key2
                return RESULT_KEY1_ONLY;
            }
            String realKey2 = key2[n];
            sp = new StringSpliter(sp.getRight());
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKey(realKey2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_BOTH_KEY;
            } else {
                return RESULT_KEY1_ONLY;
            }
        } else {
            if (findMode == MODE_FROM_ENDS) {
                sp.setMode(StringSpliter.MODE_FROM_RIGHT);
            }
            sp.setKeyOfAny(key2);
            this.keyword2 = sp.getKeyword();
            if (sp.getKeyword() != null) {
                return RESULT_KEY2_ONLY;
            } else {
                return RESULT_NO_KEY;
            }
        }
    }

    public int getMode() {
        return findMode;
    }

    public void setMode(int mode) {
        this.findMode = mode;
    }
}