particlepainter.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
 *
 * Painting particle systems
 *
 ******************************************************************************/

#ifndef _PARTICLEPAINTER_H_
#define _PARTICLEPAINTER_H_

#include "painter.h"
#include "particle.h"

namespace Manta {

class ParticlePainter : public LockedObjPainter {
    Q_OBJECT
public:
    ParticlePainter(GridPainter<int>* gridRef, QWidget* par = 0);
    ~ParticlePainter();
    
    void paint();
    void attachWidget(QLayout* layout);

    enum PaintModes { PaintOff=0, PaintVel=1, PaintPos=2 };
    
protected:
    std::string getID();
    Real getScale();
    void update();
    void updateText();
    void processKeyEvent(PainterEvent e, int param);
    
    GridPainter<int>* mGridRef;
    ParticleBase* mLocal;
    QLabel* mInfo;

    int mMode; 
    int mDisplayMode; 

    int mLastPdata;
    bool mHavePdata;
    Real mMaxVal;
    std::string mPdataInfo;
    std::map<PbClass*, Real> mValScale;

    void paintBasicSys();
};    
    
} // namespace

#endif