summaryrefslogtreecommitdiffstats
path: root/generic/tclVar.c
diff options
context:
space:
mode:
authordkf <donal.k.fellows@manchester.ac.uk>2003-07-03 23:16:25 (GMT)
committerdkf <donal.k.fellows@manchester.ac.uk>2003-07-03 23:16:25 (GMT)
commit9ef344f3f4c962cb0710eabd4bdbca23381e8088 (patch)
treeca5afc3a8605289ba88ad9ad8302878fa4aac2bf /generic/tclVar.c
parent77697542b70f5ce1907d19ef2612f600a19ceb43 (diff)
downloadtcl-9ef344f3f4c962cb0710eabd4bdbca23381e8088.zip
tcl-9ef344f3f4c962cb0710eabd4bdbca23381e8088.tar.gz
tcl-9ef344f3f4c962cb0710eabd4bdbca23381e8088.tar.bz2
Made [array get] return a dictionary, and allowed [array set] to accept one
without having to convert it to a list (though lists are still preferred.) [Bug 759935]
Diffstat (limited to 'generic/tclVar.c')
-rw-r--r--generic/tclVar.c113
1 files changed, 83 insertions, 30 deletions
diff --git a/generic/tclVar.c b/generic/tclVar.c
index d62c160..bb3d38e 100644
--- a/generic/tclVar.c
+++ b/generic/tclVar.c
@@ -15,7 +15,7 @@
* See the file "license.terms" for information on usage and redistribution
* of this file, and for a DISCLAIMER OF ALL WARRANTIES.
*
- * RCS: @(#) $Id: tclVar.c,v 1.74 2003/06/26 08:43:15 dkf Exp $
+ * RCS: @(#) $Id: tclVar.c,v 1.75 2003/07/03 23:16:25 dkf Exp $
*/
#include "tclInt.h"
@@ -2856,11 +2856,7 @@ Tcl_ArrayObjCmd(dummy, interp, objc, objv)
goto errorInArrayGet;
}
}
- result = Tcl_ListObjAppendElement(interp, tmpResPtr, namePtr);
- if (result != TCL_OK) {
- goto errorInArrayGet;
- }
- result = Tcl_ListObjAppendElement(interp, tmpResPtr, valuePtr);
+ result = Tcl_DictObjPut(interp, tmpResPtr, namePtr, valuePtr);
if (result != TCL_OK) {
goto errorInArrayGet;
}
@@ -2986,7 +2982,7 @@ Tcl_ArrayObjCmd(dummy, interp, objc, objv)
Tcl_WrongNumArgs(interp, 2, objv, "arrayName list");
return TCL_ERROR;
}
- return(TclArraySet(interp, objv[2], objv[3]));
+ return TclArraySet(interp, objv[2], objv[3]);
}
case ARRAY_SIZE: {
Tcl_HashSearch search;
@@ -3133,8 +3129,8 @@ int
TclArraySet(interp, arrayNameObj, arrayElemObj)
Tcl_Interp *interp; /* Current interpreter. */
Tcl_Obj *arrayNameObj; /* The array name. */
- Tcl_Obj *arrayElemObj; /* The array elements list. If this is
- * NULL, create an empty array. */
+ Tcl_Obj *arrayElemObj; /* The array elements list or dict. If
+ * this is NULL, create an empty array. */
{
Var *varPtr, *arrayPtr;
Tcl_Obj **elemPtrs;
@@ -3159,7 +3155,61 @@ TclArraySet(interp, arrayNameObj, arrayElemObj)
return TCL_ERROR;
}
- if (arrayElemObj != NULL) {
+ if (arrayElemObj == NULL) {
+ goto ensureArray;
+ }
+
+ /*
+ * Install the contents of the dictionary or list into the array.
+ */
+
+ if (arrayElemObj->typePtr == &tclDictType) {
+ Tcl_Obj *keyPtr, *valuePtr;
+ Tcl_DictSearch search;
+ int done;
+
+ if (Tcl_DictObjSize(interp, arrayElemObj, &done) != TCL_OK) {
+ return TCL_ERROR;
+ }
+ if (done == 0) {
+ /*
+ * Empty, so we'll just force the array to be properly
+ * existing instead.
+ */
+ goto ensureArray;
+ }
+
+ /*
+ * Don't need to look at result of Tcl_DictObjFirst as we've
+ * just successfully used a dictionary operation on the same
+ * object.
+ */
+
+ for (Tcl_DictObjFirst(interp, arrayElemObj, &search,
+ &keyPtr, &valuePtr, &done) ; !done ;
+ Tcl_DictObjNext(&search, &keyPtr, &valuePtr, &done)) {
+ /*
+ * At this point, it would be nice if the key was directly
+ * usable by the array. This isn't the case though.
+ */
+ char *part2 = TclGetString(keyPtr);
+ Var *elemVarPtr = TclLookupArrayElement(interp, varName,
+ part2, TCL_LEAVE_ERR_MSG, "set", 1, 1, varPtr);
+
+ if ((elemVarPtr == NULL) ||
+ (TclPtrSetVar(interp, elemVarPtr, varPtr, varName,
+ part2, valuePtr, TCL_LEAVE_ERR_MSG) == NULL)) {
+ Tcl_DictObjDone(&search);
+ return TCL_ERROR;
+ }
+ }
+ return TCL_OK;
+ } else {
+ /*
+ * Not a dictionary, so assume (and convert to, for
+ * backward-compatability reasons) a list.
+ */
+
result = Tcl_ListObjGetElements(interp, arrayElemObj,
&elemLen, &elemPtrs);
if (result != TCL_OK) {
@@ -3171,33 +3221,36 @@ TclArraySet(interp, arrayNameObj, arrayElemObj)
"list must have an even number of elements", -1);
return TCL_ERROR;
}
- if (elemLen > 0) {
- /*
- * We needn't worry about traces invalidating arrayPtr:
- * should that be the case, TclPtrSetVar will return NULL
- * so that we break out of the loop and return an error.
- */
+ if (elemLen == 0) {
+ goto ensureArray;
+ }
- for (i = 0; i < elemLen; i += 2) {
- char *part2 = TclGetString(elemPtrs[i]);
- Var *elemVarPtr = TclLookupArrayElement(interp, varName,
- part2, TCL_LEAVE_ERR_MSG, "set", 1, 1, varPtr);
- if ((elemVarPtr == NULL) ||
- (TclPtrSetVar(interp, elemVarPtr, varPtr, varName,
- part2, elemPtrs[i+1], TCL_LEAVE_ERR_MSG) == NULL)) {
- result = TCL_ERROR;
- break;
- }
+ /*
+ * We needn't worry about traces invalidating arrayPtr: should
+ * that be the case, TclPtrSetVar will return NULL so that we
+ * break out of the loop and return an error.
+ */
+
+ for (i = 0; i < elemLen; i += 2) {
+ char *part2 = TclGetString(elemPtrs[i]);
+ Var *elemVarPtr = TclLookupArrayElement(interp, varName,
+ part2, TCL_LEAVE_ERR_MSG, "set", 1, 1, varPtr);
+ if ((elemVarPtr == NULL) ||
+ (TclPtrSetVar(interp, elemVarPtr, varPtr, varName, part2,
+ elemPtrs[i+1], TCL_LEAVE_ERR_MSG) == NULL)) {
+ result = TCL_ERROR;
+ break;
}
- return result;
}
+ return result;
}
-
+
/*
* The list is empty make sure we have an array, or create
* one if necessary.
*/
-
+
+ ensureArray:
if (varPtr != NULL) {
if (!TclIsVarUndefined(varPtr) && TclIsVarArray(varPtr)) {
/*
@@ -3219,7 +3272,7 @@ TclArraySet(interp, arrayNameObj, arrayElemObj)
TclSetVarArray(varPtr);
TclClearVarUndefined(varPtr);
varPtr->value.tablePtr =
- (Tcl_HashTable *) ckalloc(sizeof(Tcl_HashTable));
+ (Tcl_HashTable *) ckalloc(sizeof(Tcl_HashTable));
Tcl_InitHashTable(varPtr->value.tablePtr, TCL_STRING_KEYS);
return TCL_OK;
}