fix pivot not becoming last active refernce

Author: mTvare6

editor/src/messages/tool/common_functionality/pivot.rs

@@ -16,6 +16,8 @@ pub struct Pivot {
 	pivot: Option<DVec2>,
 	/// The old pivot position in the GUI, used to reduce refreshes of the document bar
 	pub old_pivot_position: ReferencePoint,
+	/// The last ReferencePoint which wasn't none
+	pub last_non_none_reference: ReferencePoint,
 	/// Used to enable and disable the pivot
 	active: bool,
 }
@@ -27,6 +29,7 @@ impl Default for Pivot {
 			transform_from_normalized: Default::default(),
 			pivot: Default::default(),
 			old_pivot_position: ReferencePoint::Center,
+			last_non_none_reference: ReferencePoint::Center,
 			active: true,
 		}
 	}

editor/src/messages/tool/tool_messages/select_tool.rs

@@ -1076,6 +1076,8 @@ impl Fsm for SelectToolFsmState {
 					let extend = input.keyboard.key(extend_selection);
 					if !extend && !input.keyboard.key(remove_from_selection) {
 						responses.add(DocumentMessage::DeselectAllLayers);
+						let position = tool_data.pivot.last_non_none_reference;
+						responses.add(SelectToolMessage::SetPivot { position });
 						tool_data.layers_dragging.clear();
 					}
 
@@ -1571,6 +1573,7 @@ impl Fsm for SelectToolFsmState {
 			(_, SelectToolMessage::SetPivot { position }) => {
 				responses.add(DocumentMessage::StartTransaction);
 
+				tool_data.pivot.last_non_none_reference = position;
 				let pos: Option<DVec2> = position.into();
 				tool_data.pivot.set_normalized_position(pos.unwrap());
 				let dot = tool_data.get_as_dot();