356 lines
13 KiB
Java
356 lines
13 KiB
Java
/*
|
|
* Copyright (C) 2023 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 android.federatedcompute.common;
|
|
|
|
import android.annotation.NonNull;
|
|
import android.annotation.Nullable;
|
|
import android.content.ComponentName;
|
|
import android.os.Parcelable;
|
|
|
|
import com.android.internal.util.Preconditions;
|
|
import com.android.ondevicepersonalization.internal.util.AnnotationValidations;
|
|
import com.android.ondevicepersonalization.internal.util.DataClass;
|
|
|
|
/**
|
|
* Training options when schedule federated computation job.
|
|
*
|
|
* @hide
|
|
*/
|
|
@DataClass(genBuilder = true, genEqualsHashCode = true)
|
|
public final class TrainingOptions implements Parcelable {
|
|
/**
|
|
* The task name to be provided to the federated compute server during checkin. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@NonNull private String mPopulationName = "";
|
|
|
|
/**
|
|
* The remote federated compute server address that federated compute client need to checkin.
|
|
* The field is required and should not be empty.
|
|
*/
|
|
@NonNull private String mServerAddress = "";
|
|
|
|
/**
|
|
* Indicated the component of the application requesting federated learning. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@Nullable private ComponentName mOwnerComponentName = null;
|
|
|
|
@Nullable private TrainingInterval mTrainingInterval = null;
|
|
|
|
/**
|
|
* The context data that federatedcompute will pass back to client when bind to
|
|
* ExampleStoreService and ResultHandlingService.
|
|
*/
|
|
@Nullable private final byte[] mContextData;
|
|
|
|
// Code below generated by codegen v1.0.23.
|
|
//
|
|
// DO NOT MODIFY!
|
|
// CHECKSTYLE:OFF Generated code
|
|
//
|
|
// To regenerate run:
|
|
// $ codegen
|
|
// $ANDROID_BUILD_TOP/packages/modules/OnDevicePersonalization/framework/java/android/federatedcompute/common/TrainingOptions.java
|
|
//
|
|
// To exclude the generated code from IntelliJ auto-formatting enable (one-time):
|
|
// Settings > Editor > Code Style > Formatter Control
|
|
// @formatter:off
|
|
|
|
@DataClass.Generated.Member
|
|
/* package-private */ TrainingOptions(
|
|
@NonNull String populationName,
|
|
@NonNull String serverAddress,
|
|
@Nullable ComponentName ownerComponentName,
|
|
@Nullable TrainingInterval trainingInterval,
|
|
@Nullable byte[] contextData) {
|
|
this.mPopulationName = populationName;
|
|
AnnotationValidations.validate(NonNull.class, null, mPopulationName);
|
|
this.mServerAddress = serverAddress;
|
|
AnnotationValidations.validate(NonNull.class, null, mServerAddress);
|
|
this.mOwnerComponentName = ownerComponentName;
|
|
this.mTrainingInterval = trainingInterval;
|
|
this.mContextData = contextData;
|
|
|
|
// onConstructed(); // You can define this method to get a callback
|
|
}
|
|
|
|
/**
|
|
* The task name to be provided to the federated compute server during checkin. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull String getPopulationName() {
|
|
return mPopulationName;
|
|
}
|
|
|
|
/**
|
|
* The remote federated compute server address that federated compute client need to checkin.
|
|
* The field is required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull String getServerAddress() {
|
|
return mServerAddress;
|
|
}
|
|
|
|
/**
|
|
* Indicated the component of the application requesting federated learning. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @Nullable ComponentName getOwnerComponentName() {
|
|
return mOwnerComponentName;
|
|
}
|
|
|
|
@DataClass.Generated.Member
|
|
public @Nullable TrainingInterval getTrainingInterval() {
|
|
return mTrainingInterval;
|
|
}
|
|
|
|
/**
|
|
* The context data that federatedcompute will pass back to client when bind to
|
|
* ExampleStoreService and ResultHandlingService.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @Nullable byte[] getContextData() {
|
|
return mContextData;
|
|
}
|
|
|
|
@Override
|
|
@DataClass.Generated.Member
|
|
public boolean equals(@Nullable Object o) {
|
|
// You can override field equality logic by defining either of the methods like:
|
|
// boolean fieldNameEquals(TrainingOptions other) { ... }
|
|
// boolean fieldNameEquals(FieldType otherValue) { ... }
|
|
|
|
if (this == o) return true;
|
|
if (o == null || getClass() != o.getClass()) return false;
|
|
@SuppressWarnings("unchecked")
|
|
TrainingOptions that = (TrainingOptions) o;
|
|
//noinspection PointlessBooleanExpression
|
|
return true
|
|
&& java.util.Objects.equals(mPopulationName, that.mPopulationName)
|
|
&& java.util.Objects.equals(mServerAddress, that.mServerAddress)
|
|
&& java.util.Objects.equals(mOwnerComponentName, that.mOwnerComponentName)
|
|
&& java.util.Objects.equals(mTrainingInterval, that.mTrainingInterval)
|
|
&& java.util.Arrays.equals(mContextData, that.mContextData);
|
|
}
|
|
|
|
@Override
|
|
@DataClass.Generated.Member
|
|
public int hashCode() {
|
|
// You can override field hashCode logic by defining methods like:
|
|
// int fieldNameHashCode() { ... }
|
|
|
|
int _hash = 1;
|
|
_hash = 31 * _hash + java.util.Objects.hashCode(mPopulationName);
|
|
_hash = 31 * _hash + java.util.Objects.hashCode(mServerAddress);
|
|
_hash = 31 * _hash + java.util.Objects.hashCode(mOwnerComponentName);
|
|
_hash = 31 * _hash + java.util.Objects.hashCode(mTrainingInterval);
|
|
_hash = 31 * _hash + java.util.Arrays.hashCode(mContextData);
|
|
return _hash;
|
|
}
|
|
|
|
@Override
|
|
@DataClass.Generated.Member
|
|
public void writeToParcel(@NonNull android.os.Parcel dest, int flags) {
|
|
// You can override field parcelling by defining methods like:
|
|
// void parcelFieldName(Parcel dest, int flags) { ... }
|
|
|
|
byte flg = 0;
|
|
if (mOwnerComponentName != null) flg |= 0x4;
|
|
if (mTrainingInterval != null) flg |= 0x8;
|
|
dest.writeByte(flg);
|
|
dest.writeString(mPopulationName);
|
|
dest.writeString(mServerAddress);
|
|
if (mOwnerComponentName != null) dest.writeTypedObject(mOwnerComponentName, flags);
|
|
if (mTrainingInterval != null) dest.writeTypedObject(mTrainingInterval, flags);
|
|
dest.writeByteArray(mContextData);
|
|
}
|
|
|
|
@Override
|
|
@DataClass.Generated.Member
|
|
public int describeContents() {
|
|
return 0;
|
|
}
|
|
|
|
/** @hide */
|
|
@SuppressWarnings({"unchecked", "RedundantCast"})
|
|
@DataClass.Generated.Member
|
|
/* package-private */ TrainingOptions(@NonNull android.os.Parcel in) {
|
|
// You can override field unparcelling by defining methods like:
|
|
// static FieldType unparcelFieldName(Parcel in) { ... }
|
|
|
|
byte flg = in.readByte();
|
|
String populationName = in.readString();
|
|
String serverAddress = in.readString();
|
|
ComponentName ownerComponentName =
|
|
(flg & 0x4) == 0 ? null : (ComponentName) in.readTypedObject(ComponentName.CREATOR);
|
|
TrainingInterval trainingInterval =
|
|
(flg & 0x8) == 0
|
|
? null
|
|
: (TrainingInterval) in.readTypedObject(TrainingInterval.CREATOR);
|
|
byte[] contextData = in.createByteArray();
|
|
|
|
this.mPopulationName = populationName;
|
|
AnnotationValidations.validate(NonNull.class, null, mPopulationName);
|
|
this.mServerAddress = serverAddress;
|
|
AnnotationValidations.validate(NonNull.class, null, mServerAddress);
|
|
this.mOwnerComponentName = ownerComponentName;
|
|
this.mTrainingInterval = trainingInterval;
|
|
this.mContextData = contextData;
|
|
|
|
// onConstructed(); // You can define this method to get a callback
|
|
}
|
|
|
|
@DataClass.Generated.Member
|
|
public static final @NonNull Parcelable.Creator<TrainingOptions> CREATOR =
|
|
new Parcelable.Creator<TrainingOptions>() {
|
|
@Override
|
|
public TrainingOptions[] newArray(int size) {
|
|
return new TrainingOptions[size];
|
|
}
|
|
|
|
@Override
|
|
public TrainingOptions createFromParcel(@NonNull android.os.Parcel in) {
|
|
return new TrainingOptions(in);
|
|
}
|
|
};
|
|
|
|
/** A builder for {@link TrainingOptions} */
|
|
@SuppressWarnings("WeakerAccess")
|
|
@DataClass.Generated.Member
|
|
public static final class Builder {
|
|
|
|
private @NonNull String mPopulationName;
|
|
private @NonNull String mServerAddress;
|
|
private @Nullable ComponentName mOwnerComponentName;
|
|
private @Nullable TrainingInterval mTrainingInterval;
|
|
private @Nullable byte[] mContextData;
|
|
|
|
private long mBuilderFieldsSet = 0L;
|
|
|
|
public Builder() {}
|
|
|
|
/**
|
|
* The task name to be provided to the federated compute server during checkin. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull Builder setPopulationName(@NonNull String value) {
|
|
checkNotUsed();
|
|
Preconditions.checkStringNotEmpty(value);
|
|
mBuilderFieldsSet |= 0x1;
|
|
mPopulationName = value;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* The remote federated compute server address that federated compute client need to
|
|
* checkin. The field is required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull Builder setServerAddress(@NonNull String value) {
|
|
checkNotUsed();
|
|
Preconditions.checkStringNotEmpty(value);
|
|
mBuilderFieldsSet |= 0x2;
|
|
mServerAddress = value;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Indicated the component of the application requesting federated learning. The field is
|
|
* required and should not be empty.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull Builder setOwnerComponentName(@NonNull ComponentName value) {
|
|
checkNotUsed();
|
|
mBuilderFieldsSet |= 0x4;
|
|
mOwnerComponentName = value;
|
|
return this;
|
|
}
|
|
|
|
@DataClass.Generated.Member
|
|
public @NonNull Builder setTrainingInterval(@NonNull TrainingInterval value) {
|
|
checkNotUsed();
|
|
mBuilderFieldsSet |= 0x8;
|
|
mTrainingInterval = value;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* The context data that federatedcompute will pass back to client when bind to
|
|
* ExampleStoreService and ResultHandlingService.
|
|
*/
|
|
@DataClass.Generated.Member
|
|
public @NonNull Builder setContextData(@NonNull byte... value) {
|
|
checkNotUsed();
|
|
mBuilderFieldsSet |= 0x10;
|
|
mContextData = value;
|
|
return this;
|
|
}
|
|
|
|
/** Builds the instance. This builder should not be touched after calling this! */
|
|
public @NonNull TrainingOptions build() {
|
|
checkNotUsed();
|
|
mBuilderFieldsSet |= 0x20; // Mark builder used
|
|
|
|
if ((mBuilderFieldsSet & 0x1) == 0) {
|
|
mPopulationName = "";
|
|
}
|
|
if ((mBuilderFieldsSet & 0x2) == 0) {
|
|
mServerAddress = "";
|
|
}
|
|
if ((mBuilderFieldsSet & 0x4) == 0) {
|
|
mOwnerComponentName = null;
|
|
}
|
|
if ((mBuilderFieldsSet & 0x8) == 0) {
|
|
mTrainingInterval = null;
|
|
}
|
|
TrainingOptions o =
|
|
new TrainingOptions(
|
|
mPopulationName,
|
|
mServerAddress,
|
|
mOwnerComponentName,
|
|
mTrainingInterval,
|
|
mContextData);
|
|
return o;
|
|
}
|
|
|
|
private void checkNotUsed() {
|
|
if ((mBuilderFieldsSet & 0x20) != 0) {
|
|
throw new IllegalStateException(
|
|
"This Builder should not be reused. Use a new Builder instance instead");
|
|
}
|
|
}
|
|
}
|
|
|
|
@DataClass.Generated(
|
|
time = 1706145720044L,
|
|
codegenVersion = "1.0.23",
|
|
sourceFile =
|
|
"packages/modules/OnDevicePersonalization/framework/java/android/federatedcompute/common/TrainingOptions.java",
|
|
inputSignatures =
|
|
"private @android.annotation.NonNull java.lang.String mPopulationName\nprivate @android.annotation.NonNull java.lang.String mServerAddress\nprivate @android.annotation.Nullable android.content.ComponentName mOwnerComponentName\nprivate @android.annotation.Nullable android.federatedcompute.common.TrainingInterval mTrainingInterval\nprivate final @android.annotation.Nullable byte[] mContextData\nclass TrainingOptions extends java.lang.Object implements [android.os.Parcelable]\n@com.android.ondevicepersonalization.internal.util.DataClass(genBuilder=true, genEqualsHashCode=true)")
|
|
@Deprecated
|
|
private void __metadata() {}
|
|
|
|
// @formatter:on
|
|
// End of generated code
|
|
|
|
}
|