src/main/java/com/android/tools/r8/ir/analysis/proto/schema/ProtoMessageInfo.java

// Copyright (c) 2019, the R8 project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

package com.android.tools.r8.ir.analysis.proto.schema;

import static com.android.tools.r8.ir.analysis.proto.RawMessageInfoDecoder.IS_PROTO_2_MASK;

import com.android.tools.r8.utils.Pair;
import java.util.ArrayList;
import java.util.List;

public class ProtoMessageInfo {

  public static final int BITS_PER_HAS_BITS_WORD = 32;

  public static class Builder {

    private int flags;

    private List<ProtoFieldInfo> fields;
    private List<ProtoObject> hasBitsObjects;
    private List<Pair<ProtoObject, ProtoObject>> oneOfObjects;

    public void setFlags(int value) {
      this.flags = value;
    }

    public void addField(ProtoFieldInfo field) {
      if (fields == null) {
        fields = new ArrayList<>();
      }
      fields.add(field);
    }

    public void addHasBitsObject(ProtoObject hasBitsObject) {
      if (hasBitsObjects == null) {
        hasBitsObjects = new ArrayList<>();
      }
      hasBitsObjects.add(hasBitsObject);
    }

    public void setNumberOfHasBitsObjects(int number) {
      if (number > 0) {
        hasBitsObjects = new ArrayList<>(number);
      }
    }

    public void addOneOfObject(ProtoObject first, ProtoObject second) {
      if (oneOfObjects == null) {
        oneOfObjects = new ArrayList<>();
      }
      oneOfObjects.add(new Pair<>(first, second));
    }

    public void setNumberOfOneOfObjects(int number) {
      if (number > 0) {
        oneOfObjects = new ArrayList<>(number);
      }
    }

    public ProtoMessageInfo build() {
      return new ProtoMessageInfo(flags, fields, hasBitsObjects, oneOfObjects);
    }
  }

  private final int flags;

  private final List<ProtoFieldInfo> fields;
  private final List<ProtoObject> hasBitsObjects;
  private final List<Pair<ProtoObject, ProtoObject>> oneOfObjects;

  private ProtoMessageInfo(
      int flags,
      List<ProtoFieldInfo> fields,
      List<ProtoObject> hasBitsObjects,
      List<Pair<ProtoObject, ProtoObject>> oneOfObjects) {
    this.flags = flags;
    this.fields = fields;
    this.hasBitsObjects = hasBitsObjects;
    this.oneOfObjects = oneOfObjects;
  }

  public static ProtoMessageInfo.Builder builder() {
    return new ProtoMessageInfo.Builder();
  }

  public boolean isProto2() {
    return (flags & IS_PROTO_2_MASK) != 0;
  }

  public List<ProtoFieldInfo> getFields() {
    return fields;
  }

  public int getFlags() {
    return flags;
  }

  public List<ProtoObject> getHasBitsObjects() {
    return hasBitsObjects;
  }

  public List<Pair<ProtoObject, ProtoObject>> getOneOfObjects() {
    return oneOfObjects;
  }

  public boolean hasFields() {
    return fields != null && !fields.isEmpty();
  }

  public int numberOfFields() {
    return fields != null ? fields.size() : 0;
  }

  public int numberOfHasBitsObjects() {
    return hasBitsObjects != null ? hasBitsObjects.size() : 0;
  }

  public int numberOfOneOfObjects() {
    return oneOfObjects != null ? oneOfObjects.size() : 0;
  }
}