com.wangjiegulu.rapidooo.library.compiler.part.statement.parcelable.ParcelableBoxPrimitiveStatementBrew Maven / Gradle / Ivy

Go to download
package com.wangjiegulu.rapidooo.library.compiler.part.statement.parcelable;

import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeName;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.type.OOOTypeEntry;
import com.wangjiegulu.rapidooo.library.compiler.part.statement.contact.IParcelableStatementBrew;
import com.wangjiegulu.rapidooo.library.compiler.util.LogUtil;

import java.util.HashMap;

/**
 * Author: wangjie Email: [email protected] Date: 2019-06-17.
 */
public class ParcelableBoxPrimitiveStatementBrew implements IParcelableStatementBrew {
    private HashMap primitiveMap = new HashMap<>();

    public ParcelableBoxPrimitiveStatementBrew() {
        primitiveMap.put(TypeName.DOUBLE, "Double");
        primitiveMap.put(TypeName.FLOAT, "Float");
        primitiveMap.put(TypeName.BYTE, "Byte");
        primitiveMap.put(TypeName.INT, "Integer");
        primitiveMap.put(TypeName.BOOLEAN, "Boolean");
        primitiveMap.put(TypeName.SHORT, "Short");
        primitiveMap.put(TypeName.CHAR, "Character");
        primitiveMap.put(TypeName.LONG, "Long");
    }

    @Override
    public boolean match(OOOTypeEntry typeEntry) {
        TypeName fieldTypeName = typeEntry.getTypeName();
        LogUtil.logger("fieldTypeName.isBoxedPrimitive(): " + fieldTypeName.isBoxedPrimitive() + ", " + fieldTypeName);
        if(fieldTypeName.isBoxedPrimitive()){
            TypeName _fieldTypeName = fieldTypeName.unbox();
            return primitiveMap.containsKey(_fieldTypeName);
        }
        return false;
    }

    @Override
    public void read(MethodSpec.Builder methodBuilder, String statementPrefix, Object[] statementPrefixTypes, String fieldCode, OOOTypeEntry oooTypeEntry, String fieldName) {
        String name = primitiveMap.get(oooTypeEntry.getTypeName().unbox());
        methodBuilder.addStatement(statementPrefix + fieldCode + " = (" + name + ") parcel.readValue(" + name + ".class.getClassLoader())", statementPrefixTypes);
    }

    @Override
    public void write(MethodSpec.Builder methodBuilder, String statementPrefix, Object[] statementPrefixTypes, String fieldCode, OOOTypeEntry oooTypeEntry, String fieldName) {
        methodBuilder.addStatement(statementPrefix + "dest.writeValue(" + fieldCode + ")", statementPrefixTypes);
    }
}