v1.cpp

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <string>
#include "plstream.h"

using namespace std;

vector<string> split(const string &str, char delim){
  vector<string> res;
  size_t current = 0, found;
  while((found = str.find_first_of(delim, current)) != string::npos){
    res.push_back(string(str, current, found - current));
    current = found + 1;
  }
  res.push_back(string(str, current, str.size() - current));
  return res;
}
std::string basename(const std::string& path) {
    return path.substr(path.find_last_of('/') + 1);
}

struct DataOpts
{
  int nx;
  int ny;
  double xmin;
  double xmax;
  double ymin;
  double ymax;
};

DataOpts filecheck(std::vector<string> const& fs)
{
  DataOpts opts = {0, 0, 0, 0, 0, 0};

  set<double> xs, ys, xs_, ys_;

  bool isfirst = true;

  for (std::vector<string>::const_iterator it = fs.begin(); it != fs.end(); ++it)
  {
    std::string const& f = *it;
    xs_.clear(); ys_.clear();

    std::ifstream ifs(f.c_str());
    while (ifs)
    {
      double x, y, u, v;
      ifs >> x >> y >> u >> v;

      opts.xmin = std::min(opts.xmin, x);
      opts.ymin = std::min(opts.ymin, y);
      opts.xmax = std::max(opts.xmax, x);
      opts.ymax = std::max(opts.ymax, y);

      xs_.insert(x);
      ys_.insert(y);
    }

    if (isfirst)
    {
      isfirst = false;
    }
    else
    {
      if (xs.size() != xs_.size() || ys.size() != ys_.size())
      {
        std::cerr << "X要素数とY要素数が一致しません：ファイルのエラー" << std::endl;
      }
    }
    xs.swap(xs_);
    ys.swap(ys_);
  }

  opts.nx = xs.size();
  opts.ny = ys.size();

  return opts;
}

void plot( int argc, const char ** argv, char const* device, char const* inpath, char const* outpath, DataOpts const& opts)
{
  plstream *pls;
  PLFLT    **z;
  PLcGrid2 cgrid2;

  pls = new plstream();
  pls->sdev(device);
  pls->sfnam(outpath);
  pls->parseopts(&argc,argv,PL_PARSE_FULL);
  pls->scolbg(255, 255, 255);
  pls ->scol0(15, 0, 0, 0);
  // Initialize plplot
  pls->init();

  {
    //set color map
    PLFLT y[4]={-1.0,-1.0, 1.0, 1.0};
    PLFLT r[2]={0.0,1.0};
    PLFLT g[2]={0.0,0.0};
    PLFLT b[2]={1.0,0.0};
    PLFLT pos[2]={0.0,1.0};
    pls->scmap1l(true,2,pos,r,g,b,NULL);
  }
  { // plot
    double xmin=opts.xmin;
    double xmax=opts.xmax;
    double ymin=opts.ymin;
    double ymax=opts.ymax;
    int nx = opts.nx;
    int ny = opts.ny;
    // Allocate arrays
    PLFLT **us;
    PLFLT **vs;
    pls->Alloc2dGrid(&us,nx,ny);
    pls->Alloc2dGrid(&vs,nx,ny);
    PLcGrid2  cgrid2;//PLcGrid2  xg,yg,nx,ny
    pls->Alloc2dGrid(&cgrid2.xg,nx,ny);
    pls->Alloc2dGrid(&cgrid2.yg,nx,ny);
    cgrid2.nx=nx;
    cgrid2.ny=ny;
    {
      std::ifstream ifs(inpath);
      for(int i=0;i<nx;i++){
        for(int j=0;j<ny;j++){
          double x, y, u, v;
          ifs >> x >> y >> u >> v;
          cgrid2.xg[i][j] = x;
          cgrid2.yg[i][j] = y;
          us[i][j]         = u;
          vs[i][j]         = v;
        }
      }
    }


    //Plot
    pls->col0(15);
    pls->env(xmin,xmax,ymin,ymax,1,0);
    pls->lab(getenv("X"),getenv("Y"),getenv("TITLE"));
    pls->vect(us,vs,nx,ny,0.0, plstream::tr2, (void *) &cgrid2 );

    //Free grid
    pls->Free2dGrid(cgrid2.xg,nx,ny);
    pls->Free2dGrid(cgrid2.yg,nx,ny);
    pls->Free2dGrid(us,nx,ny);
    pls->Free2dGrid(vs,nx,ny);

    delete pls;
  }
}


int main( int argc, const char ** argv )
{
    std::vector<string> const fs = split(getenv("F"), ',');

    DataOpts const opts = filecheck(fs);

    for (std::vector<string>::const_iterator it = fs.begin(); it != fs.end(); ++it)
    {
      std::string const& f = *it;
      std::string const outpath = ("public/"+basename(f)+".png");
      plot( argc, argv, getenv("D"), f.c_str(), outpath.c_str(), opts);
      std::cout << outpath << ",";
    }

    return 0;
}