213 lines
6.9 KiB
C++
213 lines
6.9 KiB
C++
|
/*
|
||
|
|
Copyright (c) 2005-2019 Intel Corporation
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "seismic_video.h"
|
||
|
|
#include "universe.h"
|
||
|
|
#include <taskflow/taskflow.hpp>
|
||
|
|
#include <tbb/task_scheduler_init.h>
|
||
|
|
#include <tbb/flow_graph.h>
|
||
|
|
#include <tbb/partitioner.h>
|
||
|
|
#include <tbb/blocked_range.h>
|
||
|
|
#include <tbb/parallel_for.h>
|
||
|
|
|
||
|
|
|
||
|
|
const char * const SeismicVideo::titles[2] = {"Seismic Simulation: Serial", "Seismic Simulation: Parallel"};
|
||
|
|
void SeismicVideo::on_mouse(int x, int y, int key) {
|
||
|
|
if(key == 1){
|
||
|
|
u_.TryPutNewPulseSource(x,y);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void SeismicVideo::on_key(int key) {
|
||
|
|
key &= 0xff;
|
||
|
|
if(char(key) == ' ') initIsParallel = !initIsParallel;
|
||
|
|
else if(char(key) == 'p') initIsParallel = true;
|
||
|
|
else if(char(key) == 's') initIsParallel = false;
|
||
|
|
else if(char(key) == 'e') updating = true;
|
||
|
|
else if(char(key) == 'd') updating = false;
|
||
|
|
else if(key == 27) running = false;
|
||
|
|
title = titles[initIsParallel?1:0];
|
||
|
|
}
|
||
|
|
|
||
|
|
void SeismicVideo::on_process() {
|
||
|
|
if(initIsParallel) {
|
||
|
|
if(u_.get_model() == "tf") {
|
||
|
|
tf::Executor executor(threadsHigh);
|
||
|
|
tf::Taskflow taskflow;
|
||
|
|
int UniverseHeight = u_.UniverseHeight;
|
||
|
|
auto stress_tasks = taskflow.dynamic_parallel_for(1, UniverseHeight, 1, [&](int i) mutable {
|
||
|
|
u_.UpdateStress(Universe::Rectangle(0, i, u_.UniverseWidth-1, 1));
|
||
|
|
}, executor.num_workers());
|
||
|
|
|
||
|
|
auto velocity_tasks = taskflow.dynamic_parallel_for(1, UniverseHeight, 1, [&](int i) mutable {
|
||
|
|
u_.UpdateVelocity(Universe::Rectangle(1, i, u_.UniverseWidth-1, 1));
|
||
|
|
}, executor.num_workers());
|
||
|
|
|
||
|
|
std::get<1>(stress_tasks).precede(std::get<0>(velocity_tasks));
|
||
|
|
std::get<0>(stress_tasks).work([&](){ u_.UpdatePulse(); });
|
||
|
|
|
||
|
|
for( int frames = 0; numberOfFrames_==0 || frames<numberOfFrames_; ++frames ) {
|
||
|
|
executor.run(taskflow).wait();
|
||
|
|
if( !next_frame() ) break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if(u_.get_model() == "tbb") {
|
||
|
|
struct UpdateStressBody {
|
||
|
|
Universe & u_;
|
||
|
|
UpdateStressBody(Universe & u):u_(u){}
|
||
|
|
void operator()( const tbb::blocked_range<int>& range ) const {
|
||
|
|
u_.UpdateStress(Universe::Rectangle(0, range.begin(), u_.UniverseWidth-1, range.size()));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
struct UpdateVelocityBody {
|
||
|
|
Universe & u_;
|
||
|
|
UpdateVelocityBody(Universe & u):u_(u){}
|
||
|
|
void operator()( const tbb::blocked_range<int>& y_range ) const {
|
||
|
|
u_.UpdateVelocity(Universe::Rectangle(1,y_range.begin(),u_.UniverseWidth-1,y_range.size()));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
using namespace tbb;
|
||
|
|
using namespace tbb::flow;
|
||
|
|
|
||
|
|
tbb::task_scheduler_init init(threadsHigh);
|
||
|
|
|
||
|
|
static tbb::affinity_partitioner affinity;
|
||
|
|
|
||
|
|
for( int frames = 0; numberOfFrames_==0 || frames<numberOfFrames_; ++frames ) {
|
||
|
|
u_.UpdatePulse();
|
||
|
|
tbb::parallel_for(tbb::blocked_range<int>( 0, u_.UniverseHeight-1 ), // Index space for loop
|
||
|
|
UpdateStressBody(u_), // Body of loop
|
||
|
|
affinity); // Affinity hint
|
||
|
|
tbb::parallel_for(tbb::blocked_range<int>( 1, u_.UniverseHeight ), // Index space for loop
|
||
|
|
UpdateVelocityBody(u_), // Body of loop
|
||
|
|
affinity); // Affinity hint
|
||
|
|
if( !next_frame() ) break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
for( int frames = 0; numberOfFrames_==0 || frames<numberOfFrames_; ++frames ) {
|
||
|
|
u_.SerialUpdateUniverse();
|
||
|
|
if( !next_frame() ) break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
//tbb::task_scheduler_init Init(threadsHigh);
|
||
|
|
//for( int frames = 0; numberOfFrames_==0 || frames<numberOfFrames_; ++frames ) {
|
||
|
|
// if( initIsParallel ) {
|
||
|
|
// u_.ParallelUpdateUniverse();
|
||
|
|
// else
|
||
|
|
// u_.SerialUpdateUniverse();
|
||
|
|
// if( !next_frame() ) break;
|
||
|
|
//}
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef _WINDOWS
|
||
|
|
#include "msvs/resource.h"
|
||
|
|
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
|
||
|
|
SeismicVideo * gVideo = NULL;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
SeismicVideo::SeismicVideo( Universe &u, int number_of_frames, int threads_high, bool init_is_parallel)
|
||
|
|
:numberOfFrames_(number_of_frames), initIsParallel(init_is_parallel), u_(u), threadsHigh(threads_high)
|
||
|
|
{
|
||
|
|
title = titles[initIsParallel?1:0];
|
||
|
|
#ifdef _WINDOWS
|
||
|
|
gVideo = this;
|
||
|
|
LoadStringA(video::win_hInstance, IDC_SEISMICSIMULATION, szWindowClass, MAX_LOADSTRING);
|
||
|
|
memset(&wcex, 0, sizeof(wcex));
|
||
|
|
wcex.lpfnWndProc = (WNDPROC)WndProc;
|
||
|
|
wcex.hIcon = LoadIcon(video::win_hInstance, MAKEINTRESOURCE(IDI_SEISMICSIMULATION));
|
||
|
|
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
|
||
|
|
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
|
||
|
|
wcex.lpszMenuName = LPCTSTR(IDC_SEISMICSIMULATION);
|
||
|
|
wcex.lpszClassName = szWindowClass;
|
||
|
|
wcex.hIconSm = LoadIcon(video::win_hInstance, MAKEINTRESOURCE(IDI_SMALL));
|
||
|
|
win_set_class(wcex); // ascii convention here
|
||
|
|
win_load_accelerators(IDC_SEISMICSIMULATION);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#ifdef _WINDOWS
|
||
|
|
//
|
||
|
|
// FUNCTION: WndProc(HWND, unsigned, WORD, LONG)
|
||
|
|
//
|
||
|
|
// PURPOSE: Processes messages for the main window.
|
||
|
|
//
|
||
|
|
// WM_COMMAND - process the application menu
|
||
|
|
// WM_PAINT - Paint the main window
|
||
|
|
// WM_DESTROY - post a quit message and return
|
||
|
|
//
|
||
|
|
//
|
||
|
|
LRESULT CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) {
|
||
|
|
switch (message)
|
||
|
|
{
|
||
|
|
case WM_INITDIALOG: return TRUE;
|
||
|
|
case WM_COMMAND:
|
||
|
|
if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL) {
|
||
|
|
EndDialog(hDlg, LOWORD(wParam));
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
|
||
|
|
{
|
||
|
|
int wmId;
|
||
|
|
switch (message) {
|
||
|
|
case WM_COMMAND:
|
||
|
|
wmId = LOWORD(wParam);
|
||
|
|
// Parse the menu selections:
|
||
|
|
switch (wmId)
|
||
|
|
{
|
||
|
|
case IDM_ABOUT:
|
||
|
|
DialogBox(video::win_hInstance, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, (DLGPROC)About);
|
||
|
|
break;
|
||
|
|
case IDM_EXIT:
|
||
|
|
PostQuitMessage(0);
|
||
|
|
break;
|
||
|
|
case ID_FILE_PARALLEL:
|
||
|
|
gVideo->on_key('p');
|
||
|
|
break;
|
||
|
|
case ID_FILE_SERIAL:
|
||
|
|
gVideo->on_key('s');
|
||
|
|
break;
|
||
|
|
case ID_FILE_ENABLEGUI:
|
||
|
|
gVideo->on_key('e');
|
||
|
|
break;
|
||
|
|
case ID_FILE_DISABLEGUI:
|
||
|
|
gVideo->on_key('d');
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
return DefWindowProc(hWnd, message, wParam, lParam);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
return DefWindowProc(hWnd, message, wParam, lParam);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|