116 lines
4.4 KiB
TypeScript
116 lines
4.4 KiB
TypeScript
import { useState, useMemo, useEffect } from "react"
|
|
|
|
import { ClientType, DefaultWorkletNodeSetting, DefaultWorkletNodeSetting_DDSP_SVC, DefaultWorkletNodeSetting_RVC, DefaultWorkletNodeSetting_so_vits_svc_40, DefaultWorkletNodeSetting_so_vits_svc_40v2, INDEXEDDB_KEY_WORKLETNODE, WorkletNodeSetting } from "../const"
|
|
import { VoiceChangerClient } from "../VoiceChangerClient"
|
|
import { useIndexedDB } from "./useIndexedDB"
|
|
|
|
export type UseWorkletNodeSettingProps = {
|
|
clientType: ClientType | null
|
|
voiceChangerClient: VoiceChangerClient | null
|
|
}
|
|
|
|
export type WorkletNodeSettingState = {
|
|
workletNodeSetting: WorkletNodeSetting;
|
|
clearSetting: () => Promise<void>
|
|
updateWorkletNodeSetting: (setting: WorkletNodeSetting) => void
|
|
startOutputRecording: () => void
|
|
stopOutputRecording: () => Promise<Float32Array>
|
|
trancateBuffer: () => Promise<void>
|
|
}
|
|
|
|
export const useWorkletNodeSetting = (props: UseWorkletNodeSettingProps): WorkletNodeSettingState => {
|
|
const defaultWorkletNodeSetting = useMemo(() => {
|
|
if (props.clientType == "MMVCv13") {
|
|
return DefaultWorkletNodeSetting
|
|
} else if (props.clientType == "MMVCv15") {
|
|
return DefaultWorkletNodeSetting
|
|
} else if (props.clientType == "so-vits-svc-40" || props.clientType == "so-vits-svc-40_c") {
|
|
return DefaultWorkletNodeSetting_so_vits_svc_40
|
|
} else if (props.clientType == "so-vits-svc-40v2") {
|
|
return DefaultWorkletNodeSetting_so_vits_svc_40v2
|
|
} else if (props.clientType == "DDSP-SVC") {
|
|
return DefaultWorkletNodeSetting_DDSP_SVC
|
|
} else if (props.clientType == "RVC") {
|
|
return DefaultWorkletNodeSetting_RVC
|
|
} else {
|
|
return DefaultWorkletNodeSetting
|
|
}
|
|
}, [])
|
|
|
|
const [workletNodeSetting, _setWorkletNodeSetting] = useState<WorkletNodeSetting>(defaultWorkletNodeSetting)
|
|
const { setItem, getItem, removeItem } = useIndexedDB({ clientType: props.clientType })
|
|
|
|
// 初期化 その１ DBから取得
|
|
useEffect(() => {
|
|
const loadCache = async () => {
|
|
const setting = await getItem(INDEXEDDB_KEY_WORKLETNODE) as WorkletNodeSetting
|
|
if (setting) {
|
|
_setWorkletNodeSetting({ ...setting, sendingSampleRate: 48000 }) // sample rateは時限措置
|
|
}
|
|
}
|
|
loadCache()
|
|
}, [])
|
|
|
|
// 初期化 その２ クライアントに設定
|
|
useEffect(() => {
|
|
if (!props.voiceChangerClient) return
|
|
props.voiceChangerClient.setServerUrl(workletNodeSetting.serverUrl)
|
|
props.voiceChangerClient.updateWorkletNodeSetting(workletNodeSetting)
|
|
}, [props.voiceChangerClient])
|
|
|
|
|
|
|
|
const clearSetting = async () => {
|
|
await removeItem(INDEXEDDB_KEY_WORKLETNODE)
|
|
}
|
|
|
|
//////////////
|
|
// 設定
|
|
/////////////
|
|
|
|
const updateWorkletNodeSetting = useMemo(() => {
|
|
return (_workletNodeSetting: WorkletNodeSetting) => {
|
|
if (!props.voiceChangerClient) return
|
|
for (let k in _workletNodeSetting) {
|
|
const cur_v = workletNodeSetting[k as keyof WorkletNodeSetting]
|
|
const new_v = _workletNodeSetting[k as keyof WorkletNodeSetting]
|
|
if (cur_v != new_v) {
|
|
_setWorkletNodeSetting(_workletNodeSetting)
|
|
setItem(INDEXEDDB_KEY_WORKLETNODE, _workletNodeSetting)
|
|
props.voiceChangerClient.updateWorkletNodeSetting(_workletNodeSetting)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}, [props.voiceChangerClient, workletNodeSetting])
|
|
|
|
const startOutputRecording = useMemo(() => {
|
|
return () => {
|
|
if (!props.voiceChangerClient) return
|
|
props.voiceChangerClient.startOutputRecording()
|
|
}
|
|
}, [props.voiceChangerClient])
|
|
|
|
const stopOutputRecording = useMemo(() => {
|
|
return async () => {
|
|
if (!props.voiceChangerClient) return new Float32Array()
|
|
return props.voiceChangerClient.stopOutputRecording()
|
|
}
|
|
}, [props.voiceChangerClient])
|
|
|
|
const trancateBuffer = useMemo(() => {
|
|
return async () => {
|
|
if (!props.voiceChangerClient) return
|
|
props.voiceChangerClient.trancateBuffer()
|
|
}
|
|
}, [props.voiceChangerClient])
|
|
|
|
return {
|
|
workletNodeSetting,
|
|
clearSetting,
|
|
updateWorkletNodeSetting,
|
|
startOutputRecording,
|
|
stopOutputRecording,
|
|
trancateBuffer
|
|
}
|
|
} |