not block so long
[enigma2.git] / lib / base / ebase.cpp
index f990f6b89aa35af987b0a97d478704dfbafd742d..75307e852dea743c957e2c80a751fc64a91b5240 100644 (file)
@@ -116,20 +116,25 @@ ePtrList<eMainloop> eMainloop::existing_loops;
 
 void eMainloop::addSocketNotifier(eSocketNotifier *sn)
 {
-       notifiers.insert(std::pair<int,eSocketNotifier*> (sn->getFD(), sn));
+       int fd = sn->getFD();
+       ASSERT(notifiers.find(fd) == notifiers.end());
+       ASSERT(new_notifiers.find(fd) == new_notifiers.end());
+       new_notifiers[fd]=sn;
 }
 
 void eMainloop::removeSocketNotifier(eSocketNotifier *sn)
 {
-       for (std::multimap<int,eSocketNotifier*>::iterator i = notifiers.find(sn->getFD());
-                       i != notifiers.end();
-                       ++i)
-               if (i->second == sn)
-                       return notifiers.erase(i);
+       int fd = sn->getFD();
+       std::map<int,eSocketNotifier*>::iterator i(notifiers.find(fd));
+       if (i != notifiers.end())
+               return notifiers.erase(i);
+       i = new_notifiers.find(fd);
+       if (i != new_notifiers.end())
+               return new_notifiers.erase(i);
        eFatal("removed socket notifier which is not present");
 }
 
-int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, PyObject *additional)
+int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, ePyObject additional)
 {
        int return_reason = 0;
                /* get current time */
@@ -162,96 +167,71 @@ int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, PyObje
                poll_timeout = user_timeout;
                return_reason = 1;
        }
-               
-       int ret = 0;
-               
-       std::multimap<int,eSocketNotifier*>::iterator it;
-       std::map<int,int> fd_merged;
-       std::map<int,int>::const_iterator fd_merged_it;
-               
-       for (it = notifiers.begin(); it != notifiers.end(); ++it)
-               fd_merged[it->first] |= it->second->getRequested();
-               
-       fd_merged_it = fd_merged.begin();
-               
-       int nativecount, fdcount;
-               
-       nativecount = fdcount = fd_merged.size();
-               
-       if (additional)
+
+       for (std::map<int, eSocketNotifier*>::iterator it(new_notifiers.begin()); it != new_notifiers.end();)
        {
-               additional = PyDict_Items(additional);
-               fdcount += PyList_Size(additional);
+               notifiers[it->first]=it->second;
+               new_notifiers.erase(it++);
        }
+
+       int nativecount=notifiers.size(),
+               fdcount=nativecount,
+               ret=0;
+
+       if (additional)
+               fdcount += PyDict_Size(additional);
                
                // build the poll aray
-       pollfd* pfd = new pollfd[fdcount];  // make new pollfd array
-               
-       for (int i=0; i < nativecount; i++, fd_merged_it++)
+       pollfd pfd[fdcount];  // make new pollfd array
+       std::map<int,eSocketNotifier*>::iterator it = notifiers.begin();
+       int i=0;
+       for (; i < nativecount; ++i, ++it)
        {
-               pfd[i].fd = fd_merged_it->first;
-               pfd[i].events = fd_merged_it->second;
+               pfd[i].fd = it->first;
+               pfd[i].events = it->second->getRequested();
        }
