/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
* vim: set ts=2 sw=2 et tw=78:
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

#ifndef InkCollector_h__
#define InkCollector_h__

#include <msinkaut.h>
#include "mozilla/StaticPtr.h"

#define MOZ_WM_PEN_LEAVES_HOVER_OF_DIGITIZER  WM_USER + 0x83

class InkCollectorEvent final : public _IInkCollectorEvents
{
public:
  // IUnknown
  HRESULT __stdcall QueryInterface(REFIID aRiid, void **aObject);
  virtual ULONG STDMETHODCALLTYPE AddRef() { return ++mRefCount; }
  virtual ULONG STDMETHODCALLTYPE Release()
  {
    MOZ_ASSERT(mRefCount);
    if (!--mRefCount) {
      delete this;
      return 0;
    }
    return mRefCount;
  }

protected:
  // IDispatch
  STDMETHOD(GetTypeInfoCount)(UINT* aInfo) { return E_NOTIMPL; }
  STDMETHOD(GetTypeInfo)(UINT aInfo, LCID aId, ITypeInfo** aTInfo) { return E_NOTIMPL; }
  STDMETHOD(GetIDsOfNames)(REFIID aRiid, LPOLESTR* aStrNames, UINT aNames,
                           LCID aId, DISPID* aDispId) { return E_NOTIMPL; }
  STDMETHOD(Invoke)(DISPID aDispIdMember, REFIID aRiid,
                    LCID aId, WORD wFlags,
                    DISPPARAMS* aDispParams, VARIANT* aVarResult,
                    EXCEPINFO* aExcepInfo, UINT* aArgErr);

  // InkCollectorEvent
  void CursorOutOfRange(IInkCursor* aCursor) const;
  bool IsHardProximityTablet(IInkTablet* aTablet) const;

private:
  uint32_t  mRefCount = 0;

  ~InkCollectorEvent() = default;
};

class InkCollector
{
public:
  ~InkCollector();
  void Shutdown();

  HWND GetTarget();
  void SetTarget(HWND aTargetWindow);
  void ClearTarget();

  uint16_t GetPointerId(); // 0 shows that there is no existing pen.
  void SetPointerId(uint16_t aPointerId);
  void ClearPointerId();

  static StaticAutoPtr<InkCollector> sInkCollector;

protected:
  void Initialize();
  void OnInitialize();
  void Enable(bool aNewState);

private:
  RefPtr<IUnknown>          mMarshaller;
  RefPtr<IInkCollector>     mInkCollector;
  RefPtr<IConnectionPoint>  mConnectionPoint;
  RefPtr<InkCollectorEvent> mInkCollectorEvent;

  HWND                        mTargetWindow     = 0;
  DWORD                       mCookie           = 0;
  bool                        mComInitialized   = false;
  bool                        mEnabled          = false;

  // This value holds the previous pointerId of the pen, and is used by the
  // nsWindow when processing a MOZ_WM_PEN_LEAVES_HOVER_OF_DIGITIZER which
  // indicates that a pen leaves the digitizer.

  // TODO: If we move our implementation to window pointer input messages, then
  // we no longer need this value, since the pointerId can be retrieved from the
  // window message, please refer to
  // https://msdn.microsoft.com/en-us/library/windows/desktop/hh454916(v=vs.85).aspx

  // NOTE: The pointerId of a pen shouldn't be 0 on a Windows platform, since 0
  // is reserved of the mouse, please refer to
  // https://msdn.microsoft.com/en-us/library/windows/desktop/ms703320(v=vs.85).aspx
  uint16_t mPointerId = 0;
};

#endif // InkCollector_h__
