X-Git-Url: http://code.vuplus.com/gitweb/?p=vuplus_dvbapp;a=blobdiff_plain;f=lib%2Fbase%2Febase.cpp;h=480c8ea8d2a8d48e70ff7e6b8e5ddc0bb697aef7;hp=ffd35eceae6cc6cb5d4adfac1c42bf37e5e4585e;hb=6dcb75c26bbfec04b381bb99d75404f9fe50635d;hpb=d026012d8ba6198d879df68cb4bf6c92c95a633a diff --git a/lib/base/ebase.cpp b/lib/base/ebase.cpp index ffd35ec..480c8ea 100644 --- a/lib/base/ebase.cpp +++ b/lib/base/ebase.cpp @@ -116,12 +116,14 @@ ePtrList eMainloop::existing_loops; void eMainloop::addSocketNotifier(eSocketNotifier *sn) { - notifiers.insert(std::pair (sn->getFD(), sn)); + int fd = sn->getFD(); + ASSERT(notifiers.find(fd) == notifiers.end()); + notifiers[fd]=sn; } void eMainloop::removeSocketNotifier(eSocketNotifier *sn) { - for (std::multimap::iterator i = notifiers.find(sn->getFD()); + for (std::map::iterator i = notifiers.find(sn->getFD()); i != notifiers.end(); ++i) if (i->second == sn) @@ -129,16 +131,22 @@ void eMainloop::removeSocketNotifier(eSocketNotifier *sn) eFatal("removed socket notifier which is not present"); } -int eMainloop::processOneEvent(unsigned int user_timeout) +int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, ePyObject additional) { int return_reason = 0; /* get current time */ timeval now; gettimeofday(&now, 0); m_now_is_invalid = 0; - + + if (additional && !PyDict_Check(additional)) + eFatal("additional, but it's not dict"); + + if (additional && !res) + eFatal("additional, but no res"); + int poll_timeout = -1; /* infinite in case of empty timer list */ - + if (m_timer_list) { singleLock s(recalcLock); @@ -151,98 +159,100 @@ int eMainloop::processOneEvent(unsigned int user_timeout) poll_timeout /= 1000; } - if ((user_timeout > 0) && (poll_timeout > user_timeout)) + if ((user_timeout > 0) && (poll_timeout > 0) && ((unsigned int)poll_timeout > user_timeout)) { poll_timeout = user_timeout; return_reason = 1; } - int ret = 0; - - if (poll_timeout) - { + int nativecount=notifiers.size(), + fdcount=nativecount, + ret=0; - std::multimap::iterator it; - std::map fd_merged; - std::map::const_iterator fd_merged_it; - - for (it = notifiers.begin(); it != notifiers.end(); ++it) - fd_merged[it->first] |= it->second->getRequested(); + if (additional) + fdcount += PyDict_Size(additional); - fd_merged_it = fd_merged.begin(); - - int fdcount = fd_merged.size(); - - // build the poll aray - pollfd* pfd = new pollfd[fdcount]; // make new pollfd array - - for (int i=0; i < fdcount; i++, fd_merged_it++) - { - pfd[i].fd = fd_merged_it->first; - pfd[i].events = fd_merged_it->second; + // build the poll aray + pollfd pfd[fdcount]; // make new pollfd array + std::map::iterator it = notifiers.begin(); + int i=0; + for (; i < nativecount; ++i, ++it) + { + pfd[i].fd = it->first; + pfd[i].events = it->second->getRequested(); + } + + if (additional) + { + 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); } - - ret = poll(pfd, fdcount, poll_timeout); - + } + + ret = ::poll(pfd, fdcount, poll_timeout); /* ret > 0 means that there are some active poll entries. */ - if (ret > 0) + if (ret > 0) + { + int i=0; + return_reason = 0; + for (; i < nativecount ; ++i) { - return_reason = 0; - for (int i=0; i < fdcount ; i++) + if (pfd[i].revents) { - it = notifiers.begin(); - int handled = 0; - - std::multimap::iterator - l = notifiers.lower_bound(pfd[i].fd), - u = notifiers.upper_bound(pfd[i].fd); - - ePtrList n; - - for (; l != u; ++l) - n.push_back(l->second); - - for (ePtrList::iterator li(n.begin()); li != n.end(); ++li) + it = notifiers.find(pfd[i].fd); + if (it != notifiers.end()) { - int req = li->getRequested(); - + int req = it->second->getRequested(); handled |= req; - if (pfd[i].revents & req) - (*li)->activate(pfd[i].revents); + it->second->activate(pfd[i].revents); } - if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL)) - eFatal("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, 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); } - - ret = 1; /* poll did not timeout. */ - } else if (ret < 0) + } + for (; i < fdcount; ++i) { - /* when we got a signal, we get EINTR. */ - if (errno != EINTR) - eDebug("poll made error (%m)"); - else + if (pfd[i].revents) { - return_reason = 2; - ret = -1; /* don't assume the timeout has passed when we got a signal */ + if (!*res) + *res = PyList_New(0); + 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); + Py_DECREF(it); } } - delete [] pfd; + + ret = 1; /* poll did not timeout. */ + } else if (ret < 0) + { + /* when we got a signal, we get EINTR. */ + if (errno != EINTR) + eDebug("poll made error (%m)"); + else + { + return_reason = 2; + ret = -1; /* don't assume the timeout has passed when we got a signal */ + } } /* when we not processed anything, check timers. */ - if (!ret) + if (!m_timer_list.empty()) { /* we know that this time has passed. */ - now += poll_timeout; - singleLock s(recalcLock); - /* this will never change while we have the recalcLock */ - /* we can savely return here, the timer will be re-checked soon. */ - if (m_now_is_invalid) - return 0; + if (ret || m_now_is_invalid) + gettimeofday(&now, 0); + else + now += poll_timeout; /* process all timers which are ready. first remove them out of the list. */ while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() <= now)) @@ -262,14 +272,37 @@ void eMainloop::removeTimer(eTimer* e) m_timer_list.remove(e); } -int eMainloop::iterate(unsigned int user_timeout) +int eMainloop::iterate(unsigned int user_timeout, PyObject **res, ePyObject dict) { int ret = 0; + timeval user_timer; + gettimeofday(&user_timer, 0); + user_timer += user_timeout; + + /* TODO: this code just became ugly. fix that. */ do - { - if (app_quit_now) break; - ret = processOneEvent(user_timeout); + { + if (m_interrupt_requested) + { + m_interrupt_requested = 0; + return 0; + } + if (app_quit_now) return -1; + timeval now, timeout; + gettimeofday(&now, 0); + timeout = user_timer - now; + + if (user_timeout && (user_timer <= now)) + return 0; + + int to = 0; + if (user_timeout) + to = timeout.tv_sec * 1000 + timeout.tv_usec / 1000; + + ret = processOneEvent(to, res, dict); + if (res && *res) + return ret; } while (ret == 0); return ret; @@ -282,6 +315,37 @@ int eMainloop::runLoop() return retval; } +void eMainloop::reset() +{ + app_quit_now=false; +} + +PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict) +{ + PyObject *res=0; + + if (app_quit_now) + { + Py_INCREF(Py_None); + return Py_None; + } + + int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout); + + iterate(user_timeout, &res, dict); + ePyObject ret(res); + + if (!ret) /* return empty list on timeout */ + return PyList_New(0); + + return ret; +} + +void eMainloop::interruptPoll() +{ + m_interrupt_requested = 1; +} + void eMainloop::quit(int ret) { retval = ret;