com.reandroid.arsc.pool.ResXmlStringPool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ARSCLib Show documentation
Android binary resources read/write library
There is a newer version: 1.3.5
/*
 *  Copyright (C) 2022 github.com/REAndroid
 *
 *  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 com.reandroid.arsc.pool;

import com.reandroid.arsc.array.*;
import com.reandroid.arsc.chunk.xml.ResXmlDocument;
import com.reandroid.arsc.chunk.xml.ResXmlIDMap;
import com.reandroid.arsc.group.StringGroup;
import com.reandroid.arsc.item.*;

import java.util.Iterator;
import java.util.Objects;

public class ResXmlStringPool extends StringPool {
    public ResXmlStringPool(boolean is_utf8) {
        super(is_utf8, false);
    }
    @Override
    public ResXmlString removeReference(ReferenceItem referenceItem){
        if(referenceItem==null){
            return null;
        }
        ResXmlString stringItem = super.removeReference(referenceItem);
        removeNotUsedItem(stringItem);
        return stringItem;
    }
    private void removeNotUsedItem(ResXmlString xmlString){
        if(xmlString == null || xmlString.hasReference()){
            return;
        }
        ResXmlIDMap idMap = getResXmlIDMap();
        int lastIdIndex = -1;
        if(idMap!=null){
            lastIdIndex = idMap.countId() - 1;
        }
        if(idMap!=null && xmlString.getIndex()>lastIdIndex){
            removeString(xmlString);
        }else {
            xmlString.set("");
        }
    }
    @Override
    StringArray newInstance(OffsetArray offsets, IntegerItem itemCount, IntegerItem itemStart, boolean is_utf8) {
        return new ResXmlStringArray(offsets, itemCount, itemStart, is_utf8);
    }
    public ResXmlString getOrCreate(String str){
        return getOrCreateAttribute(0, str);
    }
    public ResXmlString createNew(String str){
        StringArray stringsArray = getStringsArray();
        ResXmlString xmlString = stringsArray.createNext();
        xmlString.set(str);
        return xmlString;
    }
    public ResXmlString getOrCreateAttribute(int resourceId, String str){
        ResXmlIDMap resXmlIDMap = getResXmlIDMap();
        if(resXmlIDMap == null){
            return super.getOrCreate(str);
        }
        ResXmlIDArray idArray = resXmlIDMap.getResXmlIDArray();
        int count = idArray.childesCount();
        if(resourceId == 0){
            return getOrCreateAfter(count, str);
        }
        StringArray stringsArray = getStringsArray();
        ResXmlID xmlID = idArray.getByResId(resourceId);
        if(xmlID != null){
            ResXmlString xmlString = stringsArray.get(xmlID.getIndex());
            if(xmlString!=null && Objects.equals(str, xmlString.get())){
                return xmlString;
            }
        }
        count = idArray.childesCount() + 1;
        stringsArray.ensureSize(count);
        idArray.setChildesCount(count);
        int index = count - 1;
        xmlID = idArray.get(index);
        assert xmlID != null;
        xmlID.set(resourceId);
        idArray.refreshIdMap();

        ResXmlString xmlString = stringsArray.newInstance();
        xmlString.set(str);
        stringsArray.insertItem(index, xmlString);

        updateUniqueIdMap(xmlString);
        return xmlString;
    }
    private ResXmlString getOrCreateAfter(int position, String str){
        if(position<0){
            position=0;
        }
        StringGroup group = get(str);
        if(group!=null){
            Iterator itr = group.iterator();
            while (itr.hasNext()){
                ResXmlString xmlString = itr.next();
                int index = xmlString.getIndex();
                if(index > position || (position==0 && position == index)){
                    return xmlString;
                }
            }
        }
        StringArray stringsArray = getStringsArray();
        int count = stringsArray.childesCount();
        if(count < position){
            count = position;
        }
        stringsArray.ensureSize(count+1);
        ResXmlString xmlString = stringsArray.get(count);
        assert xmlString != null;
        xmlString.set(str);
        super.updateUniqueIdMap(xmlString);
        return xmlString;
    }
    private ResXmlIDMap getResXmlIDMap(){
        ResXmlDocument resXmlDocument = getParentInstance(ResXmlDocument.class);
        if(resXmlDocument!=null){
            return resXmlDocument.getResXmlIDMap();
        }
        return null;
    }
    @Override
    public void onChunkLoaded() {
        super.onChunkLoaded();
        StyleArray styleArray = getStyleArray();
        if(styleArray.childesCount()>0){
            notifyResXmlStringPoolHasStyles(styleArray.childesCount());
        }
    }
    private static void notifyResXmlStringPoolHasStyles(int styleArrayCount){
        if(HAS_STYLE_NOTIFIED){
            return;
        }
        String msg="Not expecting ResXmlStringPool to have styles count="
                +styleArrayCount+",\n please create issue along with this apk/file on https://github.com/REAndroid/ARSCLib";
        System.err.println(msg);
        HAS_STYLE_NOTIFIED=true;
    }
    private static boolean HAS_STYLE_NOTIFIED;
}