1
0
mirror of https://github.com/VDR4Arch/vdr.git synced 2023-10-10 13:36:52 +02:00
vdr/cutter.c

273 lines
8.1 KiB
C
Raw Normal View History

2002-06-22 10:11:59 +02:00
/*
* cutter.c: The video cutting facilities
*
* See the main source file 'vdr.c' for copyright information and
* how to reach the author.
*
* $Id: cutter.c 2.2 2009/01/24 15:19:26 kls Exp $
2002-06-22 10:11:59 +02:00
*/
#include "cutter.h"
#include "recording.h"
#include "remux.h"
#include "thread.h"
#include "videodir.h"
// --- cCuttingThread --------------------------------------------------------
class cCuttingThread : public cThread {
private:
const char *error;
cUnbufferedFile *fromFile, *toFile;
2002-06-22 10:11:59 +02:00
cFileName *fromFileName, *toFileName;
cIndexFile *fromIndex, *toIndex;
cMarks fromMarks, toMarks;
off_t maxVideoFileSize;
2002-06-22 10:11:59 +02:00
protected:
virtual void Action(void);
public:
cCuttingThread(const char *FromFileName, const char *ToFileName);
virtual ~cCuttingThread();
const char *Error(void) { return error; }
};
cCuttingThread::cCuttingThread(const char *FromFileName, const char *ToFileName)
:cThread("video cutting")
2002-06-22 10:11:59 +02:00
{
error = NULL;
fromFile = toFile = NULL;
2002-06-22 10:11:59 +02:00
fromFileName = toFileName = NULL;
fromIndex = toIndex = NULL;
cRecording Recording(FromFileName);
bool isPesRecording = Recording.IsPesRecording();
if (fromMarks.Load(FromFileName, Recording.FramesPerSecond(), isPesRecording) && fromMarks.Count()) {
fromFileName = new cFileName(FromFileName, false, true, isPesRecording);
toFileName = new cFileName(ToFileName, true, true, isPesRecording);
fromIndex = new cIndexFile(FromFileName, false, isPesRecording);
toIndex = new cIndexFile(ToFileName, true, isPesRecording);
toMarks.Load(ToFileName, Recording.FramesPerSecond(), isPesRecording); // doesn't actually load marks, just sets the file name
maxVideoFileSize = MEGABYTE(Setup.MaxVideoFileSize);
if (isPesRecording && maxVideoFileSize > MEGABYTE(MAXVIDEOFILESIZEPES))
maxVideoFileSize = MEGABYTE(MAXVIDEOFILESIZEPES);
2002-06-22 10:11:59 +02:00
Start();
}
else
esyslog("no editing marks found for %s", FromFileName);
}
cCuttingThread::~cCuttingThread()
{
Cancel(3);
delete fromFileName;
delete toFileName;
delete fromIndex;
delete toIndex;
}
void cCuttingThread::Action(void)
{
cMark *Mark = fromMarks.First();
if (Mark) {
fromFile = fromFileName->Open();
toFile = toFileName->Open();
if (!fromFile || !toFile)
2005-08-13 13:17:24 +02:00
return;
2006-02-04 14:12:17 +01:00
fromFile->SetReadAhead(MEGABYTE(20));
2002-06-22 10:11:59 +02:00
int Index = Mark->position;
Mark = fromMarks.Next(Mark);
off_t FileSize = 0;
2002-06-22 10:11:59 +02:00
int CurrentFileNumber = 0;
int LastIFrame = 0;
toMarks.Add(0);
toMarks.Save();
uchar buffer[MAXFRAMESIZE];
bool LastMark = false;
bool cutIn = true;
2005-08-14 11:24:57 +02:00
while (Running()) {
uint16_t FileNumber;
off_t FileOffset;
int Length;
bool Independent;
2002-06-22 10:11:59 +02:00
// Make sure there is enough disk space:
AssertFreeDiskSpace(-1);
2002-06-22 10:11:59 +02:00
// Read one frame:
if (fromIndex->Get(Index++, &FileNumber, &FileOffset, &Independent, &Length)) {
2002-06-22 10:11:59 +02:00
if (FileNumber != CurrentFileNumber) {
fromFile = fromFileName->SetOffset(FileNumber, FileOffset);
2006-02-04 14:12:17 +01:00
fromFile->SetReadAhead(MEGABYTE(20));
2002-06-22 10:11:59 +02:00
CurrentFileNumber = FileNumber;
}
if (fromFile) {
2002-06-22 10:11:59 +02:00
int len = ReadFrame(fromFile, buffer, Length, sizeof(buffer));
if (len < 0) {
error = "ReadFrame";
break;
}
if (len != Length) {
CurrentFileNumber = 0; // this re-syncs in case the frame was larger than the buffer
Length = len;
}
}
else {
error = "fromFile";
break;
}
}
else {
// Error, unless we're past the last cut-in and there's no cut-out
if (Mark || LastMark)
error = "index";
2002-06-22 10:11:59 +02:00
break;
}
2002-06-22 10:11:59 +02:00
// Write one frame:
if (Independent) { // every file shall start with an independent frame
if (LastMark) // edited version shall end before next I-frame
2002-06-22 10:11:59 +02:00
break;
if (FileSize > maxVideoFileSize) {
2002-06-22 10:11:59 +02:00
toFile = toFileName->NextFile();
if (!toFile) {
2002-06-22 10:11:59 +02:00
error = "toFile 1";
break;
}
FileSize = 0;
}
LastIFrame = 0;
if (cutIn) {
cRemux::SetBrokenLink(buffer, Length);
cutIn = false;
}
2002-06-22 10:11:59 +02:00
}
if (toFile->Write(buffer, Length) < 0) {
2002-06-22 10:11:59 +02:00
error = "safe_write";
break;
}
if (!toIndex->Write(Independent, toFileName->Number(), FileSize)) {
2002-06-22 10:11:59 +02:00
error = "toIndex";
break;
}
FileSize += Length;
if (!LastIFrame)
LastIFrame = toIndex->Last();
// Check editing marks:
if (Mark && Index >= Mark->position) {
Mark = fromMarks.Next(Mark);
toMarks.Add(LastIFrame);
if (Mark)
toMarks.Add(toIndex->Last() + 1);
toMarks.Save();
if (Mark) {
Index = Mark->position;
Mark = fromMarks.Next(Mark);
CurrentFileNumber = 0; // triggers SetOffset before reading next frame
cutIn = true;
2002-06-22 10:11:59 +02:00
if (Setup.SplitEditedFiles) {
toFile = toFileName->NextFile();
if (!toFile) {
2002-06-22 10:11:59 +02:00
error = "toFile 2";
break;
}
FileSize = 0;
}
}
else
LastMark = true;
2002-06-22 10:11:59 +02:00
}
}
Recordings.TouchUpdate();
2002-06-22 10:11:59 +02:00
}
else
esyslog("no editing marks found!");
}
// --- cCutter ---------------------------------------------------------------
char *cCutter::editedVersionName = NULL;
cCuttingThread *cCutter::cuttingThread = NULL;
bool cCutter::error = false;
bool cCutter::ended = false;
bool cCutter::Start(const char *FileName)
{
if (!cuttingThread) {
error = false;
ended = false;
cRecording Recording(FileName);
const char *evn = Recording.PrefixFileName('%');
if (evn && RemoveVideoFile(evn) && MakeDirs(evn, true)) {
// XXX this can be removed once RenameVideoFile() follows symlinks (see videodir.c)
// remove a possible deleted recording with the same name to avoid symlink mixups:
char *s = strdup(evn);
char *e = strrchr(s, '.');
if (e) {
if (strcmp(e, ".rec") == 0) {
strcpy(e, ".del");
RemoveVideoFile(s);
}
}
free(s);
2002-06-22 10:11:59 +02:00
// XXX
editedVersionName = strdup(evn);
Recording.WriteInfo();
Recordings.AddByName(editedVersionName, false);
2002-06-22 10:11:59 +02:00
cuttingThread = new cCuttingThread(FileName, editedVersionName);
return true;
}
}
return false;
}
void cCutter::Stop(void)
{
2005-08-14 11:24:57 +02:00
bool Interrupted = cuttingThread && cuttingThread->Active();
2002-06-22 10:11:59 +02:00
const char *Error = cuttingThread ? cuttingThread->Error() : NULL;
delete cuttingThread;
cuttingThread = NULL;
if ((Interrupted || Error) && editedVersionName) {
if (Interrupted)
isyslog("editing process has been interrupted");
if (Error)
esyslog("ERROR: '%s' during editing process", Error);
RemoveVideoFile(editedVersionName); //XXX what if this file is currently being replayed?
Recordings.DelByName(editedVersionName);
2002-06-22 10:11:59 +02:00
}
}
bool cCutter::Active(void)
{
if (cuttingThread) {
2005-08-14 11:24:57 +02:00
if (cuttingThread->Active())
2002-06-22 10:11:59 +02:00
return true;
error = cuttingThread->Error();
Stop();
if (!error)
cRecordingUserCommand::InvokeCommand(RUC_EDITEDRECORDING, editedVersionName);
free(editedVersionName);
2002-06-22 10:11:59 +02:00
editedVersionName = NULL;
ended = true;
}
return false;
}
bool cCutter::Error(void)
{
bool result = error;
error = false;
return result;
}
bool cCutter::Ended(void)
{
bool result = ended;
ended = false;
return result;
}