/*

 * libopenraw - rw2file.cpp

 *

 * Copyright (C) 2011-2017 Hubert Figuiere

 *

 * This library is free software: you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public License

 * as published by the Free Software Foundation, either version 3 of

 * the License, or (at your option) any later version.

 *

 * This library is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this library.  If not, see

 * <http://www.gnu.org/licenses/>.

 */

#include <stddef.h>

#include <algorithm>

#include <cstdint>

#include <memory>

#include <libopenraw/cameraids.h>

#include <libopenraw/debug.h>

#include "rawdata.hpp"

#include "trace.hpp"

#include "io/streamclone.hpp"

#include "rw2file.hpp"

#include "rw2container.hpp"

#include "jfifcontainer.hpp"

#include "rawfile_private.hpp"

using namespace Debug;

namespace OpenRaw {

namespace Internals {

#define OR_MAKE_PANASONIC_TYPEID(camid) \

    OR_MAKE_FILE_TYPEID(OR_TYPEID_VENDOR_PANASONIC,camid)

#define OR_MAKE_LEICA_TYPEID(camid) \

    OR_MAKE_FILE_TYPEID(OR_TYPEID_VENDOR_LEICA,camid)

/* taken from dcraw, by default */

static const BuiltinColourMatrix s_matrices[] = {

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_CM1),

    15,

    0,

    { 8770, -3194, -820, -2871, 11281, 1803, -513, 1552, 4434 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF1),

    15,

    0xf92,

    { 7888,-1902,-1011,-8106,16085,2099,-2353,2866,7330 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF2),

    15,

    0xfff,

    { 7888,-1902,-1011,-8106,16085,2099,-2353,2866,7330 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF3),

    15,

    0xfff,

    { 9051,-2468,-1204,-5212,13276,2121,-1197,2510,6890 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF5),

    15,

    0xfff,

    { 8228,-2945,-660,-3938,11792,2430,-1094,2278,5793 } },

  // Adobe DNG convert 7.4

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF6),

    15,

    0xfff,

    { 8130,-2801,-946,-3520,11289,2552,-1314,2511,5791 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF7),

    15,

    0,

    { 7610, -2780, -576, -4614, 12195, 2733, -1375, 2393, 6490 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX1),

    15,

    0,

    { 6763,-1919,-863,-3868,11515,2684,-1216,2387,5879 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX7),

    15,

    0,

    { 7610,-2780,-576,-4614,12195,2733,-1375,2393,6490 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX8),

    15,

    0,

    { 7564, -2263, -606, -3148, 11239, 2177, -540, 1435, 4853 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ8),

    0,

    0xf7f,

    { 8986,-2755,-802,-6341,13575,3077,-1476,2144,6379 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ18),

    0,

    0,

    { 9932,-3060,-935,-5809,13331,2753,-1267,2155,5575 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ28),

    15,

    0xf96,

    { 10109,-3488,-993,-5412,12812,2916,-1305,2140,5543 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ200),

    143,

    0xfff,

    { 8112,-2563,-740,-3730,11784,2197,-941,2075,4933 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ30),

    0,

    0xf94,

    { 10976,-4029,-1141,-7918,15491,2600,-1670,2071,8246 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ330),

    15,

    0,

    { 8378, -2798, -769, -3068, 11410, 1877, -538, 1792, 4623 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ50),

    0,

    0,

    { 7906,-2709,-594,-6231,13351,3220,-1922,2631,6537 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ100),

    143,

    0xfff,

    { 16197,-6146,-1761,-2393,10765,1869,366,2238,5248 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G1),

    15,

    0xf94,

    { 8199,-2065,-1056,-8124,16156,2033,-2458,3022,7220 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G2),

    15,

    0xf3c,

    { 10113,-3400,-1114,-4765,12683,2317,-377,1437,6710 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G3),

    143,

    0xfff,

    { 6763,-1919,-863,-3868,11515,2684,-1216,2387,5879 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G5),

    143,

    0xfff,

    { 7798,-2562,-740,-3879,11584,2613,-1055,2248,5434 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G10),

    0,

    0,

    { 10113,-3400,-1114,-4765,12683,2317,-377,1437,6710 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH1),

    15,

    0xf92,

    { 6299,-1466,-532,-6535,13852,2969,-2331,3112,5984 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH2),

    15,

    0xf95,

    { 7780,-2410,-806,-3913,11724,2484,-1018,2390,5298 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH3),

    144,

    0,

    { 6559,-1752,-491,-3672,11407,2586,-962,1875,5130 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH4),

    15,

    0,

    { 7122,-2108,-512,-3155,11201,2231,-541,1423,5045 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GM1),

    15,

    0,

    { 6770,-1895,-744,-5232,13145,2303,-1664,2691,5703 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX2),

    0,

    0,

    { 8048,-2810,-623,-6450,13519,3272,-1700,2146,7049 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX3),

    15,

    0,

    { 8128,-2668,-655,-6134,13307,3161,-1782,2568,6083 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX5),

    143,

    0,

    { 10909,-4295,-948,-1333,9306,2399,22,1738,4582 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX7),

    143,

    0,

    { 10148,-3743,-991,-2837,11366,1659,-701,1893,4899 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX100),

    143,

    0,

    { 8844,-3538,-768,-3709,11762,2200,-698,1792,5220 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_L1),

    0,

    0xf7f,

    { 8054,-1885,-1025,-8349,16367,2040,-2805,3542,7629 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_L10),

    15,

    0xf96,

    { 8025,-1942,-1050,-7920,15904,2100,-2456,3005,7039 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ70),

    15,

    0,

    { 8802, -3135, -789, -3151, 11468, 1904, -550, 1745, 4810 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ80),

    15,

    0,

    { 8550, -2908, -842, -3195, 11529, 1881, -338, 1603, 4631 } },

  { OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ100),

    15,

    0,

    {  7790, -2736, -755, -3452, 11870, 1769, -628, 1647, 4898 } },

  { OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_DIGILUX2),

    0,

    0,

    { 11340,-4069,-1275,-7555,15266,2448,-2960,3426,7685 } },

  { OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_DLUX_3),

    0,

    0,

    { 8048,-2810,-623,-6450,13519,3272,-1700,2146,7049 } },

  { OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_VLUX_1),

    0,

    0,

    { 7906,-2709,-594,-6231,13351,3220,-1922,2631,6537 } },

  { 0, 0, 0, { 0, 0, 0, 0, 0, 0, 0, 0, 0 } }

};

