kinect/codes/Azure-Kinect-Sensor-SDK/tools/k4aviewer/k4asourceselectiondockcontr...

154 lines
4.1 KiB
C++
Raw Permalink Normal View History

2024-03-06 18:05:53 +00:00
// 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());
}
}