数据可视化 第4章



  • 第4章 添加表格QTableView

    1.添加table_model.py,里面子类化QAbstractTableModel,实现自定义table model

    from PySide2.QtCore import Qt, QAbstractTableModel, QModelIndex
    from PySide2.QtGui import QColor
    
    #实现CustromTableModel类
    class CustomTableModel(QAbstractTableModel):
        def __init__(self, data=None):
            QAbstractTableModel.__init__(self)
            self.load_data(data)
    
        def load_data(self, data):
            self.input_datas = data[0].values
            self.input_magnitudes = data[1].values
    
            self.column_count = 2
            self.row_count = len(self.input_magnitudes)
    
        def rowCount(self, parent=QModelIndex()):
            return self.row_count
    
        def columnCount(self, parent=QModelIndex()):
                return self.column_count
    
        def headerData(self, section, orientation, role):
            if role != Qt.DisplayRole:
                return None
            if orientation == Qt.Horizontal:
                return ("Date", "Magnitude")[section]
            else:
                return "{}".format(section)
    
        def data(self, index, role=Qt.DisplayRole):
            column = index.column()
            row = index.row()
    
            if role == Qt.DisplayRole:
                if column == 0:
                    raw_data = self.input_datas[row]
                    data = "{}".format(raw_data.toPython())
                    return data[:-3]
                elif column == 1:
                    return "{:.2f}".format(self.input_magnitudes[row])
            elif role == Qt.BackgroundRole:
               return QColor(Qt.white)
            elif role == Qt.TextAlignmentRole:
               return Qt.AlignRight
    
            return None
    
    

    2.添加main_widget.py,使用QTableView+TableModel显示表格,设置widget布局,自适应窗口

    from PySide2.QtWidgets import (QHBoxLayout, QHeaderView, QSizePolicy, QTableView, QWidget)
    from table_model import *
    
    #含TableView的窗口类
    class Widget(QWidget):
        def __init__(self, data):
            QWidget.__init__(self)
    
            #获取model
            self.model = CustomTableModel(data)
    
            #创建QTableView
            self.table_view = QTableView()
            self.table_view.setModel(self.model)
    
            #QTableView 表头自适应以及拉伸
            self.horizontal_header = self.table_view.horizontalHeader()
            self.vertical_header = self.table_view.verticalHeader()
            self.horizontal_header.setSectionResizeMode(QHeaderView.ResizeToContents)
            self.vertical_header.setSectionResizeMode(QHeaderView.ResizeToContents)
            self.horizontal_header.setStretchLastSection(True)
    
            #布局
            self.main_layout = QHBoxLayout()
            size = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
    
            #应用size策略
            size.setHorizontalStretch(1)
            self.table_view.setSizePolicy(size)
            self.main_layout.addWidget(self.table_view)
    
            #应用布局
            self.setLayout(self.main_layout)
    
    

    3.添加main_window.py,在主窗口里使用定制好的widget

    from PySide2.QtCore import Slot, qApp
    from PySide2.QtGui import QKeySequence
    from PySide2.QtWidgets import QMainWindow, QAction, QApplication
    
    #主窗口
    class MainWindow(QMainWindow):
        def __init__(self, widget):
            QMainWindow.__init__(self)
            self.setWindowTitle("地震信息")
            self.setCentralWidget(widget)
    
            # Menu
            self.menu = self.menuBar()
            self.file_menu = self.menu.addMenu("文件")
    
            # Exit QAction
            exit_action = QAction("退出", self)
            exit_action.setShortcut(QKeySequence.Quit)
            exit_action.triggered.connect(self.close)
    
            self.file_menu.addAction(exit_action)
    
            # Status Bar
            self.status = self.statusBar()
            self.status.showMessage("数据加载和绘制")
    
            # Window dimensions
            geometry = qApp.desktop().availableGeometry(self)
            self.setFixedSize(geometry.width() * 0.6, geometry.height() * 0.4)
    
    

    4.添加chapter4.py,这是我们要运行的py文件,读取命令行参数,解析csv文件,将data数据传递给窗口显示

    import sys, argparse
    import pandas as pd
    from PySide2.QtCore import QDateTime, QTimeZone
    
    from main_widget import Widget
    from main_window import MainWindow
    
    from PySide2.QtWidgets import QApplication
    
    def transform_date(utc, timezone=None):
        utc_fmt = "yyyy-MM-ddTHH:mm:ss.zzzZ"
        new_date = QDateTime().fromString(utc, utc_fmt)
    #    print (new_date.toString(utc_fmt))
        if timezone:
            new_date.setTimeZone(timezone)
    
        return new_date
    
    def read_data(fname):
        df = pd.read_csv(fname)
    
        #移除错误的mag
        df = df.drop(df[df.mag < 0].index)
        magnitudes = df["mag"]
    
        #本地时区
        timezone = QTimeZone(b"Asia/Beijing")
    
        #将时间戳转换为本地时区
        times = df["time"].apply(lambda x: transform_date(x, timezone))
    
        return times, magnitudes
    
    
    
    if __name__ == "__main__":
        options = argparse.ArgumentParser()
        options.description = "窗口显示csv信息"
        options.add_argument("-f", "--file", type=str, required=True, help="CSV文件路径")
        args = options.parse_args()
        data = read_data(args.file)
    
    #    print (data)
        #qt 应用
        app = QApplication(sys.argv)
    
        widget = Widget(data)
        window = MainWindow(widget)
        window.show()
    
        sys.exit(app.exec_())
    
    

    5.运行效果
    4.png



  • QTableView还是比较成熟的,因为很简单。但是如果是树状图,那么只能用QAbstractItemModel的子类了。因为有父子层级关系,所以要自己写。


