src/image-load-cr3.cc

   1 /*
   2  * Copyright (C) 2020 The Geeqie Team
   3  *
   4  * Authors: Colin Clark
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include "image-load-cr3.h"
  22
  23 #include <cstring>
  24
  25 #include <glib.h>
  26
  27 #include "image-load-jpeg.h"
  28 #include "image-load.h"
  29
  30 namespace
  31 {
  32
  33 struct ImageLoaderCr3 : public ImageLoaderJpeg
  34 {
  35 public:
  36         gboolean write(const guchar *buf, gsize &chunk_size, gsize count, GError **error) override;
  37         gchar *get_format_name() override;
  38         gchar **get_format_mime_types() override;
  39 };
  40
  41 gboolean ImageLoaderCr3::write(const guchar *buf, gsize &chunk_size, gsize count, GError **error)
  42 {
  43         /** @FIXME Just start search at where full size jpeg should be,
  44          * then search through the file looking for a jpeg end-marker
  45          */
  46         gboolean found = FALSE;
  47         gint i;
  48         guint n;
  49
  50         n = 0;
  51         while (n < count - 4 && !found)
  52                 {
  53                 if (memcmp(&buf[n], "mdat", 4) == 0)
  54                         {
  55                         if (memcmp(&buf[n + 12], "\xFF\xD8", 2) == 0)
  56                                 {
  57                                 i = 0;
  58                                 while (!found )
  59                                         {
  60                                         if (memcmp(&buf[n + 12 + i], "\xFF\xD9", 2) == 0)
  61                                                 {
  62                                                 found = TRUE;
  63                                                 }
  64                                         i++;
  65                                         }
  66                                 }
  67                         else
  68                                 {
  69                                 break;
  70                                 }
  71                         }
  72                 else
  73                         {
  74                         n++;
  75                         }
  76                 }
  77
  78         if (!found)
  79                 {
  80                 return FALSE;
  81                 }
  82
  83         gboolean ret = ImageLoaderJpeg::write(buf + n + 12, chunk_size, i, error);
  84         if (ret)
  85                 {
  86                 chunk_size = count;
  87                 }
  88         return ret;
  89 }
  90
  91 gchar *ImageLoaderCr3::get_format_name()
  92 {
  93         return g_strdup("cr3");
  94 }
  95
  96 gchar **ImageLoaderCr3::get_format_mime_types()
  97 {
  98         static const gchar *mime[] = {"image/x-canon-cr3", nullptr};
  99         return g_strdupv(const_cast<gchar **>(mime));
 100 }
 101
 102 } // namespace
 103
 104 std::unique_ptr<ImageLoaderBackend> get_image_loader_backend_cr3()
 105 {
 106         return std::make_unique<ImageLoaderCr3>();
 107 }
 108
 109 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */