/**
* @brief circular buffer test application
*/
int main(int argc, char **argv)
{
unsigned char data = {0};
int size = 0;
printf("How much characteres you want to put? ");
scanf("%d", &size);
for(unsigned int i = 0; i < size; ++i) {
printf("Enter the %d value: ",i);
scanf("%hu", &data);
/* put the new byte on buffer
* NOTE! We dont need to manage our buffer
*/
buffer_insert(&console_buffer,&data, 1);
}
/* prints the buffer contents */
printf("The buffer contents are \n\r");
while(buffer_empty(&console_buffer) == 0) {
/* extracts byte to byte from buffer, higher sizes streams are supported
* but this mode shows better the capabilities of circ buffer
*/
buffer_retrieve(&console_buffer, &data, 1);
printf("%hu",&data );
}
/* wait for a key */
scanf("%hu", &data);
return(0);
}
static
void connection_onevent(int fd, int event, void* userdata)
{
tcp_connection_t *connection = (tcp_connection_t*)userdata;
inetaddr_t *peer_addr = &connection->peer_addr;
buffer_t* in_buffer;
buffer_t* out_buffer;
void* data;
unsigned size;
unsigned written;
int error;
log_debug("connection_onevent: fd(%d), event(%d), peer addr: %s:%u", fd, event, peer_addr->ip, peer_addr->port);
if (event & EPOLLHUP)
{
connection->is_connected = 0;
connection->is_in_callback = 1;
connection->closecb(connection, connection->userdata);
connection->is_in_callback = 0;
}
else
{
if (event & EPOLLIN)
{
if (connection->need_closed_after_sent_done == 0)
{
in_buffer = connection->in_buffer;
size = buffer_readFd(in_buffer, connection->fd);
if (0 == size)
{
assert(NULL != connection->closecb);
connection->is_connected = 0;
connection->is_in_callback = 1;
connection->closecb(connection, connection->userdata);
connection->is_in_callback = 0;
}
else
{
assert(NULL != connection->datacb);
connection->is_in_callback = 1;
connection->datacb(connection, in_buffer, connection->userdata);
connection->is_in_callback = 0;
}
}
else
{
char temp[256];
while (read(fd, temp, sizeof(temp)) > 0);
}
}
if (event & EPOLLOUT)
{
out_buffer = connection->out_buffer;
data = buffer_peek(out_buffer);
size = buffer_readablebytes(out_buffer);
written = write(connection->fd, data, size);
if (written < 0)
{
error = errno;
if (error != EAGAIN && error != EINTR)
{
log_error("connection_onevent: write() failed, fd(%d), errno(%d), peer addr: %s:%u", fd, error, peer_addr->ip, peer_addr->port);
return;
}
else
{
written = 0;
}
}
buffer_retrieve(out_buffer, written);
if(written >= size)
{
channel_clearevent(connection->channel, EPOLLOUT);
if (connection->need_closed_after_sent_done)
{
connection->is_alive = 0;
}
}
}
}
if (0 == connection->is_alive)
{
delete_connection(connection);
}
return;
}
static
int tcp_connection_sendInLoop(tcp_connection_t* connection, const void* data, unsigned size)
{
inetaddr_t *peer_addr = &connection->peer_addr;
int fd;
void* buffer_data;
channel_t *channel;
buffer_t* out_buffer;
int written;
unsigned buffer_left_data_size;
struct iovec vecs[2];
int error;
out_buffer = connection->out_buffer;
channel = connection->channel;
fd = connection->fd;
buffer_left_data_size = buffer_readablebytes(out_buffer);
if (buffer_left_data_size > 0)
{
buffer_data = buffer_peek(out_buffer);
vecs[0].iov_base = buffer_data;
vecs[0].iov_len = buffer_left_data_size;
vecs[1].iov_base = (void*)data;
vecs[1].iov_len = size;
written = writev(fd, vecs, 2);
if (written < 0)
{
error = errno;
if (error != EAGAIN)
{
log_error("tcp_connection_sendInLoop: writev() failed, errno: %d, peer addr: %s:%u", errno, peer_addr->ip, peer_addr->port);
return -1;
}
}
else if (written == (buffer_left_data_size+size))
{
/* 当前所有的数据都发送完毕，一切安好则去除EPOLLOUT事件 */
channel_clearevent(channel, EPOLLOUT);
}
else if (written < buffer_left_data_size)
{
/* out_buffer中的数据尚未发送完毕，则将本次的数据放入out_buffer后再发送 */
buffer_retrieve(out_buffer, written);
buffer_append(out_buffer, data, size);
channel_setevent(channel, EPOLLOUT);
}
else if (written < (buffer_left_data_size+size))
{
buffer_retrieveall(out_buffer);
buffer_append(out_buffer, ((const char*)data+written-buffer_left_data_size), ((buffer_left_data_size+size)-written));
channel_setevent(channel, EPOLLOUT);
}
}
else
{
written = write(fd, data, size);
if (written < 0)
{
log_error("tcp_connection_sendInLoop: write() failed, errno: %d, peer addr: %s:%u", errno, peer_addr->ip, peer_addr->port);
return -1;
}
else if (written < size)
{
buffer_append(out_buffer, ((const char*)data+written), (size-written));
channel_setevent(channel, EPOLLOUT);
}
}
return 0;
}
int main(int argc, char *argv[])
{
avi_t *avifile1=NULL;
avi_t *avifile2=NULL;
avi_t *avifile3=NULL;
char *in_file=NULL, *out_file=NULL;
long frames, bytes;
double fps;
char *codec;
int track_num=0, aud_tracks;
int encode_null=0;
int i, j, n, key, shift=0;
int ch, preload=0;
long rate, mp3rate;
int width, height, format, chan, bits;
int be_quiet = 0;
FILE *status_fd = stderr;
/* for null frame encoding */
char nulls[32000];
long nullbytes=0;
char tmp0[] = "/tmp/nullfile.00.avi"; /* XXX: use mktemp*() */
buffer_list_t *ptr;
double vid_ms = 0.0, shift_ms = 0.0, one_vid_ms = 0.0;
double aud_ms [ AVI_MAX_TRACKS ];
int aud_bitrate = 0;
int aud_chunks = 0;
ac_init(AC_ALL);
if(argc==1) usage(EXIT_FAILURE);
while ((ch = getopt(argc, argv, "a:b:vi:o:n:Nq?h")) != -1)
{
switch (ch) {
case 'i':
if(optarg[0]=='-') usage(EXIT_FAILURE);
in_file=optarg;
break;
case 'a':
if(optarg[0]=='-') usage(EXIT_FAILURE);
track_num = atoi(optarg);
if(track_num<0) usage(EXIT_FAILURE);
break;
case 'b':
if(optarg[0]=='-') usage(EXIT_FAILURE);
is_vbr = atoi(optarg);
if(is_vbr<0) usage(EXIT_FAILURE);
break;
case 'o':
if(optarg[0]=='-') usage(EXIT_FAILURE);
out_file=optarg;
break;
case 'f':
if(optarg[0]=='-') usage(EXIT_FAILURE);
comfile = optarg;
break;
case 'n':
if(sscanf(optarg,"%d", &shift)!=1) {
fprintf(stderr, "invalid parameter for option -n\n");
usage(EXIT_FAILURE);
}
break;
case 'N':
encode_null=1;
break;
case 'q':
be_quiet = 1;
break;
case 'v':
version();
exit(0);
break;
case 'h':
usage(EXIT_SUCCESS);
default:
usage(EXIT_FAILURE);
}
}
// check
if(in_file==NULL || out_file == NULL) usage(EXIT_FAILURE);
if(shift == 0) fprintf(stderr, "no sync requested - exit");
memset (nulls, 0, sizeof(nulls));
// open file
if(NULL == (avifile1 = AVI_open_input_file(in_file,1))) {
AVI_print_error("AVI open");
exit(1);
}
if(strcmp(in_file, out_file)==0) {
printf("error: output filename conflicts with input filename\n");
exit(1);
}
if(NULL == (avifile2 = AVI_open_output_file(out_file))) {
AVI_print_error("AVI open");
exit(1);
}
if (be_quiet) {
if (!(status_fd = fopen("/dev/null", "w"))) {
fprintf(stderr, "Can't open /dev/null\n");
exit(1);
}
}
// read video info;
AVI_info(avifile1);
// read video info;
frames = AVI_video_frames(avifile1);
width = AVI_video_width(avifile1);
height = AVI_video_height(avifile1);
fps = AVI_frame_rate(avifile1);
codec = AVI_video_compressor(avifile1);
//set video in outputfile
AVI_set_video(avifile2, width, height, fps, codec);
if (comfile!=NULL)
AVI_set_comment_fd(avifile2, open(comfile, O_RDONLY));
aud_tracks = AVI_audio_tracks(avifile1);
for(j=0; j<aud_tracks; ++j) {
AVI_set_audio_track(avifile1, j);
rate = AVI_audio_rate(avifile1);
chan = AVI_audio_channels(avifile1);
bits = AVI_audio_bits(avifile1);
format = AVI_audio_format(avifile1);
mp3rate= AVI_audio_mp3rate(avifile1);
//set next track of output file
AVI_set_audio_track(avifile2, j);
AVI_set_audio(avifile2, chan, rate, bits, format, mp3rate);
AVI_set_audio_vbr(avifile2, is_vbr);
}
//switch to requested audio_channel
if(AVI_set_audio_track(avifile1, track_num)<0) {
fprintf(stderr, "invalid auto track\n");
}
AVI_set_audio_track(avifile2, track_num);
if (encode_null) {
char cmd[1024];
rate = AVI_audio_rate(avifile2);
chan = AVI_audio_channels(avifile2);
bits = AVI_audio_bits(avifile2);
format = AVI_audio_format(avifile2);
mp3rate= AVI_audio_mp3rate(avifile2);
if (bits==0) bits=16;
if (mp3rate%2) mp3rate++;
fprintf(status_fd, "Creating silent mp3 frame with current parameter\n");
memset (cmd, 0, sizeof(cmd));
tc_snprintf(cmd, sizeof(cmd), "transcode -i /dev/zero -o %s -x raw,raw"
" -n 0x1 -g 16x16 -y raw,raw -c 0-5 -e %ld,%d,%d -b %ld -q0",
tmp0, rate,bits,chan, mp3rate);
printf(cmd);
system(cmd);
if(NULL == (avifile3 = AVI_open_input_file(tmp0,1))) {
AVI_print_error("AVI open");
exit(1);
}
nullbytes = AVI_audio_size(avifile3, 3);
/* just read a few frames */
if(AVI_read_audio(avifile3, nulls, nullbytes) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
memset (nulls, 0, sizeof(nulls));
if(AVI_read_audio(avifile3, nulls, nullbytes) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
memset (nulls, 0, sizeof(nulls));
if(AVI_read_audio(avifile3, nulls, nullbytes) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
/*
printf("\nBytes (%ld): \n", nullbytes);
{
int asd=0;
for (asd=0; asd<nullbytes; asd++){
printf("%x ",(unsigned char)nulls[asd]);
}
printf("\n");
}
*/
}
vid_ms = 0.0;
shift_ms = 0.0;
for (n=0; n<AVI_MAX_TRACKS; ++n)
aud_ms[n] = 0.0;
// ---------------------------------------------------------------------
for (n=0; n<frames; ++n) {
// video unchanged
bytes = AVI_read_frame(avifile1, data, &key);
if(bytes < 0) {
AVI_print_error("AVI read video frame");
return(-1);
}
if(AVI_write_frame(avifile2, data, bytes, key)<0) {
AVI_print_error("AVI write video frame");
return(-1);
}
vid_ms = (n+1)*1000.0/fps;
// Pass-through all other audio tracks.
for(j=0; j<aud_tracks; ++j) {
// skip track we want to modify
if (j == track_num) continue;
// switch to track
AVI_set_audio_track(avifile1, j);
AVI_set_audio_track(avifile2, j);
sync_audio_video_avi2avi(vid_ms, &aud_ms[j], avifile1, avifile2);
}
//switch to requested audio_channel
if(AVI_set_audio_track(avifile1, track_num)<0) {
fprintf(stderr, "invalid auto track\n");
}
AVI_set_audio_track(avifile2, track_num);
shift_ms = (double)shift*1000.0/fps;
one_vid_ms = 1000.0/fps;
format = AVI_audio_format(avifile1);
rate = AVI_audio_rate(avifile1);
chan = AVI_audio_channels(avifile1);
bits = AVI_audio_bits(avifile1);
bits = bits==0?16:bits;
mp3rate= AVI_audio_mp3rate(avifile1);
if(shift>0) {
// for n < shift, shift audio frames are discarded
if(!preload) {
if (tc_format_ms_supported(format)) {
for(i=0;i<shift;++i) {
//fprintf (stderr, "shift (%d) i (%d) n (%d) a (%d)\n", shift, i, n, aud_chunks);
while (aud_ms[track_num] < vid_ms + one_vid_ms*(double)i) {
aud_bitrate = (format==0x1||format==0x2000)?1:0;
aud_chunks++;
if( (bytes = AVI_read_audio_chunk(avifile1, data)) <= 0) {
aud_ms[track_num] = vid_ms + one_vid_ms*i;
if (bytes == 0) continue;
AVI_print_error("AVI 2 audio read frame");
break;
}
if ( !aud_bitrate && tc_get_audio_header(data, bytes, format, NULL, NULL, &aud_bitrate)<0) {
// if this is the last frame of the file, slurp in audio chunks
if (n == frames-1) continue;
aud_ms[track_num] = vid_ms + one_vid_ms*i;
} else
aud_ms[track_num] += (bytes*8.0)/(format==0x1?((double)(rate*chan*bits)/1000.0):
(format==0x2000?(double)(mp3rate):aud_bitrate));
}
}
} else { // fallback
bytes=0;
for(i=0;i<shift;++i) {
do {
if( (bytes = AVI_read_audio_chunk(avifile1, data)) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
} while (AVI_can_read_audio(avifile1));
}
}
preload=1;
}
// copy rest of the track
if(n<frames-shift) {
if (tc_format_ms_supported(format)) {
while (aud_ms[track_num] < vid_ms + shift_ms) {
aud_chunks++;
aud_bitrate = (format==0x1||format==0x2000)?1:0;
if( (bytes = AVI_read_audio_chunk(avifile1, data)) < 0) {
aud_ms[track_num] = vid_ms + shift_ms;
AVI_print_error("AVI 3 audio read frame");
break;
}
if(AVI_write_audio(avifile2, data, bytes) < 0) {
AVI_print_error("AVI 3 write audio frame");
return(-1);
}
fprintf(status_fd, "V [%05d][%08.2f] | A [%05d][%08.2f] [%05ld]\r", n, vid_ms, aud_chunks, aud_ms[track_num], bytes);
if (bytes == 0) {
aud_ms[track_num] = vid_ms + shift_ms;
continue;
}
if(n>=frames-2*shift) {
// save audio frame for later
ptr = buffer_register(n);
if(ptr==NULL) {
fprintf(stderr,"buffer allocation failed\n");
break;
}
ac_memcpy(ptr->data, data, bytes);
ptr->size = bytes;
ptr->status = BUFFER_READY;
}
if ( !aud_bitrate && tc_get_audio_header(data, bytes, format, NULL, NULL, &aud_bitrate)<0) {
if (n == frames-1) continue;
aud_ms[track_num] = vid_ms + shift_ms;
} else
aud_ms[track_num] += (bytes*8.0)/(format==0x1?((double)(rate*chan*bits)/1000.0):
(format==0x2000?(double)(mp3rate):aud_bitrate));
}
} else { // fallback
bytes = AVI_audio_size(avifile1, n+shift-1);
do {
if( (bytes = AVI_read_audio_chunk(avifile1, data)) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
if(AVI_write_audio(avifile2, data, bytes) < 0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
fprintf(status_fd, "V [%05d] | A [%05d] [%05ld]\r", n, n+shift, bytes);
if(n>=frames-2*shift) {
// save audio frame for later
ptr = buffer_register(n);
if(ptr==NULL) {
fprintf(stderr,"buffer allocation failed\n");
break;
}
ac_memcpy(ptr->data, data, bytes);
ptr->size = bytes;
ptr->status = BUFFER_READY;
}
} while (AVI_can_read_audio(avifile1));
}
}
// padding at the end
if(n>=frames-shift) {
if (!ptrlen) {
ptr = buffer_retrieve();
ac_memcpy (ptrdata, ptr->data, ptr->size);
ptrlen = ptr->size;
}
if (tc_format_ms_supported(format)) {
while (aud_ms[track_num] < vid_ms + shift_ms) {
aud_bitrate = (format==0x1||format==0x2000)?1:0;
// mute this -- check if can mute (valid A header)!
if (tc_probe_audio_header(ptrdata, ptrlen) > 0)
tc_format_mute(ptrdata, ptrlen, format);
if(AVI_write_audio(avifile2, ptrdata, ptrlen) < 0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
fprintf(status_fd, " V [%05d][%08.2f] | A [%05d][%08.2f] [%05ld]\r", n, vid_ms, n+shift, aud_ms[track_num], bytes);
if ( !aud_bitrate && tc_get_audio_header(ptrdata, ptrlen, format, NULL, NULL, &aud_bitrate)<0) {
//if (n == frames-1) continue;
aud_ms[track_num] = vid_ms + shift_ms;
} else
aud_ms[track_num] += (ptrlen*8.0)/(format==0x1?((double)(rate*chan*bits)/1000.0):
(format==0x2000?(double)(mp3rate):aud_bitrate));
}
} else { // fallback
// get next audio frame
ptr = buffer_retrieve();
while (1) {
printf("ptr->id (%d) ptr->size (%d)\n", ptr->id, ptr->size);
if(ptr==NULL) {
fprintf(stderr,"no buffer found\n");
break;
}
if (encode_null) {
if(AVI_write_audio(avifile2, nulls, nullbytes)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
} else {
// simple keep old frames to force exact time delay
if(AVI_write_audio(avifile2, ptr->data, ptr->size)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
}
fprintf(status_fd, "V [%05d] | padding\r", n);
if (ptr->next && ptr->next->id == ptr->id) {
buffer_remove(ptr);
ptr = buffer_retrieve();
continue;
}
buffer_remove(ptr);
break;
} // 1
}
}
// *************************************
// negative shift (pad audio at start)
// *************************************
} else {
if (tc_format_ms_supported(format)) {
/*
fprintf(status_fd, "n(%d) -shift(%d) shift_ms (%.2lf) vid_ms(%.2lf) aud_ms[%d](%.2lf) v-s(%.2lf)\n",
n, -shift, shift_ms, vid_ms, track_num, aud_ms[track_num], vid_ms + shift_ms);
*/
// shift<0 -> shift_ms<0 !
while (aud_ms[track_num] < vid_ms) {
/*
fprintf(stderr, " 1 (%02d) %s frame_read len=%4ld (A/V) (%8.2f/%8.2f)\n",
n, format==0x55?"MP3":"AC3", bytes, aud_ms[track_num], vid_ms);
*/
aud_bitrate = (format==0x1||format==0x2000)?1:0;
if( (bytes = AVI_read_audio_chunk(avifile1, data)) < 0) {
AVI_print_error("AVI 2 audio read frame");
aud_ms[track_num] = vid_ms;
break;
//return(-1);
}
// save audio frame for later
ptr = buffer_register(n);
if(ptr==NULL) {
fprintf(stderr,"buffer allocation failed\n");
break;
}
ac_memcpy(ptr->data, data, bytes);
ptr->size = bytes;
ptr->status = BUFFER_READY;
if(n<-shift) {
// mute this -- check if can mute!
if (tc_probe_audio_header(data, bytes) > 0)
tc_format_mute(data, bytes, format);
// simple keep old frames to force exact time delay
if(AVI_write_audio(avifile2, data, bytes)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
fprintf(status_fd, "V [%05d] | padding\r", n);
} else {
if (n==-shift)
fprintf(status_fd, "\n");
// get next audio frame
ptr = buffer_retrieve();
if(ptr==NULL) {
fprintf(stderr,"no buffer found\n");
break;
}
if(AVI_write_audio(avifile2, ptr->data, ptr->size)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
bytes = ptr->size;
ac_memcpy (data, ptr->data, bytes);
fprintf(status_fd, "V [%05d] | A [%05d]\r", n, ptr->id);
buffer_remove(ptr);
}
if ( !aud_bitrate && tc_get_audio_header(data, bytes, format, NULL, NULL, &aud_bitrate)<0) {
if (n == frames-1) continue;
aud_ms[track_num] = vid_ms;
} else
aud_ms[track_num] += (bytes*8.0)/(format==0x1?((double)(rate*chan*bits)/1000.0):
(format==0x2000?(double)(mp3rate):aud_bitrate));
/*
fprintf(stderr, " 1 (%02d) %s frame_read len=%4ld (A/V) (%8.2f/%8.2f)\n",
n, format==0x55?"MP3":"AC3", bytes, aud_ms[track_num], vid_ms);
*/
}
} else { // no supported format
bytes = AVI_audio_size(avifile1, n);
if(bytes > SIZE_RGB_FRAME) {
fprintf(stderr, "invalid frame size\n");
return(-1);
}
if(AVI_read_audio(avifile1, data, bytes) < 0) {
AVI_print_error("AVI audio read frame");
return(-1);
}
// save audio frame for later
ptr = buffer_register(n);
if(ptr==NULL) {
fprintf(stderr,"buffer allocation failed\n");
break;
}
ac_memcpy(ptr->data, data, bytes);
ptr->size = bytes;
ptr->status = BUFFER_READY;
if(n<-shift) {
if (encode_null) {
if(AVI_write_audio(avifile2, nulls, nullbytes)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
} else {
// simple keep old frames to force exact time delay
if(AVI_write_audio(avifile2, data, bytes)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
}
fprintf(status_fd, "V [%05d] | padding\r", n);
} else {
// get next audio frame
ptr = buffer_retrieve();
if(ptr==NULL) {
fprintf(stderr,"no buffer found\n");
break;
}
if(AVI_write_audio(avifile2, ptr->data, ptr->size)<0) {
AVI_print_error("AVI write audio frame");
return(-1);
}
fprintf(status_fd, "V [%05d] | A [%05d]\r", n, ptr->id);
buffer_remove(ptr);
}
}
}
}
fprintf(status_fd, "\n");
if (be_quiet) {
fclose(status_fd);
}
AVI_close(avifile1);
AVI_close(avifile2);
if (avifile3) {
memset(nulls, 0, sizeof(nulls));
tc_snprintf(nulls, sizeof(nulls), "rm -f %s", tmp0);
system(nulls);
AVI_close(avifile3);
}
return(0);
}
