ntm

ppg.cpp

@@ -1,34 +1,67 @@
-
+#include <iostream>
 #include "opencv2/opencv.hpp"
 #include "opencv2/videoio.hpp"
-#include "opencv2/highgui.hpp
-#include <iostream>
+#include "opencv2/highgui.hpp"
 
-int main()
-{
-        std::cout <<"PPG algorithm"<< std::endl;
-	return 0;
-        cv::VideoCapture cap;
-        cap.open(0);
-        if (!cap.isOpened())
-        {
+#define FPS 30.0
+#define temps 1.0
+
+bool isDiscardData = true;
+int countDiscard = 0;
+
+int main() {
+    std::cout << "PPG, j'adore gitbash même si avec github desktop c'est mieux" << std::endl;
+
+    cv::VideoCapture cap;
+    cap.open(0);
+
+    cv::CascadeClassifier faceDetector;
+    if (!faceDetector.load("./haarcascade_frontalface_alt.xml")) {
+        std::cerr << "[ERROR] Unable to load face cascade" << std::endl;
+        return -1;
+    }
+
+    if (!cap.isOpened()) {
         std::cerr << "[ERROR] Unable to open camera!" << std::endl;
         return -2;
+    }
+
+    while (true) {
+        if (isDiscardData) {
+            countDiscard++;
+            std::cout << "countDiscard = " << countDiscard << std::endl;
+            if (countDiscard == temps*FPS) {
+                isDiscardData = false;
+            }
+        } else {
+            // create a matrix to store the image from the cam
+            cv::Mat frame;
+            // wait for a new frame from camera and store it into 'frame'
+            cap.read(frame);
+            // check if we succeeded
+            if (frame.empty()) {
+                std::cerr << "[ERROR] blank frame grabbed" << std::endl;
+                break;
+            }
+
+            cv::Mat frame_gray;
+            cv::cvtColor(frame, frame_gray, cv::COLOR_BGR2GRAY);
+            std::vector<cv::Rect> faceRectangles;
+            faceDetector.detectMultiScale(frame_gray, faceRectangles, 1.1, 3, 0, cv::Size(20, 20)); 
+            for (size_t i = 0; i < faceRectangles.size(); i++) {
+                cv::rectangle(frame, faceRectangles[i], cv::Scalar(255, 0, 0), 2);
+                cv::Rect foreheadROI = faceRectangles[i];
+                foreheadROI.height *= 0.3;
+                cv::rectangle(frame, foreheadROI, cv::Scalar(0, 255, 0), 2);
+            }
+            cv::imshow("Color", frame);
         }
-        while (true)
-        {
-        // create a matrix to store the image from the cam
-        cv::Mat frame;
-        // wait for a new frame from camera and store it into 'frame'
-        cap.read(frame);
-        // check if we succeeded
-        if (frame.empty())
-        {
-        std::cerr << "[ERROR] blank frame grabbed" << std::endl;
-        break;
-        }
-        cv::imshow("Color", frame);
-        if (cv::waitKey(1000.0/FPS) >= 0)
-        break;
+
+        if (cv::waitKey(1000.0/FPS) >= 0) {
+            break;
         }
+    }
+
+    return 0;
 }
+