const IfdFile::camera_ids_t Rw2File::s_def[] = {

	{ "DMC-CM1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_CM1) },

	{ "DMC-GF1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF1) },

	{ "DMC-GF2", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF2) },

	{ "DMC-GF3", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF3) },

	{ "DMC-GF5", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF5) },

	{ "DMC-GF6", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF6) },

	{ "DMC-GF7", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GF7) },

	{ "DMC-GX1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX1) },

	{ "DMC-GX7", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX7) },

	{ "DMC-GX8", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX8) },

	{ "DMC-GX80", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX80) },

	{ "DC-GX850", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GX850) },

	{ "DMC-FZ8", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ8) },

	{ "DMC-FZ18", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ18) },

	{ "DMC-FZ28", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ28) },

	{ "DMC-FZ30", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ30) },

	{ "DMC-FZ50", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ50) },

	{ "DMC-FZ100", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ100) },

	{ "DMC-FZ200", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ200) },

	{ "DMC-FZ2500", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ2500) },

	{ "DMC-FZ330", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ330) },

	{ "DC-FZ82", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_FZ82) },

	{ "DMC-G1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G1) },

	{ "DMC-G2", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G2) },

	{ "DMC-G3", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G3) },

	{ "DMC-G5", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G5) },

	{ "DMC-G10", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G10) },

	{ "DMC-G80", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_G80) },

	{ "DMC-GH1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH1) },

	{ "DMC-GH2", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH2) },

	{ "DMC-GH3", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH3) },

	{ "DMC-GH4", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH4) },

	{ "DC-GH5", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GH5) },

	{ "DMC-GM1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GM1) },

	{ "DMC-GM5", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_GM5) },

	{ "DMC-LX2", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX2) },

	{ "DMC-LX3", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX3) },

	{ "DMC-LX5", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX5) },

	{ "DMC-LX7", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX7) },

	{ "DMC-LX10", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX10) },

	{ "DMC-LX100", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_LX100) },

	{ "DMC-L1", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_L1) },

	{ "DMC-L10", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_L10) },

	{ "DMC-TZ100", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ100) },

	{ "DMC-TZ70", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ70) },

	{ "DMC-TZ80", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_TZ80) },

	{ "DMC-ZS100", OR_MAKE_PANASONIC_TYPEID(OR_TYPEID_PANASONIC_ZS100) },

	{ "DIGILUX 2", OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_DIGILUX2) },

	{ "D-LUX 3", OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_DLUX_3) },

	{ "V-LUX 1", OR_MAKE_LEICA_TYPEID(OR_TYPEID_LEICA_VLUX_1) },

	{ 0, 0 }

};

RawFile *Rw2File::factory(const IO::Stream::Ptr & s)

{

	return new Rw2File(s);

}

Rw2File::Rw2File(const IO::Stream::Ptr & s)

	: IfdFile(s, OR_RAWFILE_TYPE_RW2, false)

{

  _setIdMap(s_def);

  _setMatrices(s_matrices);

  m_container = new Rw2Container(m_io, 0);

}

Rw2File::~Rw2File()

{

}

IfdDir::Ref  Rw2File::_locateCfaIfd()

{

	return mainIfd();

}

IfdDir::Ref  Rw2File::_locateMainIfd()

{

	return m_container->setDirectory(0);

}

::or_error Rw2File::_locateThumbnail(const IfdDir::Ref & dir,

                                     std::vector<uint32_t> &list)

