ReadModelComputer.cpp
No OneTemporary
Actions

Subscribers

None

File Metadata


	#include <utility> // std::move()

	#include <ReadModelComputer.hpp>
	#include <Matrix2D.hpp>
	#include <Matrix4D.hpp>
	#include <ReadLayer.hpp>
	#include <ThreadPool.hpp>

	#include <iostream>

	ReadModelComputer::ReadModelComputer(Matrix2D<int>&& data,
	const Matrix4D<double>& post_prob,
	size_t n_threads)
	: ModelComputer(),
	threads(nullptr)
	{ // parameters
	size_t n_class = post_prob.get_dim()[1] ;
	size_t n_shift = post_prob.get_dim()[2] ;
	size_t n_flip = post_prob.get_dim()[3] ;
	bool flip = n_flip == 2 ;

	// the threads
	if(n_threads)
	{ this->threads = new ThreadPool(n_threads) ; }

	// the data and the model
	this->data_layer = new ReadLayer(std::move(data),
	n_class,
	n_shift,
	flip) ;
	this->data_layer->update_model(post_prob,
	this->threads) ;
	}

	ReadModelComputer::~ReadModelComputer()
	{ // threads
	if(this->threads != nullptr)
	{ this->threads->join() ;
	delete this->threads ;
	this->threads = nullptr ;
	}
	// data and model
	if(this->data_layer != nullptr)
	{ delete this->data_layer ;
	this->data_layer = nullptr ;
	}
	}