IFACEMETHODIMP CTriColorFragmentProvider::Navigate(_In_ NavigateDirection direction, _Outptr_result_maybenull_ IRawElementProviderFragment ** retVal)
{
*retVal = nullptr;
ComPtr<IInspectable> spFragment;
HRESULT hr = CheckDisconnected();
if (SUCCEEDED(hr))
{
if (direction == NavigateDirection_Parent)
{
hr = m_control->GetTriColorControlProvider(&spFragment);
}
else if (direction == NavigateDirection_NextSibling)
{
if (!TriColorValueHelper::IsLast(m_value))
{
hr = m_control->GetTriColorFragmentProvider(TriColorValueHelper::NextValue(m_value), &spFragment);
}
}
else if (direction == NavigateDirection_PreviousSibling)
{
if (!TriColorValueHelper::IsFirst(m_value))
{
hr = m_control->GetTriColorFragmentProvider(TriColorValueHelper::PreviousValue(m_value), &spFragment);
}
}
}
if (SUCCEEDED(hr) && spFragment != nullptr)
{
hr = spFragment.Get()->QueryInterface(IID_PPV_ARGS(retVal));
}
// For the other directions (first child, last child) the default of nullptr is correct
return hr;
}
IFACEMETHODIMP CTriColorControlProvider::Navigate(_In_ NavigateDirection direction, _Outptr_result_maybenull_ IRawElementProviderFragment ** retVal)
{
*retVal = nullptr;
ComPtr<IInspectable> spFragment;
HRESULT hr = CheckDisconnected();
if (SUCCEEDED(hr))
{
if (direction == NavigateDirection_Parent)
{
hr = m_control->GetAppWindowProvider(&spFragment);
}
else if (direction == NavigateDirection_FirstChild)
{
hr = m_control->GetTriColorFragmentProvider(TriColorValue::Red, &spFragment);
}
else if (direction == NavigateDirection_LastChild)
{
hr = m_control->GetTriColorFragmentProvider(TriColorValue::Green, &spFragment);
}
}
if (SUCCEEDED(hr) && spFragment != nullptr)
{
hr = spFragment.Get()->QueryInterface(IID_PPV_ARGS(retVal));
}
// For the other directions (next, previous) the default of nullptr is correct
return hr;
}
IFACEMETHODIMP CTriColorControlProvider::GetSelection(_Outptr_result_maybenull_ SAFEARRAY * *retVal)
{
*retVal = nullptr;
HRESULT hr = CheckDisconnected();
if (SUCCEEDED(hr))
{
TriColorValue controlValue;
hr = m_control->get_ControlValue(&controlValue);
if (SUCCEEDED(hr))
{
ComPtr<IInspectable> spFragmentProvider;
hr = m_control->GetTriColorFragmentProvider(controlValue, &spFragmentProvider);
if (SUCCEEDED(hr))
{
*retVal = SafeArrayCreateVector(VT_UNKNOWN, 0, 1);
if (*retVal != nullptr)
{
long index = 0;
hr = SafeArrayPutElement(*retVal, &index, spFragmentProvider.Get());
if (FAILED(hr))
{
SafeArrayDestroy(*retVal);
*retVal = nullptr;
}
}
else
{
hr = E_OUTOFMEMORY;
}
}
}
}
return hr;
}