use re_data_store::LatestAtQuery;
use re_query::{query_archetype, QueryError};
use re_types::{
archetypes::{self, TextDocument},
components,
};
use re_viewer_context::{
IdentifiedViewSystem, SpaceViewSystemExecutionError, ViewContextCollection, ViewQuery,
ViewerContext, VisualizerQueryInfo, VisualizerSystem,
};
#[derive(Debug, Clone)]
pub struct TextDocumentEntry {
pub body: components::Text,
pub media_type: components::MediaType,
}
#[derive(Default)]
pub struct TextDocumentSystem {
pub text_entries: Vec<TextDocumentEntry>,
}
impl IdentifiedViewSystem for TextDocumentSystem {
fn identifier() -> re_viewer_context::ViewSystemIdentifier {
"TextDocument".into()
}
}
impl VisualizerSystem for TextDocumentSystem {
fn visualizer_query_info(&self) -> VisualizerQueryInfo {
VisualizerQueryInfo::from_archetype::<TextDocument>()
}
fn execute(
&mut self,
ctx: &ViewerContext<'_>,
query: &ViewQuery<'_>,
_view_ctx: &ViewContextCollection,
) -> Result<Vec<re_renderer::QueueableDrawData>, SpaceViewSystemExecutionError> {
let store = ctx.recording_store();
let timeline_query = LatestAtQuery::new(query.timeline, query.latest_at);
for data_result in query.iter_visible_data_results(ctx, Self::identifier()) {
match query_archetype::<archetypes::TextDocument>(
store,
&timeline_query,
&data_result.entity_path,
) {
Ok(arch_view) => {
let bodies = arch_view.iter_required_component::<components::Text>()?;
let media_types =
arch_view.iter_optional_component::<components::MediaType>()?;
for (body, media_type) in itertools::izip!(bodies, media_types) {
let media_type = media_type.unwrap_or(components::MediaType::plain_text());
self.text_entries
.push(TextDocumentEntry { body, media_type });
}
}
Err(QueryError::PrimaryNotFound(_)) => {}
Err(err) => {
re_log::error_once!(
"Unexpected error querying {:?}: {err}",
&data_result.entity_path
);
}
};
}
Ok(Vec::new())
}
fn as_any(&self) -> &dyn std::any::Any {
self
}
}