use crate::class::ApplyClass;
use bevy_ui::{RepeatedGridTrack, Style};
use derive_more::From;
#[derive(Debug, Clone, From, PartialEq)]
pub struct GridTemplateColumns(pub RepeatedGridTrack);
pub fn grid_cols(repeat: u16) -> GridTemplateColumns {
GridTemplateColumns(RepeatedGridTrack::flex(repeat, 1.))
}
impl ApplyClass<GridTemplateColumns> for Style {
#[inline]
fn apply_class(&mut self, class: &GridTemplateColumns) {
if !self.grid_template_columns.is_empty() {
self.grid_template_columns.clear();
}
self.grid_template_columns.push(class.0.clone());
}
}