116 lines
3.8 KiB
C++
Executable File
116 lines
3.8 KiB
C++
Executable File
/**********
|
|
This library is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU Lesser General Public License as published by the
|
|
Free Software Foundation; either version 2.1 of the License, or (at your
|
|
option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.)
|
|
|
|
This library is distributed in the hope that it will be useful, but WITHOUT
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for
|
|
more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with this library; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
**********/
|
|
// "liveMedia"
|
|
// Copyright (c) 1996-2015 Live Networks, Inc. All rights reserved.
|
|
// A sink that generates an AVI file from a composite media session
|
|
// C++ header
|
|
|
|
#ifndef _AVI_FILE_SINK_HH
|
|
#define _AVI_FILE_SINK_HH
|
|
|
|
#ifndef _MEDIA_SESSION_HH
|
|
#include "MediaSession.hh"
|
|
#endif
|
|
|
|
class AVIFileSink: public Medium {
|
|
public:
|
|
static AVIFileSink* createNew(UsageEnvironment& env,
|
|
MediaSession& inputSession,
|
|
char const* outputFileName,
|
|
unsigned bufferSize = 20000,
|
|
unsigned short movieWidth = 240,
|
|
unsigned short movieHeight = 180,
|
|
unsigned movieFPS = 15,
|
|
Boolean packetLossCompensate = False);
|
|
|
|
typedef void (afterPlayingFunc)(void* clientData);
|
|
Boolean startPlaying(afterPlayingFunc* afterFunc,
|
|
void* afterClientData);
|
|
|
|
unsigned numActiveSubsessions() const { return fNumSubsessions; }
|
|
|
|
private:
|
|
AVIFileSink(UsageEnvironment& env, MediaSession& inputSession,
|
|
char const* outputFileName, unsigned bufferSize,
|
|
unsigned short movieWidth, unsigned short movieHeight,
|
|
unsigned movieFPS, Boolean packetLossCompensate);
|
|
// called only by createNew()
|
|
virtual ~AVIFileSink();
|
|
|
|
Boolean continuePlaying();
|
|
static void afterGettingFrame(void* clientData, unsigned frameSize,
|
|
unsigned numTruncatedBytes,
|
|
struct timeval presentationTime,
|
|
unsigned durationInMicroseconds);
|
|
static void onSourceClosure(void* clientData);
|
|
void onSourceClosure1();
|
|
static void onRTCPBye(void* clientData);
|
|
void addIndexRecord(class AVIIndexRecord* newIndexRecord);
|
|
void completeOutputFile();
|
|
|
|
private:
|
|
friend class AVISubsessionIOState;
|
|
MediaSession& fInputSession;
|
|
FILE* fOutFid;
|
|
class AVIIndexRecord *fIndexRecordsHead, *fIndexRecordsTail;
|
|
unsigned fNumIndexRecords;
|
|
unsigned fBufferSize;
|
|
Boolean fPacketLossCompensate;
|
|
Boolean fAreCurrentlyBeingPlayed;
|
|
afterPlayingFunc* fAfterFunc;
|
|
void* fAfterClientData;
|
|
unsigned fNumSubsessions;
|
|
unsigned fNumBytesWritten;
|
|
struct timeval fStartTime;
|
|
Boolean fHaveCompletedOutputFile;
|
|
|
|
private:
|
|
///// Definitions specific to the AVI file format:
|
|
|
|
unsigned addWord(unsigned word); // outputs "word" in little-endian order
|
|
unsigned addHalfWord(unsigned short halfWord);
|
|
unsigned addByte(unsigned char byte) {
|
|
putc(byte, fOutFid);
|
|
return 1;
|
|
}
|
|
unsigned addZeroWords(unsigned numWords);
|
|
unsigned add4ByteString(char const* str);
|
|
void setWord(unsigned filePosn, unsigned size);
|
|
|
|
// Define member functions for outputting various types of file header:
|
|
#define _header(name) unsigned addFileHeader_##name()
|
|
_header(AVI);
|
|
_header(hdrl);
|
|
_header(avih);
|
|
_header(strl);
|
|
_header(strh);
|
|
_header(strf);
|
|
_header(JUNK);
|
|
// _header(JUNK);
|
|
_header(movi);
|
|
private:
|
|
unsigned short fMovieWidth, fMovieHeight;
|
|
unsigned fMovieFPS;
|
|
unsigned fRIFFSizePosition, fRIFFSizeValue;
|
|
unsigned fAVIHMaxBytesPerSecondPosition;
|
|
unsigned fAVIHFrameCountPosition;
|
|
unsigned fMoviSizePosition, fMoviSizeValue;
|
|
class AVISubsessionIOState* fCurrentIOState;
|
|
unsigned fJunkNumber;
|
|
};
|
|
|
|
#endif
|