xref: /aosp_15_r20/external/flatbuffers/samples/sample_binary.swift (revision 890232f25432b36107d06881e0a25aaa6b473652)
1*890232f2SAndroid Build Coastguard Worker /*
2*890232f2SAndroid Build Coastguard Worker  * Copyright 2021 Google Inc. All rights reserved.
3*890232f2SAndroid Build Coastguard Worker  *
4*890232f2SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*890232f2SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*890232f2SAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*890232f2SAndroid Build Coastguard Worker  *
8*890232f2SAndroid Build Coastguard Worker  *     http://www.apache.org/licenses/LICENSE-2.0
9*890232f2SAndroid Build Coastguard Worker  *
10*890232f2SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*890232f2SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*890232f2SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*890232f2SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*890232f2SAndroid Build Coastguard Worker  * limitations under the License.
15*890232f2SAndroid Build Coastguard Worker  */
16*890232f2SAndroid Build Coastguard Worker 
17*890232f2SAndroid Build Coastguard Worker import FlatBuffers
18*890232f2SAndroid Build Coastguard Worker 
19*890232f2SAndroid Build Coastguard Worker typealias Monster = MyGame_Sample_Monster
20*890232f2SAndroid Build Coastguard Worker typealias Weapon = MyGame_Sample_Weapon
21*890232f2SAndroid Build Coastguard Worker typealias Color = MyGame_Sample_Color
22*890232f2SAndroid Build Coastguard Worker typealias Vec3 = MyGame_Sample_Vec3
23*890232f2SAndroid Build Coastguard Worker 
mainnull24*890232f2SAndroid Build Coastguard Worker func main() {
25*890232f2SAndroid Build Coastguard Worker   let expectedDMG: [Int16] = [3, 5]
26*890232f2SAndroid Build Coastguard Worker   let expectedNames = ["Sword", "Axe"]
27*890232f2SAndroid Build Coastguard Worker 
28*890232f2SAndroid Build Coastguard Worker   var builder = FlatBufferBuilder(initialSize: 1024)
29*890232f2SAndroid Build Coastguard Worker   let weapon1Name = builder.create(string: expectedNames[0])
30*890232f2SAndroid Build Coastguard Worker   let weapon2Name = builder.create(string: expectedNames[1])
31*890232f2SAndroid Build Coastguard Worker 
32*890232f2SAndroid Build Coastguard Worker   let weapon1Start = Weapon.startWeapon(&builder)
33*890232f2SAndroid Build Coastguard Worker   Weapon.add(name: weapon1Name, &builder)
34*890232f2SAndroid Build Coastguard Worker   Weapon.add(damage: expectedDMG[0], &builder)
35*890232f2SAndroid Build Coastguard Worker   let sword = Weapon.endWeapon(&builder, start: weapon1Start)
36*890232f2SAndroid Build Coastguard Worker   let weapon2Start = Weapon.startWeapon(&builder)
37*890232f2SAndroid Build Coastguard Worker   Weapon.add(name: weapon2Name, &builder)
38*890232f2SAndroid Build Coastguard Worker   Weapon.add(damage: expectedDMG[1], &builder)
39*890232f2SAndroid Build Coastguard Worker   let axe = Weapon.endWeapon(&builder, start: weapon2Start)
40*890232f2SAndroid Build Coastguard Worker 
41*890232f2SAndroid Build Coastguard Worker   let name = builder.create(string: "Orc")
42*890232f2SAndroid Build Coastguard Worker   let inventory: [Byte] = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
43*890232f2SAndroid Build Coastguard Worker   let inventoryOffset = builder.createVector(inventory)
44*890232f2SAndroid Build Coastguard Worker 
45*890232f2SAndroid Build Coastguard Worker   let weaponsOffset = builder.createVector(ofOffsets: [sword, axe])
46*890232f2SAndroid Build Coastguard Worker 
47*890232f2SAndroid Build Coastguard Worker   let orc = Monster.createMonster(
48*890232f2SAndroid Build Coastguard Worker     &builder,
49*890232f2SAndroid Build Coastguard Worker     pos: MyGame_Sample_Vec3(x: 1, y: 2, z: 3),
50*890232f2SAndroid Build Coastguard Worker     hp: 300,
51*890232f2SAndroid Build Coastguard Worker     nameOffset: name,
52*890232f2SAndroid Build Coastguard Worker     inventoryVectorOffset: inventoryOffset,
53*890232f2SAndroid Build Coastguard Worker     color: .red,
54*890232f2SAndroid Build Coastguard Worker     weaponsVectorOffset: weaponsOffset,
55*890232f2SAndroid Build Coastguard Worker     equippedType: .weapon,
56*890232f2SAndroid Build Coastguard Worker     equippedOffset: axe)
57*890232f2SAndroid Build Coastguard Worker   builder.finish(offset: orc)
58*890232f2SAndroid Build Coastguard Worker 
59*890232f2SAndroid Build Coastguard Worker   let buf = builder.sizedByteArray
60*890232f2SAndroid Build Coastguard Worker   let monster = Monster.getRootAsMonster(bb: ByteBuffer(bytes: buf))
61*890232f2SAndroid Build Coastguard Worker 
62*890232f2SAndroid Build Coastguard Worker   assert(monster.mana == 150)
63*890232f2SAndroid Build Coastguard Worker   assert(monster.hp == 300)
64*890232f2SAndroid Build Coastguard Worker   assert(monster.name == "Orc")
65*890232f2SAndroid Build Coastguard Worker   assert(monster.color == MyGame.Sample.Color.red)
66*890232f2SAndroid Build Coastguard Worker   assert(monster.pos != nil)
67*890232f2SAndroid Build Coastguard Worker   assert(monster.mutablePos != nil)
68*890232f2SAndroid Build Coastguard Worker   for i in 0..<monster.inventoryCount {
69*890232f2SAndroid Build Coastguard Worker     assert(i == monster.inventory(at: i))
70*890232f2SAndroid Build Coastguard Worker   }
71*890232f2SAndroid Build Coastguard Worker 
72*890232f2SAndroid Build Coastguard Worker   for i in 0..<monster.weaponsCount {
73*890232f2SAndroid Build Coastguard Worker     let weap = monster.weapons(at: i)
74*890232f2SAndroid Build Coastguard Worker     let index = Int(i)
75*890232f2SAndroid Build Coastguard Worker     assert(weap?.damage == expectedDMG[index])
76*890232f2SAndroid Build Coastguard Worker     assert(weap?.name == expectedNames[index])
77*890232f2SAndroid Build Coastguard Worker   }
78*890232f2SAndroid Build Coastguard Worker   assert(monster.equippedType == .weapon)
79*890232f2SAndroid Build Coastguard Worker   let equipped = monster.equipped(type: Weapon.self)
80*890232f2SAndroid Build Coastguard Worker   assert(equipped?.name == "Axe")
81*890232f2SAndroid Build Coastguard Worker   assert(equipped?.damage == 5)
82*890232f2SAndroid Build Coastguard Worker   print("Monster Object is Verified")
83*890232f2SAndroid Build Coastguard Worker }
84