aboutsummaryrefslogtreecommitdiffstats
path: root/include/grgsm/plotting.hpp
blob: 21890e435dd0142a886ac721d3366c55c1e8baee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/* -*- c++ -*- */
/*
 * @file
 * @author (C) 2014 by Piotr Krysik <ptrkrysik@gmail.com>
 * @section LICENSE
 *
 * Gr-gsm is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * Gr-gsm is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with gr-gsm; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#define USE_CXX (__cplusplus >= 201103)

#include <vector>
#include <armadillo>
#include <string>
#include <boost/make_shared.hpp>

#include "gnuplot-iostream.h"

boost::shared_ptr<Gnuplot> current_figure;

void imagesc(arma::mat & x){
    Gnuplot gp;
	gp << "set palette rgb 3,2,2;";
   gp << "plot ";
 	gp << gp.file1d(x) << "matrix with image";
	gp << std::endl;
}

void plot(arma::cx_mat & x, std::string title){
   arma::mat y = arma::abs(x);
   if(current_figure.get()==NULL){
      current_figure = boost::make_shared<Gnuplot>();
   }
   (*current_figure) << "plot ";
   
   (*current_figure) << current_figure->file1d(y) <<"title \'" << title << "\' with lines ";
   (*current_figure) << std::endl; 
}

void replot(arma::cx_mat & x, std::string title){
   arma::mat y = arma::abs(x);
   if(current_figure.get()==NULL){
      current_figure = boost::make_shared<Gnuplot>();
   }
   (*current_figure) << "replot ";
   (*current_figure) << current_figure->file1d(y) <<"title \'"  << title << "\' with lines ";
   (*current_figure) << std::endl; 
}

template<typename T>
void plot(std::vector<T> & x){
   arma::cx_mat y = arma::conv_to<arma::cx_mat>::from(x);
   plot(y,"");
}

template<typename T>
void plot(std::vector<T> & x, std::string title){
   arma::cx_mat y = arma::conv_to<arma::cx_mat>::from(x);
   plot(y,title);
}

template<typename T>
void replot(std::vector<T> & x, std::string title){
   arma::cx_mat y = arma::conv_to<arma::cx_mat>::from(x);
   replot(y,title);
}