dx/src/com/android/jack/dx/dex/file/MemberIdsSection.java

/*
 * Copyright (C) 2007 The Android Open Source Project
 *
 * 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.android.jack.dx.dex.file;

import com.android.jack.dx.util.DexException;

import java.util.Formatter;
import java.util.Map;
import java.util.TreeMap;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Member (field or method) refs list section of a {@code .dex} file.
 */
public abstract class MemberIdsSection extends UniformItemSection {
  /** The largest addressable member is 0xffff, in the dex spec as field@CCCC or meth@CCCC. */
  private static final int MAX_MEMBERS = 0x10000;

  /**
   * Constructs an instance. The file offset is initially unknown.
   *
   * @param name {@code null-ok;} the name of this instance, for annotation
   * purposes
   * @param file {@code non-null;} file that this instance is part of
   */
  public MemberIdsSection(String name, DexFile file) {
    super(name, file, 4);
  }

  /** {@inheritDoc} */
  @Override
  protected void orderItems() {
    int idx = 0;

    if (items().size() > MAX_MEMBERS) {
      throw new DexException(tooManyMembersMessage());
    }

    for (Object i : items()) {
      ((MemberIdItem) i).setIndex(idx);
      idx++;
    }
  }

  private String tooManyMembersMessage() {
    Map<String, AtomicInteger> membersByPackage = new TreeMap<String, AtomicInteger>();
    for (Object member : items()) {
      String packageName = ((MemberIdItem) member).getDefiningClass().getPackageName();
      AtomicInteger count = membersByPackage.get(packageName);
      if (count == null) {
        count = new AtomicInteger();
        membersByPackage.put(packageName, count);
      }
      count.incrementAndGet();
    }

    Formatter formatter = null;
    try {
      formatter = new Formatter();
      String memberType = this instanceof MethodIdsSection ? "methods" : "fields";
      formatter.format("Too many %s: %d; max is %d. By package:", memberType, items().size(),
          MAX_MEMBERS);
      for (Map.Entry<String, AtomicInteger> entry : membersByPackage.entrySet()) {
        formatter.format("%n%6d %s", entry.getValue().get(), entry.getKey());
      }
      return formatter.toString();
    } finally {
      formatter.close();
    }
  }
}