// Code generated by sidekick. DO NOT EDIT.
//
// Copyright 2026 Google LLC
//
// 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
//
//     https://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.

#if JobService
  import Foundation
  import GoogleCloudWkt

  /// Active learning data sampling config. For every active learning labeling
  /// iteration, it will select a batch of data based on the sampling strategy.
  public struct SampleConfig: Codable, Equatable, GoogleCloudWkt._AnyPackable,
    Sendable
  {
    /// Field to choose sampling strategy. Sampling strategy will decide which data
    /// should be selected for human labeling in every batch.
    public var sampleStrategy: SampleConfig.SampleStrategy = SampleConfig.SampleStrategy()

    /// Decides sample size for the initial batch. initial_batch_sample_percentage
    /// is used by default.
    public var initialBatchSampleSize: OneOf_InitialBatchSampleSize? = nil

    /// Decides sample size for the following batches.
    /// following_batch_sample_percentage is used by default.
    public var followingBatchSampleSize: OneOf_FollowingBatchSampleSize? = nil

    /// Initialize a new instance of `SampleConfig`.
    public init() {}

    /// Use `config` to return a new instance of this object, with some fields updated.
    ///
    /// Commonly used to initialize the value, for example:
    ///
    /// ```
    /// let value = SampleConfig().with { $0.initialBatchSamplePercentage = ... }
    /// ```
    public func with(_ config: (inout Self) throws -> Swift.Void) rethrows -> Self {
      var copy = self
      try config(&copy)
      return copy
    }

    private enum CodingKeys: Swift.String, CodingKey {
      case initialBatchSamplePercentage = "initialBatchSamplePercentage"
      case followingBatchSamplePercentage = "followingBatchSamplePercentage"
      case sampleStrategy = "sampleStrategy"
    }

    public init(from decoder: Decoder) throws {
      let container = try decoder.container(keyedBy: CodingKeys.self)
      self.sampleStrategy = try container.decode(
        SampleConfig.SampleStrategy.self, forKey: .sampleStrategy)

      var initialBatchSampleSize: OneOf_InitialBatchSampleSize? = nil
      let initialBatchSampleSizeCheckAndSet = {
        if initialBatchSampleSize != nil {
          throw DecodingError.dataCorrupted(
            DecodingError.Context(
              codingPath: decoder.codingPath,
              debugDescription: "Multiple values set for oneof 'initialBatchSampleSize'"))
        }
        initialBatchSampleSize = $0
      }
      if let initialBatchSamplePercentage = try container.decodeIfPresent(
        Swift.Int32.self, forKey: .initialBatchSamplePercentage)
      {
        try initialBatchSampleSizeCheckAndSet(
          .initialBatchSamplePercentage(initialBatchSamplePercentage))
      }
      self.initialBatchSampleSize = initialBatchSampleSize

      var followingBatchSampleSize: OneOf_FollowingBatchSampleSize? = nil
      let followingBatchSampleSizeCheckAndSet = {
        if followingBatchSampleSize != nil {
          throw DecodingError.dataCorrupted(
            DecodingError.Context(
              codingPath: decoder.codingPath,
              debugDescription: "Multiple values set for oneof 'followingBatchSampleSize'"))
        }
        followingBatchSampleSize = $0
      }
      if let followingBatchSamplePercentage = try container.decodeIfPresent(
        Swift.Int32.self, forKey: .followingBatchSamplePercentage)
      {
        try followingBatchSampleSizeCheckAndSet(
          .followingBatchSamplePercentage(followingBatchSamplePercentage))
      }
      self.followingBatchSampleSize = followingBatchSampleSize
    }

    public func encode(to encoder: Encoder) throws {
      var container = encoder.container(keyedBy: CodingKeys.self)
      try container.encode(self.sampleStrategy, forKey: .sampleStrategy)

      if let choice = self.initialBatchSampleSize {
        switch choice {
        case .initialBatchSamplePercentage(let value):
          try container.encode(value, forKey: .initialBatchSamplePercentage)
        }
      }

      if let choice = self.followingBatchSampleSize {
        switch choice {
        case .followingBatchSamplePercentage(let value):
          try container.encode(value, forKey: .followingBatchSamplePercentage)
        }
      }
    }

    /// Sample strategy decides which subset of DataItems should be selected for
    /// human labeling in every batch.
    public enum SampleStrategy: Codable, Equatable, Sendable {
      /// Default will be treated as UNCERTAINTY.
      case unspecified
      /// Sample the most uncertain data to label.
      case uncertainty
      /// Encodes an unknown integer value.
      ///
      /// The most common cause for an unknown values is for the service to send
      /// a value unknown to the library. We recommend you update your library to
      /// the latest version.
      case unknownIntValue(Int)
      /// Encodes an unknown string value.
      ///
      /// The most common cause for an unknown values is for the service to send
      /// a value unknown to the library. We recommend you update your library to
      /// the latest version.
      case unknownStringValue(String)

      public init() {
        self = .unspecified
      }

      /// Returns the integer value associated with the enumeration.
      ///
      /// If the enumeration was initialized with an unknown string value, this returns `nil`.
      public var intValue: Int? {
        switch self {
        case .unspecified: return 0
        case .uncertainty: return 1
        case .unknownIntValue(let v): return v
        case .unknownStringValue: return nil
        }
      }

      /// Returns the string value (or name) associated with the enumeration.
      ///
      /// If the enumeration was initialized with an unknown integer value, this returns `nil`.
      public var stringValue: Swift.String? {
        switch self {
        case .unspecified: return "SAMPLE_STRATEGY_UNSPECIFIED"
        case .uncertainty: return "UNCERTAINTY"
        case .unknownIntValue: return nil
        case .unknownStringValue(let v): return v
        }
      }

      /// Initialize from a string value.
      ///
      /// If the value is unknown, this initializes to [`unknownStringValue`](doc:SampleStrategy/unknownStringValue(_:)).
      public init(stringValue: Swift.String) {
        switch stringValue {
        case "SAMPLE_STRATEGY_UNSPECIFIED": self = .unspecified
        case "UNCERTAINTY": self = .uncertainty
        default: self = .unknownStringValue(stringValue)
        }
      }

      /// Initialize from an integer value.
      ///
      /// If the value is unknown, this initializes to [`unknownIntValue`](doc:SampleStrategy/unknownIntValue(_:)).
      public init(intValue: Int) {
        switch intValue {
        case 0: self = .unspecified
        case 1: self = .uncertainty
        default: self = .unknownIntValue(intValue)
        }
      }

      public init(from decoder: Decoder) throws {
        let container = try decoder.singleValueContainer()
        if let v = try? container.decode(Int.self) {
          self.init(intValue: v)
          return
        }
        if let s = try? container.decode(String.self) {
          if let v = Int(s) {
            self.init(intValue: v)
          } else {
            self.init(stringValue: s)
          }
          return
        }
        throw DecodingError.dataCorruptedError(
          in: container, debugDescription: "Expected enum value, must be integer or string.")
      }

      public func encode(to encoder: Encoder) throws {
        var container = encoder.singleValueContainer()
        switch self {
        case .unspecified: return try container.encode(0)
        case .uncertainty: return try container.encode(1)
        case .unknownIntValue(let v): return try container.encode(v)
        case .unknownStringValue(let v): return try container.encode(v)
        }
      }
    }

    /// Decides sample size for the initial batch. initial_batch_sample_percentage
    /// is used by default.
    public enum OneOf_InitialBatchSampleSize: Codable, Equatable, Sendable {
      /// The percentage of data needed to be labeled in the first batch.
      case initialBatchSamplePercentage(Swift.Int32)
    }

    /// Decides sample size for the following batches.
    /// following_batch_sample_percentage is used by default.
    public enum OneOf_FollowingBatchSampleSize: Codable, Equatable, Sendable {
      /// The percentage of data needed to be labeled in each following batch
      /// (except the first batch).
      case followingBatchSamplePercentage(Swift.Int32)
    }

    public static var _anyTypeUrl: Swift.String {
      return "type.googleapis.com/google.cloud.aiplatform.v1.SampleConfig"
    }
    public init(fromAny any: GoogleCloudWkt.`Any`) throws {
      self = try GoogleCloudWkt._slowAnyDeserialize(Self.self, from: any)
    }
    public func _pack() throws -> GoogleCloudWkt.Struct {
      return try GoogleCloudWkt._slowAnySerialize(message: self)
    }
  }
#endif
