mirror of
https://github.com/google/pebble.git
synced 2025-03-15 08:41:21 +00:00
133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
# Copyright 2024 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
|
|
#
|
|
# 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.
|
|
|
|
|
|
import argparse
|
|
import os
|
|
import json
|
|
|
|
from libpebble2.protocol.base import PebblePacket
|
|
from libpebble2.protocol.base.types import *
|
|
|
|
|
|
class VibeNote(PebblePacket):
|
|
vibe_duration_ms = Uint16()
|
|
brake_duration_ms = Uint8()
|
|
strength = Int8()
|
|
|
|
|
|
class VibeNoteList(PebblePacket):
|
|
notes = FixedList(VibeNote)
|
|
|
|
|
|
class VibePattern(PebblePacket):
|
|
indices = FixedList(Uint8())
|
|
|
|
|
|
class VibePatternRepeatDelay(PebblePacket):
|
|
duration = Uint16()
|
|
|
|
|
|
class VibeAttribute(PebblePacket):
|
|
id = Uint8()
|
|
length = Uint16()
|
|
attribute = Union(id, {
|
|
0x01: VibeNoteList,
|
|
0x02: VibePattern,
|
|
0x03: VibePatternRepeatDelay,
|
|
}, length=length)
|
|
|
|
|
|
class VibeAttributeList(PebblePacket):
|
|
num_attributes = Uint8()
|
|
attributes = FixedList(VibeAttribute, count=num_attributes)
|
|
|
|
|
|
class VibeScore(PebblePacket):
|
|
version = Uint16()
|
|
reserved = Padding(4)
|
|
length = Uint16()
|
|
attr_list = Embed(VibeAttributeList, length=length)
|
|
|
|
|
|
class VibeFile(PebblePacket):
|
|
class Meta:
|
|
endianness = '<'
|
|
|
|
fourcc = FixedString(length=4, default="VIBE")
|
|
score = Embed(VibeScore)
|
|
|
|
|
|
def serialize(json_data):
|
|
CURRENT_VERSION = 1
|
|
NEGATIVE_VIBE_STRENGTH_MAX = -100
|
|
POSITIVE_VIBE_STRENGTH_MAX = 100
|
|
|
|
for note in json_data['notes']:
|
|
if not (NEGATIVE_VIBE_STRENGTH_MAX <= note['strength'] <= POSITIVE_VIBE_STRENGTH_MAX):
|
|
raise ValueError('"strength" {} out of bounds. Values between -100 and 100 only.'
|
|
.format(note['strength']))
|
|
|
|
# construct an object to be fed into the VibeFileAdapter
|
|
note_dictionary = {note['id']: i for i, note in enumerate(json_data['notes'])}
|
|
|
|
vibe_attribute_list = [
|
|
VibeAttribute(
|
|
attribute=VibeNoteList(
|
|
notes=[VibeNote(
|
|
vibe_duration_ms=note['vibe_duration_ms'],
|
|
brake_duration_ms=note['brake_duration_ms'],
|
|
strength=note['strength']
|
|
) for note in json_data['notes']]
|
|
)
|
|
),
|
|
VibeAttribute(
|
|
attribute=VibePattern(indices=[note_dictionary[x]
|
|
for x in json_data['pattern']])
|
|
)]
|
|
if 'repeat_delay_ms' in json_data:
|
|
vibe_attribute_list.append(
|
|
VibeAttribute(
|
|
attribute=VibePatternRepeatDelay(duration=json_data['repeat_delay_ms'])))
|
|
|
|
obj = VibeFile(score=VibeScore(
|
|
version=CURRENT_VERSION,
|
|
attr_list=VibeAttributeList(attributes=vibe_attribute_list)))
|
|
|
|
# do the dirty work
|
|
return obj.serialise()
|
|
|
|
|
|
def convert(file_path, out_path=None):
|
|
if out_path is None:
|
|
out_path = os.path.splitext(file_path)[0] + ".vibe"
|
|
|
|
with open(out_path, 'wb') as o:
|
|
convert_to_file(file_path, o)
|
|
|
|
|
|
def convert_to_file(input_file_path, output_file):
|
|
with open(input_file_path, 'r') as f:
|
|
data = json.loads(f.read())
|
|
|
|
output_file.write(serialize(data))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('json_file')
|
|
args = parser.parse_args()
|
|
|
|
convert(args.json_file)
|