154 lines
4.1 KiB
C++
154 lines
4.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
// Associated header
|
|
//
|
|
#include "k4asourceselectiondockcontrol.h"
|
|
|
|
// System headers
|
|
//
|
|
#include <sstream>
|
|
|
|
// Library headers
|
|
//
|
|
#include "k4aimgui_all.h"
|
|
#include <k4a/k4a.hpp>
|
|
#include <k4arecord/playback.hpp>
|
|
|
|
// Project headers
|
|
//
|
|
#include "filesystem17.h"
|
|
#include "k4aaudiomanager.h"
|
|
#include "k4aimguiextensions.h"
|
|
#include "k4aviewererrormanager.h"
|
|
#include "k4arecordingdockcontrol.h"
|
|
#include "k4aviewerutil.h"
|
|
#include "k4awindowmanager.h"
|
|
|
|
using namespace k4aviewer;
|
|
|
|
K4ASourceSelectionDockControl::K4ASourceSelectionDockControl()
|
|
{
|
|
RefreshDevices();
|
|
}
|
|
|
|
K4ADockControlStatus K4ASourceSelectionDockControl::Show()
|
|
{
|
|
ImGui::SetNextTreeNodeOpen(true, ImGuiCond_FirstUseEver);
|
|
if (ImGui::TreeNode("Open Device"))
|
|
{
|
|
ImGuiExtensions::K4AComboBox("Device S/N",
|
|
"(No available devices)",
|
|
ImGuiComboFlags_None,
|
|
m_connectedDevices,
|
|
&m_selectedDevice);
|
|
|
|
if (ImGui::Button("Refresh Devices"))
|
|
{
|
|
RefreshDevices();
|
|
}
|
|
|
|
ImGui::SameLine();
|
|
|
|
const bool openAvailable = !m_connectedDevices.empty();
|
|
{
|
|
ImGuiExtensions::ButtonColorChanger colorChanger(ImGuiExtensions::ButtonColor::Green, openAvailable);
|
|
if (ImGuiExtensions::K4AButton("Open Device", openAvailable))
|
|
{
|
|
OpenDevice();
|
|
}
|
|
}
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
|
|
ImGui::NewLine();
|
|
ImGui::Separator();
|
|
ImGui::NewLine();
|
|
|
|
if (ImGui::TreeNode("Open Recording"))
|
|
{
|
|
if (m_filePicker.Show())
|
|
{
|
|
OpenRecording(m_filePicker.GetPath());
|
|
}
|
|
|
|
ImGui::TreePop();
|
|
}
|
|
|
|
return K4ADockControlStatus::Ok;
|
|
}
|
|
|
|
void K4ASourceSelectionDockControl::RefreshDevices()
|
|
{
|
|
m_selectedDevice = -1;
|
|
|
|
const uint32_t installedDevices = k4a_device_get_installed_count();
|
|
|
|
m_connectedDevices.clear();
|
|
|
|
for (uint32_t i = 0; i < installedDevices; i++)
|
|
{
|
|
try
|
|
{
|
|
k4a::device device = k4a::device::open(i);
|
|
m_connectedDevices.emplace_back(std::make_pair(i, device.get_serialnum()));
|
|
}
|
|
catch (const k4a::error &)
|
|
{
|
|
// We can't have 2 handles to the same device, and we need to open a device handle to check
|
|
// its serial number, so we expect devices we already have open to fail here. Ignore those.
|
|
//
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (!m_connectedDevices.empty())
|
|
{
|
|
m_selectedDevice = m_connectedDevices[0].first;
|
|
}
|
|
|
|
const int audioRefreshStatus = K4AAudioManager::Instance().RefreshDevices();
|
|
if (audioRefreshStatus != SoundIoErrorNone)
|
|
{
|
|
std::stringstream errorBuilder;
|
|
errorBuilder << "Failed to refresh audio devices: " << soundio_strerror(audioRefreshStatus) << "!" << std::endl
|
|
<< "Attempting to open microphones may fail!";
|
|
|
|
K4AViewerErrorManager::Instance().SetErrorStatus(errorBuilder.str());
|
|
}
|
|
}
|
|
|
|
void K4ASourceSelectionDockControl::OpenDevice()
|
|
{
|
|
try
|
|
{
|
|
if (m_selectedDevice < 0)
|
|
{
|
|
K4AViewerErrorManager::Instance().SetErrorStatus("No device selected!");
|
|
return;
|
|
}
|
|
|
|
k4a::device device = k4a::device::open(static_cast<uint32_t>(m_selectedDevice));
|
|
K4AWindowManager::Instance().PushLeftDockControl(std14::make_unique<K4ADeviceDockControl>(std::move(device)));
|
|
}
|
|
catch (const k4a::error &e)
|
|
{
|
|
K4AViewerErrorManager::Instance().SetErrorStatus(e.what());
|
|
}
|
|
}
|
|
|
|
void K4ASourceSelectionDockControl::OpenRecording(const std17::filesystem::path &path)
|
|
{
|
|
try
|
|
{
|
|
k4a::playback recording = k4a::playback::open(path.c_str());
|
|
K4AWindowManager::Instance().PushLeftDockControl(
|
|
std14::make_unique<K4ARecordingDockControl>(path.string(), std::move(recording)));
|
|
}
|
|
catch (const k4a::error &e)
|
|
{
|
|
K4AViewerErrorManager::Instance().SetErrorStatus(e.what());
|
|
}
|
|
}
|