3D_RFUltrasound_Reconstruction/main.cpp at master · TJKlein/3D_RFUltrasound_Reconstruction · GitHub

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#include "compounding.h"
//#include <QtWidgets/QApplication>
#include <iostream>
#include <sstream>
#include "Reconstruction.h"

#include <QtWidgets\QApplication>
#include <QtDebug>
#include <QtGlobal>

#include <stdio.h>
#include <stdlib.h>
#include <Windows.h>

std::string computeElapsedTime(const clock_t &_startTimer, const clock_t &_stopTimer)
{
	int _iHours =(int)floor((float)(_stopTimer - _startTimer)/(CLOCKS_PER_SEC*60*60));
	int _iMinutes = (int)floor((float)(_stopTimer - _startTimer))/(CLOCKS_PER_SEC*60)-_iHours*60;
	int _iSeconds = ((float)(_stopTimer - _startTimer))/CLOCKS_PER_SEC - _iMinutes*60 - _iHours*60*60;
		char _cHours[3], _cMinutes[3], _cSeconds[3];
		sprintf(_cHours,"%d",_iHours);
		if ( strlen(_cHours) < 2)
			sprintf(_cHours,"0%d",_iHours);
		sprintf(_cMinutes,"%d",_iMinutes);
		if ( strlen(_cMinutes) < 2)
			sprintf(_cMinutes,"0%d",_iMinutes);
		sprintf(_cSeconds,"%d",_iSeconds);
		if ( strlen(_cSeconds) < 2)
			sprintf(_cSeconds,"0%d",_iSeconds);

	std::stringstream _ss;
	_ss <<  _cHours << ":" << _cMinutes << ":" << _cSeconds;

	return _ss.str();

}

int main(int argc, char *argv[])
{

	// Create the application's console
  // ISSUE_X64
	#if NDEBUG
    {


  if ( AllocConsole() == 0 )
      {
      ::MessageBox( 0, L"Can't allocate console!\n",
                       L"** PROGRAM NAME ***",
                       MB_OK | MB_ICONSTOP );

      return FALSE;
      }

   // We assume this stuff never fails
   //(!) add better error handling code later...   freopen( "CON", "w", stdout);

   freopen( "CON", "r", stdin );
   freopen( "CON", "w", stderr );
   freopen( "CON", "w", stdout );

   std::ios::sync_with_stdio();
   }
#endif

	QApplication a(argc, argv);

	//std::cout << argv[0] << " " << argv[1] << std::endl;
	Compounding w;
	w.show();

	if ( argc > 1 )
	{

		if ( strcmp(argv[1],"RF")==0 )
		{
			// MODE calibrationfile configurationfile trackingfile voxelSize voxelExtent=0,voxelDiameter=1 distance(0=0.5, 1=1,...) compoundingMode(0=GAUSSIAN, 1=NEAREST_NEIGHBOR, 2=MEDIAN, 3=WEIGHTED_MEDIAN, 4=INVERSE_DISTANCE, 5=MEAN) targetName referenceName(optional)
			BATCH_MODE mode;
			mode = BATCH_MODE::RF;

			std::string _reference = "";
			if (argc == 11 )
				_reference = std::string(argv[10]);

		w.batchRun(mode, std::string(argv[2]), std::string(argv[3]), std::string(argv[4]), atof(argv[5]),  (BATCH_DISTANCE_TYPE)atoi(argv[6]), (BATCH_DISTANCE)atoi(argv[7]),  (COMPOUNDING_MODE)atoi(argv[8]), std::string(argv[9]), _reference);

		}
		else if ( strcmp(argv[1],"BMODE")==0 )
		{

			// BMODE <Number of data sets> ouputFilename calibrationFilename video1 tracking1 ...
			BATCH_MODE mode;
			mode = BATCH_MODE::BMODE;


			std::vector<std::string> _trackingFiles;
			std::vector<std::string> _videoFiles;
		for(int k=0;k<atoi(argv[2])*2;k+=2)
		{
			_trackingFiles.push_back(std::string(argv[5+k+1]));
			_videoFiles.push_back(std::string(argv[5+k]));
		}


		int _offset = atoi(argv[2])*2 +5;

		std::string _reference = "";
			if (argc == _offset+8-1 )
				_reference = std::string(argv[_offset+6]);

			w.batchRunBMODE(mode, std::string(argv[4]), std::string(argv[3]), _videoFiles, _trackingFiles, atof(argv[_offset+0]),  (BATCH_DISTANCE_TYPE)atoi(argv[_offset+1]), (BATCH_DISTANCE)atoi(argv[_offset+2]),  (COMPOUNDING_MODE)atoi(argv[_offset+3]),(bool)atoi(argv[_offset+4]), std::string(argv[_offset+5]),  _reference);

		}
		else if ( strcmp(argv[1],"BUTTERWORTHENVELOPE")==0 )
		{
			// filename lowCut highCut order frequency
			w.batchButterwothEnvelope(std::string(argv[2]), atoi(argv[3]), atoi(argv[4]), atoi(argv[5]), atoi(argv[6]));
		}
		else if ( strcmp(argv[1],"INTENSITYCOMPOUND")==0 )
		{
			// filename voxelExtent=0,voxelDiameter=1 distance(0=0.5, 1=1,...) logCompress(0=false, 1=true)
			w.batchIntensityCompound(std::string(argv[2]),  (COMPOUNDING_MODE)atoi(argv[3]), (BATCH_DISTANCE_TYPE)atoi(argv[4]), (BATCH_DISTANCE)atoi(argv[5]), atoi(argv[6]));

		}

		return 0;
	}
	return a.exec();
}