painter.h

Go to the documentation of this file.

/******************************************************************************
 *
 * MantaFlow fluid solver framework
 * Copyright 2011 Tobias Pfaff, Nils Thuerey 
 *
 * This program is free software, distributed under the terms of the
 * GNU General Public License (GPL) 
 * http://www.gnu.org/licenses
 *
 * Base class for objects painting into the GL widget
 *
 ******************************************************************************/

#ifndef _PAINTER_H_
#define _PAINTER_H_

#include <QWidget>
#include <QLabel>
#include <map>
#include "grid.h"

namespace Manta {
    
// forward decl.
class PbClass;
    

class Painter : public QObject {
    Q_OBJECT
public:
    enum PainterEvent { 
        EventNone = 0, UpdateRequest, UpdateFull, UpdateStep,
        EventScaleVecUpSm, EventScaleVecDownSm, EventScaleVecUp, EventScaleVecDown, 
        EventNextRealDisplayMode, EventScaleRealUp, EventScaleRealDown, EventScaleRealUpSm, EventScaleRealDownSm, EventChangePlane, 
        EventSetPlane, EventSetDim, EventNextInt, EventNextReal, EventNextVec, EventNextVecDisplayMode,
        EventNextMesh, EventMeshMode, EventToggleGridDisplay, EventScaleMeshUp, EventScaleMeshDown, EventMeshColorMode,
        EventNextSystem, EventToggleParticles, EventNextParticleDisplayMode, EventToggleBackgroundMesh, EventSetMax,
        EventScalePdataDown, EventScalePdataUp };

    enum RealDisplayModes { RealDispOff=0, RealDispStd, RealDispLevelset, RealDispShadeVol, RealDispShadeSurf, NumRealDispModes }; 
    enum VecDisplayModes  { VecDispOff=0, VecDispCentered, VecDispStaggered, VecDispUv, NumVecDispModes };
    
    Painter(QWidget* par = 0) : QObject(par) {}
    virtual ~Painter() {}
    
    virtual std::string clickLine(const Vec3& p0, const Vec3& p1) { return ""; }
    virtual void attachWidget(QLayout* layout) {}
signals:
    void setViewport(const Vec3i& gridsize);
    
public slots:
    virtual void paint() = 0;
    virtual void doEvent(int e, int param=0) = 0;
};


class LockedObjPainter : public Painter {
    Q_OBJECT
public:
    LockedObjPainter(QWidget* par = 0) : Painter(par), mRequestUpdate(false), mObject(NULL), mObjIndex(-1) {}

    void doEvent(int e, int param=0); // don't overload, use processKeyEvent and update instead
    
protected:
    void nextObject();
    virtual std::string getID() = 0;
    virtual void update() = 0;
    virtual void processKeyEvent(PainterEvent e, int param) = 0;
    
    bool     mRequestUpdate;
    PbClass* mObject;
    int      mObjIndex;
};

template<class T>
class GridPainter : public LockedObjPainter {
public:
    GridPainter(FlagGrid** flags = NULL, QWidget* par = 0);
    ~GridPainter();
    
    void paint();
    void attachWidget(QLayout* layout);
    Grid<T>** getGridPtr() { return &mLocalGrid; }
    int getPlane() { return mPlane; }
    int getDim()   { return mDim; }
    int getMax()   { return mMax; }
    virtual std::string clickLine(const Vec3& p0, const Vec3& p1);

protected:
    std::string getID();
    Real getScale();
    void setScale(Real v);
    int  getDispMode();
    void setDispMode(int dm);
    void update();
    void updateText();
    void processKeyEvent(PainterEvent e, int param);
    void processSpecificKeyEvent(PainterEvent e, int param);
    //void paintGridLines(bool lines, bool box);
    
    Real        mMaxVal;       
    int         mDim, mPlane, mMax;
    Grid<T>*    mLocalGrid;    
    FlagGrid**  mFlags;        
    QLabel*     mInfo;         
    bool        mHide;         
    bool        mHideLocal;    
    std::map< void*, int > mDispMode; 
    std::map< std::pair<void*, int>, Real> mValScale; 
};

}

#endif