use crate::kurbo::{Rect, Size};
use crate::{
BoxConstraints, Data, Env, Event, EventCtx, LayoutCtx, LifeCycle, LifeCycleCtx, PaintCtx,
UpdateCtx, Widget, WidgetPod,
};
use crate::piet::UnitPoint;
pub struct Align<T> {
align: UnitPoint,
child: WidgetPod<T, Box<dyn Widget<T>>>,
width_factor: Option<f64>,
height_factor: Option<f64>,
}
impl<T> Align<T> {
pub fn new(align: UnitPoint, child: impl Widget<T> + 'static) -> Align<T> {
Align {
align,
child: WidgetPod::new(child).boxed(),
width_factor: None,
height_factor: None,
}
}
pub fn centered(child: impl Widget<T> + 'static) -> Align<T> {
Align::new(UnitPoint::CENTER, child)
}
pub fn right(child: impl Widget<T> + 'static) -> Align<T> {
Align::new(UnitPoint::RIGHT, child)
}
pub fn left(child: impl Widget<T> + 'static) -> Align<T> {
Align::new(UnitPoint::LEFT, child)
}
pub fn horizontal(align: UnitPoint, child: impl Widget<T> + 'static) -> Align<T> {
Align {
align,
child: WidgetPod::new(child).boxed(),
width_factor: None,
height_factor: Some(1.0),
}
}
pub fn vertical(align: UnitPoint, child: impl Widget<T> + 'static) -> Align<T> {
Align {
align,
child: WidgetPod::new(child).boxed(),
width_factor: Some(1.0),
height_factor: None,
}
}
}
impl<T: Data> Widget<T> for Align<T> {
fn event(&mut self, ctx: &mut EventCtx, event: &Event, data: &mut T, env: &Env) {
self.child.event(ctx, event, data, env)
}
fn lifecycle(&mut self, ctx: &mut LifeCycleCtx, event: &LifeCycle, data: &T, env: &Env) {
self.child.lifecycle(ctx, event, data, env)
}
fn update(&mut self, ctx: &mut UpdateCtx, _old_data: &T, data: &T, env: &Env) {
self.child.update(ctx, data, env);
}
fn layout(
&mut self,
layout_ctx: &mut LayoutCtx,
bc: &BoxConstraints,
data: &T,
env: &Env,
) -> Size {
bc.debug_check("Align");
let size = self.child.layout(layout_ctx, &bc.loosen(), data, env);
log_size_warnings(size);
let mut my_size = size;
if bc.is_width_bounded() {
my_size.width = bc.max().width;
}
if bc.is_height_bounded() {
my_size.height = bc.max().height;
}
if let Some(width) = self.width_factor {
my_size.width = size.width * width;
}
if let Some(height) = self.height_factor {
my_size.height = size.height * height;
}
my_size = bc.constrain(my_size);
let extra_width = (my_size.width - size.width).max(0.);
let extra_height = (my_size.height - size.height).max(0.);
let origin = self
.align
.resolve(Rect::new(0., 0., extra_width, extra_height));
self.child
.set_layout_rect(Rect::from_origin_size(origin, size));
let my_insets = self.child.compute_parent_paint_insets(my_size);
layout_ctx.set_paint_insets(my_insets);
my_size
}
fn paint(&mut self, ctx: &mut PaintCtx, data: &T, env: &Env) {
self.child.paint_with_offset(ctx, data, env);
}
}
fn log_size_warnings(size: Size) {
if size.width.is_infinite() {
log::warn!("Align widget's child has an infinite width.");
}
if size.height.is_infinite() {
log::warn!("Align widget's child has an infinite height.");
}
}