{

    uint32_t offset = 0;

    uint32_t size = 0;

    offset = _getJpegThumbnailOffset(dir, size);

    if(size == 0) {

        return OR_ERROR_NOT_FOUND;

    }

    LOGDBG1("Jpeg offset: %u\n", offset);

    uint32_t x = 0;

    uint32_t y = 0;

    ::or_data_type _type = OR_DATA_TYPE_JPEG;

    IO::Stream::Ptr s(new IO::StreamClone(m_io, offset));

    std::unique_ptr<JfifContainer> jfif(new JfifContainer(s, 0));

    if (jfif->getDimensions(x,y)) {

        LOGDBG1("JPEG dimensions x=%u y=%u\n", x, y);

    }

    if(_type != OR_DATA_TYPE_NONE) {

        uint32_t dim = std::max(x, y);

        _addThumbnail(dim, ThumbDesc(x, y, _type, offset, size));

        list.push_back(dim);

    }

    return OR_ERROR_NONE;

}

uint32_t Rw2File::_getJpegThumbnailOffset(const IfdDir::Ref & dir, uint32_t & len)

{

    IfdEntry::Ref e = dir->getEntry(IFD::RW2_TAG_JPEG_FROM_RAW);

	if(!e) {

	    len = 0;

		LOGDBG1("JpegFromRaw not found\n");

		return 0;

	}

    uint32_t offset = e->offset();

    len = e->count();

    return offset;

}

::or_error Rw2File::_getRawData(RawData & data, uint32_t /*options*/)

{

	::or_error ret = OR_ERROR_NONE;

	const IfdDir::Ref & _cfaIfd = cfaIfd();

	if(!_cfaIfd) {

		LOGDBG1("cfa IFD not found\n");

		return OR_ERROR_NOT_FOUND;

	}

	LOGDBG1("_getRawData()\n");

	uint32_t offset = 0;

	uint32_t byte_length = 0;

	// RW2 file

	auto result = _cfaIfd->getIntegerValue(IFD::RW2_TAG_STRIP_OFFSETS);

	if(result.ok()) {

		offset = result.unwrap();

        byte_length = m_container->file()->filesize() - offset;

    }

    else {

        // RAW file alternative.

		result = _cfaIfd->getIntegerValue(IFD::EXIF_TAG_STRIP_OFFSETS);

        if(result.empty()) {

            LOGDBG1("offset not found\n");

            return OR_ERROR_NOT_FOUND;

        }

		offset = result.unwrap();

        result = _cfaIfd->getIntegerValue(IFD::EXIF_TAG_STRIP_BYTE_COUNTS);

        if(result.empty()) {

            LOGDBG1("byte len not found\n");

            return OR_ERROR_NOT_FOUND;

        }

		byte_length = result.unwrap();

    }

	result = _cfaIfd->getIntegerValue(IFD::RW2_TAG_SENSOR_WIDTH);

	if(result.empty()) {

		LOGDBG1("X not found\n");

		return OR_ERROR_NOT_FOUND;

	}

	uint32_t x = result.unwrap();

	result = _cfaIfd->getIntegerValue(IFD::RW2_TAG_SENSOR_HEIGHT);

	if(result.empty()) {

		LOGDBG1("Y not found\n");

		return OR_ERROR_NOT_FOUND;

	}

	uint32_t y = result.unwrap();

	// this is were things are complicated. The real size of the raw data

	// is whatever is read (if compressed)

	void *p = data.allocData(byte_length);

	size_t real_size = m_container->fetchData(p, offset,

											  byte_length);

	if (real_size / (x * 8 / 7) == y) {

		data.setDataType(OR_DATA_TYPE_COMPRESSED_RAW);

		data.setCompression(PANA_RAW_COMPRESSION);

	}

	else if (real_size < byte_length) {

		LOGWARN("Size mismatch for data: expected %u got %lu ignoring.\n",

				byte_length, real_size);

		return OR_ERROR_NOT_FOUND;

	}

	else {

		data.setDataType(OR_DATA_TYPE_RAW);

	}

	data.setCfaPatternType(OR_CFA_PATTERN_BGGR);

	// they are not all RGGB.

	// but I don't seem to see where this is encoded.

	//

	data.setDimensions(x, y);

	LOGDBG1("In size is %ux%u\n", data.width(), data.height());

	// get the sensor info

	// XXX what if it is not found?

	IfdEntry::Ref e = _cfaIfd->getEntry(IFD::RW2_TAG_SENSOR_LEFTBORDER);

	x = e->getIntegerArrayItem(0);

	e = _cfaIfd->getEntry(IFD::RW2_TAG_SENSOR_TOPBORDER);

	y = e->getIntegerArrayItem(0);

	e = _cfaIfd->getEntry(IFD::RW2_TAG_IMAGE_HEIGHT);

	uint32_t h = e->getIntegerArrayItem(0);

	e = _cfaIfd->getEntry(IFD::RW2_TAG_IMAGE_WIDTH);

	uint32_t w = e->getIntegerArrayItem(0);

	data.setRoi(x, y, w, h);

	return ret;

}

}

}

/*

  Local Variables:

  mode:c++

  c-file-style:"stroustrup"

  c-file-offsets:((innamespace . 0))

  tab-width:2

  c-basic-offset:2

  indent-tabs-mode:nil

  fill-column:80

  End:

*/