rocm-systems/rocclr/runtime/device/rocm/rocglinterop.cpp

//
// Copyright (c) 2016 Advanced Micro Devices, Inc. All rights reserved.
//

#ifndef WITHOUT_HSA_BACKEND

#include "os/os.hpp"
#include "utils/debug.hpp"
#include "utils/flags.hpp"
#include "device/rocm/rocglinterop.hpp"

#if !defined(_WIN32)
#include <dlfcn.h>
#endif

namespace roc
{

#if !defined(_WIN32)
static PFNMESAGLINTEROPGLXQUERYDEVICEINFOPROC GlxInfo   = nullptr;
static PFNMESAGLINTEROPGLXEXPORTOBJECTPROC    GlxExport = nullptr;
static PFNMESAGLINTEROPEGLQUERYDEVICEINFOPROC EglInfo   = nullptr;
static PFNMESAGLINTEROPEGLEXPORTOBJECTPROC    EglExport = nullptr;
#endif

std::atomic<uint32_t> MesaInterop::refCount(0);

bool MesaInterop::Supported()
{
#ifdef _WIN32
  return false;
#else
  return true;
#endif
}

//Attempt to locate Mesa interop APIs.  Return which of glx/egl are supported.
bool MesaInterop::Bind(MESA_INTEROP_KIND Kind, const DisplayHandle& Display, const ContextHandle& Context)
{
#if defined(_WIN32)
  return false;
#else
  if(Kind==MESA_INTEROP_NONE)
    return false;

  if(kind!=MESA_INTEROP_NONE)
  {
    LogError("Error - MesaInterop Bind while already bound.");
    return false;
  }

  void* glxinfo=dlsym(RTLD_DEFAULT, "MesaGLInteropGLXQueryDeviceInfo");
  void* eglinfo=dlsym(RTLD_DEFAULT, "MesaGLInteropEGLQueryDeviceInfo");
  
  if(((glxinfo!=GlxInfo) || (eglinfo!=EglInfo)) && (refCount!=0))
    LogWarning("Warning - Mesa changed while holding interop contexts.");

  GlxInfo=(PFNMESAGLINTEROPGLXQUERYDEVICEINFOPROC)glxinfo;
  EglInfo=(PFNMESAGLINTEROPEGLQUERYDEVICEINFOPROC)eglinfo;

  GlxExport=(PFNMESAGLINTEROPGLXEXPORTOBJECTPROC)dlsym(RTLD_DEFAULT, "MesaGLInteropGLXExportObject");
  EglExport=(PFNMESAGLINTEROPEGLEXPORTOBJECTPROC)dlsym(RTLD_DEFAULT, "MesaGLInteropEGLExportObject");

  uint32_t ret=MESA_INTEROP_NONE;
  if(GlxInfo && GlxExport)
    ret|=MESA_INTEROP_GLX;
  if(EglInfo && EglExport)
    ret|=MESA_INTEROP_EGL;

  kind = MESA_INTEROP_KIND(ret & Kind);
  display=Display;
  context=Context;

  if(kind!=MESA_INTEROP_NONE)
  {
    refCount++;
    return true;
  }
  return false;

#endif
}

bool MesaInterop::GetInfo(mesa_glinterop_device_info& info) const
{
#ifdef _WIN32
  return false;
#else
  switch(kind)
  {
  case MESA_INTEROP_GLX:
    return GlxInfo(display.glxDisplay, context.glxContext, &info)==MESA_GLINTEROP_SUCCESS;
  case MESA_INTEROP_EGL:
    return EglInfo(display.eglDisplay, context.eglContext, &info)==MESA_GLINTEROP_SUCCESS;
  default:
    return false;
  }
#endif
}

bool MesaInterop::Export (mesa_glinterop_export_in& in, mesa_glinterop_export_out& out) const
{
#ifdef _WIN32
  return false;
#else
  switch(kind)
  {
  case MESA_INTEROP_GLX:
    return GlxExport(display.glxDisplay, context.glxContext, &in, &out)==MESA_GLINTEROP_SUCCESS;
  case MESA_INTEROP_EGL:
    return EglExport(display.eglDisplay, context.eglContext, &in, &out)==MESA_GLINTEROP_SUCCESS;
  default:
    return false;
  }
#endif
}

}

#endif  // WITHOUT_HSA_BACKEND