Log in to reply
 

走马观花

最近的回复

  • 在Maya 2018安装了之后,还会出现很多的错误:比如说在脚本编辑器中会出现这个:

    file /usr/autodesk/maya2018/lib/python2.7/site-packages/maya/app/general/mayaMixin.py line 35: libpyside2.so.2.0: cannot open shared object file: No such file or directory, No module named PyQt4.QtCore

    这是因为Maya里面有一个文件叫做mayaMixin.py,判断系统是否有PySide或者是PySide2。不用担心一定要安装PySide。网上有人说遇到了No module named PyQt4.QtCore需要执行下面的操作:

    sudo apt install python-qt4

    但是我查找了mayaMixin.py的35行,发现它是先寻找是否有PySide2,如果没有PySide2才会去找PySide。其实Maya 2018默认带了PySide2,因此我们只需要编写一个运行Maya的脚本,设置好环境变量,就可以让这个问题消除了。所以我果断卸载了python-qt4。

    sudo apt remove python-qt4

    #!/bin/sh export PYTHONPATH=usr/autodesk/maya2018/lib/python2.7/site-packages:$PYTHONPATH export LD_LIBRARY_PATH=/usr/autodesk/maya2018/lib:$LD_LIBRARY_PATH /usr/autodesk/maya2018/bin/maya.bin

    把这个脚本放在桌面上吧。

    read more
  • 最近在研究怎样让USD支持Maya类似的约束功能。目前初步得到了成功。当一开始遇到这个问题的时候,我还是转向了github,问问官方人员有没有相关现成的方案,或者是有没有计划怎么做。不过我在这个issue中得知,官方并没有立即支持Maya的约束,目前只能用到的是ConstraintTarget特性。所以很遗憾地暂时没有办法使用官方的建议了,只能自己做USD的约束了。😞 😱

    不过好在USD的基础设施还比较好,我开始研究USD的约束,希望在USDView中得到类似Maya的效果。在上一篇文章中,我介绍了一种通过Schema生成cpp文件的方法,事实上,大部分基于USD的项目,都是用到了这个方便的方式快速拓展原型,并且在此基础上添加功能。

    不过USDView的功能,变得更加复杂了。我尝试了很多方案。首先参考的是USDView的通知机制。首先在USDView中打开interpreter,然后看到交互式Python界面(虽然还是有点丑):
    interpreter在USDView中

    然后输入下面的脚本,用来测试:

    stage = usdviewApi.stage pipe = stage.GetPrimAtPath( '/pPipe' ) UsdGeom.XformCommonAPI( pipe ).SetTranslate( ( 1, 3, 1 ) )

    这个脚本是用来让名为pPipe的物体移动到世界坐标1, 3, 1中的。但是如果你用Visual Studio 2017调试它,会发现一条比较严谨并且漫长的调用路线:
    XformCommonAPI -> UsdGeomXformOp -> UsdStage -> SdfLayer -> SdfChangeBlock -> Sdf_ChangeManager -> TfNotice -> UsdImagingDelegate

    这里有一个很重要的点,就是在UsdImagingDelegate在TfNotice中注册了一个处理函数_OnUsdObjectsChanged,所以一旦场景的属性更改了,都会触发_OnUsdObjectsChanged函数。所以我根据这个特性,创建了usdRigImaging模块,位置在USD\qtdream\usdImaging\lib\usdRigImaging中。这其中很重要的一个类是UsdRigImagingConstraintAdapter,并且在构造函数注册了TfNotice的回调函数。一旦场景发生变化了,我让约束本身感知被约束的物体,进行同步的更新。当然了,Maya本身是支持多个物体约束一个物体的,并且给出了一个权重的信息。其实约束的算法就是各个约束的加权平均数。因此要通过一个for循环对所有的约束物体计算平移变化,再加权平均,即可得到效果。核心的代码在这里:

    void UsdRigImagingConstraintAdapter::_OnUsdObjectsChanged( UsdNotice::ObjectsChanged const& notice ) { const UsdNotice::ObjectsChanged::PathRange& fields = notice.GetChangedInfoOnlyPaths( ); TF_FOR_ALL( it, _constraints ) { UsdRigConstraint constraint( *it ); std::vector<int> constrainedIndices; VtArray<SdfAssetPath> targets; constraint.GetTargetsAttr( ).Get( &targets ); for ( int i = 0; i < targets.size( ); ++i ) { SdfPath targetPath( targets[i].GetAssetPath( ) ); TF_FOR_ALL( field, fields ) { if ( targetPath == field->GetPrimPath( ) ) { constrainedIndices.push_back( i ); } } } TF_FOR_ALL( constrainedIndex, constrainedIndices ) { _ApplyConstrainEffect( constraint, *constrainedIndex ); } } } void UsdRigImagingConstraintAdapter::_ApplyConstrainEffect( const UsdRigConstraint& constraint, int constrainedIndex ) { const UsdPrim& constraintTarget = _GetConstraintTarget( constraint ); const UsdPrim& target = _GetTargetAtIndex( constraint, constrainedIndex ); UsdTimeCode time = UsdTimeCode::Default( ); UsdGeomXformCommonAPI ctAPI( constraintTarget ); GfVec3d translation( 0, 0, 0 ); target.GetAttribute( TfToken( "xformOp.translate" ) ).Get( &translation ); GfVec3d initialTranslation = _initialTranslationMap[target.GetPrimPath( )]; double weight = _GetWeightAtIndex( constraint, constrainedIndex ); GfVec3d finalTranslation = ( translation - initialTranslation ) * weight; ctAPI.SetTranslate( finalTranslation ); } void UsdRigImagingConstraintAdapter::_PopulateOnConstraint( const UsdPrim& prim ) { UsdStageWeakPtr& stage = prim.GetStage( ); _constraints.push_back( prim ); UsdRigConstraint constraint( prim ); VtArray<SdfAssetPath> targets; constraint.GetTargetsAttr( ).Get( &targets ); TF_FOR_ALL( tit, targets ) { SdfPath targetPath( tit->GetAssetPath( ) ); GfVec3d translation; const UsdPrim& target = stage->GetPrimAtPath( targetPath ); target.GetAttribute( TfToken( "xformOp:translate" ) ).Get( &translation ); std::pair<SdfPath, GfVec3d> translationPair = std::pair<SdfPath, GfVec3d>( targetPath, translation ); _initialTranslationMap.insert( translationPair ); } } UsdPrim UsdRigImagingConstraintAdapter::_GetConstraintTarget( const UsdRigConstraint& constraint ) { SdfAssetPath constraintTargetPath; constraint.GetTargetAttr( ).Get( &constraintTargetPath ); return _GetPrimAtPath( constraint, SdfPath( constraintTargetPath.GetAssetPath( ) ) ); } UsdPrim UsdRigImagingConstraintAdapter::_GetTargetAtIndex( const UsdRigConstraint& constraint, int constraintIndex ) { VtArray<SdfAssetPath> targets; constraint.GetTargetsAttr( ).Get( &targets ); return _GetPrimAtPath( constraint, SdfPath( targets[constraintIndex].GetAssetPath( ) ) ); } UsdPrim UsdRigImagingConstraintAdapter::_GetPrimAtPath( const UsdRigConstraint& constraint, const SdfPath& path ) { UsdStageWeakPtr& stage = constraint.GetPrim( ).GetStage( ); return stage->GetPrimAtPath( path ); } float UsdRigImagingConstraintAdapter::_GetWeightAtIndex( const UsdRigConstraint& constraint, int constraintIndex ) { VtArray<float> weights; constraint.GetWeightsAttr( ).Get( &weights ); float weight = weights[constraintIndex]; weight = weight / weights.size( ); return weight; }

    在内部的一个测试场景中,我们针对pPipe物体做出了平移变换,被约束的物体也同步地变换。

    对USD的二次开发感兴趣?或者是想要让你的三维软件添加USD的导入导出支持?可以找我们,我们研究USD很深入哦。
    上海USD研究小组

    read more

关注我们

微博
QQ群