python string conversion

- use _PyUnicode_AsStringAndSize where possible
- use %R for PyErr_Format(...) rather then running repr on the object explicitly 
- use const char
This commit is contained in:
2011-11-03 14:09:18 +00:00
parent d210703bca
commit 665f602f15
14 changed files with 55 additions and 46 deletions

View File

@@ -230,7 +230,7 @@ void IMB_moviecache_put(MovieCache *cache, void *userkey, ImBuf *ibuf)
key= BLI_mempool_alloc(cache->keys_pool); key= BLI_mempool_alloc(cache->keys_pool);
key->cache_owner= cache; key->cache_owner= cache;
key->userkey= BLI_mempool_alloc(cache->userkeys_pool);; key->userkey= BLI_mempool_alloc(cache->userkeys_pool);
memcpy(key->userkey, userkey, cache->keysize); memcpy(key->userkey, userkey, cache->keysize);
item= BLI_mempool_alloc(cache->items_pool); item= BLI_mempool_alloc(cache->items_pool);

View File

@@ -195,19 +195,22 @@ static PyObject *BPy_IDGroup_GetName(BPy_IDProperty *self, void *UNUSED(closure)
static int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value, void *UNUSED(closure)) static int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value, void *UNUSED(closure))
{ {
char *st; const char *name;
Py_ssize_t name_size;
if (!PyUnicode_Check(value)) { if (!PyUnicode_Check(value)) {
PyErr_SetString(PyExc_TypeError, "expected a string!"); PyErr_SetString(PyExc_TypeError, "expected a string!");
return -1; return -1;
} }
st = _PyUnicode_AsString(value); name = _PyUnicode_AsStringAndSize(value, &name_size);
if (BLI_strnlen(st, MAX_IDPROP_NAME) == MAX_IDPROP_NAME) {
if (name_size > MAX_IDPROP_NAME) {
PyErr_SetString(PyExc_TypeError, "string length cannot exceed 31 characters!"); PyErr_SetString(PyExc_TypeError, "string length cannot exceed 31 characters!");
return -1; return -1;
} }
BLI_strncpy(self->prop->name, st, sizeof(self->prop->name)); memcpy(self->prop->name, name, name_size);
return 0; return 0;
} }
@@ -236,7 +239,7 @@ static Py_ssize_t BPy_IDGroup_Map_Len(BPy_IDProperty *self)
static PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item) static PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item)
{ {
IDProperty *idprop; IDProperty *idprop;
char *name; const char *name;
if (self->prop->type != IDP_GROUP) { if (self->prop->type != IDP_GROUP) {
PyErr_SetString(PyExc_TypeError, "unsubscriptable object"); PyErr_SetString(PyExc_TypeError, "unsubscriptable object");
@@ -301,14 +304,22 @@ static int idp_sequence_type(PyObject *seq)
return type; return type;
} }
/* note: group can be a pointer array or a group */ /* note: group can be a pointer array or a group.
const char *BPy_IDProperty_Map_ValidateAndCreate(const char *name, IDProperty *group, PyObject *ob) * assume we already checked key is a string. */
const char *BPy_IDProperty_Map_ValidateAndCreate(PyObject *name_obj, IDProperty *group, PyObject *ob)
{ {
IDProperty *prop = NULL; IDProperty *prop = NULL;
IDPropertyTemplate val = {0}; IDPropertyTemplate val = {0};
if (strlen(name) >= sizeof(group->name)) const char *name= "";
return "the length of IDProperty names is limited to 31 characters";
if (name_obj) {
Py_ssize_t name_size;
name = _PyUnicode_AsStringAndSize(name_obj, &name_size);
if (name_size > MAX_IDPROP_NAME) {
return "the length of IDProperty names is limited to 31 characters";
}
}
if (PyFloat_Check(ob)) { if (PyFloat_Check(ob)) {
val.d = PyFloat_AsDouble(ob); val.d = PyFloat_AsDouble(ob);
@@ -364,7 +375,7 @@ const char *BPy_IDProperty_Map_ValidateAndCreate(const char *name, IDProperty *g
for (i=0; i<val.array.len; i++) { for (i=0; i<val.array.len; i++) {
const char *error; const char *error;
item = PySequence_GetItem(ob, i); item = PySequence_GetItem(ob, i);
error= BPy_IDProperty_Map_ValidateAndCreate("", prop, item); error= BPy_IDProperty_Map_ValidateAndCreate(NULL, prop, item);
Py_DECREF(item); Py_DECREF(item);
if (error) if (error)
@@ -396,7 +407,7 @@ const char *BPy_IDProperty_Map_ValidateAndCreate(const char *name, IDProperty *g
Py_XDECREF(pval); Py_XDECREF(pval);
return "invalid element in subgroup dict template!"; return "invalid element in subgroup dict template!";
} }
if (BPy_IDProperty_Map_ValidateAndCreate(_PyUnicode_AsString(key), prop, pval)) { if (BPy_IDProperty_Map_ValidateAndCreate(key, prop, pval)) {
IDP_FreeProperty(prop); IDP_FreeProperty(prop);
MEM_freeN(prop); MEM_freeN(prop);
Py_XDECREF(keys); Py_XDECREF(keys);
@@ -453,7 +464,7 @@ int BPy_Wrap_SetMapItem(IDProperty *prop, PyObject *key, PyObject *val)
return -1; return -1;
} }
err = BPy_IDProperty_Map_ValidateAndCreate(_PyUnicode_AsString(key), prop, val); err = BPy_IDProperty_Map_ValidateAndCreate(key, prop, val);
if (err) { if (err) {
PyErr_SetString(PyExc_KeyError, err ); PyErr_SetString(PyExc_KeyError, err );
return -1; return -1;
@@ -587,7 +598,7 @@ static PyObject *BPy_IDGroup_Pop(BPy_IDProperty *self, PyObject *value)
{ {
IDProperty *idprop; IDProperty *idprop;
PyObject *pyform; PyObject *pyform;
char *name = _PyUnicode_AsString(value); const char *name = _PyUnicode_AsString(value);
if (!name) { if (!name) {
PyErr_SetString(PyExc_TypeError, "pop expected at least 1 argument, got 0"); PyErr_SetString(PyExc_TypeError, "pop expected at least 1 argument, got 0");
@@ -724,7 +735,7 @@ static PyObject *BPy_IDGroup_GetItems(BPy_IDProperty *self)
static int BPy_IDGroup_Contains(BPy_IDProperty *self, PyObject *value) static int BPy_IDGroup_Contains(BPy_IDProperty *self, PyObject *value)
{ {
char *name = _PyUnicode_AsString(value); const char *name = _PyUnicode_AsString(value);
if (!name) { if (!name) {
PyErr_SetString(PyExc_TypeError, "expected a string"); PyErr_SetString(PyExc_TypeError, "expected a string");

View File

@@ -61,7 +61,7 @@ int BPy_Wrap_SetMapItem(struct IDProperty *prop, PyObject *key, PyObject *val);
PyObject *BPy_IDGroup_WrapData(struct ID *id, struct IDProperty *prop ); PyObject *BPy_IDGroup_WrapData(struct ID *id, struct IDProperty *prop );
const char *BPy_IDProperty_Map_ValidateAndCreate(const char *name, struct IDProperty *group, PyObject *ob); const char *BPy_IDProperty_Map_ValidateAndCreate(PyObject *key, struct IDProperty *group, PyObject *ob);
void IDProp_Init_Types(void); void IDProp_Init_Types(void);

View File

@@ -363,7 +363,7 @@ error_cleanup:
/* string conversion, escape non-unicode chars, coerce must be set to NULL */ /* string conversion, escape non-unicode chars, coerce must be set to NULL */
const char *PyC_UnicodeAsByte(PyObject *py_str, PyObject **coerce) const char *PyC_UnicodeAsByte(PyObject *py_str, PyObject **coerce)
{ {
char *result; const char *result;
result= _PyUnicode_AsString(py_str); result= _PyUnicode_AsString(py_str);

View File

@@ -380,7 +380,7 @@ static PyObject *pyop_getrna(PyObject *UNUSED(self), PyObject *value)
{ {
wmOperatorType *ot; wmOperatorType *ot;
PointerRNA ptr; PointerRNA ptr;
char *opname= _PyUnicode_AsString(value); const char *opname= _PyUnicode_AsString(value);
BPy_StructRNA *pyrna= NULL; BPy_StructRNA *pyrna= NULL;
if (opname==NULL) { if (opname==NULL) {
@@ -413,7 +413,7 @@ static PyObject *pyop_getinstance(PyObject *UNUSED(self), PyObject *value)
wmOperatorType *ot; wmOperatorType *ot;
wmOperator *op; wmOperator *op;
PointerRNA ptr; PointerRNA ptr;
char *opname= _PyUnicode_AsString(value); const char *opname= _PyUnicode_AsString(value);
BPy_StructRNA *pyrna= NULL; BPy_StructRNA *pyrna= NULL;
if (opname==NULL) { if (opname==NULL) {

View File

@@ -1211,7 +1211,7 @@ static StructRNA *pointer_type_from_py(PyObject *value, const char *error_prefix
if (!srna) { if (!srna) {
if (PyErr_Occurred()) { if (PyErr_Occurred()) {
PyObject *msg= PyC_ExceptionBuffer(); PyObject *msg= PyC_ExceptionBuffer();
char *msg_char= _PyUnicode_AsString(msg); const char *msg_char= _PyUnicode_AsString(msg);
PyErr_Format(PyExc_TypeError, PyErr_Format(PyExc_TypeError,
"%.200s expected an RNA type derived from PropertyGroup, failed with: %s", "%.200s expected an RNA type derived from PropertyGroup, failed with: %s",
error_prefix, msg_char); error_prefix, msg_char);

View File

@@ -344,8 +344,9 @@ bool BL_ActionActuator::Update(double curtime, bool frame)
/* Python functions */ /* Python functions */
/* ------------------------------------------------------------------------- */ /* ------------------------------------------------------------------------- */
PyObject* BL_ActionActuator::PyGetChannel(PyObject* value) { PyObject* BL_ActionActuator::PyGetChannel(PyObject* value)
char *string= _PyUnicode_AsString(value); {
const char *string= _PyUnicode_AsString(value);
if (!string) { if (!string) {
PyErr_SetString(PyExc_TypeError, "expected a single string"); PyErr_SetString(PyExc_TypeError, "expected a single string");

View File

@@ -339,10 +339,9 @@ PyObject* listvalue_mapping_subscript(PyObject* self, PyObject* pyindex)
int index = PyLong_AsSsize_t(pyindex); int index = PyLong_AsSsize_t(pyindex);
return listvalue_buffer_item(self, index); /* wont add a ref */ return listvalue_buffer_item(self, index); /* wont add a ref */
} }
PyObject *pyindex_str = PyObject_Repr(pyindex); /* new ref */ PyErr_Format(PyExc_KeyError,
PyErr_Format(PyExc_KeyError, "CList[key]: '%s' key not in list", _PyUnicode_AsString(pyindex_str)); "CList[key]: '%R' key not in list", pyindex);
Py_DECREF(pyindex_str);
return NULL; return NULL;
} }

View File

@@ -1012,8 +1012,8 @@ int PyObjectPlus::py_set_attrdef(PyObject *self_py, PyObject *value, const PyAtt
{ {
if (PyUnicode_Check(value)) if (PyUnicode_Check(value))
{ {
Py_ssize_t val_len; Py_ssize_t val_size;
char *val = _PyUnicode_AsStringAndSize(value, &val_len); const char *val = _PyUnicode_AsStringAndSize(value, &val_size);
strncpy(ptr, val, attrdef->m_size); strncpy(ptr, val, attrdef->m_size);
ptr[attrdef->m_size-1] = 0; ptr[attrdef->m_size-1] = 0;
} }
@@ -1030,7 +1030,7 @@ int PyObjectPlus::py_set_attrdef(PyObject *self_py, PyObject *value, const PyAtt
if (PyUnicode_Check(value)) if (PyUnicode_Check(value))
{ {
Py_ssize_t val_len; Py_ssize_t val_len;
char *val = _PyUnicode_AsStringAndSize(value, &val_len); const char *val = _PyUnicode_AsStringAndSize(value, &val_len); /* XXX, should be 'const' but we do a silly trick to have a shorter string */
if (attrdef->m_clamp) if (attrdef->m_clamp)
{ {
if (val_len < attrdef->m_imin) if (val_len < attrdef->m_imin)
@@ -1042,10 +1042,8 @@ int PyObjectPlus::py_set_attrdef(PyObject *self_py, PyObject *value, const PyAtt
else if (val_len > attrdef->m_imax) else if (val_len > attrdef->m_imax)
{ {
// trim the string // trim the string
char c = val[attrdef->m_imax];
val[attrdef->m_imax] = 0;
*var = val; *var = val;
val[attrdef->m_imax] = c; var->SetLength(attrdef->m_imax);
break; break;
} }
} else if (val_len < attrdef->m_imin || val_len > attrdef->m_imax) } else if (val_len < attrdef->m_imin || val_len > attrdef->m_imax)

View File

@@ -31,8 +31,7 @@ public:
CStringValue(); CStringValue();
CStringValue (const char *txt, const char *name , AllocationTYPE alloctype = CValue::HEAPVALUE); CStringValue (const char *txt, const char *name , AllocationTYPE alloctype = CValue::HEAPVALUE);
virtual ~CStringValue() { virtual ~CStringValue() {}
};
/// CValue implementation /// CValue implementation
virtual bool IsEqual(const STR_String & other); virtual bool IsEqual(const STR_String & other);
virtual const STR_String & GetText(); virtual const STR_String & GetText();
@@ -40,7 +39,7 @@ public:
virtual CValue* Calc(VALUE_OPERATOR op, CValue *val); virtual CValue* Calc(VALUE_OPERATOR op, CValue *val);
virtual CValue* CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val); virtual CValue* CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val);
virtual void SetValue(CValue* newval) { m_strString = newval->GetText(); SetModified(true); }; virtual void SetValue(CValue* newval) { m_strString = newval->GetText(); SetModified(true); }
virtual CValue* GetReplica(); virtual CValue* GetReplica();
#ifdef WITH_PYTHON #ifdef WITH_PYTHON
virtual PyObject* ConvertValueToPython() { virtual PyObject* ConvertValueToPython() {

View File

@@ -199,7 +199,7 @@ SCA_IActuator* SCA_PythonController::LinkedActuatorFromPy(PyObject *value)
if (PyUnicode_Check(value)) { if (PyUnicode_Check(value)) {
/* get the actuator from the name */ /* get the actuator from the name */
char *name= _PyUnicode_AsString(value); const char *name= _PyUnicode_AsString(value);
for(it = lacts.begin(); it!= lacts.end(); ++it) { for(it = lacts.begin(); it!= lacts.end(); ++it) {
if( name == (*it)->GetName() ) { if( name == (*it)->GetName() ) {
return *it; return *it;
@@ -214,12 +214,11 @@ SCA_IActuator* SCA_PythonController::LinkedActuatorFromPy(PyObject *value)
} }
} }
} }
/* set the exception */ /* set the exception */
PyObject *value_str = PyObject_Repr(value); /* new ref */ PyErr_Format(PyExc_ValueError,
PyErr_Format(PyExc_ValueError, "'%s' not in this python controllers actuator list", _PyUnicode_AsString(value_str)); "%R not in this python controllers actuator list", value);
Py_DECREF(value_str);
return NULL; return NULL;
} }
@@ -500,7 +499,7 @@ int SCA_PythonController::pyattr_set_script(void *self_v, const KX_PYATTRIBUTE_D
{ {
SCA_PythonController* self= static_cast<SCA_PythonController*>(self_v); SCA_PythonController* self= static_cast<SCA_PythonController*>(self_v);
char *scriptArg = _PyUnicode_AsString(value); const char *scriptArg = _PyUnicode_AsString(value);
if (scriptArg==NULL) { if (scriptArg==NULL) {
PyErr_SetString(PyExc_TypeError, "controller.script = string: Python Controller, expected a string script text"); PyErr_SetString(PyExc_TypeError, "controller.script = string: Python Controller, expected a string script text");

View File

@@ -883,7 +883,9 @@ bool ConvertPythonToCamera(PyObject * value, KX_Camera **object, bool py_none_ok
if (*object) { if (*object) {
return true; return true;
} else { } else {
PyErr_Format(PyExc_ValueError, "%s, requested name \"%s\" did not match any KX_Camera in this scene", error_prefix, _PyUnicode_AsString(value)); PyErr_Format(PyExc_ValueError,
"%s, requested name \"%s\" did not match any KX_Camera in this scene",
error_prefix, _PyUnicode_AsString(value));
return false; return false;
} }
} }

View File

@@ -705,7 +705,7 @@ static PyObject *gLibNew(PyObject*, PyObject* args)
KX_Scene *kx_scene= gp_KetsjiScene; KX_Scene *kx_scene= gp_KetsjiScene;
char *path; char *path;
char *group; char *group;
char *name; const char *name;
PyObject *names; PyObject *names;
int idcode; int idcode;

View File

@@ -189,7 +189,7 @@ static PyObject *KX_PythonSeq_getIndex(PyObject* self, int index)
return NULL; return NULL;
} }
static PyObjectPlus * KX_PythonSeq_subscript__internal(PyObject *self, char *key) static PyObjectPlus * KX_PythonSeq_subscript__internal(PyObject *self, const char *key)
{ {
PyObjectPlus *self_plus= BGE_PROXY_REF(((KX_PythonSeq *)self)->base); PyObjectPlus *self_plus= BGE_PROXY_REF(((KX_PythonSeq *)self)->base);
@@ -277,7 +277,7 @@ static PyObject * KX_PythonSeq_subscript(PyObject * self, PyObject *key)
return KX_PythonSeq_getIndex(self, PyLong_AsSsize_t( key )); return KX_PythonSeq_getIndex(self, PyLong_AsSsize_t( key ));
} }
else if ( PyUnicode_Check(key) ) { else if ( PyUnicode_Check(key) ) {
char *name = _PyUnicode_AsString(key); const char *name = _PyUnicode_AsString(key);
PyObjectPlus *ret = KX_PythonSeq_subscript__internal(self, name); PyObjectPlus *ret = KX_PythonSeq_subscript__internal(self, name);
if(ret) { if(ret) {