Home | History | Annotate | Download | only in builder 
      1 /*
      2  * Copyright 2013, Google Inc.
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions are
      7  * met:
      8  *
      9  *     * Redistributions of source code must retain the above copyright
     10  * notice, this list of conditions and the following disclaimer.
     11  *     * Redistributions in binary form must reproduce the above
     12  * copyright notice, this list of conditions and the following disclaimer
     13  * in the documentation and/or other materials provided with the
     14  * distribution.
     15  *     * Neither the name of Google Inc. nor the names of its
     16  * contributors may be used to endorse or promote products derived from
     17  * this software without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     22  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     23  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     29  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 package org.jf.dexlib2.writer.builder;
     33 
     34 import com.google.common.collect.Maps;
     35 import org.jf.dexlib2.iface.reference.FieldReference;
     36 import org.jf.dexlib2.immutable.reference.ImmutableFieldReference;
     37 import org.jf.dexlib2.writer.FieldSection;
     38 
     39 import javax.annotation.Nonnull;
     40 import java.util.Collection;
     41 import java.util.Map.Entry;
     42 import java.util.concurrent.ConcurrentMap;
     43 
     44 public class BuilderFieldPool extends BaseBuilderPool
     45         implements FieldSection<BuilderStringReference, BuilderTypeReference, BuilderFieldReference, BuilderField> {
     46     @Nonnull private final ConcurrentMap<FieldReference, BuilderFieldReference> internedItems =
     47             Maps.newConcurrentMap();
     48 
     49     public BuilderFieldPool(@Nonnull DexBuilder dexBuilder) {
     50         super(dexBuilder);
     51     }
     52 
     53     @Nonnull BuilderFieldReference internField(@Nonnull String definingClass, String name, String type) {
     54         ImmutableFieldReference fieldReference = new ImmutableFieldReference(definingClass, name, type);
     55         return internField(fieldReference);
     56     }
     57 
     58     @Nonnull public BuilderFieldReference internField(@Nonnull FieldReference fieldReference) {
     59         BuilderFieldReference ret = internedItems.get(fieldReference);
     60         if (ret != null) {
     61             return ret;
     62         }
     63 
     64         BuilderFieldReference dexPoolFieldReference = new BuilderFieldReference(
     65                 dexBuilder.typeSection.internType(fieldReference.getDefiningClass()),
     66                 dexBuilder.stringSection.internString(fieldReference.getName()),
     67                 dexBuilder.typeSection.internType(fieldReference.getType()));
     68         ret = internedItems.putIfAbsent(dexPoolFieldReference, dexPoolFieldReference);
     69         return ret==null?dexPoolFieldReference:ret;
     70     }
     71 
     72     @Nonnull @Override
     73     public BuilderTypeReference getDefiningClass(@Nonnull BuilderFieldReference key) {
     74         return key.definingClass;
     75     }
     76 
     77     @Nonnull @Override public BuilderTypeReference getFieldType(@Nonnull BuilderFieldReference key) {
     78         return key.fieldType;
     79     }
     80 
     81     @Nonnull @Override public BuilderStringReference getName(@Nonnull BuilderFieldReference key) {
     82         return key.name;
     83     }
     84 
     85     @Override public int getFieldIndex(@Nonnull BuilderField builderField) {
     86         return builderField.fieldReference.getIndex();
     87     }
     88 
     89     @Override public int getItemIndex(@Nonnull BuilderFieldReference key) {
     90         return key.index;
     91     }
     92 
     93     @Nonnull @Override public Collection<? extends Entry<? extends BuilderFieldReference, Integer>> getItems() {
     94         return new BuilderMapEntryCollection<BuilderFieldReference>(internedItems.values()) {
     95             @Override protected int getValue(@Nonnull BuilderFieldReference key) {
     96                 return key.index;
     97             }
     98 
     99             @Override protected int setValue(@Nonnull BuilderFieldReference key, int value) {
    100                 int prev = key.index;
    101                 key.index = value;
    102                 return prev;
    103             }
    104         };
    105     }
    106 
    107     @Override public int getItemCount() {
    108         return internedItems.size();
    109     }
    110 }
    111