# automatically generated by the FlatBuffers compiler, do not modify # namespace: import flatbuffers from flatbuffers.compat import import_numpy from typing import Any from typing import Optional np = import_numpy() class Control(object): __slots__ = ['_tab'] @classmethod def GetRootAs(cls, buf, offset: int = 0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = Control() x.Init(buf, n + offset) return x @classmethod def GetRootAsControl(cls, buf, offset=0): """This method is deprecated. Please switch to GetRootAs.""" return cls.GetRootAs(buf, offset) # Control def Init(self, buf: bytes, pos: int): self._tab = flatbuffers.table.Table(buf, pos) # Control def File(self) -> Optional[str]: o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.String(o + self._tab.Pos) return None # Control def Weight(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) if o != 0: return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos) return 1.0 # Control def GuidanceStart(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) if o != 0: return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos) return 0.0 # Control def GuidanceEnd(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) if o != 0: return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos) return 1.0 # Control def NoPrompt(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) if o != 0: return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) return False # Control def GlobalAveragePooling(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14)) if o != 0: return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) return True # Control def DownSamplingRate(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16)) if o != 0: return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos) return 1.0 # Control def ControlMode(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) if o != 0: return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos) return 0 # Control def TargetBlocks(self, j: int): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20)) if o != 0: a = self._tab.Vector(o) return self._tab.String(a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4)) return "" # Control def TargetBlocksLength(self) -> int: o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20)) if o != 0: return self._tab.VectorLen(o) return 0 # Control def TargetBlocksIsNone(self) -> bool: o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20)) return o == 0 # Control def InputOverride(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22)) if o != 0: return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos) return 0 def ControlStart(builder: flatbuffers.Builder): builder.StartObject(10) def Start(builder: flatbuffers.Builder): ControlStart(builder) def ControlAddFile(builder: flatbuffers.Builder, file: int): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(file), 0) def AddFile(builder: flatbuffers.Builder, file: int): ControlAddFile(builder, file) def ControlAddWeight(builder: flatbuffers.Builder, weight: float): builder.PrependFloat32Slot(1, weight, 1.0) def AddWeight(builder: flatbuffers.Builder, weight: float): ControlAddWeight(builder, weight) def ControlAddGuidanceStart(builder: flatbuffers.Builder, guidanceStart: float): builder.PrependFloat32Slot(2, guidanceStart, 0.0) def AddGuidanceStart(builder: flatbuffers.Builder, guidanceStart: float): ControlAddGuidanceStart(builder, guidanceStart) def ControlAddGuidanceEnd(builder: flatbuffers.Builder, guidanceEnd: float): builder.PrependFloat32Slot(3, guidanceEnd, 1.0) def AddGuidanceEnd(builder: flatbuffers.Builder, guidanceEnd: float): ControlAddGuidanceEnd(builder, guidanceEnd) def ControlAddNoPrompt(builder: flatbuffers.Builder, noPrompt: bool): builder.PrependBoolSlot(4, noPrompt, 0) def AddNoPrompt(builder: flatbuffers.Builder, noPrompt: bool): ControlAddNoPrompt(builder, noPrompt) def ControlAddGlobalAveragePooling(builder: flatbuffers.Builder, globalAveragePooling: bool): builder.PrependBoolSlot(5, globalAveragePooling, 1) def AddGlobalAveragePooling(builder: flatbuffers.Builder, globalAveragePooling: bool): ControlAddGlobalAveragePooling(builder, globalAveragePooling) def ControlAddDownSamplingRate(builder: flatbuffers.Builder, downSamplingRate: float): builder.PrependFloat32Slot(6, downSamplingRate, 1.0) def AddDownSamplingRate(builder: flatbuffers.Builder, downSamplingRate: float): ControlAddDownSamplingRate(builder, downSamplingRate) def ControlAddControlMode(builder: flatbuffers.Builder, controlMode: int): builder.PrependInt8Slot(7, controlMode, 0) def AddControlMode(builder: flatbuffers.Builder, controlMode: int): ControlAddControlMode(builder, controlMode) def ControlAddTargetBlocks(builder: flatbuffers.Builder, targetBlocks: int): builder.PrependUOffsetTRelativeSlot(8, flatbuffers.number_types.UOffsetTFlags.py_type(targetBlocks), 0) def AddTargetBlocks(builder: flatbuffers.Builder, targetBlocks: int): ControlAddTargetBlocks(builder, targetBlocks) def ControlStartTargetBlocksVector(builder, numElems: int) -> int: return builder.StartVector(4, numElems, 4) def StartTargetBlocksVector(builder, numElems: int) -> int: return ControlStartTargetBlocksVector(builder, numElems) def ControlAddInputOverride(builder: flatbuffers.Builder, inputOverride: int): builder.PrependInt8Slot(9, inputOverride, 0) def AddInputOverride(builder: flatbuffers.Builder, inputOverride: int): ControlAddInputOverride(builder, inputOverride) def ControlEnd(builder: flatbuffers.Builder) -> int: return builder.EndObject() def End(builder: flatbuffers.Builder) -> int: return ControlEnd(builder)