svn merge https://svn.blender.org/svnroot/bf-blender/trunk/blender -r22625:22668
[blender.git] / source / gameengine / Expressions / ListValue.cpp
1 // ListValue.cpp: implementation of the CListValue class.
2 //
3 //////////////////////////////////////////////////////////////////////
4 /*
5  * Copyright (c) 1996-2000 Erwin Coumans <coockie@acm.org>
6  *
7  * Permission to use, copy, modify, distribute and sell this software
8  * and its documentation for any purpose is hereby granted without fee,
9  * provided that the above copyright notice appear in all copies and
10  * that both that copyright notice and this permission notice appear
11  * in supporting documentation.  Erwin Coumans makes no
12  * representations about the suitability of this software for any
13  * purpose.  It is provided "as is" without express or implied warranty.
14  *
15  */
16
17 #include "ListValue.h"
18 #include "StringValue.h"
19 #include "VoidValue.h"
20 #include <algorithm>
21 #include "BoolValue.h"
22
23 #include "BLO_sys_types.h" /* for intptr_t support */
24
25 #ifdef HAVE_CONFIG_H
26 #include <config.h>
27 #endif
28
29 #if  ((PY_MAJOR_VERSION == 2) &&(PY_MINOR_VERSION < 5))
30 #define Py_ssize_t int
31 #endif
32
33 Py_ssize_t listvalue_bufferlen(PyObject* self)
34 {
35         CListValue *list= static_cast<CListValue *>(BGE_PROXY_REF(self));
36         if (list==NULL)
37                 return 0;
38         
39         return (Py_ssize_t)list->GetCount();
40 }
41
42 PyObject* listvalue_buffer_item(PyObject* self, Py_ssize_t index)
43 {
44         CListValue *list= static_cast<CListValue *>(BGE_PROXY_REF(self));
45         CValue *cval;
46         
47         if (list==NULL) {
48                 PyErr_SetString(PyExc_SystemError, "val = CList[i], "BGE_PROXY_ERROR_MSG);
49                 return NULL;
50         }
51         
52         int count = list->GetCount();
53         
54         if (index < 0)
55                 index = count+index;
56         
57         if (index < 0 || index >= count) {
58                 PyErr_SetString(PyExc_IndexError, "CList[i]: Python ListIndex out of range in CValueList");
59                 return NULL;
60         }
61         
62         cval= list->GetValue(index);
63         
64         PyObject* pyobj = cval->ConvertValueToPython();
65         if (pyobj)
66                 return pyobj;
67         else
68                 return cval->GetProxy();
69 }
70
71 PyObject* listvalue_mapping_subscript(PyObject* self, PyObject* pyindex)
72 {
73         CListValue *list= static_cast<CListValue *>(BGE_PROXY_REF(self));
74         if (list==NULL) {
75                 PyErr_SetString(PyExc_SystemError, "value = CList[i], "BGE_PROXY_ERROR_MSG);
76                 return NULL;
77         }
78         
79         if (PyUnicode_Check(pyindex))
80         {
81                 CValue *item = ((CListValue*) list)->FindValue(_PyUnicode_AsString(pyindex));
82                 if (item) {
83                         PyObject* pyobj = item->ConvertValueToPython();
84                         if(pyobj)
85                                 return pyobj;
86                         else
87                                 return item->GetProxy();
88                 }
89         }
90         else if (PyLong_Check(pyindex))
91         {
92                 int index = PyLong_AsSsize_t(pyindex);
93                 return listvalue_buffer_item(self, index); /* wont add a ref */
94         }
95         
96         PyObject *pyindex_str = PyObject_Repr(pyindex); /* new ref */
97         PyErr_Format(PyExc_KeyError, "CList[key]: '%s' key not in list", _PyUnicode_AsString(pyindex_str));
98         Py_DECREF(pyindex_str);
99         return NULL;
100 }
101
102
103 /* just slice it into a python list... */
104 PyObject* listvalue_buffer_slice(PyObject* self,Py_ssize_t ilow, Py_ssize_t ihigh)
105 {
106         CListValue *list= static_cast<CListValue *>(BGE_PROXY_REF(self));
107         if (list==NULL) {
108                 PyErr_SetString(PyExc_SystemError, "val = CList[i:j], "BGE_PROXY_ERROR_MSG);
109                 return NULL;
110         }
111         
112         int i, j;
113         PyObject *newlist;
114
115         if (ilow < 0) ilow = 0;
116
117         int n = ((CListValue*) list)->GetCount();
118
119         if (ihigh >= n)
120                 ihigh = n;
121     if (ihigh < ilow)
122         ihigh = ilow;
123
124         newlist = PyList_New(ihigh - ilow);
125         if (!newlist)
126                 return NULL;
127
128         for (i = ilow, j = 0; i < ihigh; i++, j++)
129         {
130                 PyObject* pyobj = list->GetValue(i)->ConvertValueToPython();
131                 if (!pyobj)
132                         pyobj = list->GetValue(i)->GetProxy();
133                 PyList_SET_ITEM(newlist, i, pyobj);
134         }       
135         return newlist;
136 }
137
138
139 /* clist + list, return a list that python owns */
140 static PyObject *listvalue_buffer_concat(PyObject * self, PyObject * other)
141 {
142         CListValue *listval= static_cast<CListValue *>(BGE_PROXY_REF(self));
143         int i, numitems, numitems_orig;
144         
145         if (listval==NULL) {
146                 PyErr_SetString(PyExc_SystemError, "CList+other, "BGE_PROXY_ERROR_MSG);
147                 return NULL;
148         }
149         
150         numitems_orig= listval->GetCount();
151         
152         // for now, we support CListValue concatenated with items
153         // and CListValue concatenated to Python Lists
154         // and CListValue concatenated with another CListValue
155         
156         /* Shallow copy, dont use listval->GetReplica(), it will screw up with KX_GameObjects */
157         CListValue* listval_new = new CListValue();
158         
159         if (PyList_Check(other))
160         {
161                 CValue* listitemval;
162                 bool error = false;
163                 
164                 numitems = PyList_Size(other);
165                 
166                 /* copy the first part of the list */
167                 listval_new->Resize(numitems_orig + numitems);
168                 for (i=0;i<numitems_orig;i++)
169                         listval_new->SetValue(i, listval->GetValue(i)->AddRef());
170                 
171                 for (i=0;i<numitems;i++)
172                 {
173                         listitemval = listval->ConvertPythonToValue(PyList_GetItem(other,i), "cList + pyList: CListValue, ");
174                         
175                         if (listitemval) {
176                                 listval_new->SetValue(i+numitems_orig, listitemval);
177                         } else {
178                                 error= true;
179                                 break;
180                         }
181                 }
182                 
183                 if (error) {
184                         listval_new->Resize(numitems_orig+i); /* resize so we dont try release NULL pointers */
185                         listval_new->Release();
186                         return NULL; /* ConvertPythonToValue above sets the error */ 
187                 }
188         
189         }
190         else if (PyObject_TypeCheck(other, &CListValue::Type)) {
191                 // add items from otherlist to this list
192                 CListValue* otherval = static_cast<CListValue *>(BGE_PROXY_REF(other));
193                 if(otherval==NULL) {
194                         listval_new->Release();
195                         PyErr_SetString(PyExc_SystemError, "CList+other, "BGE_PROXY_ERROR_MSG);
196                         return NULL;
197                 }
198                 
199                 numitems = otherval->GetCount();
200                 
201                 /* copy the first part of the list */
202                 listval_new->Resize(numitems_orig + numitems); /* resize so we dont try release NULL pointers */
203                 for (i=0;i<numitems_orig;i++)
204                         listval_new->SetValue(i, listval->GetValue(i)->AddRef());
205                 
206                 /* now copy the other part of the list */
207                 for (i=0;i<numitems;i++)
208                         listval_new->SetValue(i+numitems_orig, otherval->GetValue(i)->AddRef());
209                 
210         }
211         return listval_new->NewProxy(true); /* python owns this list */
212 }
213
214 static int listvalue_buffer_contains(PyObject *self_v, PyObject *value)
215 {
216         CListValue *self= static_cast<CListValue *>(BGE_PROXY_REF(self_v));
217         
218         if (self==NULL) {
219                 PyErr_SetString(PyExc_SystemError, "val in CList, "BGE_PROXY_ERROR_MSG);
220                 return -1;
221         }
222         
223         if (PyUnicode_Check(value)) {
224                 if (self->FindValue((const char *)_PyUnicode_AsString(value))) {
225                         return 1;
226                 }
227         }
228         else if (PyObject_TypeCheck(value, &CValue::Type)) { /* not dict like at all but this worked before __contains__ was used */
229                 CValue *item= static_cast<CValue *>(BGE_PROXY_REF(value));
230                 for (int i=0; i < self->GetCount(); i++)
231                         if (self->GetValue(i) == item) // Com
232                                 return 1;
233                 
234         } // not using CheckEqual
235         
236         return 0;
237 }
238
239
240 static  PySequenceMethods listvalue_as_sequence = {
241         listvalue_bufferlen,//(inquiry)buffer_length, /*sq_length*/
242         listvalue_buffer_concat, /*sq_concat*/
243         NULL, /*sq_repeat*/
244         listvalue_buffer_item, /*sq_item*/
245 // TODO, slicing in py3
246         NULL, // listvalue_buffer_slice, /*sq_slice*/
247         NULL, /*sq_ass_item*/
248         NULL, /*sq_ass_slice*/
249         (objobjproc)listvalue_buffer_contains,  /* sq_contains */
250 };
251
252
253
254 /* Is this one used ? */
255 static  PyMappingMethods instance_as_mapping = {
256         listvalue_bufferlen, /*mp_length*/
257         listvalue_mapping_subscript, /*mp_subscript*/
258         NULL /*mp_ass_subscript*/
259 };
260
261
262
263 PyTypeObject CListValue::Type = {
264         PyVarObject_HEAD_INIT(NULL, 0)
265         "CListValue",                   /*tp_name*/
266         sizeof(PyObjectPlus_Proxy), /*tp_basicsize*/
267         0,                              /*tp_itemsize*/
268         /* methods */
269         py_base_dealloc,                        /*tp_dealloc*/
270         0,                              /*tp_print*/
271         0,                      /*tp_getattr*/
272         0,                      /*tp_setattr*/
273         0,                              /*tp_compare*/
274         py_base_repr,                           /*tp_repr*/
275         0,                              /*tp_as_number*/
276         &listvalue_as_sequence, /*tp_as_sequence*/
277         &instance_as_mapping,           /*tp_as_mapping*/
278         0,                              /*tp_hash*/
279         0,                              /*tp_call */
280         0,
281         NULL,
282         NULL,
283         0,
284         Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
285         0,0,0,0,0,0,0,
286         Methods,
287         0,
288         0,
289         &CValue::Type,
290         0,0,0,0,0,0,
291         py_base_new
292 };
293
294 PyMethodDef CListValue::Methods[] = {
295         /* List style access */
296         {"append", (PyCFunction)CListValue::sPyappend,METH_O},
297         {"reverse", (PyCFunction)CListValue::sPyreverse,METH_NOARGS},
298         {"index", (PyCFunction)CListValue::sPyindex,METH_O},
299         {"count", (PyCFunction)CListValue::sPycount,METH_O},
300         
301         /* Dict style access */
302         {"get", (PyCFunction)CListValue::sPyget,METH_VARARGS},
303         {"has_key", (PyCFunction)CListValue::sPyhas_key,METH_O},
304         
305         /* Own cvalue funcs */
306         {"from_id", (PyCFunction)CListValue::sPyfrom_id,METH_O},
307         
308         {NULL,NULL} //Sentinel
309 };
310
311 PyAttributeDef CListValue::Attributes[] = {
312         { NULL }        //Sentinel
313 };
314
315 //////////////////////////////////////////////////////////////////////
316 // Construction/Destruction
317 //////////////////////////////////////////////////////////////////////
318
319 CListValue::CListValue()
320 : CPropValue()
321 {
322         m_bReleaseContents=true;        
323 }
324
325
326
327 CListValue::~CListValue()
328 {
329
330         if (m_bReleaseContents) {
331                 for (unsigned int i=0;i<m_pValueArray.size();i++) {
332                         m_pValueArray[i]->Release();
333                 }
334         }
335 }
336
337
338 static STR_String gstrListRep=STR_String("List");
339
340 const STR_String & CListValue::GetText()
341 {
342         gstrListRep = "[";
343         STR_String commastr = "";
344
345         for (int i=0;i<GetCount();i++)
346         {
347                 gstrListRep += commastr;
348                 gstrListRep += GetValue(i)->GetText();
349                 commastr = ",";
350         }
351         gstrListRep += "]";
352
353         return gstrListRep;
354 }
355
356
357
358 CValue* CListValue::GetReplica() { 
359         CListValue* replica = new CListValue(*this);
360
361         replica->ProcessReplica();
362
363         replica->m_bReleaseContents=true; // for copy, complete array is copied for now...
364         // copy all values
365         int numelements = m_pValueArray.size();
366         unsigned int i=0;
367         replica->m_pValueArray.resize(numelements);
368         for (i=0;i<m_pValueArray.size();i++)
369                 replica->m_pValueArray[i] = m_pValueArray[i]->GetReplica();
370
371
372         return replica;
373 };
374
375
376
377 void CListValue::SetValue(int i, CValue *val)
378 {
379         assertd(i < m_pValueArray.size());
380         m_pValueArray[i]=val;
381 }
382
383
384
385 void CListValue::Resize(int num)
386 {
387         m_pValueArray.resize(num);
388 }
389
390
391
392 void CListValue::Remove(int i)
393 {
394         assertd(i<m_pValueArray.size());
395         m_pValueArray.erase(m_pValueArray.begin()+i);
396 }
397
398
399
400 void CListValue::ReleaseAndRemoveAll()
401 {
402         for (unsigned int i=0;i<m_pValueArray.size();i++)
403                 m_pValueArray[i]->Release();
404         m_pValueArray.clear();//.Clear();
405 }
406
407
408
409 CValue* CListValue::FindValue(const STR_String & name)
410 {
411         for (int i=0; i < GetCount(); i++)
412                 if (GetValue(i)->GetName() == name)
413                         return GetValue(i);
414         
415         return NULL;
416 }
417
418 CValue* CListValue::FindValue(const char * name)
419 {
420         for (int i=0; i < GetCount(); i++)
421                 if (GetValue(i)->GetName() == name)
422                         return GetValue(i);
423         
424         return NULL;
425 }
426
427 bool CListValue::SearchValue(CValue *val)
428 {
429         for (int i=0;i<GetCount();i++)
430                 if (val == GetValue(i))
431                         return true;
432         return false;
433 }
434
435
436
437 void CListValue::SetReleaseOnDestruct(bool bReleaseContents)
438 {
439         m_bReleaseContents = bReleaseContents;
440 }
441
442
443
444 bool CListValue::RemoveValue(CValue *val)
445 {
446         bool result=false;
447
448         for (int i=GetCount()-1;i>=0;i--)
449                 if (val == GetValue(i))
450                 {
451                         Remove(i);
452                         result=true;
453                 }
454         return result;
455 }
456
457
458
459 void CListValue::MergeList(CListValue *otherlist)
460 {
461
462         int numelements = this->GetCount();
463         int numotherelements = otherlist->GetCount();
464
465
466         Resize(numelements+numotherelements);
467
468         for (int i=0;i<numotherelements;i++)
469         {
470                 SetValue(i+numelements,otherlist->GetValue(i)->AddRef());
471         }
472 }
473
474
475 PyObject* CListValue::Pyappend(PyObject* value)
476 {
477         CValue* objval = ConvertPythonToValue(value, "CList.append(i): CValueList, ");
478
479         if (!objval) /* ConvertPythonToValue sets the error */
480                 return NULL;
481         
482         if (!BGE_PROXY_PYOWNS(m_proxy)) {
483                 PyErr_SetString(PyExc_TypeError, "CList.append(i): this CValueList is used internally for the game engine and can't be modified");
484                 return NULL;
485         }
486         
487         Add(objval);
488         
489         Py_RETURN_NONE;
490 }
491
492
493
494 PyObject* CListValue::Pyreverse()
495 {
496         std::reverse(m_pValueArray.begin(),m_pValueArray.end());
497         Py_RETURN_NONE;
498 }
499
500
501
502 bool CListValue::CheckEqual(CValue* first,CValue* second)
503 {
504         bool result = false;
505         
506         CValue* eqval =  ((CValue*)first)->Calc(VALUE_EQL_OPERATOR,(CValue*)second);
507         
508         if (eqval==NULL)
509                 return false;
510         const STR_String& text = eqval->GetText();
511         if (&text==&CBoolValue::sTrueString)
512         {
513                 result = true;
514         }
515         eqval->Release();
516         return result;
517
518 }
519
520
521
522 PyObject* CListValue::Pyindex(PyObject *value)
523 {
524         PyObject* result = NULL;
525
526         CValue* checkobj = ConvertPythonToValue(value, "val = cList[i]: CValueList, ");
527         if (checkobj==NULL)
528                 return NULL; /* ConvertPythonToValue sets the error */
529
530         int numelem = GetCount();
531         for (int i=0;i<numelem;i++)
532         {
533                 CValue* elem =                  GetValue(i);
534                 if (checkobj==elem || CheckEqual(checkobj,elem))
535                 {
536                         result = PyLong_FromSsize_t(i);
537                         break;
538                 }
539         }
540         checkobj->Release();
541
542         if (result==NULL) {
543                 PyErr_SetString(PyExc_ValueError, "CList.index(x): x not in CListValue");
544         }
545         return result;
546         
547 }
548
549
550
551 PyObject* CListValue::Pycount(PyObject* value)
552 {
553         int numfound = 0;
554
555         CValue* checkobj = ConvertPythonToValue(value, ""); /* error ignored */
556         
557         if (checkobj==NULL) { /* in this case just return that there are no items in the list */
558                 PyErr_Clear();
559                 return PyLong_FromSsize_t(0);
560         }
561
562         int numelem = GetCount();
563         for (int i=0;i<numelem;i++)
564         {
565                 CValue* elem =                  GetValue(i);
566                 if (checkobj==elem || CheckEqual(checkobj,elem))
567                 {
568                         numfound ++;
569                 }
570         }
571         checkobj->Release();
572
573         return PyLong_FromSsize_t(numfound);
574 }
575
576 /* Matches python dict.get(key, [default]) */
577 PyObject* CListValue::Pyget(PyObject *args)
578 {
579         char *key;
580         PyObject* def = Py_None;
581
582         if (!PyArg_ParseTuple(args, "s|O:get", &key, &def))
583                 return NULL;
584         
585         CValue *item = FindValue((const char *)key);
586         if (item) {     
587                 PyObject* pyobj = item->ConvertValueToPython();
588                 if (pyobj)
589                         return pyobj;
590                 else
591                         return item->GetProxy();
592         }
593         Py_INCREF(def);
594         return def;
595 }
596
597 /* Matches python dict.has_key() */
598 PyObject* CListValue::Pyhas_key(PyObject* value)
599 {
600         if (PyUnicode_Check(value) && FindValue((const char *)_PyUnicode_AsString(value)))
601                 Py_RETURN_TRUE;
602         
603         Py_RETURN_FALSE;
604 }
605
606 PyObject* CListValue::Pyfrom_id(PyObject* value)
607 {
608         uintptr_t id= (uintptr_t)PyLong_AsVoidPtr(value);
609         
610         if (PyErr_Occurred())
611                 return NULL;
612
613         int numelem = GetCount();
614         for (int i=0;i<numelem;i++)
615         {
616                 if (reinterpret_cast<uintptr_t>(m_pValueArray[i]->m_proxy) == id)
617                         return GetValue(i)->GetProxy();
618         }
619         PyErr_SetString(PyExc_IndexError, "from_id(#): id not found in CValueList");
620         return NULL;    
621
622 }
623
624
625 /* --------------------------------------------------------------------- 
626  * Some stuff taken from the header
627  * --------------------------------------------------------------------- */
628 CValue* CListValue::Calc(VALUE_OPERATOR op,CValue *val) 
629 {
630         //assert(false); // todo: implement me!
631         static int error_printed =  0;
632         if (error_printed==0) {
633                 fprintf(stderr, "CValueList::Calc not yet implimented\n");
634                 error_printed = 1;
635         }
636         return NULL;
637 }
638
639 CValue* CListValue::CalcFinal(VALUE_DATA_TYPE dtype,
640                                                           VALUE_OPERATOR op, 
641                                                           CValue* val) 
642 {
643         //assert(false); // todo: implement me!
644         static int error_printed =  0;
645         if (error_printed==0) {
646                 fprintf(stderr, "CValueList::CalcFinal not yet implimented\n");
647                 error_printed = 1;
648         }
649         return NULL;
650 }
651
652
653
654 void CListValue::Add(CValue* value)
655 {
656         m_pValueArray.push_back(value);
657 }
658
659
660
661 double CListValue::GetNumber()
662 {
663         return -1;
664 }
665
666
667
668 void CListValue::SetModified(bool bModified)
669 {       
670         CValue::SetModified(bModified);
671         int numels = GetCount();
672
673         for (int i=0;i<numels;i++)
674                 GetValue(i)->SetModified(bModified);
675 }
676
677
678
679 bool CListValue::IsModified()
680 {
681         bool bmod = CValue::IsModified(); //normal own flag
682         int numels = GetCount();
683
684         for (int i=0;i<numels;i++)
685                 bmod = bmod || GetValue(i)->IsModified();
686
687         return bmod;
688 }