/**

* Copyright 2015 Google Inc. All Rights Reserved.

*

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

// Ensure that Python.h is included before any other header.

#include "common.h"

#include "python_callback.h"

namespace devtools {

namespace cdbg {

PyTypeObject PythonCallback::python_type_ =

DefaultTypeDefinition(CDBG_SCOPED_NAME("_Callback"));

PyMethodDef PythonCallback::callback_method_def_ = {

const_cast<char*>("Callback"), // ml_name

reinterpret_cast<PyCFunction>(PythonCallback::Run), // ml_meth

METH_NOARGS, // ml_flags

const_cast<char*>("") // ml_doc

};

ScopedPyObject PythonCallback::Wrap(std::function<void()> callback) {

ScopedPyObject callback_obj = NewNativePythonObject<PythonCallback>();

py_object_cast<PythonCallback>(callback_obj.get())->callback_ = callback;

ScopedPyObject callback_method(PyCFunction_NewEx(

&callback_method_def_,

callback_obj.get(),

GetDebugletModule()));

return callback_method;

}

void PythonCallback::Disable(PyObject* method) {

DCHECK(PyCFunction_Check(method));

auto instance = py_object_cast<PythonCallback>(PyCFunction_GET_SELF(method));

DCHECK(instance);

instance->callback_ = nullptr;

}

PyObject* PythonCallback::Run(PyObject* self) {

auto instance = py_object_cast<PythonCallback>(self);

if (instance->callback_ != nullptr) {

instance->callback_();

}

Py_RETURN_NONE;

}

} // namespace cdbg

} // namespace devtools