blob: 58f7f6f666e370497b0b68d2afa52f5fd107902d (
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
|
// This file declares the functions and structures for memory I/O with libjpeg
// These functions are not meant to be used directly, see jpeg_mem.h isntead.
#ifndef TENSORFLOW_LIB_JPEG_JPEG_HANDLE_H_
#define TENSORFLOW_LIB_JPEG_JPEG_HANDLE_H_
extern "C" {
#include "external/jpeg_archive/jpeg-9a/jinclude.h"
#include "external/jpeg_archive/jpeg-9a/jpeglib.h"
#include "external/jpeg_archive/jpeg-9a/jerror.h"
#include "external/jpeg_archive/jpeg-9a/transupp.h" // for rotations
}
#include "tensorflow/core/platform/port.h"
namespace tensorflow {
namespace jpeg {
// Handler for fatal JPEG library errors: clean up & return
void CatchError(j_common_ptr cinfo);
typedef struct {
struct jpeg_destination_mgr pub;
JOCTET *buffer;
int bufsize;
int datacount;
string *dest;
} MemDestMgr;
typedef struct {
struct jpeg_source_mgr pub;
const unsigned char *data;
unsigned long int datasize;
bool try_recover_truncated_jpeg;
} MemSourceMgr;
void SetSrc(j_decompress_ptr cinfo, const void *data,
unsigned long int datasize, bool try_recover_truncated_jpeg);
// JPEG destination: we will store all the data in a buffer "buffer" of total
// size "bufsize", if the buffer overflows, we will be in trouble.
void SetDest(j_compress_ptr cinfo, void *buffer, int bufsize);
// Same as above, except that buffer is only used as a temporary structure and
// is emptied into "destination" as soon as it fills up.
void SetDest(j_compress_ptr cinfo, void *buffer, int bufsize,
string *destination);
} // namespace jpeg
} // namespace tensorflow
#endif // TENSORFLOW_LIB_JPEG_JPEG_HANDLE_H_
|
