writer.h

Go to the documentation of this file.

#ifndef WRITER_H
#define WRITER_H

#include <QObject>
#include <QPainter>
#include <QColor>
#include <QImage>

#include "output.h"

namespace LangmuirCore
{

class ChargeAgent;
class FluxAgent;
class World;
class Grid;

class XYZWriter : public QObject
{
    Q_OBJECT
public:
    XYZWriter(World &world,
              const QString& name,
              QObject *parent = 0);

    void write();
protected:
    World &m_world;

    OutputStream m_stream;

    void writeVMDInitFile();
};

class FluxWriter : public QObject
{
    Q_OBJECT
public:
    FluxWriter(World &world,
               const QString& name,
               QObject *parent = 0);

    void write();
protected:
    World &m_world;

    OutputStream m_stream;
};

class CarrierWriter : public QObject
{
    Q_OBJECT
public:
    CarrierWriter(World &world,
                  const QString& name,
                  QObject *parent = 0);

    void write(ChargeAgent &charge);
protected:
    World &m_world;

    OutputStream m_stream;
};

class ExcitonWriter : public QObject
{
    Q_OBJECT
public:
    ExcitonWriter(World &world,
                  const QString& name,
                  QObject *parent = 0);

    void write(ChargeAgent &charge1, ChargeAgent &charge2, bool recombined = false);
protected:
    World &m_world;

    OutputStream m_stream;
};

class GridImage : public QObject
{
    Q_OBJECT
public:

    GridImage(World &world, QColor bg = Qt::black, QObject *parent=0);


    void drawSites(QList<int> &sites, QColor color, int layer);


    void drawCharges(QList<ChargeAgent *> &charges, QColor color, int layer);


    void save(QString name, int scale=3);
private:
    QPainter m_painter;

    QImage m_image;

    World &m_world;
};

class Logger : public QObject
{
    Q_OBJECT
public:
    Logger(World &world, QObject *parent = 0);

    virtual void saveTrapImage(const QString& name = "%stub-traps.png");

    virtual void saveHoleImage(const QString& name = "%stub-%step-holes.png");

    virtual void saveElectronImage(const QString& name = "%stub-%step-electrons.png");

    virtual void saveCarriersImage(const QString& name = "%stub-%step-carriers.png");

    virtual void saveDefectImage(const QString& name = "%stub-defects.png");

    virtual void saveImage(const QString& name = "%stub-%step-all.png");

    virtual void saveGridPotential(const QString& name = "%stub.grid");

    virtual void saveCoulombEnergy(const QString& name = "%stub-%step.coulomb");

    virtual void reportFluxStream();

    virtual void reportXYZStream();

    virtual void reportCarrier(ChargeAgent &charge);

    virtual void reportExciton(ChargeAgent &charge1, ChargeAgent &charge2, bool recombined = false);

    virtual void initialize();

protected:
    World &m_world;

    XYZWriter *m_xyzWriter;

    FluxWriter *m_fluxWriter;

    CarrierWriter *m_carrierWriter;

    ExcitonWriter *m_excitonWriter;
};

}
#endif // WRITER_H