-               
+       
        if (additional)
        {
-               for (int i=0; i < PyList_Size(additional); ++i)
-               {
-                       PyObject *it = PyList_GET_ITEM(additional, i);
-                       if (!PyTuple_Check(it))
-                               eFatal("poll item is not a tuple");
-                       if (PyTuple_Size(it) != 2)
-                               eFatal("poll tuple size is not 2");
-                       int fd = PyObject_AsFileDescriptor(PyTuple_GET_ITEM(it, 0));
-                       if (fd == -1)
-                               eFatal("poll tuple not a filedescriptor");
-                       pfd[nativecount + i].fd = fd;
-                       pfd[nativecount + i].events = PyInt_AsLong(PyTuple_GET_ITEM(it, 1));
+               PyObject *key, *val;
+               int pos=0;
+               while (PyDict_Next(additional, &pos, &key, &val)) {
+                       pfd[i].fd = PyObject_AsFileDescriptor(key);
+                       pfd[i++].events = PyInt_AsLong(val);
                }
-               Py_DECREF(additional);
        }
-               
+       
        ret = ::poll(pfd, fdcount, poll_timeout);
-               
                        /* ret > 0 means that there are some active poll entries. */
        if (ret > 0)
        {
+               int i=0;
                return_reason = 0;
-               for (int i=0; i < nativecount ; i++)
+               for (; i < nativecount ; ++i)
                {
-                       it = notifiers.begin();
-                               
-                       int handled = 0;
-                               
-                       std::multimap<int,eSocketNotifier*>::iterator 
-                               l = notifiers.lower_bound(pfd[i].fd),
-                               u = notifiers.upper_bound(pfd[i].fd);
-                               
-                       ePtrList<eSocketNotifier> n;
-                               
-                       for (; l != u; ++l)
-                               n.push_back(l->second);
-                               
-                       for (ePtrList<eSocketNotifier>::iterator li(n.begin()); li != n.end(); ++li)
+                       if (pfd[i].revents)
                        {
-                               int req = li->getRequested();
-                                       
-                               handled |= req;
-                               
-                               if (pfd[i].revents & req)
-                                       (*li)->activate(pfd[i].revents);
+                               int handled = 0;
+                               it = notifiers.find(pfd[i].fd);
+                               if (it != notifiers.end())
+                               {
+                                       int req = it->second->getRequested();
+                                       handled |= req;
+                                       if (pfd[i].revents & req)
+                                               it->second->activate(pfd[i].revents);
+                               }
+                               pfd[i].revents &= ~handled;
+                               if (pfd[i].revents & (POLLERR|POLLHUP|POLLNVAL))
+                                       eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
                        }
-                       if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL))
-                               eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
                }
-                       
-               for (int i = nativecount; i < fdcount; ++i)
+               for (; i < fdcount; ++i)
                {
                        if (pfd[i].revents)
                        {
                                if (!*res)
                                        *res = PyList_New(0);
-                               PyObject *it = PyTuple_New(2);
+                               ePyObject it = PyTuple_New(2);
                                PyTuple_SET_ITEM(it, 0, PyInt_FromLong(pfd[i].fd));
                                PyTuple_SET_ITEM(it, 1, PyInt_FromLong(pfd[i].revents));
                                PyList_Append(*res, it);
@@ -271,7 +251,6 @@ int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, PyObje
                        ret = -1; /* don't assume the timeout has passed when we got a signal */
                }
        }
-       delete [] pfd;
        
                /* when we not processed anything, check timers. */
        if (!m_timer_list.empty())
@@ -302,7 +281,7 @@ void eMainloop::removeTimer(eTimer* e)
        m_timer_list.remove(e);
 }
 
-int eMainloop::iterate(unsigned int user_timeout, PyObject **res, PyObject *dict)
+int eMainloop::iterate(unsigned int user_timeout, PyObject **res, ePyObject dict)
 {
        int ret = 0;
        
@@ -350,9 +329,9 @@ void eMainloop::reset()
        app_quit_now=false;
 }
 
-PyObject *eMainloop::poll(PyObject *timeout, PyObject *dict)
+PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict)
 {
-       PyObject *res = 0;
+       PyObject *res=0;
        
        if (app_quit_now)
        {
@@ -363,11 +342,12 @@ PyObject *eMainloop::poll(PyObject *timeout, PyObject *dict)
        int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);
 
        iterate(user_timeout, &res, dict);
+       ePyObject ret(res);
        
-       if (!res) /* return empty list on timeout */
-               res = PyList_New(0);
+       if (!ret) /* return empty list on timeout */
+               return PyList_New(0);
        
-       return res;
+       return ret;
 }
 
 void eMainloop::interruptPoll()