use super::proxy_cpp_bridge::{QTableModelProxyCpp, ffi};
use crate::{RustObjAccess, call_rust_trait_impl, call_cpp_impl};
use qtbridge_runtime::qrustproxy::{QRustProxy, ConstructionMode};
use qtbridge_runtime::QObjectHolder;
use qtbridge_runtime::QModelItem;
use qtbridge_type_lib::{QByteArray, QHash, QMetaObject, QMetaType, QModelIndex, QVariant};
use std::cell::RefCell;
use std::rc::Rc;
#[doc(hidden)]
pub trait QTableModelAdapter {
fn index(&self, row: i32, column: i32, parent: &QModelIndex) -> QModelIndex;
fn parent(&self, child: &QModelIndex) -> QModelIndex;
fn row_count(&self, parent: &QModelIndex) -> i32;
fn column_count(&self, parent: &QModelIndex) -> i32;
fn data(&self, index: &QModelIndex, role: i32) -> QVariant;
fn role_names(&self) -> QHash<i32, QByteArray>;
fn set_data(&mut self, index: &QModelIndex, value: &QVariant, role: i32) -> bool;
fn remove_columns(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool;
fn remove_rows(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool;
fn sibling(&self, row: i32, column: i32, idx: &QModelIndex) -> QModelIndex;
}
impl<T> QTableModelAdapter for T
where
T: QTableModel + QObjectHolder<ProxyRust = QTableModelProxyRust> {
fn index(&self, row: i32, column: i32, _: &QModelIndex) -> QModelIndex {
let proxy = <Self as QObjectHolder>::get_rust_proxy(self);
proxy.base_create_index(row, column, 0)
}
fn parent(&self, _: &QModelIndex) -> QModelIndex {
QModelIndex::default()
}
fn row_count(&self, _: &QModelIndex) -> i32 {
self.row_count() as i32
}
fn column_count(&self, _: &QModelIndex) -> i32 {
<Self as QTableModel>::column_count(&self) as i32
}
fn data(&self, index: &QModelIndex, role: i32) -> QVariant {
let Some(item) = self.get((index.row() as usize, index.column() as usize))
else {
return QVariant::default();
};
item.get_role(role)
}
fn role_names(&self) -> QHash<i32, QByteArray> {
let names = T::Item::role_names();
let mut result = QHash::default();
names.iter()
.for_each(|(k, v)| result.insert(k, &QByteArray::from(v)));
result
}
fn set_data(&mut self, index: &QModelIndex, value: &QVariant, role: i32) -> bool {
if !index.is_valid() {
return false;
}
let Some(mut item) = self.get((index.row() as usize, index.column() as usize))
.cloned()
else {
return false;
};
let updated = item.set_role(role, value);
if updated {
self.set_unnotified((index.row() as usize, index.column() as usize), item);
self.get_rust_proxy_mut().base_data_changed(index, index);
}
updated
}
fn remove_columns(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
let first = first as usize;
let last = first + count as usize;
if last > self.column_count() {
return false;
}
self.get_rust_proxy_mut().base_begin_remove_columns(parent, first as i32, (last - 1) as i32);
for index in (first..last).rev() {
self.remove_row_unnotified(index);
}
self.get_rust_proxy_mut().base_end_remove_columns();
true
}
fn remove_rows(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
let first = first as usize;
let last = first + count as usize;
if last > self.row_count() {
return false;
}
self.get_rust_proxy_mut().base_begin_remove_rows(parent, first as i32, (last - 1) as i32);
for index in (first..last).rev() {
self.remove_row_unnotified(index);
}
self.get_rust_proxy_mut().base_end_remove_rows();
true
}
fn sibling(&self, row: i32, column: i32, idx: &QModelIndex) -> QModelIndex {
let proxy = self.get_rust_proxy();
proxy.base_sibling(row, column, idx)
}
}
pub trait QTableModel {
type Item: QModelItem + Default + Clone;
fn row_count(&self) -> usize;
fn column_count(&self) -> usize;
fn get(&self, index: (usize, usize)) -> Option<&Self::Item>;
fn set_unnotified(&mut self, _index: (usize, usize), _value: Self::Item) -> bool {
false
}
fn push_row_unnotified(&mut self, values: &[Self::Item]) {
self.insert_row_unnotified(self.row_count(), values);
}
fn push_column_unnotified(&mut self, values: &[Self::Item]) {
self.insert_column_unnotified(self.column_count(), values);
}
fn insert_row_unnotified(&mut self, _index: usize, _value: &[Self::Item]) {
panic!("In order to use insert, implement insert_unnotified")
}
fn insert_column_unnotified(&mut self, _index: usize, _value: &[Self::Item]) {
panic!("In order to use insert, implement insert_unnotified")
}
fn pop_row_unnotified(&mut self) -> Option<Vec<Self::Item>> {
(self.row_count() > 0)
.then(|| self.remove_row_unnotified(self.row_count() - 1))
}
fn pop_column_unnotified(&mut self) -> Option<Vec<Self::Item>> {
(self.column_count() > 0)
.then(|| self.remove_column_unnotified(self.column_count() - 1))
}
fn remove_row_unnotified(&mut self, _index: usize) -> Vec<Self::Item> {
panic!("In order to use remove, implement remove_unnotified")
}
fn remove_column_unnotified(&mut self, _index: usize) -> Vec<Self::Item> {
panic!("In order to use remove, implement remove_unnotified")
}
fn reset_unnotified(&mut self) {
panic!("In order to use reset, implement reset_unnotified")
}
}
pub trait QTableModelBase : QTableModel + QObjectHolder<ProxyRust = QTableModelProxyRust> {
fn set(&mut self, index: (usize, usize), value: <Self as QTableModel>::Item) -> bool {
if self.set_unnotified(index, value) {
let model_index = self.get_rust_proxy().index(index.0 as i32, index.1 as i32 , &QModelIndex::default());
self.get_rust_proxy_mut().base_data_changed(&model_index, &model_index);
true
} else {
false
}
}
fn push_row(&mut self, values: &[Self::Item]) {
self.get_rust_proxy_mut().base_begin_insert_rows(&QModelIndex::default(), self.row_count() as i32, self.row_count() as i32);
self.push_row_unnotified(values);
self.get_rust_proxy_mut().base_end_insert_rows();
}
fn push_column(&mut self, values: &[Self::Item]) {
self.get_rust_proxy_mut().base_begin_insert_columns(&QModelIndex::default(), self.column_count() as i32, self.column_count() as i32);
self.push_column_unnotified(values);
self.get_rust_proxy_mut().base_end_insert_columns();
}
fn insert_row(&mut self, index: usize, values: &[Self::Item]) {
self.get_rust_proxy_mut().base_begin_insert_rows(&QModelIndex::default(), index as i32, index as i32);
self.insert_row_unnotified(index, values);
self.get_rust_proxy_mut().base_end_insert_rows();
}
fn insert_column(&mut self, index: usize, values: &[Self::Item]) {
self.get_rust_proxy_mut().base_begin_insert_columns(&QModelIndex::default(), index as i32, index as i32);
self.insert_column_unnotified(index, values);
self.get_rust_proxy_mut().base_end_insert_columns();
}
fn pop_row(&mut self) -> Option<Vec<Self::Item>> {
if self.row_count() == 0 {
return None;
}
self.get_rust_proxy_mut().base_begin_remove_rows(&QModelIndex::default(), self.row_count() as i32 - 1, self.row_count() as i32 - 1);
let values = self.pop_row_unnotified();
self.get_rust_proxy_mut().base_end_remove_rows();
values
}
fn pop_column(&mut self) -> Option<Vec<Self::Item>> {
if self.column_count() == 0 {
return None;
}
self.get_rust_proxy_mut().base_begin_remove_columns(&QModelIndex::default(), self.column_count() as i32 - 1, self.column_count() as i32 - 1);
let values = self.pop_column_unnotified();
self.get_rust_proxy_mut().base_end_remove_columns();
values
}
fn remove_row(&mut self, index: usize) -> Vec<Self::Item> {
self.get_rust_proxy_mut().base_begin_remove_rows(&QModelIndex::default(), index as i32, index as i32);
let values = self.remove_row_unnotified(index);
self.get_rust_proxy_mut().base_end_remove_rows();
values
}
fn remove_column(&mut self, index: usize) -> Vec<Self::Item> {
self.get_rust_proxy_mut().base_begin_remove_columns(&QModelIndex::default(), index as i32, index as i32);
let values = self.remove_column_unnotified(index);
self.get_rust_proxy_mut().base_end_remove_columns();
values
}
fn reset(&mut self) {
self.get_rust_proxy_mut().base_begin_reset_model();
self.reset_unnotified();
self.get_rust_proxy_mut().base_end_reset_model();
}
}
impl<T> QTableModelBase for T
where T: QTableModel + QObjectHolder<ProxyRust = QTableModelProxyRust> { }
pub struct QTableModelProxyRust {
cpp_proxy: *mut QTableModelProxyCpp,
#[allow(dead_code)]
rust_obj: RustObjAccess<dyn QTableModelAdapter>,
on_drop: fn(rust_obj: *const u8),
}
impl QRustProxy for QTableModelProxyRust {
type ProxyCppType = QTableModelProxyCpp;
type AdapterType = dyn QTableModelAdapter;
fn new(rust_obj: &Rc<RefCell<dyn QTableModelAdapter>>, construct: ConstructionMode, on_drop: fn(rust_obj: *const u8)) -> *mut Self {
let raw_rust_obj = rust_obj.as_ptr();
let boxed_self = Box::new(Self {
cpp_proxy: std::ptr::null_mut(),
rust_obj: match construct {
ConstructionMode::Strong | ConstructionMode::AtAddress(_) => RustObjAccess::new_strong(rust_obj.clone()),
ConstructionMode::Weak => RustObjAccess::new_weak(Rc::downgrade(rust_obj)),
},
on_drop,
});
let raw_self = Box::into_raw(boxed_self);
unsafe{ (*raw_self).cpp_proxy = match construct {
ConstructionMode::AtAddress(addr) => {
ffi::create_qtable_model_proxy_cpp_at( addr, raw_rust_obj.cast(), raw_self)
}
ConstructionMode::Strong | ConstructionMode::Weak => {
ffi::create_qtable_model_proxy_cpp(raw_rust_obj.cast(), raw_self)
}
}};
raw_self
}
fn drop_self(raw_self: *mut Self, rust_obj_ptr: *const u8) {
Self::drop_self_impl(raw_self, rust_obj_ptr)
}
fn get_static_meta_object() -> &'static QMetaObject {
ffi::static_qmeta_object_of_qtable_model_proxy_cpp()
}
fn get_size_of_cpp_proxy() -> usize {
ffi::size_of_qtable_model_proxy_cpp()
}
fn get_align_of_cpp_proxy() -> usize {
ffi::align_of_qtable_model_proxy_cpp()
}
fn get_qmetatype_list_of_cpp_proxy() -> QMetaType {
ffi::qmetatype_list_of_qtable_model_proxy_cpp()
}
fn get_cpp_proxy(&self) -> *const QTableModelProxyCpp {
self.cpp_proxy as *const _
}
fn get_cpp_proxy_mut(&self) -> *mut QTableModelProxyCpp {
self.cpp_proxy
}
}
impl QTableModelProxyRust {
pub fn drop_self_impl(raw_self: *mut Self, rust_obj_ptr: *const u8) {
let boxed_self = unsafe { Box::from_raw(raw_self) };
(boxed_self.on_drop)(rust_obj_ptr);
}
pub fn index(&self, row: i32, column: i32, parent: &QModelIndex) -> QModelIndex {
call_rust_trait_impl!(self, index(row, column, parent))
}
pub fn parent(&self, child: &QModelIndex) -> QModelIndex {
call_rust_trait_impl!(self, parent(child))
}
pub fn row_count(&self, parent: &QModelIndex) -> i32 {
call_rust_trait_impl!(self, row_count(parent))
}
pub fn column_count(&self, parent: &QModelIndex) -> i32 {
call_rust_trait_impl!(self, column_count(parent))
}
pub fn data(&self, index: &QModelIndex, role: i32) -> QVariant {
call_rust_trait_impl!(self, data(index, role))
}
pub fn role_names(&self) -> QHash<i32, QByteArray> {
call_rust_trait_impl!(self, role_names())
}
pub fn set_data(&mut self, index: &QModelIndex, value: &QVariant, role: i32) -> bool {
call_rust_trait_impl!(mut self, set_data(index, value, role))
}
pub fn remove_columns(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
call_rust_trait_impl!(mut self, remove_columns(first, count, parent))
}
pub fn remove_rows(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
call_rust_trait_impl!(mut self, remove_rows(first, count, parent))
}
pub fn sibling(&self, row: i32, column: i32, idx: &QModelIndex) -> QModelIndex {
call_rust_trait_impl!(self, sibling(row, column, idx))
}
pub fn base_role_names(&self) -> QHash<i32, QByteArray> {
call_cpp_impl!(self, base_role_names())
}
pub fn base_set_data(&mut self, index: &QModelIndex, value: &QVariant, role: i32) -> bool {
call_cpp_impl!(mut self, base_set_data(index, value, role))
}
pub fn base_remove_columns(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
call_cpp_impl!(mut self, base_remove_columns(first, count, parent))
}
pub fn base_remove_rows(&mut self, first: i32, count: i32, parent: &QModelIndex) -> bool {
call_cpp_impl!(mut self, base_remove_rows(first, count, parent))
}
pub fn base_sibling(&self, row: i32, column: i32, idx: &QModelIndex) -> QModelIndex {
call_cpp_impl!(self, base_sibling(row, column, idx))
}
pub fn base_data_changed(&mut self, top_left: &QModelIndex, bottom_right: &QModelIndex) {
call_cpp_impl!(mut self, base_data_changed(top_left, bottom_right))
}
pub fn base_begin_insert_columns(&mut self, parent: &QModelIndex, first: i32, last: i32) {
call_cpp_impl!(mut self, base_begin_insert_columns(parent, first, last))
}
pub fn base_end_insert_columns(&mut self) {
call_cpp_impl!(mut self, base_end_insert_columns())
}
pub fn base_begin_insert_rows(&mut self, parent: &QModelIndex, first: i32, last: i32) {
call_cpp_impl!(mut self, base_begin_insert_rows(parent, first, last))
}
pub fn base_end_insert_rows(&mut self) {
call_cpp_impl!(mut self, base_end_insert_rows())
}
pub fn base_begin_move_columns(
&mut self,
source_parent: &QModelIndex,
source_first: i32,
source_last: i32,
destination_parent: &QModelIndex,
destination_child: i32,
) {
call_cpp_impl!(mut self, base_begin_move_columns(source_parent, source_first, source_last, destination_parent, destination_child))
}
pub fn base_end_move_columns(&mut self) {
call_cpp_impl!(mut self, base_end_move_columns())
}
pub fn base_begin_move_rows(&mut self, source_parent: &QModelIndex, source_first: i32, source_last: i32, destination_parent: &QModelIndex, destination_child: i32) {
call_cpp_impl!(mut self, base_begin_move_rows(source_parent, source_first, source_last, destination_parent, destination_child))
}
pub fn base_end_move_rows(&mut self) {
call_cpp_impl!(mut self, base_end_move_rows())
}
pub fn base_begin_remove_columns(&mut self, parent: &QModelIndex, first: i32, last: i32) {
call_cpp_impl!(mut self, base_begin_remove_columns(parent, first, last))
}
pub fn base_end_remove_columns(&mut self) {
call_cpp_impl!(mut self, base_end_remove_columns())
}
pub fn base_begin_remove_rows(&mut self, parent: &QModelIndex, first: i32, last: i32) {
call_cpp_impl!(mut self, base_begin_remove_rows(parent, first, last))
}
pub fn base_end_remove_rows(&mut self) {
call_cpp_impl!(mut self, base_end_remove_rows())
}
pub fn base_begin_reset_model(&mut self) {
call_cpp_impl!(mut self, base_begin_reset_model())
}
pub fn base_end_reset_model(&mut self) {
call_cpp_impl!(mut self, base_end_reset_model())
}
pub fn base_create_index(&self, row: i32, column: i32, ptr: usize) -> QModelIndex {
call_cpp_impl!(self, base_create_index(row, column, ptr))
}
}