1HEADmaster

1 files changed, 127 insertions, 0 deletions

diff --git a/hammer/ToolPickEntity.h b/hammer/ToolPickEntity.h
new file mode 100644
index 0000000..59c63c3
--- /dev/null
+++ b/hammer/ToolPickEntity.h
@@ -0,0 +1,127 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Tool used for picking entities for filling out entity properties and
+//			I/O connections. Anywhere you see an entity name field there should
+//			be an eyedropper button next to it for picking the entity.
+//
+//			TODO: make the entity field / picker button a single control?
+//			TODO: combine the face picker with the entity picker?
+//
+//=============================================================================//
+
+#ifndef TOOLPICKENTITY_H
+#define TOOLPICKENTITY_H
+#ifdef _WIN32
+#pragma once
+#endif
+
+#include "MapEntity.h"
+#include "ToolInterface.h"
+
+
+class CMapView3D;
+class CMapViewLogical;
+class CToolPickEntity;
+
+
+//
+// Selection states for entries in our list of selected faces.
+//
+enum EntityState_t
+{
+	EntityState_Select = 0,			// 
+	EntityState_Partial,			// Used for multiselect; the face is in at least one of the face lists being edited.
+	EntityState_None,				// Used for multiselect; to deselect partially selected faces. Otherwise they are removed from the list.
+};
+
+
+//
+// An entry in our list of selected entities.
+//
+struct SelectedEntity_t
+{
+	CMapEntity *pEntity;				// Pointer to the entity.
+	EntityState_t eState;				// The current selection state of this entity.
+	EntityState_t eOriginalState;		// The original selection state of this entity.
+};
+
+
+//
+// Interface for notification by the entity picking tool. Inherit from this if you
+// are a client of the entity picker.
+//
+class IPickEntityTarget
+{
+public:
+	virtual void OnNotifyPickEntity(CToolPickEntity *pTool) = 0;
+};
+
+
+class CToolPickEntity : public CBaseTool
+{
+public:
+
+	//
+	// Constructor/Destructor
+	//
+    CToolPickEntity();
+    ~CToolPickEntity();
+
+	//
+	// CBaseTool virtual implementations
+	//
+	virtual void OnDeactivate();
+	virtual ToolID_t GetToolID(void) { return TOOL_PICK_ENTITY; }
+
+	virtual bool OnLMouseUp3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+    virtual bool OnLMouseDown3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+    virtual bool OnLMouseDblClk3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+	virtual bool OnRMouseUp3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+    virtual bool OnRMouseDown3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+	virtual bool OnMouseMove3D(CMapView3D *pView, UINT nFlags, const Vector2D &vPoint);
+
+	virtual bool OnLMouseUpLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint) { return true; }
+	virtual bool OnLMouseDownLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint);
+    virtual bool OnLMouseDblClkLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint) { return true; }
+	virtual bool OnRMouseUpLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint) { return true; }
+    virtual bool OnRMouseDownLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint) { return true; }
+	virtual bool OnMouseMoveLogical(CMapViewLogical *pView, UINT nFlags, const Vector2D &vPoint);
+
+	//
+	// Functions specific to this tool.
+	//
+	inline void Attach(IPickEntityTarget *pTarget);
+	void AllowMultiSelect(bool bAllow);
+	void GetSelectedEntities(CMapEntityList &EntityListFull, CMapEntityList &EntityListPartial);
+	void SetSelectedEntities(CMapEntityList &EntityListFull, CMapEntityList &EntityListPartial);
+
+protected:
+
+	void CycleSelectEntity(CMapEntity *pEntity);
+	void DeselectAll(void);
+	void DeselectEntity(int nIndex);
+	void DeselectEntity(CMapEntity *pEntity);
+	int FindEntity(CMapEntity *pEntity);
+	void SelectEntity(CMapEntity *pEntity);
+
+	void AddToList(CMapEntity *pEntity, EntityState_t eState);
+	void RemoveFromList(int nIndex);
+
+	void SetEyedropperCursor(void);
+
+	IPickEntityTarget *m_pNotifyTarget;			// Object to notify when selection events occur.
+	bool m_bAllowMultiSelect;					// If false, only one entity can be selected at a time.
+	CUtlVector <SelectedEntity_t> m_Entities;	// Picked entities and their selection state (partial or full).
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Attaches the given notification target to this tool. That object
+//			will be used for all future notifications and updates by the tool.
+//-----------------------------------------------------------------------------
+void CToolPickEntity::Attach(IPickEntityTarget *pNotifyTarget)
+{
+	m_pNotifyTarget = pNotifyTarget;
+}
+
+#endif // TOOLPICKENTITY_H