xref: /external/lldb/include/lldb/Interpreter/PythonDataObjects.h

//===-- PythonDataObjects.h----------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef liblldb_PythonDataObjects_h_
#define liblldb_PythonDataObjects_h_

// C Includes
// C++ Includes

// Other libraries and framework includes
// Project includes
#include "lldb/lldb-defines.h"
#include "lldb/Core/ConstString.h"
#include "lldb/Core/Flags.h"
#include "lldb/Interpreter/OptionValue.h"

namespace lldb_private {

    class PythonRefCountedObject
    {
    public:
        PythonRefCountedObject (PyObject* obj) : m_object(obj)
        {
            Py_XINCREF(m_object);
        }

        ~PythonRefCountedObject ()
        {
            Py_XDECREF(m_object);
        }

        void
        Reset (PyObject* object = NULL)
        {
            Py_XDECREF(m_object);
            m_object = object;
            Py_XINCREF(m_object);
        }

        PyObject*
        GetPyhonObject ()
        {
            return m_object;
        }

        operator bool ()
        {
            return m_object != NULL;
        }

    private:
        PyObject* m_object;
    };

    class PythonDataString
    {
    public:

        PythonDataString (PyObject* object);
        PythonDataString (const char* string);
        ~PythonDataString ();

        const char*
        GetString();

        void
        SetString (const char* string);

        operator bool ()
        {
            return m_object.operator bool();
        }

        PyObject*
        GetPythonObject() { return m_object.GetPyhonObject(); }
    private:
        PythonRefCountedObject m_object;
    };

    class PythonDataInteger
    {
    public:

        PythonDataInteger (PyObject* object);
        PythonDataInteger (int64_t value);
        ~PythonDataInteger ();

        int64_t
        GetInteger();

        void
        SetInteger (int64_t value);

        operator bool ()
        {
            return m_object.operator bool();
        }

        PyObject*
        GetPythonObject() { return m_object.GetPyhonObject(); }
    private:
        PythonRefCountedObject m_object;
    };

    class PythonDataArray
    {
    public:

        PythonDataArray (uint32_t count);
        PythonDataArray (PyObject* object);
        ~PythonDataArray ();

        uint32_t
        GetSize();

        PythonDataObject*
        GetItemAtIndex (uint32_t index);

        void
        SetItemAtIndex (uint32_t index, PythonDataObject* object);

        void
        AppendItem (PythonDataObject* object);

        operator bool ()
        {
            return m_object.operator bool();
        }

        PyObject*
        GetPythonObject() { return m_object.GetPyhonObject(); }
    private:
        PythonRefCountedObject m_object;
    };

    class PythonDataDictionary
    {
    public:

        PythonDataDictionary ();
        PythonDataDictionary (PyObject* object);
        ~PythonDataDictionary ();

        uint32_t GetSize();

        PythonDataObject*
        GetItemForKey (PythonDataString* key);

        typedef bool (*DictionaryIteratorCallback)(PythonDataString* key, PythonDataDictionary* dict);

        PythonDataArray*
        GetKeys ();

        PythonDataString*
        GetKeyAtPosition (uint32_t pos);

        PythonDataObject*
        GetValueAtPosition (uint32_t pos);

        void
        SetItemForKey (PythonDataString* key, PythonDataObject* value);

        operator bool ()
        {
            return m_object.operator bool();
        }

        PyObject*
        GetPythonObject() { return m_object.GetPyhonObject(); }
    private:
        PythonRefCountedObject m_object;
    };

    class PythonDataObject
    {
    public:

        PythonDataObject (PyObject* object);

        ~PythonDataObject ();

        PythonDataString*
        GetStringObject ();

        PythonDataInteger*
        GetIntegerObject ();

        PythonDataArray*
        GetArrayObject();

        PythonDataDictionary*
        GetDictionaryObject();

        operator bool ()
        {
            return m_object.operator bool();
        }

        PyObject*
        GetPythonObject() { return m_object.GetPyhonObject(); }

    private:
        PythonRefCountedObject m_object;
    };

} // namespace lldb_private

#endif  // liblldb_PythonDataObjects_h_