summaryrefslogtreecommitdiffstats
path: root/xbmc/pictures/JpegParse.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'xbmc/pictures/JpegParse.cpp')
-rw-r--r--xbmc/pictures/JpegParse.cpp316
1 files changed, 316 insertions, 0 deletions
diff --git a/xbmc/pictures/JpegParse.cpp b/xbmc/pictures/JpegParse.cpp
new file mode 100644
index 0000000..718ac54
--- /dev/null
+++ b/xbmc/pictures/JpegParse.cpp
@@ -0,0 +1,316 @@
+/*
+ * Copyright (C) 2005-2018 Team Kodi
+ * This file is part of Kodi - https://kodi.tv
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ * See LICENSES/README.md for more information.
+ */
+
+//--------------------------------------------------------------------------
+// This module gathers information about a digital image file. This includes:
+// - File name and path
+// - File size
+// - Resolution (if available)
+// - IPTC information (if available)
+// - EXIF information (if available)
+// All gathered information is stored in a vector of 'description' and 'value'
+// pairs (where both description and value fields are of CStdString types).
+//--------------------------------------------------------------------------
+
+#include "JpegParse.h"
+
+#include "filesystem/File.h"
+
+#ifdef TARGET_WINDOWS
+#include <windows.h>
+#else
+#include <memory.h>
+#include <cstring>
+typedef unsigned char BYTE;
+#endif
+
+#ifndef min
+#define min(a,b) (a)>(b)?(b):(a)
+#endif
+
+using namespace XFILE;
+
+//--------------------------------------------------------------------------
+#define JPEG_PARSE_STRING_ID_BASE 21500
+enum {
+ ProcessUnknown = JPEG_PARSE_STRING_ID_BASE,
+ ProcessSof0,
+ ProcessSof1,
+ ProcessSof2,
+ ProcessSof3,
+ ProcessSof5,
+ ProcessSof6,
+ ProcessSof7,
+ ProcessSof9,
+ ProcessSof10,
+ ProcessSof11,
+ ProcessSof13,
+ ProcessSof14,
+ ProcessSof15,
+};
+
+
+
+
+//--------------------------------------------------------------------------
+// Constructor
+//--------------------------------------------------------------------------
+CJpegParse::CJpegParse():
+ m_SectionBuffer(NULL)
+{
+ memset(&m_ExifInfo, 0, sizeof(m_ExifInfo));
+ memset(&m_IPTCInfo, 0, sizeof(m_IPTCInfo));
+}
+
+//--------------------------------------------------------------------------
+// Process a SOFn marker. This is useful for the image dimensions
+//--------------------------------------------------------------------------
+void CJpegParse::ProcessSOFn (void)
+{
+ m_ExifInfo.Height = CExifParse::Get16(m_SectionBuffer+3);
+ m_ExifInfo.Width = CExifParse::Get16(m_SectionBuffer+5);
+
+ unsigned char num_components = m_SectionBuffer[7];
+ if (num_components != 3)
+ {
+ m_ExifInfo.IsColor = 0;
+ }
+ else
+ {
+ m_ExifInfo.IsColor = 1;
+ }
+}
+
+
+//--------------------------------------------------------------------------
+// Read a section from a JPEG file. Note that this function allocates memory.
+// It must be called in pair with ReleaseSection
+//--------------------------------------------------------------------------
+bool CJpegParse::GetSection (CFile& infile, const unsigned short sectionLength)
+{
+ if (sectionLength < 2)
+ {
+ printf("JpgParse: invalid section length");
+ return false;
+ }
+
+ m_SectionBuffer = new unsigned char[sectionLength];
+ if (m_SectionBuffer == NULL)
+ {
+ printf("JpgParse: could not allocate memory");
+ return false;
+ }
+ // Store first two pre-read bytes.
+ m_SectionBuffer[0] = (unsigned char)(sectionLength >> 8);
+ m_SectionBuffer[1] = (unsigned char)(sectionLength & 0x00FF);
+
+ unsigned int len = (unsigned int)sectionLength;
+
+ size_t bytesRead = infile.Read(m_SectionBuffer+sizeof(sectionLength), len-sizeof(sectionLength));
+ if (bytesRead != sectionLength-sizeof(sectionLength))
+ {
+ printf("JpgParse: premature end of file?");
+ ReleaseSection();
+ return false;
+ }
+ return true;
+}
+
+//--------------------------------------------------------------------------
+// Deallocate memory allocated in GetSection. This function must always
+// be paired by a preceding GetSection call.
+//--------------------------------------------------------------------------
+void CJpegParse::ReleaseSection (void)
+{
+ delete[] m_SectionBuffer;
+ m_SectionBuffer = NULL;
+}
+
+//--------------------------------------------------------------------------
+// Parse the marker stream until SOS or EOI is seen; infile has already been
+// successfully open
+//--------------------------------------------------------------------------
+bool CJpegParse::ExtractInfo (CFile& infile)
+{
+ // Get file marker (two bytes - must be 0xFFD8 for JPEG files
+ BYTE a;
+ size_t bytesRead = infile.Read(&a, sizeof(BYTE));
+ if ((bytesRead != sizeof(BYTE)) || (a != 0xFF))
+ {
+ return false;
+ }
+ bytesRead = infile.Read(&a, sizeof(BYTE));
+ if ((bytesRead != sizeof(BYTE)) || (a != M_SOI))
+ {
+ return false;
+ }
+
+ for(;;)
+ {
+ BYTE marker = 0;
+ for (a=0; a<7; a++) {
+ bytesRead = infile.Read(&marker, sizeof(BYTE));
+ if (marker != 0xFF)
+ break;
+
+ if (a >= 6)
+ {
+ printf("JpgParse: too many padding bytes");
+ return false;
+ }
+ marker = 0;
+ }
+
+ // Read the length of the section.
+ unsigned short itemlen = 0;
+ bytesRead = infile.Read(&itemlen, sizeof(itemlen));
+ itemlen = CExifParse::Get16(&itemlen);
+
+ if ((bytesRead != sizeof(itemlen)) || (itemlen < sizeof(itemlen)))
+ {
+ printf("JpgParse: invalid marker");
+ return false;
+ }
+
+ switch(marker)
+ {
+ case M_SOS: // stop before hitting compressed data
+ return true;
+
+ case M_EOI: // in case it's a tables-only JPEG stream
+ printf("JpgParse: No image in jpeg!");
+ return false;
+ break;
+
+ case M_COM: // Comment section
+ GetSection(infile, itemlen);
+ if (m_SectionBuffer != NULL)
+ {
+ // CExifParse::FixComment(comment); // Ensure comment is printable
+ unsigned short length = min(itemlen - 2, MAX_COMMENT);
+ strncpy(m_ExifInfo.FileComment, (char *)&m_SectionBuffer[2], length);
+ m_ExifInfo.FileComment[length] = '\0';
+ }
+ ReleaseSection();
+ break;
+
+ case M_SOF0:
+ case M_SOF1:
+ case M_SOF2:
+ case M_SOF3:
+ case M_SOF5:
+ case M_SOF6:
+ case M_SOF7:
+ case M_SOF9:
+ case M_SOF10:
+ case M_SOF11:
+ case M_SOF13:
+ case M_SOF14:
+ case M_SOF15:
+ GetSection(infile, itemlen);
+ if ((m_SectionBuffer != NULL) && (itemlen >= 7))
+ {
+ ProcessSOFn();
+ m_ExifInfo.Process = marker;
+ }
+ ReleaseSection();
+ break;
+
+ case M_IPTC:
+ GetSection(infile, itemlen);
+ if (m_SectionBuffer != NULL)
+ {
+ CIptcParse::Process(m_SectionBuffer, itemlen, &m_IPTCInfo);
+ }
+ ReleaseSection();
+ break;
+
+ case M_EXIF:
+ // Seen files from some 'U-lead' software with Vivitar scanner
+ // that uses marker 31 for non exif stuff. Thus make sure
+ // it says 'Exif' in the section before treating it as exif.
+ GetSection(infile, itemlen);
+ if (m_SectionBuffer != NULL)
+ {
+ CExifParse exif;
+ exif.Process(m_SectionBuffer, itemlen, &m_ExifInfo);
+ }
+ ReleaseSection();
+ break;
+
+ case M_JFIF:
+ // Regular jpegs always have this tag, exif images have the exif
+ // marker instead, although ACDsee will write images with both markers.
+ // this program will re-create this marker on absence of exif marker.
+ // hence no need to keep the copy from the file.
+ // fall through to default case
+ default:
+ // Skip any other sections.
+ GetSection(infile, itemlen);
+ ReleaseSection();
+ break;
+ }
+ }
+ return true;
+}
+
+//--------------------------------------------------------------------------
+// Process a file. Check if it is JPEG. Extract exif/iptc info if it is.
+//--------------------------------------------------------------------------
+bool CJpegParse::Process (const char *picFileName)
+{
+ CFile file;
+
+ if (!file.Open(picFileName))
+ return false;
+
+ // File exists and successfully opened. Start processing
+ // Gather all information about the file
+
+/* // Get file name...
+ CStdString tmp, urlFName, path;
+ CURL url(picFileName);
+ url.GetURLWithoutUserDetails(urlFName);
+ CUtil::Split(urlFName, path, tmp);
+ m_JpegInfo[SLIDESHOW_FILE_NAME] = tmp;
+ // ...then path...
+ m_JpegInfo[SLIDESHOW_FILE_PATH] = path;
+
+ // ...then size...
+ __stat64 fileStat;
+ CFile::Stat(picFileName, &fileStat);
+ float fileSize = (float)fileStat.st_size;
+ tmp = "";
+ if (fileSize > 1024)
+ {
+ fileSize /= 1024;
+ tmp = "KB";
+ }
+ if (fileSize > 1024)
+ {
+ fileSize /= 1024;
+ tmp = "MB";
+ }
+ if (fileSize > 1024)
+ {
+ fileSize /= 1024;
+ tmp = "GB";
+ }
+ tmp.Format("%.2f %s", fileSize, tmp);
+ m_JpegInfo[SLIDESHOW_FILE_SIZE] = tmp;
+
+ // ...then date and time...
+ CDateTime date((time_t)fileStat.st_mtime);
+ tmp.Format("%s %s", date.GetAsLocalizedDate(), date.GetAsLocalizedTime());
+ m_JpegInfo[SLIDESHOW_FILE_DATE] = tmp;*/
+
+ bool result = ExtractInfo(file);
+ file.Close();
+ return result;
+}
+
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-10 22:22:07 +0000