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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
|
//
// C++ Interface: k9vamps
//
// Description: A transcription from Vamps in C++
//
//
// Author: Jean-Michel PETIT <k9copy@free.fr>, (C) 2006
//
// Copyright: See COPYING file that comes with this distribution
//
//
#ifndef K9VAMPS_H
#define K9VAMPS_H
#include "k9common.h"
#include <tqfile.h>
#include <tqthread.h>
#include <tqobject.h>
#include <stdio.h>
#include <errno.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <getopt.h>
#include <pthread.h>
#include <sys/stat.h>
#include <tqmutex.h>
#include <tqwaitcondition.h>
#include "k9dvdbackup.h"
#include "k9requant.h"
#include "k9fifo.h"
#include "k9saveimage.h"
// DVD sector size
#define SECT_SIZE 2048
// read buffer size (4MB)
#define RBUF_SIZE (0x1000*1024)
// write buffer size (4MB)
#define WBUF_SIZE (0x1000*1024)
// initial video buffer size (1MB)
#define VBUF_SIZE (1024*1024)
class k9bgUpdate : public TQThread
{
private:
uchar * m_buffer;
k9DVDBackup *m_backup;
uint32_t m_size;
TQMutex mutex;
public:
k9bgUpdate(k9DVDBackup * _backup);
void update(uchar *_buffer, uint32_t size);
protected:
void run();
};
class k9vamps:public TQThread
{
private:
uchar *rbuf;// [RBUF_SIZE]; // the PS read buffer
uchar wbuf [WBUF_SIZE]; // the PS write buffer
uchar *vibuf; // the video ES requant input buffer
uchar *vobuf; // the video ES requant output buffer
uchar *rptr ; // pointer to current char in read buf
uchar *rhwp ; // read buffer high water pointer
uchar *wptr ; // pointer to first unused char in wbuf
uint64_t bytes_read; // total PS bytes read
uint64_t bytes_written; // total PS bytes written
uint64_t padding_bytes; // total padding bytes written
uint64_t vin_bytes; // total unshrinked video ES bytes
uint64_t vout_bytes; // total shrinked video ES bytes
uint64_t ps_size; // total PS size in bytes
uint32_t vbuf_size; // the video ES requant buffers' size
uint32_t rbuf_size;
uint32_t vilen; // current GOP's unshrinked vidES bytes
uint32_t volen; // current GOP's shrinked vidES bytes
int total_packs; // total no. PS packs
int video_packs; // no. video packs in PS
int skipped_video_packs; // skipped thereof
int aux_packs; // no. audio and subpicture packs in PS
int skipped_aux_packs; // skipped thereof
int sequence_headers; // no. sequence headers (== #GOPs)
int nav_packs; // no. nav packs
int eof; // end of file flag
int spu_track_map [32]; // subpicture track# translation map
int audio_track_map [8]; // audio track# translation map
int verbose; // level of verbosity
int calc_ps_vap; // calc vaporization based on PS size
bool m_preserve; // preserve audio/spu track numbers
float vap_fact; // vaporization factor from cmd line
bool noData;
TQMutex mutex;
k9bgUpdate *m_bgUpdate;
k9fifo m_fifo;
TQString m_errMsg;
bool m_error;
double avgdiff;
double m_totfact,m_nbfact,m_avgfact;
TQFile *m_output;
private:
// prototypes
void vaporize (void);
void fatal (TQString _msg);
int lock(int size);
void copy(int size);
void skip (int size);
void flush();
uint64_t wtell (uchar *ptr);
uint64_t rtell (uchar *ptr);
bool check_pack (uchar *ptr);
int check_video_packet (uchar *ptr);
int requant (uchar *dst, uchar *src, int n, float fact);
int new_private_1_type (uchar *ptr);
void copy_private_1 (uchar *ptr);
int new_mpeg_audio_id (int id);
void copy_mpeg_audio (uchar *ptr);
void vap_leader ();
void vap_trailer (int length);
int vap_phase1 (void);
int gen_video_packet (uchar *ptr, uchar *voptr, int avail);
void vap_phase2 (int seq_length);
pthread_t thread;
int readData(uchar *data,uint size);
TQWaitCondition wDataRead;
TQWaitCondition wDataReady;
k9DVDBackup *m_dvdbackup;
k9requant *m_requant;
k9SaveImage *m_saveImage;
protected:
void run();
public:
k9vamps(k9DVDBackup *dvdbackup);
void addData(uchar* data,uint size);
void setNoData();
void addSubpicture(uint id);
void addAudio(uint id);
void addAudio(uint id,uint newId);
void reset();
void setInputSize(uint64_t size);
void setVapFactor(float factor);
void setSaveImage(k9SaveImage*);
void setOutput(TQFile *_output);
uint64_t getOutputBytes();
TQString & geterrMsg();
bool geterror();
void abort();
void setPreserve(bool _value);
~k9vamps();
};
#endif
|
