summaryrefslogtreecommitdiff
path: root/gpr/source/lib/gpr_sdk/private/gpr_read_image.cpp
blob: 84c2fe979deaa847def321f9a8c3ac421a6ef88d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

#include "gpr_read_image.h"

#include "dng_camera_profile.h"
#include "dng_color_space.h"
#include "dng_date_time.h"
#include "dng_exceptions.h"
#include "dng_file_stream.h"
#include "dng_globals.h"
#include "dng_host.h"
#include "dng_ifd.h"
#include "dng_image_writer.h"
#include "dng_info.h"
#include "dng_linearization_info.h"
#include "dng_mosaic_info.h"
#include "dng_negative.h"
#include "dng_preview.h"
#include "dng_render.h"
#include "dng_simple_image.h"
#include "dng_tag_codes.h"
#include "dng_tag_types.h"
#include "dng_tag_values.h"
#include "dng_xmp.h"
#include "dng_xmp_sdk.h"
#include "dng_memory_stream.h"
#include "dng_bottlenecks.h"

#include "gpr_utils.h"

#if GPR_READING

#include "vc5_decoder.h"

static bool DecodeVC5(dng_image &image, gpr_buffer_auto& vc5_buffer, VC5_DECODER_PIXEL_FORMAT pixel_format )
{
    gpr_buffer_auto raw_buffer( malloc, free );
    
    vc5_decoder_parameters vc5_decoder_params;
    
    vc5_decoder_parameters_set_default(&vc5_decoder_params);
    
    vc5_decoder_params.mem_alloc        = raw_buffer.get_malloc();
    vc5_decoder_params.mem_free         = raw_buffer.get_free();
    vc5_decoder_params.pixel_format     = pixel_format;
    
    gpr_buffer vc5_image = { vc5_buffer.get_buffer(), vc5_buffer.get_size()  };
    gpr_buffer raw_image = { raw_buffer.get_buffer(), raw_buffer.get_size()  };
    
    if( vc5_decoder_process( &vc5_decoder_params, &vc5_image, &raw_image, NULL ) != CODEC_ERROR_OKAY )
    {
        assert(0);
        return false;
    }
    
    raw_buffer.set( raw_image.buffer, raw_image.size, true );
    
    dng_point size = image.Bounds().Size();
    
    CopyBufferToRawImage( raw_buffer, size.h, image );

    return true;
}

gpr_read_image::gpr_read_image( gpr_buffer_auto* vc5_buffer ) : _vc5_buffer(vc5_buffer)
{
    fReadVC5 = true;
    fDecodeVC5 = true;
}

void gpr_read_image::ReadTile (dng_host &host,
                                   const dng_ifd &ifd,
                                   dng_stream &stream,
                                   dng_image &image,
                                   const dng_rect &tileArea,
                                   uint32 plane,
                                   uint32 planes,
                                   uint32 tileByteCount,
                                   AutoPtr<dng_memory_block> &compressedBuffer,
                                   AutoPtr<dng_memory_block> &uncompressedBuffer,
                                   AutoPtr<dng_memory_block> &subTileBlockBuffer)
{
    
    if( ifd.fCompression == ccVc5 )
    {
        if( GetReadVC5() && _vc5_buffer != NULL )
        {
            _vc5_buffer->allocate( tileByteCount );
            
            stream.Get(_vc5_buffer->get_buffer(), _vc5_buffer->get_size() );
            
            if( GetDecodeVC5() == true )
            {
                bool rggb_raw = (ifd.fCFAPattern[0][0] == 0) && (ifd.fCFAPattern[0][1] == 1) && (ifd.fCFAPattern[1][0] == 1) && (ifd.fCFAPattern[1][1] == 2);
                
                VC5_DECODER_PIXEL_FORMAT pixel_format;
                
                if( rggb_raw )
                {
                    pixel_format = VC5_DECODER_PIXEL_FORMAT_RGGB_14;
                }
                else
                {
                    pixel_format = VC5_DECODER_PIXEL_FORMAT_GBRG_12;
                }
                
                if( DecodeVC5( image, *_vc5_buffer, pixel_format ) )
                {
                    return;
                }
            }
        }
    }
    else
    {
        dng_read_image::ReadTile (host,
                                  ifd,
                                  stream,
                                  image,
                                  tileArea,
                                  plane,
                                  planes,
                                  tileByteCount,
                                  compressedBuffer,
                                  uncompressedBuffer,
                                  subTileBlockBuffer);
    }
}

#endif