From 88833fb5db7e757fb06b37f843df033fb94f82af Mon Sep 17 00:00:00 2001
From: Cole Brinsfield <brinsfield.cole@gmail.com>
Date: Fri, 4 Dec 2020 19:44:53 -0800
Subject: [PATCH] decode pi revision to grab model (#17)

---
 CMakeLists.txt | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index c115505..ade9108 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -60,11 +60,12 @@ if(GST_BUILD)
         add_definitions(-DRPI)
         include(${CMAKE_CURRENT_SOURCE_DIR}/cmake_modules/functions.cmake)
         findRpiRevision( RPI_REVISION )
-        set(RPI_4_REVISIONS a03111 b03111 b03112 c03111 c03112 d03114)
-        if(RPI_REVISION IN_LIST RPI_4_REVISIONS)
+	math(EXPR RPI_MODEL "(0x${RPI_REVISION}>>4)&0xFF")
+        message( "-- Raspberry Pi Model: ${RPI_MODEL}" )
+        if(RPI_MODEL EQUAL 17)
             message("Raspberry Pi 4 Found")
             add_definitions(-DPI4)
-        endif(RPI_REVISION IN_LIST RPI_4_REVISIONS)
+        endif(RPI_REVISION EQUAL 17)
     endif(RPI_BUILD)
 message(STATUS "${GST_LIBRARIES}")
 endif(GST_BUILD)