source: trunk/src/Plotter2.h@ 2874

#include <cstdlib>
#include <iostream>
#include <math.h>
#include <stdio.h>
#include <string>
#include <unistd.h>
#include <vector>

#include <cpgplot.h>

namespace asap {

class Plotter2RectInfo {
public: 
    Plotter2RectInfo();
    ~Plotter2RectInfo();

    float xmin;
    float xmax;
    float ymin;
    float ymax;

    int color;
    int fill;
    int width;
    float hsep;
};

class Plotter2DataInfo {
public:
    Plotter2DataInfo();
    ~Plotter2DataInfo();

    std::vector<float> xData;
    std::vector<float> yData;

    bool drawLine;
    int lineColor;
    int lineWidth;
    int lineStyle;

    bool drawMarker;
    int markerType;
    float markerSize;
    int markerColor;

    bool hasData;
};

class Plotter2ViewportInfo {
public:
    Plotter2ViewportInfo();
    ~Plotter2ViewportInfo();

    // show the whole viewport
    bool showViewport;

    // viewport position in window-coordinate
    float vpPosXMin;
    float vpPosXMax;
    float vpPosYMin;
    float vpPosYMax;

    // plotting range in world-coordinate
    float vpRangeXMin;
    float vpRangeXMax;
    float vpRangeYMin;
    float vpRangeYMax;

    // set plotting range automatic
    bool isAutoRangeX;
    bool isAutoRangeY;
    float autoRangeMarginX;
    float autoRangeMarginY;
    void adjustRange();
    std::vector<float> getRangeX();
    std::vector<float> getRangeY();

    // tick intervals
    bool isAutoTickIntervalX;
    bool isAutoTickIntervalY;
    float majorTickIntervalX;
    float majorTickIntervalY;
    int nMajorTickWithinTickNumsX;
    int nMajorTickWithinTickNumsY;
    int nMinorTickWithinMajorTicksX;
    int nMinorTickWithinMajorTicksY;
    void adjustTickInterval();

    // location of value strings along axes
    std::string numLocationX;
    std::string numLocationY;

    // default font size
    float fontSizeDef;

    // data to be plotted
    std::vector<Plotter2DataInfo> vData;
    void setData(const std::vector<float>& inXData, const std::vector<float>& inYData, const int id);

    // rectangles
    std::vector<Plotter2RectInfo> vRect;

    // x-label
    std::string labelXString;
    float labelXPosX;
    float labelXPosY;
    float labelXAngle;
    float labelXFJust;
    float labelXSize;
    int labelXColor;
    int labelXBColor;

    // y-label
    std::string labelYString;
    float labelYPosX;
    float labelYPosY;
    float labelYAngle;
    float labelYFJust;
    float labelYSize;
    int labelYColor;
    int labelYBColor;

    // title
    std::string titleString;
    float titlePosX;
    float titlePosY;
    float titleAngle;
    float titleFJust;
    float titleSize;
    int titleColor;
    int titleBColor;

    // background colour
    int vpBColor;

    void getWorldCoordByWindowCoord(const float winX, const float winY, float* worldX, float* worldY);

private:
    float minXData;
    float maxXData;
    float minYData;
    float maxYData;
    bool hasDataRange;
    void updateXDataRange(const float data);
    void updateYDataRange(const float data);
    void updateAllDataRanges();
    void adjustRangeX(float* xmin, float* xmax);
    void adjustRangeY(float* ymin, float* ymax);
    void adjustTickIntervalX(const float xmin, const float xmax);
    void adjustTickIntervalY(const float ymin, const float ymax);
};

class Plotter2 {
public:
    Plotter2();
    ~Plotter2();

    std::string getFileName();
    void setFileName(const std::string& inFilename);
    std::string getDevice();
    void setDevice(const std::string& inDevice);

    int addViewport(const float xmin, const float xmax, const float ymin, const float ymax);
    void setViewport(const float xmin, const float xmax, const float ymin, const float ymax, const int id);
    void showViewport(const int inVpid);
    void hideViewport(const int inVpid);
    bool getHasDefaultViewport();
    int getCurrentViewportId();
    void getViewInfo();
    void setRange(const float xmin, const float xmax, const float ymin, const float ymax, const int inVpid);
    void setRangeX(const float xmin, const float xmax, const int inVpid);
    void setRangeY(const float ymin, const float ymax, const int inVpid);
    std::vector<float> getRangeX(const int inVpid);
    std::vector<float> getRangeY(const int inVpid);
    void setAutoRange(const int inVpid);
    void setAutoRangeX(const int inVpid);
    void setAutoRangeY(const int inVpid);
    void setFontSizeDef(const float size, const int inVpid);
    void setTicksX(const float interval, const int num, const int inVpid);
    void setTicksY(const float interval, const int num, const int inVpid);
    void setAutoTicks(const int inVpid);
    void setAutoTicksX(const int inVpid);
    void setAutoTicksY(const int inVpid);
    void setNumIntervalX(const float interval, const int inVpid);
    void setNumIntervalY(const float interval, const int inVpid);
    void setNumLocationX(const std::string& side, const int inVpid);
    void setNumLocationY(const std::string& side, const int inVpid);
    void setData(const std::vector<float>& xdata, const std::vector<float>& ydata, const int inVpid, const int inDataid);
    void setLine(const int color, const int width, const int style, const int inVpid, const int inDataid);
    void showLine(const int inVpid, const int inDataid);
    void hideLine(const int inVpid, const int inDataid);
    void setPoint(const int type, const float size, const int color, const int inVpid, const int inDataid);
    void showPoint(const int inVpid, const int inDataid);
    void hidePoint(const int inVpid, const int inDataid);
    void setMaskX(const float xmin, const float xmax, const int color, const int fill, const int width, const float hsep, const int inVpid);
    void setLabelX(const std::string& label, const float posx, const float posy, const float size, const std::string& style, const int color, const int bgcolor, const int inVpid);
    void setLabelY(const std::string& label, const float posx, const float posy, const float size, const std::string& style, const int color, const int bgcolor, const int inVpid);
    void setTitle(const std::string& label, const float posx, const float posy, const float size, const std::string& style, const int color, const int bgcolor, const int inVpid);
    void setViewportBackgroundColor(const int bgcolor, const int inVpid);
    //void setAnnotation(const std::string& label, const float posx, const float posy, const float angle, const float fjust, const float size, const std::string& style, const int color, const int bgcolor, const int inVpid);
    void plot();
private:
    std::string filename;
    std::string device;
    bool hasDevice;
    std::vector<Plotter2ViewportInfo> vInfo;
    bool hasDefaultViewport;
    int currentViewportId;
    void open();
    void close();
};

} // namespace asap