src/OrbitControls.h - main - redgpu/CuRast

src/OrbitControls.h
180 строк4 KB
m-schuetz
initial commit
23 апр 2026, 18:12
23 апр 2026, 18:1204cf9f3

#pragma once

// #include <iostream>

 #include "glm/common.hpp"
 #include "glm/matrix.hpp"
 #include <glm/gtx/transform.hpp>

// #include "Runtime.h"

// using namespace std;

using glm::mat4;
using glm::vec3;
using glm::length;

struct OrbitControls{

	double yaw = 0.0;
	double pitch = 0.0;
	double radius = 2;
	glm::dvec3 target = {0.0, 0.0, 0.0};
	glm::dmat4 world;

	bool isLeftDown = false;
	bool isMiddleDown = false;
	bool isRightDown = false;

	glm::dvec2 mousePos;

	OrbitControls(){
	
	}

	glm::dvec3 getDirection(){
		auto rotation = getRotation();

		auto dir = rotation * glm::dvec4(0, 1, 0, 1.0);

		return dir;
	}

	glm::dvec3 getPosition(){
		auto dir = getDirection();

		auto pos = target - (radius * dir);

		return pos;
	}

	glm::dmat4 getRotation(){
		glm::dvec3 up    = {0, 0, 1};
		glm::dvec3 right = {1, 0, 0};

		auto rotYaw = glm::rotate(yaw, up);
		auto rotPitch = glm::rotate(pitch, right);

		//auto rotation = rotYaw;
		//auto rotation = rotPitch * rotYaw;
		auto rotation = rotPitch * rotYaw;

		return rotation;
	}

	void translate_local(double x, double y, double z){
		auto _pos = glm::dvec3(0, 0, 0);
		auto _right = glm::dvec3(1, 0, 0);
		auto _forward = glm::dvec3(0, 1, 0);
		auto _up = glm::dvec3(0, 0, 1);

		_pos = world * glm::dvec4(_pos, 1);
		_right = world * glm::dvec4(_right, 1);
		_forward = world * glm::dvec4(_forward, 1);
		_up = world * glm::dvec4(_up, 1);

		_right = glm::normalize(_right - _pos) * x;
		_forward = glm::normalize(_forward - _pos) * y;
		_up = glm::normalize(_up - _pos) * (-z);

		this->target = this->target + _right + _forward + _up;
	}

	void onMouseButton(int button, int action, int mods){
		//cout << "button: " << button << ", action: " << action << ", mods: " << mods << endl;

		if(button == 0 && action == 1){
			isLeftDown = true;
		}else if(action == 0){
			isLeftDown = false;
		}

		if(button == 2 && action == 1){
			isMiddleDown = true;
		}else if(action == 0){
			isMiddleDown = false;
		}

		if(button == 1 && action == 1){
			isRightDown = true;
		}else if(action == 0){
			isRightDown = false;
		}
	}

	void onMouseMove(double xpos, double ypos){

		glm::dvec2 newMousePos = {xpos, ypos};
		glm::dvec2 diff = newMousePos - mousePos;

		if(isLeftDown){
			yaw -= double(diff.x) / 400.0;
			pitch += double(diff.y) / 400.0;
		}else if(isRightDown){
			auto ux = diff.x / 1000.0;
			auto uy = -diff.y / 1000.0;

			translate_local(-ux * radius, uy * radius, 0);
		}
		
		mousePos = newMousePos;
	}

	void onMouseScroll(double xoffset, double yoffset){

		// +1: zoom in
		// -1: zoom out

		if(yoffset < -0.0f){
			radius = radius * 1.1;
		}else if(yoffset > 0.0f){
			radius = radius / 1.1;
		}

	}

	void update(){
		glm::dvec3 up    = {0, 0, 1};
		glm::dvec3 right = {1, 0, 0};

		auto translateRadius = glm::translate(
			glm::dmat4(), 
			glm::dvec3(0.0, 0.0, radius));
		auto translateTarget = glm::translate(glm::dmat4(), target);
		auto rotYaw = glm::rotate(yaw, up);
		auto rotPitch = glm::rotate(pitch, right);

		auto flip = glm::dmat4(
			1.0, 0.0, 0.0, 0.0,
			0.0, 0.0, 1.0, 0.0,
			0.0, -1.0, 0.0, 0.0,
			0.0, 0.0, 0.0, 1.0
		);

		world = translateTarget * rotYaw * rotPitch * flip * translateRadius;
	}

	// focus view onto given bounding box
	void focus(vec3 min, vec3 max, float factor = 1.0f){
		vec3 center = (min + max) / 2.0f;
		float radius = factor * length(max - min);

		this->target = center;
		this->radius = radius;
	}

	// void clear(){
	// 	wheel_x = 0;
	// 	wheel_y = 0;
	// 	// pos_x = 0;
	// 	// pos_y = 0;
	// 	button = 0;
	// 	action = 0;
	// 	mods = 0;

	// 	buttonPressed = false;
	// 	mouseMoved = false;
	// 	mouseScrolled = false;
	// }

};