blob: efecba0e1d57f36d5ceb7c9e8e8be1ac317ce5cf [file] [log] [blame]
// Copyright 2016 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "base/message_pump_ui_starboard.h"
#include "base/logging.h"
#include "base/run_loop.h"
#include "base/time.h"
#include "nb/memory_scope.h"
#include "starboard/event.h"
#include "starboard/system.h"
namespace base {
namespace {
void CallMessagePumpImmediate(void* context) {
DCHECK(context);
MessagePumpUIStarboard* pump =
reinterpret_cast<MessagePumpUIStarboard*>(context);
pump->RunOneAndReschedule(false /*delayed*/);
}
void CallMessagePumpDelayed(void* context) {
DCHECK(context);
MessagePumpUIStarboard* pump =
reinterpret_cast<MessagePumpUIStarboard*>(context);
pump->RunOneAndReschedule(true /*delayed*/);
}
} // namespace
MessagePumpUIStarboard::MessagePumpUIStarboard() : delegate_(NULL) {}
void MessagePumpUIStarboard::RunOneAndReschedule(bool delayed) {
DCHECK(delegate_);
if (delayed) {
CancelDelayed();
} else {
CancelImmediate();
}
TimeTicks delayed_work_time;
for (;;) {
TimeTicks next_time;
if (!RunOne(&next_time)) {
delayed_work_time = next_time;
break;
}
}
if (!delayed_work_time.is_null()) {
ScheduleDelayedWork(delayed_work_time);
}
}
void MessagePumpUIStarboard::Run(Delegate* delegate) {
// This should never be called because we are not like a normal message pump
// where we loop until told to quit. We are providing a MessagePump interface
// on top of an externally-owned message pump. We want to exist and be able to
// schedule work, but the actual for(;;) loop is owned by Starboard.
NOTREACHED();
}
void MessagePumpUIStarboard::Start(Delegate* delegate) {
run_loop_.reset(new base::RunLoop());
delegate_ = delegate;
// Since the RunLoop was just created above, BeforeRun should be guaranteed to
// return true (it only returns false if the RunLoop has been Quit already).
if (!run_loop_->BeforeRun())
NOTREACHED();
}
void MessagePumpUIStarboard::Quit() {
delegate_ = NULL;
CancelAll();
if (run_loop_) {
run_loop_->AfterRun();
run_loop_.reset();
}
}
void MessagePumpUIStarboard::ScheduleWork() {
base::AutoLock auto_lock(outstanding_events_lock_);
if (!outstanding_events_.empty()) {
// No need, already an outstanding event.
return;
}
outstanding_events_.insert(
SbEventSchedule(&CallMessagePumpImmediate, this, 0));
}
void MessagePumpUIStarboard::ScheduleDelayedWork(
const base::TimeTicks& delayed_work_time) {
TRACK_MEMORY_SCOPE("MessageLoop");
base::TimeDelta delay;
if (!delayed_work_time.is_null()) {
delay = delayed_work_time - base::TimeTicks::Now();
if (delay <= base::TimeDelta()) {
delay = base::TimeDelta();
}
}
{
base::AutoLock auto_lock(outstanding_events_lock_);
// Make sure any outstanding delayed event is canceled.
CancelDelayedLocked();
TRACK_MEMORY_SCOPE("MessageLoop");
outstanding_delayed_events_.insert(
SbEventSchedule(&CallMessagePumpDelayed, this, delay.ToSbTime()));
}
}
void MessagePumpUIStarboard::CancelAll() {
base::AutoLock auto_lock(outstanding_events_lock_);
CancelImmediateLocked();
CancelDelayedLocked();
}
void MessagePumpUIStarboard::CancelImmediate() {
base::AutoLock auto_lock(outstanding_events_lock_);
CancelImmediateLocked();
}
void MessagePumpUIStarboard::CancelDelayed() {
base::AutoLock auto_lock(outstanding_events_lock_);
CancelDelayedLocked();
}
void MessagePumpUIStarboard::CancelImmediateLocked() {
outstanding_events_lock_.AssertAcquired();
for (SbEventIdSet::iterator it = outstanding_events_.begin();
it != outstanding_events_.end(); ++it) {
SbEventCancel(*it);
}
outstanding_events_.erase(outstanding_events_.begin(),
outstanding_events_.end());
}
void MessagePumpUIStarboard::CancelDelayedLocked() {
TRACK_MEMORY_SCOPE("MessageLoop");
outstanding_events_lock_.AssertAcquired();
for (SbEventIdSet::iterator it = outstanding_delayed_events_.begin();
it != outstanding_delayed_events_.end(); ++it) {
SbEventCancel(*it);
}
outstanding_delayed_events_.erase(outstanding_delayed_events_.begin(),
outstanding_delayed_events_.end());
}
bool MessagePumpUIStarboard::RunOne(TimeTicks* out_delayed_work_time) {
DCHECK(out_delayed_work_time);
// We expect to start with a delegate, so we can DCHECK it, but any task we
// run could call Quit and remove it.
DCHECK(delegate_);
// Do immediate work.
bool did_work = delegate_->DoWork();
// Do all delayed work. Unlike Chromium, we drain all due delayed work before
// going back to the loop. See message_pump_io_starboard.cc for more
// information.
while (delegate_ && delegate_->DoDelayedWork(out_delayed_work_time)) {
did_work = true;
}
// If we did work, and we still have a delegate, return true, so we will be
// called again.
if (did_work) {
return !!delegate_;
}
// If the delegate has been removed, Quit() has been called, so no more work.
if (!delegate_) {
return false;
}
// No work was done, so only call back if there was idle work done, otherwise
// go to sleep. ScheduleWork or ScheduleDelayedWork will be called if new work
// is scheduled.
return delegate_->DoIdleWork();
}
} // namespace base