Initial commit

include/LiveScanClient/liveScanClient.h

@@ -0,0 +1,90 @@
+//   Copyright (C) 2015  Marek Kowalski (M.Kowalski@ire.pw.edu.pl), Jacek Naruniec (J.Naruniec@ire.pw.edu.pl)
+//   License: MIT Software License   See LICENSE.txt for the full license.
+
+//   If you use this software in your research, then please use the following citation:
+
+//    Kowalski, M.; Naruniec, J.; Daniluk, M.: "LiveScan3D: A Fast and Inexpensive 3D Data
+//    Acquisition System for Multiple Kinect v2 Sensors". in 3D Vision (3DV), 2015 International Conference on, Lyon, France, 2015
+
+//    @INPROCEEDINGS{Kowalski15,
+//        author={Kowalski, M. and Naruniec, J. and Daniluk, M.},
+//        booktitle={3D Vision (3DV), 2015 International Conference on},
+//        title={LiveScan3D: A Fast and Inexpensive 3D Data Acquisition System for Multiple Kinect v2 Sensors},
+//        year={2015},
+//    }
+#pragma once
+
+#include "resource.h"
+#include "ImageRenderer.h"
+#include "SocketCS.h"
+#include "calibration.h"
+#include "utils.h"
+#include "KinectCapture.h"
+#include <thread>
+#include <mutex>
+
+class LiveScanClient
+{
+public:
+    LiveScanClient();
+    ~LiveScanClient();
+
+
+    static LRESULT CALLBACK MessageRouter(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
+    LRESULT CALLBACK        DlgProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
+    int                     Run(HINSTANCE hInstance, int nCmdShow);
+
+	bool m_bSocketThread;
+private:
+	Calibration calibration;
+
+	bool m_bCalibrate;
+	bool m_bFilter;
+
+	ICapture *pCapture;
+
+	int m_nFilterNeighbors;
+	float m_fFilterThreshold;
+
+	bool m_bCaptureFrame;
+	bool m_bConnected;
+	bool m_bConfirmCaptured;
+	bool m_bConfirmCalibrated;
+	bool m_bShowDepth;
+	
+	SocketClient *m_pClientSocket;
+	std::vector<float> m_vBounds;
+
+	std::vector<Point3f> m_vLastFrameVertices;
+	std::vector<RGB> m_vLastFrameRGB;
+	std::vector<std::vector<Point3f>> m_vGatheredVertices;
+	std::vector<std::vector<RGB>> m_vGatheredRGBPoints;
+
+	HWND m_hWnd;
+    INT64 m_nLastCounter;
+    double m_fFreq;
+    INT64 m_nNextStatusTime;
+    DWORD m_nFramesSinceUpdate;	
+
+	Point3f* m_pCameraSpaceCoordinates;
+	Point2f* m_pColorCoordinates;
+
+    // Direct2D
+    ImageRenderer* m_pDrawColor;
+    ID2D1Factory* m_pD2DFactory;
+	RGB* m_pDepthRGBX;
+
+	void UpdateFrame();
+    void ProcessColor(RGB* pBuffer, int nWidth, int nHeight);
+	void ProcessDepth(const UINT16* pBuffer, int nHeight, int nWidth);
+
+    bool SetStatusMessage(_In_z_ WCHAR* szMessage, DWORD nShowTimeMsec, bool bForce);
+
+	void HandleSocket();
+	void SocketThreadFunction();
+	void StoreFrame(Point3f *vertices, Point2f *mapping, RGB *color);
+	void ShowFPS();
+	void ReadIPFromFile();
+	void WriteIPToFile();
+};
+