Imports Microsoft.DirectX
Imports Microsoft.DirectX.DirectInput

Public Class CKeyboard

Private ref_form As Windows.Forms.Form
Private p_keyboard As DirectInput.Device
Private p_keystate As DirectInput.KeyboardState


Public Sub New(ByRef frm As Windows.Forms.Form)
ref_form = frm
End Sub

Public Function Init() As Boolean
Try
'create new keyboard device
p_keyboard = New DirectInput.Device(DirectInput.SystemGuid.Keyboard)

'set cooperative level for keyboard
Dim cooplevel As CooperativeLevelFlags = _
CooperativeLevelFlags.Background Or _
CooperativeLevelFlags.NonExclusive

p_keyboard.SetCooperativeLevel(ref_form, cooplevel)

'try to acquire the keyboard
p_keyboard.Acquire()

'initialize key state
Poll()

Return True

Catch ex As Exception
Return False
End Try
End Function

Public Sub Poll()
p_keystate = p_keyboard.GetCurrentKeyboardState()
End Sub

Public Function KeyState(ByVal key As Microsoft.DirectX.DirectInput.Key) As Boolean
Return p_keystate(key)
End Function

Protected Overrides Sub Finalize()
MyBase.Finalize()
p_keyboard.Unacquire()
p_keyboard.Dispose()
p_keyboard = Nothing
End Sub
End Class

Imports Microsoft.DirectX
Imports Microsoft.DirectX.DirectInput

Public Class CMouse

Private p_mouse As DirectInput.Device
Private p_mousestate As DirectInput.MouseState
Private buttons() As Byte

Public Sub New(ByRef frm As Windows.Forms.Form)
Try
'create new mouse device
p_mouse = New DirectInput.Device(SystemGuid.Mouse)

'set cooperative level for mouse
Dim cooplevel As CooperativeLevelFlags = _
CooperativeLevelFlags.Background Or _
CooperativeLevelFlags.NonExclusive
p_mouse.SetCooperativeLevel(frm, cooplevel)

'try to acquire the mouse
p_mouse.Acquire()

Catch ex As Exception
MessageBox.Show("Error initializing mouse", "DirectInput Error")
End Try

End Sub

Public Sub Poll()
p_mousestate = p_mouse.CurrentMouseState
buttons = p_mousestate.GetMouseButtons()
End Sub

Public Function Button() As Integer
If (buttons(0) > 0) Then
Return 1
ElseIf (buttons(1) > 0) Then
Return 2
ElseIf (buttons(2) > 0) Then
Return 3
Else
Return 0
End If
End Function

Public Function X() As Integer
Return p_mousestate.X()
End Function

Public Function Y() As Integer
Return p_mousestate.Y()
End Function

Public Function Wheel() As Integer
Return p_mousestate.Z
End Function

Public Sub Dispose()
p_mouse.Unacquire()
p_mouse.Dispose()
End Sub

End Class