KylesFilmScanner/Scanner.cpp at master · kylemikableh/KylesFilmScanner · 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
132
133
134
135
136
137
138
/*
*   Scanner.cpp
*	Film Scanner Master PC Control Software by Kyle Mikolajczyk
*   kyle@kylem.org
*/

// #define ARDUINO
#define MDRIVE

#ifdef ARDUINO
    #define ARDUINO_BAUD_RATE 115200
    #define ARDUINO_PORT "COM6"
    #define ARDUINO_MSG_START_DELIM '('
    #define ARDUINO_MSG_END_DELIM ')'
#endif

#ifdef MDRIVE
    #define MDRIVE_PORT "COM5"
    #define MDRIVE_BAUD_RATE 9600
#endif

#include "SerialConn.h"
#include "ImageCaptureController.h"
#include "MDriveConn.h"

#include <OpenImageIO/imagebuf.h>

bool useCamera = false;
bool enableSerialComms = false;

/*
* Create the main Image Controller to handle the scans for this ID.
* (ID will we configurable later)
*/
ImageCaptureController* initializeImageController() {
    ImageCaptureController* imageCaptureController = nullptr;
    ImageCaptureController::initializePylon();
    imageCaptureController = new ImageCaptureController("EK00001");

    return imageCaptureController;
}

/*
* If we have enabled the serial connection, initialize it.
* If its not available, bail out
*/
#ifdef ARDUINO
SerialConn* getArudinoConnection() {
    SerialConn* arduinoConnection = nullptr;
    if (enableSerialComms) {
        try {
            arduinoConnection = new SerialConn(ARDUINO_BAUD_RATE, ARDUINO_PORT);
        }
        catch (const std::exception& e) {
            std::cerr << "Failed to open serial port: " << ARDUINO_PORT << ". Please check connection or change port." << std::endl;
            std::exit(EXIT_FAILURE);
        }
    }
    return arduinoConnection;
}
#endif

int main(int argc, char* argv[])
{
    int exitCode = 0;

#ifdef ARDUINO
    SerialConn* arduinoConnection = getArudinoConnection();
#endif

#ifdef MDRIVE
	// Initialize the MDrive connection
	MDriveConn* mDriveConnection = new MDriveConn(MDRIVE_PORT, MDRIVE_BAUD_RATE);
	mDriveConnection->initializeAndHome();
#endif


    ImageCaptureController* imageCaptureController;

    if (useCamera) {
        imageCaptureController = initializeImageController();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
        imageCaptureController->captureFrame();
    }

    int i = 0;
    while (enableSerialComms) {

#ifdef ARDUINO
        if (i > 10) { break; }
        cout << "Sent command to set color to RED" << endl;
        // Test out sending command to arduino with pause
		if (i == 0) {
			arduinoConnection->sendCommand(SerialConn::SET_COLOR_RED);
		}
		else if (i == 1) {
			arduinoConnection->sendCommand(SerialConn::SET_COLOR_BLUE);
		}
		else if (i == 2) {
			arduinoConnection->sendCommand(SerialConn::GET_FRAME_ID);
		}
		else if (i == 3) {
			arduinoConnection->sendCommand(SerialConn::RESET_FRAME_ID);
		}
		else {
			arduinoConnection->sendCommand(SerialConn::SET_COLOR_GREEN);
		}
		arduinoConnection->sendCommand(SerialConn::SET_COLOR_RED);

		cout << "Waiting for response" << endl;
        char* buffer = arduinoConnection->readMessage(ARDUINO_MSG_START_DELIM, ARDUINO_MSG_END_DELIM);
		cout << "Response recieved" << endl;
        if (buffer != nullptr) {
            cout << buffer << endl;
            arduinoConnection->parseMessage(buffer);
        }
        else {
            cout << "No response recieved for try " << i << endl;
        }
#endif

		Sleep(25);
        i++;
    }

	if (useCamera) {
		delete imageCaptureController; // Clean up the dynamically allocated memory
	}

    return exitCode;
}