add slide_model.rs

This file will eventually be the model of slides to be used in the app.
This commit is contained in:
Chris Cochrun 2023-03-24 16:36:05 -05:00
parent f17a88ff68
commit ec95ba0d89

193
src/rust/slide_model.rs Normal file
View file

@ -0,0 +1,193 @@
#[cxx_qt::bridge(cxx_file_stem = "custom_base_class")]
mod ffi {
unsafe extern "C++" {
include!(< QAbstractListModel >);
include!("cxx-qt-lib/qhash.h");
type QHash_i32_QByteArray = cxx_qt_lib::QHash<cxx_qt_lib::QHashPair_i32_QByteArray>;
include!("cxx-qt-lib/qvariant.h");
type QVariant = cxx_qt_lib::QVariant;
include!("cxx-qt-lib/qmodelindex.h");
type QModelIndex = cxx_qt_lib::QModelIndex;
include!("cxx-qt-lib/qvector.h");
type QVector_i32 = cxx_qt_lib::QVector<i32>;
}
#[cxx_qt::qobject(
base = "QAbstractListModel",
qml_uri = "com.kdab.cxx_qt.demo",
qml_version = "1.0"
)]
#[derive(Default)]
pub struct SlideModel {
id: i32,
vector: Vec<(i32, f64)>,
}
#[cxx_qt::qsignals(SlideModel)]
pub enum Signals<'a> {
#[inherit]
DataChanged {
top_left: &'a QModelIndex,
bottom_right: &'a QModelIndex,
roles: &'a QVector_i32,
},
}
impl qobject::SlideModel {
#[qinvokable]
pub fn add(self: Pin<&mut Self>) {
self.add_cpp_context();
}
#[qinvokable]
pub fn add_on_thread(self: Pin<&mut Self>, mut counter: i32) {
let qt_thread = self.qt_thread();
std::thread::spawn(move || {
while counter > 0 {
counter -= 1;
std::thread::sleep(std::time::Duration::from_millis(250));
// Use our add helper to add a row on the Qt event loop
// as seen in the threading demo channels could be used to pass info
qt_thread
.queue(|custom_base_class| {
custom_base_class.add_cpp_context();
})
.unwrap();
}
});
}
fn add_cpp_context(mut self: Pin<&mut Self>) {
let count = self.vector().len();
unsafe {
self.as_mut().begin_insert_rows(
&QModelIndex::default(),
count as i32,
count as i32,
);
let id = *self.id();
self.as_mut().set_id(id + 1);
self.as_mut().vector_mut().push((id, (id as f64) / 3.0));
self.as_mut().end_insert_rows();
}
}
#[qinvokable]
pub fn clear(mut self: Pin<&mut Self>) {
unsafe {
self.as_mut().begin_reset_model();
self.as_mut().set_id(0);
self.as_mut().vector_mut().clear();
self.as_mut().end_reset_model();
}
}
#[qinvokable]
pub fn multiply(mut self: Pin<&mut Self>, index: i32, factor: f64) {
if let Some((_, value)) = self.as_mut().vector_mut().get_mut(index as usize) {
*value *= factor;
// Emit dataChanged for the index and value role
let model_index = self.index(index, 0, &QModelIndex::default());
let mut vector_roles = QVector_i32::default();
vector_roles.append(1);
self.as_mut().emit(Signals::DataChanged {
top_left: &model_index,
bottom_right: &model_index,
roles: &vector_roles,
});
}
}
#[qinvokable]
pub fn remove(mut self: Pin<&mut Self>, index: i32) {
if index < 0 || (index as usize) >= self.vector().len() {
return;
}
unsafe {
self.as_mut()
.begin_remove_rows(&QModelIndex::default(), index, index);
self.as_mut().vector_mut().remove(index as usize);
self.as_mut().end_remove_rows();
}
}
}
// Create Rust bindings for C++ functions of the base class (QAbstractItemModel)
#[cxx_qt::inherit]
extern "C++" {
unsafe fn begin_insert_rows(
self: Pin<&mut qobject::SlideModel>,
parent: &QModelIndex,
first: i32,
last: i32,
);
unsafe fn end_insert_rows(self: Pin<&mut qobject::SlideModel>);
unsafe fn begin_remove_rows(
self: Pin<&mut qobject::SlideModel>,
parent: &QModelIndex,
first: i32,
last: i32,
);
unsafe fn end_remove_rows(self: Pin<&mut qobject::SlideModel>);
unsafe fn begin_reset_model(self: Pin<&mut qobject::SlideModel>);
unsafe fn end_reset_model(self: Pin<&mut qobject::SlideModel>);
}
#[cxx_qt::inherit]
unsafe extern "C++" {
#[cxx_name = "canFetchMore"]
fn base_can_fetch_more(self: &qobject::SlideModel, parent: &QModelIndex) -> bool;
fn index(
self: &qobject::SlideModel,
row: i32,
column: i32,
parent: &QModelIndex,
) -> QModelIndex;
}
// QAbstractListModel implementation
impl qobject::SlideModel {
#[qinvokable(cxx_override)]
fn data(&self, index: &QModelIndex, role: i32) -> QVariant {
if let Some((id, value)) = self.rust().vector.get(index.row() as usize) {
return match role {
0 => QVariant::from(id),
1 => QVariant::from(value),
_ => QVariant::default(),
};
}
QVariant::default()
}
// Example of overriding a C++ virtual method and calling the base class implementation.
#[qinvokable(cxx_override)]
pub fn can_fetch_more(&self, parent: &QModelIndex) -> bool {
self.base_can_fetch_more(parent)
}
#[qinvokable(cxx_override)]
pub fn role_names(&self) -> QHash_i32_QByteArray {
let mut roles = QHash_i32_QByteArray::default();
roles.insert(0, cxx_qt_lib::QByteArray::from("id"));
roles.insert(1, cxx_qt_lib::QByteArray::from("value"));
roles
}
#[qinvokable(cxx_override)]
pub fn row_count(&self, _parent: &QModelIndex) -> i32 {
self.rust().vector.len() as i32
}
}
}