Skip to content

Commit

Permalink
(1) Added gaze model; (2) added capacity to print; (3) fixed issue with
Browse files Browse the repository at this point in the history 
noise generation that prohibited reproducing experiments when the
duration changed; (4) other minor bug fixes
  • Loading branch information
@whit-schonbein
whit-schonbein committed Feb 5, 2017
1 parent 8bde391 commit 8126038
Show file tree
Hide file tree
Showing 16 changed files with 965 additions and 95 deletions.
35 changes: 29 additions & 6 deletions chart_direct_britton.cpp
View file
Expand Up @@ -11,7 +11,7 @@ ChartDirectBritton::ChartDirectBritton(params_direct_britton_t *p, QWidget *pare

/* populate noise arrays using seed */
std::default_random_engine generator(params->seed);
int length = (int)(params->d/params->h);
int length = ceil(params->d/params->h);

params->cn_q1 = (double *)malloc(length * sizeof(*(params->cn_q1)));
params->cn_q2 = (double *)malloc(length * sizeof(*(params->cn_q2)));
Expand Down Expand Up @@ -68,9 +68,8 @@ ChartDirectBritton::ChartDirectBritton(params_direct_britton_t *p, QWidget *pare

/* create chart */
QtCharts::QChart *chart = new QtCharts::QChart();
//chart->legend()->hide();
chart->legend()->setVisible(true);
chart->setTitle("Simplified Direct Britton Model");
//chart->setTitle("Simplified Direct Britton Model");

/* this will animate the series as it is displayed */
chart->setAnimationOptions(QtCharts::QChart::SeriesAnimations);
Expand All @@ -80,11 +79,13 @@ ChartDirectBritton::ChartDirectBritton(params_direct_britton_t *p, QWidget *pare
/* time axis */
QtCharts::QValueAxis *axisX = new QtCharts::QValueAxis;
axisX->setTitleText("Time");
axisX->setGridLineVisible(false);

/* activation axis */
QtCharts::QValueAxis *axisY = new QtCharts::QValueAxis;
axisY->setRange(-1.0,params->population);
axisY->setTitleText("Population");
axisY->setGridLineVisible(false);

/* add axes to chart */
chart->addAxis(axisX, Qt::AlignBottom);
Expand All @@ -104,26 +105,29 @@ ChartDirectBritton::ChartDirectBritton(params_direct_britton_t *p, QWidget *pare
source_population->attachAxis(axisY);

/* chart view goes on top */
QtCharts::QChartView *chartView = new QtCharts::QChartView(chart);
chartView->setRenderHint(QPainter::Antialiasing);
chart_view = new QtCharts::QChartView(chart);
chart_view->setRenderHint(QPainter::Antialiasing);

/* button box comes next */
button_box = new QGroupBox;
QHBoxLayout *button_layout = new QHBoxLayout;
m_button_print = new QPushButton("print", this);
m_button_close = new QPushButton("close", this);
button_layout->addWidget(m_button_print);
button_layout->addWidget(m_button_close);
button_box->setLayout(button_layout);

/* assemble window: main layout */
QVBoxLayout *main_layout = new QVBoxLayout;
main_layout->addWidget(chartView);
main_layout->addWidget(chart_view);
main_layout->addWidget(button_box);

setLayout(main_layout);
setWindowTitle(tr("Simplified Direct Britton Model"));
show();

/* wire the signals */
connect(m_button_print, SIGNAL (clicked()), this, SLOT(slot_print()));
connect(m_button_close, SIGNAL (clicked()), this, SLOT(slot_close()));
}

Expand All @@ -141,3 +145,22 @@ void ChartDirectBritton::slot_close() {
this->close();
}

void ChartDirectBritton::slot_print() {
/* create a printer */
QPrinter *printer = new QPrinter(QPrinter::HighResolution);
//printer->setOrientation(QPrinter::Landscape);
QPrintDialog *dialog = new QPrintDialog(printer);
dialog->setWindowTitle("Print Chart");
if (dialog->exec() != QDialog::Accepted)
return;

/* make a painter to paint onto the pages of the printer */
QPainter *painter = new QPainter();
painter->begin(printer);
painter->setRenderHint(QPainter::Antialiasing);
chart_view->render(painter, printer->pageRect());
painter->end();
delete(dialog);
delete(painter);
delete(printer);
}
7 changes: 7 additions & 0 deletions chart_direct_britton.h
View file
Expand Up @@ -8,6 +8,8 @@
#include <QFlags>
#include <QGroupBox>
#include <QLineSeries>
#include <QPrinter>
#include <QPrintDialog>
#include <QPushButton>
#include <QValueAxis>
#include <QVBoxLayout>
Expand All @@ -28,8 +30,12 @@ class ChartDirectBritton : public QWidget
double * results_y1;
double * results_y2;

/* window component: the chart */
QtCharts::QChartView *chart_view;

/* window component: button box */
QGroupBox *button_box;
QPushButton *m_button_print;
QPushButton *m_button_close;

signals:
Expand All @@ -38,6 +44,7 @@ public slots:

private slots:
void slot_close();
void slot_print();
};

#endif // CHART_DIRECT_BRITTON_H
185 changes: 185 additions & 0 deletions chart_gaze.cpp
View file
@@ -0,0 +1,185 @@
#include "chart_gaze.h"

/* for debugging */
#include <iostream>

ChartGaze::ChartGaze(params_gaze_t *p, QWidget *parent)
: QWidget(parent), params(p)
{
/* make the window sort of large */
setFixedSize(1000,600);

/* populate noise arrays using seed */
// if we wanted a seed from time: unsigned seed = std::chrono::system_clock::now().time_since_epoch().count();
std::default_random_engine generator(params->seed);
int length = ceil(params->d/params->h);
params->n_I1 = (double *)malloc(length * sizeof(*(params->n_I1)));
params->n_I2 = (double *)malloc(length * sizeof(*(params->n_I2)));
params->n_w1 = (double *)malloc(length * sizeof(*(params->n_w1)));
params->n_w2 = (double *)malloc(length * sizeof(*(params->n_w2)));
params->n_g1 = (double *)malloc(length * sizeof(*(params->n_g1)));
params->n_g2 = (double *)malloc(length * sizeof(*(params->n_g2)));
params->n_l1 = (double *)malloc(length * sizeof(*(params->n_l1)));
params->n_l2 = (double *)malloc(length * sizeof(*(params->n_l2)));

gaze_set_noise(&generator,
0.0,
params->n_std_dev,
length,
params->n_I1,
params->n_I2,
params->n_w1,
params->n_w2,
params->n_g1,
params->n_g2,
params->n_l1,
params->n_l2);

/* get results of approximation */
results_y1 = (double *)malloc(length * sizeof(*results_y1));
results_y2 = (double *)malloc(length * sizeof(*results_y2));

// note: function call sets initial conditions
gaze_rk4(&generator, params, results_y1, results_y2);

/* create series to chart */
double t;
QtCharts::QLineSeries *series1 = new QtCharts::QLineSeries();
QtCharts::QLineSeries *series2 = new QtCharts::QLineSeries();
QtCharts::QLineSeries *series_diff = new QtCharts::QLineSeries();
QtCharts::QLineSeries *series_gaze = new QtCharts::QLineSeries();
t = 0.;
for(int i=0;i<length;i++) {
series_diff->append(t,results_y1[i]-results_y2[i]);
series1->append(t,results_y1[i]);
series2->append(t,results_y2[i]);
if(params->gaze_start >= params->gaze_end) {
series_gaze->append(t,-1.5);
} else if (t >= params->gaze_start && t <= params->gaze_end) {
series_gaze->append(t,-1.0);
} else series_gaze->append(t,-1.5);
t += params->h;
}

/* set series options */
series1->setName("y1 activation");
series2->setName("y2 activation");
series_diff->setName("y1-y2");
series_gaze->setName("gaze active");

QPen diff_pen(QRgb(0xff9933));
diff_pen.setWidth(2);
diff_pen.setStyle(Qt::DashLine);
series_diff->setPen(diff_pen);

/* create main chart */
QtCharts::QChart *chart = new QtCharts::QChart();
chart->legend()->setVisible(true);
//chart->setTitle("Gaze Model");

/* this will animate the series as it is displayed */
chart->setAnimationOptions(QtCharts::QChart::SeriesAnimations);
chart->setAnimationDuration(3000); // duration in milliseconds

/* create axes */
/* time axis */
QtCharts::QValueAxis *axisX = new QtCharts::QValueAxis;
axisX->setTitleText("Time");
axisX->setGridLineVisible(false);

/* activation axis */
QtCharts::QValueAxis *axisY = new QtCharts::QValueAxis;
axisY->setRange(-2.0,2.0);
axisY->setTitleText("Activation");
axisY->setGridLineVisible(false);

/* region axis */
QtCharts::QCategoryAxis *axisY2 = new QtCharts::QCategoryAxis;
axisY2->append("t2", 2);
axisY2->append("t1", 4);
axisY2->setRange(0,4);
axisY2->setLinePenColor(series_diff->pen().color());
//axisY3->setGridLinePen((series->pen()));

/* add axes to chart */
chart->addAxis(axisX, Qt::AlignBottom);
chart->addAxis(axisY, Qt::AlignLeft);
chart->addAxis(axisY2, Qt::AlignRight);

/* add data */
chart->addSeries(series1);
chart->addSeries(series2);
chart->addSeries(series_diff);
chart->addSeries(series_gaze);

/* attach axes */
series1->attachAxis(axisX);
series1->attachAxis(axisY);
series2->attachAxis(axisY);
series2->attachAxis(axisX);
// series_diff->attachAxis(axisY2);
series_diff->attachAxis(axisY);
series_diff->attachAxis(axisX);
series_gaze->attachAxis(axisX);
series_gaze->attachAxis(axisY);

/* main chart view goes on top */
chart_view = new QtCharts::QChartView(chart);
chart_view->setRenderHint(QPainter::Antialiasing);

/* button box comes next */
button_box = new QGroupBox;
QHBoxLayout *button_layout = new QHBoxLayout;
m_button_print = new QPushButton("print", this);
m_button_close = new QPushButton("close", this);
button_layout->addWidget(m_button_print);
button_layout->addWidget(m_button_close);
button_box->setLayout(button_layout);

/* assemble window: main layout */
QVBoxLayout *main_layout = new QVBoxLayout;
main_layout->addWidget(chart_view);
main_layout->addWidget(button_box);

setLayout(main_layout);
setWindowTitle(tr("Gaze Model"));
show();

/* wire the signals */
connect(m_button_close, SIGNAL (clicked()), this, SLOT(slot_close()));
connect(m_button_print, SIGNAL (clicked()), this, SLOT(slot_print()));
}

void ChartGaze::slot_close() {
free(params->n_I1);
free(params->n_I2);
free(params->n_w1);
free(params->n_w2);
free(params->n_g1);
free(params->n_g2);
free(params->n_l1);
free(params->n_l2);
free(results_y1);
free(results_y2);
this->close();
}

void ChartGaze::slot_print() {
/* create a printer */
QPrinter *printer = new QPrinter(QPrinter::HighResolution);
//printer->setOrientation(QPrinter::Landscape);
QPrintDialog *dialog = new QPrintDialog(printer);
dialog->setWindowTitle("Print Chart");
if (dialog->exec() != QDialog::Accepted)
return;

/* make a painter to paint onto the pages of the printer */
QPainter *painter = new QPainter();
painter->begin(printer);
painter->setRenderHint(QPainter::Antialiasing);
chart_view->render(painter, printer->pageRect());
painter->end();
delete(dialog);
delete(painter);
delete(printer);
}
52 changes: 52 additions & 0 deletions chart_gaze.h
View file
@@ -0,0 +1,52 @@
#ifndef CHART_GAZE_H
#define CHART_GAZE_H

#include <QApplication>
#include <QChart>
#include <QtCharts/QCategoryAxis>
#include <QChartView>
#include <QFlags>
#include <QGroupBox>
#include <QLineSeries>
//#include <QPainter>
//#include <QPdfWriter>
#include <QPrinter>
#include <QPrintDialog>
#include <QPushButton>
#include <QValueAxis>
#include <QVBoxLayout>
#include <QWidget>
#include "models.h"

class ChartGaze : public QWidget
{
Q_OBJECT
public:
explicit ChartGaze(params_gaze_t *p, QWidget *parent = 0);

private:
/* pointer to parameters */
params_gaze_t *params;

/* simulation results */
double *results_y1;
double *results_y2;

/* window component: the chart */
QtCharts::QChartView *chart_view;

/* window component: button_box */
QGroupBox *button_box;
QPushButton *m_button_close;
QPushButton *m_button_print;

signals:

public slots:

private slots:
void slot_close();
void slot_print();
};

#endif // CHART_GAZE_H

0 comments on commit 8126038

Please sign in to comment.