smali/src/main/java/org/JesusFreke/dexlib/ProtoIdItem.java

/*
 * [The "BSD licence"]
 * Copyright (c) 2009 Ben Gruver
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.JesusFreke.dexlib;

import org.JesusFreke.dexlib.ItemType;

import java.util.ArrayList;

public class ProtoIdItem extends IndexedItem<ProtoIdItem> {
    private final Field[] fields;

    private final IndexedItemReference<StringIdItem> shortyDescriptor;
    private final IndexedItemReference<TypeIdItem> returnType;
    private final OffsettedItemReference<TypeListItem> parameters;

    public ProtoIdItem(DexFile dexFile, int index) {
        super(index);
        fields = new Field[] {
                shortyDescriptor = new IndexedItemReference<StringIdItem>(dexFile.StringIdsSection, new IntegerField()),
                returnType = new IndexedItemReference<TypeIdItem>(dexFile.TypeIdsSection, new IntegerField()),
                parameters = new OffsettedItemReference<TypeListItem>(dexFile.TypeListsSection, new IntegerField())
        };
    }

    public ProtoIdItem(DexFile dexFile, TypeIdItem returnType, ArrayList<TypeIdItem> parameters)
    {
        super(-1);
        StringIdItem stringIdItem = new StringIdItem(dexFile, createShortyDescriptor(returnType, parameters));
        TypeListItem typeListItem = new TypeListItem(dexFile, parameters);

        fields = new Field[] {
                this.shortyDescriptor = new IndexedItemReference<StringIdItem>(dexFile, stringIdItem, new IntegerField()),
                this.returnType = new IndexedItemReference<TypeIdItem>(dexFile, returnType, new IntegerField()),
                this.parameters = new OffsettedItemReference<TypeListItem>(dexFile, typeListItem, new IntegerField())
        };
    }

    private String createShortyDescriptor(TypeIdItem returnType, ArrayList<TypeIdItem> parameters) {
        StringBuilder sb = new StringBuilder();
        sb.append(returnType.toShorty());

        for (TypeIdItem typeIdItem: parameters) {
            sb.append(typeIdItem.toShorty());
        }
        return sb.toString();
    }

    protected int getAlignment() {
        return 4;
    }

    public int getParameterWordCount() {
        TypeListItem typeList = parameters.getReference();
        if (typeList == null) {
            return 0;
        } else {
            return typeList.getWordCount();
        }
    }

    protected Field[] getFields() {
        return fields;
    }

    public ItemType getItemType() {
        return ItemType.TYPE_PROTO_ID_ITEM;
    }

    public int compareTo(ProtoIdItem o) {
        int result = returnType.compareTo(o.returnType);
        if (result != 0) {
            return result;
        }

        TypeListItem thisParameters = parameters.getReference();
        if (thisParameters == null) {
            return -1;
        }

        return thisParameters.compareTo(o.parameters.getReference());        
    }

    public String toString() {
        return shortyDescriptor.toString();
    }
}