Skip to content

Commit

Permalink
TreeSink impls do not explicitly rely on and methods anymore
Browse files Browse the repository at this point in the history
  • Loading branch information
nikhilshagri committed Aug 18, 2017
1 parent 99235c9 commit 2cc7199
Show file tree
Hide file tree
Showing 2 changed files with 129 additions and 88 deletions.
194 changes: 107 additions & 87 deletions components/script/dom/servoparser/async_html.rs
Expand Up @@ -69,6 +69,7 @@ enum ParseOperation {

CreateComment { text: String, node: ParseNodeId },
AppendBeforeSibling { sibling: ParseNodeId, node: NodeOrText },
AppendBasedOnParentNode { element: ParseNodeId, prev_element: ParseNodeId, node: NodeOrText },
Append { parent: ParseNodeId, node: NodeOrText },

AppendDoctypeToDocument {
Expand All @@ -81,7 +82,13 @@ enum ParseOperation {
RemoveFromParent { target: ParseNodeId },
MarkScriptAlreadyStarted { node: ParseNodeId },
ReparentChildren { parent: ParseNodeId, new_parent: ParseNodeId },
AssociateWithForm { target: ParseNodeId, form: ParseNodeId },

AssociateWithForm {
target: ParseNodeId,
form: ParseNodeId,
element: ParseNodeId,
prev_element: Option<ParseNodeId>
},

CreatePI {
node: ParseNodeId,
Expand Down Expand Up @@ -113,8 +120,6 @@ enum ToTokenizerMsg {

// From Sink
ProcessOperation(ParseOperation),
IsSameTree(ParseNodeId, ParseNodeId),
HasParentNode(ParseNodeId),
}

#[derive(HeapSizeOf)]
Expand All @@ -127,14 +132,6 @@ enum ToHtmlTokenizerMsg {
SetPlainTextState,
}

// Responses to the queries asked by the the Sink to the Tokenizer,
// using the messages types in FromSinkMsg.
#[derive(HeapSizeOf)]
enum ToSinkMsg {
IsSameTree(bool),
HasParentNode(bool),
}

fn create_buffer_queue(mut buffers: VecDeque<SendTendril<UTF8>>) -> BufferQueue {
let mut buffer_queue = BufferQueue::new();
while let Some(st) = buffers.pop_front() {
Expand All @@ -155,17 +152,17 @@ fn create_buffer_queue(mut buffers: VecDeque<SendTendril<UTF8>>) -> BufferQueue
// then executes the received actions.
//
// _____________ _______________
// | | ToHtmlTokenizerMsg | |
// | |------------------------>| |
// | | | |
// | | ToTokenizerMsg | HtmlTokenizer |
// | |<------------------------| |
// | Tokenizer | | |
// | | ToTokenizerMsg | ________ |
// | | | |
// | | ToHtmlTokenizerMsg | |
// | |------------------------>| HtmlTokenizer |
// | | | |
// | Tokenizer | ToTokenizerMsg | |
// | |<------------------------| ________ |
// | | | | | |
// | | ToTokenizerMsg | | Sink | |
// | |<------------------------|---| | |
// | | | | Sink | |
// | | ToSinkMsg | | | |
// | |-------------------------|-->|________| |
// | | | |________| |
// |_____________| |_______________|
//
#[derive(HeapSizeOf, JSTraceable)]
Expand All @@ -177,7 +174,6 @@ pub struct Tokenizer {
#[ignore_heap_size_of = "Defined in std"]
html_tokenizer_sender: Sender<ToHtmlTokenizerMsg>,
#[ignore_heap_size_of = "Defined in std"]
sink_sender: Sender<ToSinkMsg>,
nodes: HashMap<ParseNodeId, JS<Node>>,
url: ServoUrl,
}
Expand All @@ -190,22 +186,19 @@ impl Tokenizer {
-> Self {
// Messages from the Tokenizer (main thread) to HtmlTokenizer (parser thread)
let (to_html_tokenizer_sender, html_tokenizer_receiver) = channel();
// Messages from the Tokenizer (main thread) to Sink (parser thread)
let (to_sink_sender, sink_receiver) = channel();
// Messages from HtmlTokenizer and Sink (parser thread) to Tokenizer (main thread)
let (to_tokenizer_sender, tokenizer_receiver) = channel();

let mut tokenizer = Tokenizer {
document: JS::from_ref(document),
receiver: tokenizer_receiver,
html_tokenizer_sender: to_html_tokenizer_sender,
sink_sender: to_sink_sender,
nodes: HashMap::new(),
url: url
};
tokenizer.insert_node(0, JS::from_ref(document.upcast()));

let mut sink = Sink::new(to_tokenizer_sender.clone(), sink_receiver);
let mut sink = Sink::new(to_tokenizer_sender.clone());
let mut ctxt_parse_node = None;
let mut form_parse_node = None;
let mut fragment_context_is_some = false;
Expand Down Expand Up @@ -250,18 +243,6 @@ impl Tokenizer {
loop {
match self.receiver.recv().expect("Unexpected channel panic in main thread.") {
ToTokenizerMsg::ProcessOperation(parse_op) => self.process_operation(parse_op),
ToTokenizerMsg::IsSameTree(ref x_id, ref y_id) => {
let x = self.get_node(x_id);
let y = self.get_node(y_id);

let x = x.downcast::<Element>().expect("Element node expected");
let y = y.downcast::<Element>().expect("Element node expected");
self.sink_sender.send(ToSinkMsg::IsSameTree(x.is_in_same_home_subtree(y))).unwrap();
},
ToTokenizerMsg::HasParentNode(ref id) => {
let res = self.get_node(id).GetParentNode().is_some();
self.sink_sender.send(ToSinkMsg::HasParentNode(res)).unwrap();
},
ToTokenizerMsg::TokenizerResultDone { updated_input } => {
let buffer_queue = create_buffer_queue(updated_input);
*input = buffer_queue;
Expand All @@ -283,18 +264,6 @@ impl Tokenizer {
loop {
match self.receiver.recv().expect("Unexpected channel panic in main thread.") {
ToTokenizerMsg::ProcessOperation(parse_op) => self.process_operation(parse_op),
ToTokenizerMsg::IsSameTree(ref x_id, ref y_id) => {
let x = self.get_node(x_id);
let y = self.get_node(y_id);

let x = x.downcast::<Element>().expect("Element node expected");
let y = y.downcast::<Element>().expect("Element node expected");
self.sink_sender.send(ToSinkMsg::IsSameTree(x.is_in_same_home_subtree(y))).unwrap();
},
ToTokenizerMsg::HasParentNode(ref id) => {
let res = self.get_node(id).GetParentNode().is_some();
self.sink_sender.send(ToSinkMsg::HasParentNode(res)).unwrap();
},
ToTokenizerMsg::End => return,
_ => unreachable!(),
};
Expand All @@ -317,6 +286,45 @@ impl Tokenizer {
self.nodes.get(id).expect("Node not found!")
}


fn append_before_sibling(&mut self, sibling: ParseNodeId, node: NodeOrText) {
let node = match node {
NodeOrText::Node(n) => HtmlNodeOrText::AppendNode(JS::from_ref(&**self.get_node(&n.id))),
NodeOrText::Text(text) => HtmlNodeOrText::AppendText(
Tendril::from(text)
)
};
let sibling = &**self.get_node(&sibling);
let parent = &*sibling.GetParentNode().expect("append_before_sibling called on node without parent");

super::insert(parent, Some(sibling), node);
}

fn append(&mut self, parent: ParseNodeId, node: NodeOrText) {
let node = match node {
NodeOrText::Node(n) => HtmlNodeOrText::AppendNode(JS::from_ref(&**self.get_node(&n.id))),
NodeOrText::Text(text) => HtmlNodeOrText::AppendText(
Tendril::from(text)
)
};

let parent = &**self.get_node(&parent);
super::insert(parent, None, node);
}

fn has_parent_node(&self, node: ParseNodeId) -> bool {
self.get_node(&node).GetParentNode().is_some()
}

fn same_tree(&self, x: ParseNodeId, y: ParseNodeId) -> bool {
let x = self.get_node(&x);
let y = self.get_node(&y);

let x = x.downcast::<Element>().expect("Element node expected");
let y = y.downcast::<Element>().expect("Element node expected");
x.is_in_same_home_subtree(y)
}

fn process_operation(&mut self, op: ParseOperation) {
let document = Root::from_ref(&**self.get_node(&0));
let document = document.downcast::<Document>().expect("Document node should be downcasted!");
Expand Down Expand Up @@ -348,27 +356,17 @@ impl Tokenizer {
self.insert_node(node, JS::from_ref(&comment.upcast()));
}
ParseOperation::AppendBeforeSibling { sibling, node } => {
let node = match node {
NodeOrText::Node(n) => HtmlNodeOrText::AppendNode(JS::from_ref(&**self.get_node(&n.id))),
NodeOrText::Text(text) => HtmlNodeOrText::AppendText(
Tendril::from(text)
)
};
let sibling = &**self.get_node(&sibling);
let parent = &*sibling.GetParentNode().expect("append_before_sibling called on node without parent");

super::insert(parent, Some(sibling), node);
self.append_before_sibling(sibling, node);
}
ParseOperation::Append { parent, node } => {
let node = match node {
NodeOrText::Node(n) => HtmlNodeOrText::AppendNode(JS::from_ref(&**self.get_node(&n.id))),
NodeOrText::Text(text) => HtmlNodeOrText::AppendText(
Tendril::from(text)
)
};

let parent = &**self.get_node(&parent);
super::insert(parent, None, node);
self.append(parent, node);
}
ParseOperation::AppendBasedOnParentNode { element, prev_element, node } => {
if self.has_parent_node(element) {
self.append_before_sibling(element, node);
} else {
self.append(prev_element, node);
}
}
ParseOperation::AppendDoctypeToDocument { name, public_id, system_id } => {
let doctype = DocumentType::new(
Expand Down Expand Up @@ -400,7 +398,14 @@ impl Tokenizer {
new_parent.AppendChild(&child).unwrap();
}
}
ParseOperation::AssociateWithForm { target, form } => {
ParseOperation::AssociateWithForm { target, form, element, prev_element } => {
let tree_node = prev_element.map_or(element, |prev| {
if self.has_parent_node(element) { element } else { prev }
});

if !self.same_tree(tree_node, form) {
return;
}
let form = self.get_node(&form);
let form = Root::downcast::<HTMLFormElement>(Root::from_ref(&**form))
.expect("Owner must be a form element");
Expand Down Expand Up @@ -502,11 +507,10 @@ pub struct Sink {
next_parse_node_id: Cell<ParseNodeId>,
document_node: ParseNode,
sender: Sender<ToTokenizerMsg>,
receiver: Receiver<ToSinkMsg>,
}

impl Sink {
fn new(sender: Sender<ToTokenizerMsg>, receiver: Receiver<ToSinkMsg>) -> Sink {
fn new(sender: Sender<ToTokenizerMsg>) -> Sink {
let mut sink = Sink {
current_line: 1,
parse_node_data: HashMap::new(),
Expand All @@ -515,8 +519,7 @@ impl Sink {
id: 0,
qual_name: None,
},
sender: sender,
receiver: receiver,
sender: sender
};
let data = ParseNodeData::default();
sink.insert_parse_node_data(0, data);
Expand Down Expand Up @@ -583,12 +586,8 @@ impl TreeSink for Sink {
target.qual_name.as_ref().expect("Expected qual name of node!").expanded()
}

fn same_tree(&self, x: &Self::Handle, y: &Self::Handle) -> bool {
self.sender.send(ToTokenizerMsg::IsSameTree(x.id, y.id)).unwrap();
match self.receiver.recv().expect("Unexpected channel panic in html parser thread.") {
ToSinkMsg::IsSameTree(result) => result,
_ => unreachable!(),
}
fn same_tree(&self, _: &Self::Handle, _: &Self::Handle) -> bool {
unreachable!();
}

fn create_element(&mut self, name: QualName, html_attrs: Vec<HtmlAttribute>, _flags: ElementFlags)
Expand Down Expand Up @@ -633,18 +632,22 @@ impl TreeSink for Sink {
node
}

fn has_parent_node(&self, node: &Self::Handle) -> bool {
self.sender.send(ToTokenizerMsg::HasParentNode(node.id)).unwrap();
match self.receiver.recv().expect("Unexpected channel panic in html parser thread.") {
ToSinkMsg::HasParentNode(result) => result,
_ => unreachable!(),
}
fn has_parent_node(&self, _: &Self::Handle) -> bool {
unreachable!();
}

fn associate_with_form(&mut self, target: &Self::Handle, form: &Self::Handle) {
fn associate_with_form(
&mut self,
target: &Self::Handle,
form: &Self::Handle,
nodes: (&Self::Handle, Option<&Self::Handle>),
) {
let (element, prev_element) = nodes;
self.send_op(ParseOperation::AssociateWithForm {
target: target.id,
form: form.id
form: form.id,
element: element.id,
prev_element: prev_element.map(|p| p.id),
});
}

Expand All @@ -658,6 +661,23 @@ impl TreeSink for Sink {
self.send_op(ParseOperation::AppendBeforeSibling { sibling: sibling.id, node: new_node });
}

fn append_based_on_parent_node(
&mut self,
elem: &Self::Handle,
prev_elem: &Self::Handle,
child: HtmlNodeOrText<Self::Handle>,
) {
let child = match child {
HtmlNodeOrText::AppendNode(node) => NodeOrText::Node(node),
HtmlNodeOrText::AppendText(text) => NodeOrText::Text(String::from(text))
};
self.send_op(ParseOperation::AppendBasedOnParentNode {
element: elem.id,
prev_element: prev_elem.id,
node: child
});
}

fn parse_error(&mut self, msg: Cow<'static, str>) {
debug!("Parse error: {}", msg);
}
Expand Down
23 changes: 22 additions & 1 deletion components/script/dom/servoparser/mod.rs
Expand Up @@ -820,7 +820,15 @@ impl TreeSink for Sink {
node.GetParentNode().is_some()
}

fn associate_with_form(&mut self, target: &JS<Node>, form: &JS<Node>) {
fn associate_with_form(&mut self, target: &JS<Node>, form: &JS<Node>, nodes: (&JS<Node>, Option<&JS<Node>>)) {
let (element, prev_element) = nodes;
let tree_node = prev_element.map_or(element, |prev| {
if self.has_parent_node(element) { element } else { prev }
});
if !self.same_tree(tree_node, form) {
return;
}

let node = target;
let form = Root::downcast::<HTMLFormElement>(Root::from_ref(&**form))
.expect("Owner must be a form element");
Expand Down Expand Up @@ -862,6 +870,19 @@ impl TreeSink for Sink {
insert(&parent, None, child);
}

fn append_based_on_parent_node(
&mut self,
elem: &JS<Node>,
prev_elem: &JS<Node>,
child: NodeOrText<JS<Node>>,
) {
if self.has_parent_node(elem) {
self.append_before_sibling(elem, child);
} else {
self.append(prev_elem, child);
}
}

fn append_doctype_to_document(&mut self, name: StrTendril, public_id: StrTendril,
system_id: StrTendril) {
let doc = &*self.document;
Expand Down

0 comments on commit 2cc7199

Please sign in to comment.