From d95fac9ff85af653169a7f007976a7826133b277 Mon Sep 17 00:00:00 2001
From: FE <93105911+ugokutennp@users.noreply.github.com>
Date: Thu, 28 Sep 2023 12:02:24 +0900
Subject: [PATCH] v2.2

---
 FW_gen.bat | 89 +++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 68 insertions(+), 21 deletions(-)

diff --git a/FW_gen.bat b/FW_gen.bat
index 272c64d..5a97285 100644
--- a/FW_gen.bat
+++ b/FW_gen.bat
@@ -1,39 +1,86 @@
-chcp 65001
+:: Fw_gen.bat
+:: ver 2.2
+
 @echo off
 setlocal enabledelayedexpansion
 
-:: ディレクトリの変数定義
+:: SteamVR directory settings
 set "steamvrDir=C:\Program Files (x86)\Steam\steamapps\common\SteamVR"
-set "firmwareDir=%steamvrDir%\drivers\indexhmd\resources\firmware\radio\gd_1558748372_dfu"
 
-::ディレクトリを削除
+::Delete old directories
 if exist "firmware" (
     rmdir /s /q "firmware"
 )
-
-if exist "%firmwareDir%" (
-    rmdir /s /q "%firmwareDir%"
+if exist "gd_1558748372_dfu" (
+    rmdir /s /q "gd_1558748372_dfu"
 )
 
-:: ディレクトリ作成
+:: Create new directory
+echo Creating a directory...
 mkdir "firmware"
-mkdir "%firmwareDir%"
+mkdir "gd_1558748372_dfu"
 
-:: ファイルの解凍
-powershell -Command "Expand-Archive -Path '%firmwareDir%.zip' -DestinationPath '%firmwareDir%'"
+:: File unpacking
+echo Unpacking firmware...
+powershell -Command "Expand-Archive -Path '%steamvrDir%\drivers\indexhmd\resources\firmware\radio\gd_1558748372_dfu.zip' -DestinationPath 'gd_1558748372_dfu'"
 
-:: アドレスの書き換え
-powershell -Command "$filePath = '%firmwareDir%\temp_app_stamped.bin'; $newValue = 0xB9; $offset = 0x1839D; $bytes = [System.IO.File]::ReadAllBytes($filePath); $bytes[$offset] = $newValue; [IO.File]::WriteAllBytes($filePath, $bytes)"
+:: File copying
+copy gd_1558748372_dfu\temp_app_stamped.bin gd_1558748372_dfu\temp_app_stamped_lyx.bin > nul
 
-:: ファイル名変更
-rename "%firmwareDir%\temp_app_stamped.bin" "temp_app_stamped_lyx.bin"
+:: Binary rewriting (to LYX)
+powershell -Command "$filePath = 'gd_1558748372_dfu\temp_app_stamped_lyx.bin'; $newValue = 0xB9; $offset = 0x1839D; $bytes = [System.IO.File]::ReadAllBytes($filePath); $bytes[$offset] = $newValue; [IO.File]::WriteAllBytes($filePath, $bytes)"
 
-:: ファイルの変換(アプリ)
-srec_cat.exe "%firmwareDir%\temp_app_stamped_lyx.bin" -Binary -offset 0x26000 -o "firmware\temp_app_stamped_lyx.hex" -Intel
+:: Convert bin to uf2
+set args=%*
+powershell "iex((@('')*3+(cat '%~f0'|select -skip 44))-join[char]10)"
 
-:: ファイルの変換(ソフトデバイス)
-srec_cat.exe "%firmwareDir%\s140_nrf52_6.1.1_softdevice.bin" -Binary -offset 0x1000 -o "firmware\s140_nrf52_6.1.1_softdevice.hex" -Intel
+:: Exit
+pause
+exit /b %ERRORLEVEL%
 
-echo ファームウェアを生成しました。
 
-pause
\ No newline at end of file
+
+:: The following is powershell script
+Set-Location $PSScriptRoot
+
+$UF2_MAGIC_START0 = 0x0A324655  # "UF2\n"
+$UF2_MAGIC_START1 = 0x9E5D5157  # Randomly selected
+$UF2_MAGIC_END    = 0x0AB16F30  # Ditto
+$appstartaddr     = 0x26000     # offset
+$familyid         = 0xADA52840  # nrf52840
+$inputFile        = "gd_1558748372_dfu\temp_app_stamped_lyx.bin"
+$outputFile       = "firmware\temp_app_stamped_lyx.uf2"
+
+Write-Host "Converting bin to uf2..."
+
+$file_content = Get-Content -LiteralPath $inputFile -Encoding Byte
+$numblocks = ($file_content.Length + 0xFF) -shr 8
+$outp = @()
+
+foreach ($blockno in @(0..($numblocks - 1))) {
+    $ptr = 0x100 * $blockno
+    $chunk = $file_content[$ptr..($ptr + 0xFF)]
+    $flags = 0
+    if ($familyid) {
+        $flags = $flags -bor 0x2000
+    }
+    $block = @(
+        [System.BitConverter]::GetBytes([Int32]$UF2_MAGIC_START0),
+        [System.BitConverter]::GetBytes([Int32]$UF2_MAGIC_START1),
+        [System.BitConverter]::GetBytes([Int32]$flags),
+        [System.BitConverter]::GetBytes([Int32]($ptr + $appstartaddr)),
+        [System.BitConverter]::GetBytes([Int32]256),
+        [System.BitConverter]::GetBytes([Int32]$blockno),
+        [System.BitConverter]::GetBytes([Int32]$numblocks),
+        [System.BitConverter]::GetBytes([Int32]$familyid),
+        $chunk,
+        (@([Byte]0x00) * (0x200 - (32 + $chunk.Length + 4))),
+        [System.BitConverter]::GetBytes([Int32]$UF2_MAGIC_END)
+    ) | ForEach-Object { $_ }
+    $outp += $block
+}
+
+$outp | Set-Content -LiteralPath $outputFile -Encoding Byte
+
+Write-Host "Firmware generation completed. Output file: $outputFile"
+