ceph/src/arrow/r/R/enums.R

   1 # Licensed to the Apache Software Foundation (ASF) under one
   2 # or more contributor license agreements.  See the NOTICE file
   3 # distributed with this work for additional information
   4 # regarding copyright ownership.  The ASF licenses this file
   5 # to you under the Apache License, Version 2.0 (the
   6 # "License"); you may not use this file except in compliance
   7 # with the License.  You may obtain a copy of the License at
   8 #
   9 #   http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing,
  12 # software distributed under the License is distributed on an
  13 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14 # KIND, either express or implied.  See the License for the
  15 # specific language governing permissions and limitations
  16 # under the License.
  17
  18 #' @export
  19 `print.arrow-enum` <- function(x, ...) {
  20   NextMethod()
  21 }
  22
  23 enum <- function(class, ..., .list = list(...)) {
  24   structure(
  25     .list,
  26     class = c(class, "arrow-enum")
  27   )
  28 }
  29
  30 #' Arrow enums
  31 #' @name enums
  32 #' @export
  33 #' @keywords internal
  34 TimeUnit <- enum("TimeUnit::type",
  35   SECOND = 0L, MILLI = 1L, MICRO = 2L, NANO = 3L
  36 )
  37
  38 #' @rdname enums
  39 #' @export
  40 DateUnit <- enum("DateUnit", DAY = 0L, MILLI = 1L)
  41
  42 #' @rdname enums
  43 #' @export
  44 Type <- enum("Type::type",
  45   "NA" = 0L,
  46   BOOL = 1L,
  47   UINT8 = 2L,
  48   INT8 = 3L,
  49   UINT16 = 4L,
  50   INT16 = 5L,
  51   UINT32 = 6L,
  52   INT32 = 7L,
  53   UINT64 = 8L,
  54   INT64 = 9L,
  55   HALF_FLOAT = 10L,
  56   FLOAT = 11L,
  57   DOUBLE = 12L,
  58   STRING = 13L,
  59   BINARY = 14L,
  60   FIXED_SIZE_BINARY = 15L,
  61   DATE32 = 16L,
  62   DATE64 = 17L,
  63   TIMESTAMP = 18L,
  64   TIME32 = 19L,
  65   TIME64 = 20L,
  66   INTERVAL_MONTHS = 21L,
  67   INTERVAL_DAY_TIME = 22L,
  68   DECIMAL = 23L,
  69   DECIMAL256 = 24L,
  70   LIST = 25L,
  71   STRUCT = 26L,
  72   SPARSE_UNION = 27L,
  73   DENSE_UNION = 28L,
  74   DICTIONARY = 29L,
  75   MAP = 30L,
  76   EXTENSION = 31L,
  77   FIXED_SIZE_LIST = 32L,
  78   DURATION = 33L,
  79   LARGE_STRING = 34L,
  80   LARGE_BINARY = 35L,
  81   LARGE_LIST = 36L
  82 )
  83
  84 TYPES_WITH_NAN <- Type[c("HALF_FLOAT", "FLOAT", "DOUBLE")]
  85
  86 #' @rdname enums
  87 #' @export
  88 StatusCode <- enum("StatusCode",
  89   OK = 0L, OutOfMemory = 1L, KeyError = 2L, TypeError = 3L,
  90   Invalid = 4L, IOError = 5L, CapacityError = 6L, IndexError = 7L,
  91   UnknownError = 9L, NotImplemented = 10L, SerializationError = 11L,
  92   PythonError = 12L, RError = 13L,
  93   PlasmaObjectExists = 20L, PlasmaObjectNotFound = 21L,
  94   PlasmaStoreFull = 22L, PlasmaObjectAlreadySealed = 23L
  95 )
  96
  97 #' @rdname enums
  98 #' @export
  99 FileMode <- enum("FileMode",
 100   READ = 0L, WRITE = 1L, READWRITE = 2L
 101 )
 102
 103 #' @rdname enums
 104 #' @export
 105 MessageType <- enum("MessageType",
 106   NONE = 0L, SCHEMA = 1L, DICTIONARY_BATCH = 2L, RECORD_BATCH = 3L, TENSOR = 4L
 107 )
 108
 109 #' @rdname enums
 110 #' @export
 111 CompressionType <- enum("Compression::type",
 112   UNCOMPRESSED = 0L, SNAPPY = 1L, GZIP = 2L, BROTLI = 3L, ZSTD = 4L, LZ4 = 5L,
 113   LZ4_FRAME = 6L, LZO = 7L, BZ2 = 8L
 114 )
 115
 116 #' @export
 117 #' @rdname enums
 118 FileType <- enum("FileType",
 119   NotFound = 0L, Unknown = 1L, File = 2L, Directory = 3L
 120 )
 121
 122 #' @export
 123 #' @rdname enums
 124 ParquetVersionType <- enum("ParquetVersionType",
 125   PARQUET_1_0 = 0L, PARQUET_2_0 = 1L
 126 )
 127
 128 #' @export
 129 #' @rdname enums
 130 MetadataVersion <- enum("MetadataVersion",
 131   V1 = 0L, V2 = 1L, V3 = 2L, V4 = 3L, V5 = 4L
 132 )
 133
 134 #' @export
 135 #' @rdname enums
 136 QuantileInterpolation <- enum("QuantileInterpolation",
 137   LINEAR = 0L, LOWER = 1L, HIGHER = 2L, NEAREST = 3L, MIDPOINT = 4L
 138 )
 139
 140 #' @export
 141 #' @rdname enums
 142 NullEncodingBehavior <- enum("NullEncodingBehavior",
 143   ENCODE = 0L, MASK = 1L
 144 )
 145
 146 #' @export
 147 #' @rdname enums
 148 NullHandlingBehavior <- enum("NullHandlingBehavior",
 149   EMIT_NULL = 0L, SKIP = 1L, REPLACE = 2L
 150 )
 151
 152 #' @export
 153 #' @rdname enums
 154 RoundMode <- enum("RoundMode",
 155   DOWN = 0L,
 156   UP = 1L,
 157   TOWARDS_ZERO = 2L,
 158   TOWARDS_INFINITY = 3L,
 159   HALF_DOWN = 4L,
 160   HALF_UP = 5L,
 161   HALF_TOWARDS_ZERO = 6L,
 162   HALF_TOWARDS_INFINITY = 7L,
 163   HALF_TO_EVEN = 8L,
 164   HALF_TO_ODD = 9L
 165 )
 166
 167 #' @export
 168 #' @rdname enums
 169 JoinType <- enum("JoinType",
 170   LEFT_SEMI = 0L,
 171   RIGHT_SEMI = 1L,
 172   LEFT_ANTI = 2L,
 173   RIGHT_ANTI = 3L,
 174   INNER = 4L,
 175   LEFT_OUTER = 5L,
 176   RIGHT_OUTER = 6L,
 177   FULL_OUTER = 7L
 178 )