[brotlimodule] add enable_context_modeling parameter (defaults to True)

1 files changed, 7 insertions, 3 deletions

diff --git a/python/brotlimodule.cc b/python/brotlimodule.cc
index 79e48c6..67c63ac 100644
--- a/python/brotlimodule.cc
+++ b/python/brotlimodule.cc
@@ -66,20 +66,22 @@ static PyObject* brotli_compress(PyObject *self, PyObject *args, PyObject *keywd
   PyObject* enable_dictionary = NULL;
   PyObject* enable_transforms = NULL;
   PyObject* greedy_block_split = NULL;
+  PyObject* enable_context_modeling = NULL;
   uint8_t *input, *output;
   size_t length, output_length;
   BrotliParams::Mode mode = (BrotliParams::Mode) -1;
   int ok;
 
   static char *kwlist[] = {"data", "mode", "enable_dictionary", "enable_transforms",
-                           "greedy_block_split", NULL};
+                           "greedy_block_split", "enable_context_modeling", NULL};
 
-  ok = PyArg_ParseTupleAndKeywords(args, keywds, "s#|O&O!O!O!:compress", kwlist,
+  ok = PyArg_ParseTupleAndKeywords(args, keywds, "s#|O&O!O!O!O!:compress", kwlist,
                         &input, &length,
                         &mode_convertor, &mode,
                         &PyBool_Type, &enable_dictionary,
                         &PyBool_Type, &enable_transforms,
-                        &PyBool_Type, &greedy_block_split);
+                        &PyBool_Type, &greedy_block_split,
+                        &PyBool_Type, &enable_context_modeling);
 
   if (!ok)
     return NULL;
@@ -96,6 +98,8 @@ static PyObject* brotli_compress(PyObject *self, PyObject *args, PyObject *keywd
     params.enable_transforms = PyObject_IsTrue(enable_transforms);
   if (greedy_block_split)
     params.greedy_block_split = PyObject_IsTrue(greedy_block_split);
+  if (enable_context_modeling)
+    params.enable_context_modeling = PyObject_IsTrue(enable_context_modeling);
 
   ok = BrotliCompressBuffer(params, length, input,
                             &output_length, output);