chore: import upstream snapshot with attribution

cpp/include/ray/api/task_options.h

@@ -0,0 +1,116 @@
+// Copyright 2017 The Ray Authors.
+//
+// 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.
+
+#pragma once
+
+#include <ray/api/ray_exception.h>
+
+#include <cmath>
+
+namespace ray {
+namespace internal {
+
+inline void CheckTaskOptions(const std::unordered_map<std::string, double> &resources) {
+  for (auto &pair : resources) {
+    if (pair.first.empty() || pair.second == 0) {
+      throw RayException("Resource values should be positive. Specified resource: " +
+                         pair.first + " = " + std::to_string(pair.second) + ".");
+    }
+    // Note: A resource value should be an integer if it is greater than 1.0.
+    // e.g. 3.0 is a valid resource value, but 3.5 is not.
+    double intpart;
+    if (pair.second > 1 && std::modf(pair.second, &intpart) != 0.0) {
+      throw RayException(
+          "A resource value should be an integer if it is greater than 1.0. Specified "
+          "resource: " +
+          pair.first + " = " + std::to_string(pair.second) + ".");
+    }
+  }
+}
+
+}  // namespace internal
+
+enum class PlacementStrategy {
+  PACK = 0,
+  SPREAD = 1,
+  STRICT_PACK = 2,
+  STRICT_SPREAD = 3,
+  UNRECOGNIZED = -1
+};
+
+enum PlacementGroupState {
+  PENDING = 0,
+  PREPARED = 1,
+  CREATED = 2,
+  REMOVED = 3,
+  RESCHEDULING = 4,
+  UNRECOGNIZED = -1,
+};
+
+struct PlacementGroupCreationOptions {
+  std::string name;
+  std::vector<std::unordered_map<std::string, double>> bundles;
+  PlacementStrategy strategy;
+};
+
+class PlacementGroup {
+ public:
+  PlacementGroup() = default;
+  PlacementGroup(std::string id,
+                 PlacementGroupCreationOptions options,
+                 PlacementGroupState state = PlacementGroupState::UNRECOGNIZED)
+      : id_(std::move(id)), options_(std::move(options)), state_(state) {}
+  std::string GetID() const { return id_; }
+  std::string GetName() { return options_.name; }
+  std::vector<std::unordered_map<std::string, double>> GetBundles() {
+    return options_.bundles;
+  }
+  ray::PlacementGroupState GetState() { return state_; }
+  PlacementStrategy GetStrategy() { return options_.strategy; }
+  bool Wait(int timeout_seconds) { return callback_(id_, timeout_seconds); }
+  void SetWaitCallbak(std::function<bool(const std::string &, int)> callback) {
+    callback_ = std::move(callback);
+  }
+  bool Empty() const { return id_.empty(); }
+
+ private:
+  std::string id_;
+  PlacementGroupCreationOptions options_;
+  PlacementGroupState state_;
+  std::function<bool(const std::string &, int)> callback_;
+};
+
+namespace internal {
+
+struct CallOptions {
+  std::string name;
+  std::unordered_map<std::string, double> resources;
+  PlacementGroup group;
+  int bundle_index;
+  std::string serialized_runtime_env_info;
+};
+
+struct ActorCreationOptions {
+  std::string name;
+  std::string ray_namespace;
+  std::unordered_map<std::string, double> resources;
+  int max_restarts = 0;
+  int max_concurrency = 1;
+  PlacementGroup group;
+  int bundle_index;
+  std::string serialized_runtime_env_info;
+};
+}  // namespace internal
+
+}  // namespace ray