1b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org/* 2b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * Copyright (c) 2011 The WebRTC project authors. All Rights Reserved. 3b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * 4b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * Use of this source code is governed by a BSD-style license 5b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * that can be found in the LICENSE file in the root of the source 6b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * tree. An additional intellectual property rights grant can be found 7b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * in the file PATENTS. All contributing project authors may 8b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org * be found in the AUTHORS file in the root of the source tree. 9b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org */ 10b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 1120a5c465cf17718743e5c5758baf9266bbb6d852pbos@webrtc.org#include "webrtc/test/testsupport/frame_reader.h" 12b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 133f45c2e0ac4cb280f941efa3a3476895795e3dd6pbos@webrtc.org#include <assert.h> 14b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 156d4d28ee857178312495a3ea83be7fa2751bbfdekjellander@webrtc.org#include "webrtc/test/testsupport/fileutils.h" 16b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 17b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgnamespace webrtc { 18b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgnamespace test { 19b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 20b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgFrameReaderImpl::FrameReaderImpl(std::string input_filename, 216d4d28ee857178312495a3ea83be7fa2751bbfdekjellander@webrtc.org size_t frame_length_in_bytes) 22b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org : input_filename_(input_filename), 23b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org frame_length_in_bytes_(frame_length_in_bytes), 24b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_file_(NULL) { 25b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} 26b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 27b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgFrameReaderImpl::~FrameReaderImpl() { 28b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org Close(); 29b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} 30b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 31b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgbool FrameReaderImpl::Init() { 32b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (frame_length_in_bytes_ <= 0) { 331e7f77a7f681b175aff8728fc9dd2d4b98019379kjellander@webrtc.org fprintf(stderr, "Frame length must be >0, was %zu\n", 34b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org frame_length_in_bytes_); 35b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; 36b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 37b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_file_ = fopen(input_filename_.c_str(), "rb"); 38b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (input_file_ == NULL) { 39b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org fprintf(stderr, "Couldn't open input file for reading: %s\n", 40b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_filename_.c_str()); 41b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; 42b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 43b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org // Calculate total number of frames. 44b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org size_t source_file_size = GetFileSize(input_filename_); 45b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (source_file_size <= 0u) { 46b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org fprintf(stderr, "Found empty file: %s\n", input_filename_.c_str()); 47b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; 48b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 491e7f77a7f681b175aff8728fc9dd2d4b98019379kjellander@webrtc.org number_of_frames_ = static_cast<int>(source_file_size / 501e7f77a7f681b175aff8728fc9dd2d4b98019379kjellander@webrtc.org frame_length_in_bytes_); 51b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return true; 52b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} 53b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 54b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgvoid FrameReaderImpl::Close() { 55b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (input_file_ != NULL) { 56b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org fclose(input_file_); 57b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_file_ = NULL; 58b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 59b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} 60b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 6191cab7186d535be98c837285ce24918188e9c8c3pbos@webrtc.orgbool FrameReaderImpl::ReadFrame(uint8_t* source_buffer) { 62b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org assert(source_buffer); 63b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (input_file_ == NULL) { 64b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org fprintf(stderr, "FrameReader is not initialized (input file is NULL)\n"); 65b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; 66b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 67b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org size_t nbr_read = fread(source_buffer, 1, frame_length_in_bytes_, 68b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_file_); 69b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (nbr_read != static_cast<unsigned int>(frame_length_in_bytes_) && 70b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org ferror(input_file_)) { 71b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org fprintf(stderr, "Error reading from input file: %s\n", 72b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org input_filename_.c_str()); 73b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; 74b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 75b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org if (feof(input_file_) != 0) { 76b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return false; // No more frames to process. 77b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org } 78b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org return true; 79b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} 80b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org 8177c6d714d2f8f700170218784553df1a48f17885pbos@webrtc.orgsize_t FrameReaderImpl::FrameLength() { return frame_length_in_bytes_; } 8277c6d714d2f8f700170218784553df1a48f17885pbos@webrtc.orgint FrameReaderImpl::NumberOfFrames() { return number_of_frames_; } 8377c6d714d2f8f700170218784553df1a48f17885pbos@webrtc.org 84b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} // namespace test 85b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org} // namespace webrtc 86