Skip to content
Snippets Groups Projects
src_window.cpp 4.08 KiB
Newer Older
#include "src_window.hpp"

Nicolas Pope's avatar
Nicolas Pope committed
#include "screen.hpp"
#include "camera.hpp"
#include <nanogui/imageview.h>
#include <nanogui/textbox.h>
#include <nanogui/slider.h>
#include <nanogui/combobox.h>
#include <nanogui/label.h>
#include <nanogui/opengl.h>
#include <nanogui/glutil.h>
#include <nanogui/screen.h>
#include <nanogui/layout.h>

#ifdef HAVE_LIBARCHIVE
#include "ftl/rgbd/snapshot.hpp"
using ftl::gui::SourceWindow;
Nicolas Pope's avatar
Nicolas Pope committed
using ftl::gui::Screen;
using ftl::rgbd::Source;
using std::string;
Nicolas Pope's avatar
Nicolas Pope committed
using ftl::config::json_t;
Nicolas Pope's avatar
Nicolas Pope committed
SourceWindow::SourceWindow(ftl::gui::Screen *screen)
		: nanogui::Window(screen, ""), screen_(screen) {
	setLayout(new nanogui::GroupLayout());

	using namespace nanogui;
Nicolas Pope's avatar
Nicolas Pope committed
	//if (!screen->root()->get<json_t>("sources")) {
	//	screen->root()->getConfig()["sources"] = json_t::array();
	//}

	//src_ = ftl::create<Source>(ctrl->getRoot(), "source", ctrl->getNet());
Nicolas Pope's avatar
Nicolas Pope committed
	//Widget *tools = new Widget(this);
	//    tools->setLayout(new BoxLayout(Orientation::Horizontal,
	//                                   Alignment::Middle, 0, 6));
	new Label(this, "Select source","sans-bold");
Nicolas Pope's avatar
Nicolas Pope committed
	available_ = screen_->control()->getNet()->findAll<string>("list_streams");
	auto select = new ComboBox(this, available_);
	select->setCallback([this,select](int ix) {
Nicolas Pope's avatar
Nicolas Pope committed
		//src_->set("uri", available_[ix]);
		// TODO(Nick) Check camera exists first
		screen_->setActiveCamera(cameras_[available_[ix]]);
		LOG(INFO) << "Change source: " << ix;
Nicolas Pope's avatar
Nicolas Pope committed
	});
Nicolas Pope's avatar
Nicolas Pope committed
	_updateCameras();

	screen->net()->onConnect([this,select](ftl::net::Peer *p) {
		available_ = screen_->net()->findAll<string>("list_streams");
		select->setItems(available_);
Nicolas Pope's avatar
Nicolas Pope committed
		_updateCameras();
Nicolas Pope's avatar
Nicolas Pope committed
	/*new Label(this, "Source Options","sans-bold");
Nicolas Pope's avatar
Nicolas Pope committed

	auto tools = new Widget(this);
    tools->setLayout(new BoxLayout(Orientation::Horizontal,
                                       Alignment::Middle, 0, 6));

	auto button_rgb = new Button(tools, "RGB");
	button_rgb->setTooltip("RGB left image");
	button_rgb->setFlags(Button::RadioButton);
	button_rgb->setPushed(true);
	button_rgb->setChangeCallback([this](bool state) { mode_ = Mode::rgb; });

Nicolas Pope's avatar
Nicolas Pope committed
	auto button_depth = new Button(tools, "Depth");
	button_depth->setFlags(Button::RadioButton);
	button_depth->setChangeCallback([this](bool state) { mode_ = Mode::depth; });

Nicolas Pope's avatar
Nicolas Pope committed
	auto button_stddev = new Button(tools, "SD. 25");
	button_stddev->setTooltip("Standard Deviation over 25 frames");
	button_stddev->setFlags(Button::RadioButton);
	button_stddev->setChangeCallback([this](bool state) { mode_ = Mode::stddev; });
Nicolas Pope's avatar
Nicolas Pope committed
	//auto button_pose = new Button(this, "Adjust Pose", ENTYPO_ICON_COMPASS);
	//button_pose->setCallback([this]() {
	//	auto posewin = new PoseWindow(screen_, screen_->control(), src_->getURI());
	//	posewin->setTheme(theme());
	//});
#ifdef HAVE_LIBARCHIVE
Nicolas Pope's avatar
Nicolas Pope committed
	auto button_snapshot = new Button(this, "Snapshot", ENTYPO_ICON_IMAGES);
	button_snapshot->setCallback([this] {
Nicolas Pope's avatar
Nicolas Pope committed
			/*char timestamp[18];
			std::time_t t=std::time(NULL);
			std::strftime(timestamp, sizeof(timestamp), "%F-%H%M%S", std::localtime(&t));
			auto writer = ftl::rgbd::SnapshotWriter(std::string(timestamp) + ".tar.gz");
			cv::Mat rgb, depth;
			this->src_->getFrames(rgb, depth);
			if (!writer.addCameraRGBD(
					"0", // TODO
					rgb,
					depth,
					this->src_->getPose(),
					this->src_->parameters()
				)) {
				LOG(ERROR) << "Snapshot failed";
			}
		}
		catch(std::runtime_error) {
			LOG(ERROR) << "Snapshot failed (file error)";
		}
	});
#endif
Nicolas Pope's avatar
Nicolas Pope committed
	//auto imageView = new VirtualCameraView(this);
	//cam.view = imageView;
Nicolas Pope's avatar
Nicolas Pope committed
	//imageView->setGridThreshold(20);
	//imageView->setSource(src_);
Nicolas Pope's avatar
Nicolas Pope committed
	//image_ = imageView;*/
}

void SourceWindow::_updateCameras() {
	for (auto s : available_) {
		if (cameras_.find(s) == cameras_.end()) {
			json_t srcjson;
			srcjson["uri"] = s;
			screen_->root()->getConfig()["sources"].push_back(srcjson);
			std::vector<ftl::rgbd::Source*> srcs = ftl::createArray<ftl::rgbd::Source>(screen_->root(), "sources", screen_->net());
			auto *src = srcs[srcs.size()-1];

			auto *cam = new ftl::gui::Camera(screen_, src);
			cameras_[s] = cam;
		} else {
			LOG(INFO) << "Camera already exists: " << s;
		}
	}
}

SourceWindow::~SourceWindow() {

}