动态旋转变色的3D爱心

之前为了在女朋友面前露一手（啊不是 闲得无聊）,为了证明两年多的图像没有白学，总不能做个控制台黑窗口print循环打印爱心字符串吧，于是乎做了一个可以动态旋转变色的3D爱心，效果如下图所示。虽然效果也很简陋，还被鄙视了（惨兮兮），但毕竟也是花了两个小时写出来的demo嘛。

基于Win10+QT平台开发，用了一些常用的2D/3D图像库比如OpenCV、PCL和VTK，其实主体也就百来行代码，思路就是调用库绘制已知方程的爱心点云然后在对点云进行变换的同时刷新可视化窗口。3D爱心的方程是

具体程序如下：
mainwindow.h

#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <QVTKWidget.h>
#include <QMouseEvent>
#include <Windows.h>

#include <iostream>
#include <opencv2/opencv.hpp>
#include <pcl/io/pcd_io.h>
#include <pcl/io/ply_io.h>
#include <pcl/point_types.h>
#include <pcl/visualization/pcl_visualizer.h>
#include <vtkAutoInit.h>
#include <vtkRenderWindow.h>


VTK_MODULE_INIT(vtkRenderingOpenGL2)
VTK_MODULE_INIT(vtkInteractionStyle)


namespace Ui {
class MainWindow;
}

class MainWindow : public QMainWindow
{
    Q_OBJECT

public:
    explicit MainWindow(QWidget *parent = 0);
    ~MainWindow();

private:
    Ui::MainWindow *ui;

    pcl::PointCloud<pcl::PointXYZRGB>::Ptr cloud;

    boost::shared_ptr<pcl::visualization::PCLVisualizer> viewer;

    bool flag = true;

public slots:
    void stopLoop();

    void mousePressEvent(QMouseEvent *event);
};

#endif // MAINWINDOW_H

mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    this->setMouseTracking(true);
    this->setFocusPolicy(Qt::StrongFocus);
    setAttribute(Qt::WA_QuitOnClose);

    cloud.reset(new pcl::PointCloud<pcl::PointXYZRGB>);
    for (float x = -1.2; x <= 1.2; x+= 0.01)
    {
        for (float y = -0.8; y <= 0.8; y+= 0.01)
        {
            for (float z = -1; z <= 1.2; z+= 0.01)
            {
                if (pow(x*x + 9 / 4 * y*y + z*z - 1, 3) - x*x*z*z*z - 9 / 80 * y*y*z*z*z <= 0)
                {
                    pcl::PointXYZRGB p;
                    p.x = x; p.y = y; p.z = z; p.r = 255; p.g = 0; p.b =0;
                    cloud->push_back(p);
                }
            }
        }
    }

    viewer.reset(new pcl::visualization::PCLVisualizer("viewer", false));
    viewer->addPointCloud(cloud, "cloud");
    viewer->setCameraPosition(0, 12, 0, 0, -1, 0, 0, 0, 1);

    ui->qvtkWidget->SetRenderWindow(viewer->getRenderWindow());
    viewer->setupInteractor(ui->qvtkWidget->GetInteractor(), ui->qvtkWidget->GetRenderWindow());
    ui->qvtkWidget->update();
}

MainWindow::~MainWindow()
{
    delete ui;
}

void MainWindow::stopLoop()
{
    flag = false;
}

void MainWindow::mousePressEvent(QMouseEvent *event)
{
    if(event->button())
    {
        srand((unsigned int)time(0));

        while(flag)
        {
            int r = rand() % 256, g = rand() % 256, b = rand() % 256;
            for(int i = 0; i < cloud->points.size(); ++i)
            {
                float x = cloud->points[i].x, y = cloud->points[i].y;
                cloud->points[i].x = cos(0.1) * x - sin(0.1) * y;
                cloud->points[i].y = sin(0.1) * x + cos(0.1) * y;
                cloud->points[i].r = r;
                cloud->points[i].g = g;
                cloud->points[i].b = b;
            }
            viewer->updatePointCloud(cloud);
            ui->qvtkWidget->update();
            cv::waitKey(1);
        }
    }
}

main.cpp

#include "mainwindow.h"
#include <QApplication>

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    MainWindow *w = new MainWindow;
    w->show();
    a.connect(&a, SIGNAL(lastWindowClosed()), w, SLOT(stopLoop()));
    return a.exec();
}

项目工程文件在此：
链接：https://pan.baidu.com/s/1Ll8-qv98upb6ugXLi9G5pQ?pwd=d5o2
提取码：d5o2