text: /*
     jep - Java Embedded Python
  
     Copyright (c) 2015-2019 JEP AUTHORS.
  
     This file is licensed under the the zlib/libpng License.
  
     This software is provided 'as-is', without any express or implied
     warranty. In no event will the authors be held liable for any
     damages arising from the use of this software.
  
     Permission is granted to anyone to use this software for any
     purpose, including commercial applications, and to alter it and
     redistribute it freely, subject to the following restrictions:
  
     1. The origin of this software must not be misrepresented; you
     must not claim that you wrote the original software. If you use
     this software in a product, an acknowledgment in the product
     documentation would be appreciated but is not required.
  
     2. Altered source versions must be plainly marked as such, and
     must not be misrepresented as being the original software.
  
     3. This notice may not be removed or altered from any source
     distribution.
  */
  
  #include "Jep.h"
  
  /*
   * Gets the size of the map.
   */
  static Py_ssize_t pyjmap_len(PyObject *self)
  {
      Py_ssize_t    len   = 0;
      PyJObject    *pyjob = (PyJObject*) self;
      JNIEnv       *env   = pyembed_get_env();
  
      len = java_util_Map_size(env, pyjob->object);
      if (process_java_exception(env)) {
          return -1;
      }
      return len;
  }
  
  
  /*
   * Method for checking if a key is in the dictionary.  For example,
   * if key in o:
   */
  static int pyjmap_contains_key(PyObject *self, PyObject *key)
  {
      jboolean     jresult      = JNI_FALSE;
      PyJObject    *obj         = (PyJObject*) self;
      JNIEnv       *env         = pyembed_get_env();
      jobject       jkey        = NULL;
      int           result   = -1;
  
      if ((*env)->PushLocalFrame(env, JLOCAL_REFS) != 0) {
          process_java_exception(env);
          return -1;
      }
      jkey = PyObject_As_jobject(env, key, JOBJECT_TYPE);
      if (!jkey && PyErr_Occurred()) {
          goto FINALLY;
      }
  
      jresult = java_util_Map_containsKey(env, obj->object, jkey);
      if (process_java_exception(env)) {
          goto FINALLY;
      }
  
  
      if (jresult) {
          result = 1;
      } else {
          result = 0;
      }
  FINALLY:
      (*env)->PopLocalFrame(env, NULL);
      return result;
  }
  
  
  /*
   * Method for the getting items with the [key] operator on pyjmap.  For
   * example, result = o[key]
   */
  static PyObject* pyjmap_getitem(PyObject *o, PyObject *key)
  {
      jobject       jkey   = NULL;
      jobject       val    = NULL;
      PyJObject    *obj    = (PyJObject*) o;
      JNIEnv       *env    = pyembed_get_env();
      PyObject     *result = NULL;
  
      if ((*env)->PushLocalFrame(env, JLOCAL_REFS) != 0) {
          process_java_exception(env);
          return NULL;
      }
      jkey = PyObject_As_jobject(env, key, JOBJECT_TYPE);
      if (!jkey && PyErr_Occurred()) {
          goto FINALLY;
      }
  
      val = java_util_Map_get(env, obj->object, jkey);
      if (process_java_exception(env)) {
          goto FINALLY;
      }
  
      if (!val) {
          /*
           * Python docs indicate KeyError should be set if the key is not in the
           * container, but some Maps allow null values. So we have to check.
           */
          if (!pyjmap_contains_key(o, key)) {
              PyObject *pystr = PyObject_Str(key);
              PyErr_Format(PyExc_KeyError,
                           "KeyError: %s",
                           PyString_AsString(pystr));
              Py_XDECREF(pystr);
              goto FINALLY;
          }
      }
  
      result = jobject_As_PyObject(env, val);
  FINALLY:
      (*env)->PopLocalFrame(env, NULL);
      return result;
  }
  
  /*
   * Method for the setting items with the [key] operator on pyjmap.  For example,
   * o[key] = v.  Also supports del o[key]
   */
  static int pyjmap_setitem(PyObject *o, PyObject *key, PyObject *v)
  {
      jobject       jkey     = NULL;
      jobject       value    = NULL;
      PyJObject    *obj      = (PyJObject*) o;
      JNIEnv       *env      = pyembed_get_env();
      int           result   = -1;
  
      if ((*env)->PushLocalFrame(env, JLOCAL_REFS) != 0) {
          process_java_exception(env);
          return -1;
      }
  
      if (v == NULL) {
          // this is a del PyJMap[key] statement
          if (!pyjmap_contains_key(o, key)) {
              PyObject *pystr = PyObject_Str(key);
              PyErr_Format(PyExc_KeyError,
                           "KeyError: %s",
                           PyString_AsString(pystr));
              Py_XDECREF(pystr);
              goto FINALLY;
          }
  
          jkey = PyObject_As_jobject(env, key, JOBJECT_TYPE);
          if (!jkey && PyErr_Occurred()) {
              goto FINALLY;
          }
  
          java_util_Map_remove(env, obj->object, jkey);
          if (process_java_exception(env)) {
              goto FINALLY;
          }
      } else {
          value = PyObject_As_jobject(env, v, JOBJECT_TYPE);
          if (!value && PyErr_Occurred()) {
              goto FINALLY;
          }
  
          jkey = PyObject_As_jobject(env, key, JOBJECT_TYPE);
          if (!jkey && PyErr_Occurred()) {
              return -1;
          }
  
          java_util_Map_put(env, obj->object, jkey, value);
          if (process_java_exception(env)) {
              goto FINALLY;
          }
      }
      // have to return 0 on success even though it's not documented
      result = 0;
  FINALLY:
      (*env)->PopLocalFrame(env, NULL);
      return result;
  }
  
  
  /*
   * Method for iterating over the keys of the dictionary.  For example,
   * for key in o:
   */
  static PyObject* pyjmap_getiter(PyObject* obj)
  {
      jobject       set      = NULL;
      jobject       iter     = NULL;
      PyJObject    *pyjob    = (PyJObject*) obj;
      PyObject     *result   = NULL;
      JNIEnv       *env      = pyembed_get_env();
  
      if ((*env)->PushLocalFrame(env, JLOCAL_REFS) != 0) {
          process_java_exception(env);
          return NULL;
      }
      set = java_util_Map_keySet(env, pyjob->object);
      if (process_java_exception(env) || !set) {
          goto FINALLY;
      }
  
  
      iter = java_lang_Iterable_iterator(env, set);
      if (process_java_exception(env) || !iter) {
          goto FINALLY;
      }
  
      result = PyJIterator_Wrap(env, iter, NULL);
  FINALLY:
      (*env)->PopLocalFrame(env, NULL);
      return result;
  }
  
  
  static PySequenceMethods pyjmap_seq_methods = {
      0,                          /* sq_length */
      0,                          /* sq_concat */
      0,                          /* sq_repeat */
      0,                          /* sq_item */
      0,                          /* sq_slice */
      0,                          /* sq_ass_item */
      0,                          /* sq_ass_slice */
      pyjmap_contains_key,        /* sq_contains */
      0,                          /* sq_inplace_concat */
      0,                          /* sq_inplace_repeat */
  };
  
  static PyMappingMethods pyjmap_map_methods = {
      pyjmap_len,           /* mp_length */
      pyjmap_getitem,       /* mp_subscript */
      pyjmap_setitem        /* mp_ass_subscript */
  };
  
  
  /*
   * Inherits from PyJObject_Type
   */
  PyTypeObject PyJMap_Type = {
      PyVarObject_HEAD_INIT(NULL, 0)
      "jep.PyJMap",
      sizeof(PyJObject),
      0,
      0,                                        /* tp_dealloc */
      0,                                        /* tp_print */
      0,                                        /* tp_getattr */
      0,                                        /* tp_setattr */
      0,                                        /* tp_compare */
      0,                                        /* tp_repr */
      0,                                        /* tp_as_number */
      &pyjmap_seq_methods,                      /* tp_as_sequence */
      &pyjmap_map_methods,                      /* tp_as_mapping */
      0,                                        /* tp_hash  */
      0,                                        /* tp_call */
      0,                                        /* tp_str */
      0,                                        /* tp_getattro */
      0,                                        /* tp_setattro */
      0,                                        /* tp_as_buffer */
      Py_TPFLAGS_DEFAULT,                       /* tp_flags */
      "jmap",                                   /* tp_doc */
      0,                                        /* tp_traverse */
      0,                                        /* tp_clear */
      0,                                        /* tp_richcompare */
      0,                                        /* tp_weaklistoffset */
      (getiterfunc) pyjmap_getiter,             /* tp_iter */
      0,                                        /* tp_iternext */
      0,                                        /* tp_methods */
      0,                                        /* tp_members */
      0,                                        /* tp_getset */
      0, // &PyJObject_Type                     /* tp_base */
      0,                                        /* tp_dict */
      0,                                        /* tp_descr_get */
      0,                                        /* tp_descr_set */
      0,                                        /* tp_dictoffset */
      0,                                        /* tp_init */
      0,                                        /* tp_alloc */
      NULL,                                     /* tp_new */
  };
  