/*
* RED5 Open Source Flash Server - http://code.google.com/p/red5/
*
* Copyright 2006-2014 by respective authors (see below). All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.red5.server.service;
import java.io.File;
import java.io.IOException;
import java.util.TreeSet;
import junit.framework.TestCase;
import org.apache.mina.core.buffer.IoBuffer;
import org.red5.cache.impl.NoCacheImpl;
import org.red5.io.ITag;
import org.red5.io.ITagReader;
import org.red5.io.ITagWriter;
import org.red5.io.IoConstants;
import org.red5.io.amf.Output;
import org.red5.io.flv.IFLV;
import org.red5.io.flv.impl.Tag;
import org.red5.io.flv.meta.ICueType;
import org.red5.io.flv.meta.IMetaCue;
import org.red5.io.flv.meta.MetaCue;
import org.red5.io.object.Serializer;
import org.red5.server.service.flv.IFLVService;
import org.red5.server.service.flv.impl.FLVService;
/**
* @author The Red5 Project
* @author Dominick Accattato (daccattato@gmail.com)
* @author Luke Hubbard, Codegent Ltd (luke@codegent.com)
*/
public class MetaDataInjectionTest extends TestCase {
private IFLVService service;
/**
* SetUp is called before each test
*/
@Override
public void setUp() {
service = new FLVService();
}
/**
* Test MetaData injection
* @throws IOException if io exception
*/
public void testMetaDataInjection() throws IOException {
String path = "target/test-classes/fixtures/test_cue1.flv";
File f = new File(path);
System.out.println("Path: " + f.getAbsolutePath());
if (f.exists()) {
f.delete();
} else {
// try test subdirectory
path = "target/test-classes/fixtures/test_cue1.flv";
f = new File(path);
System.out.println("Path: " + f.getAbsolutePath());
if (f.exists()) {
f.delete();
}
}
// Create new file
f.createNewFile();
// Use service to grab FLV file
IFLV flv = (IFLV) service.getStreamableFile(f);
// Grab a writer for writing a new FLV
ITagWriter writer = flv.getWriter();
// Create a reader for testing
File readfile = new File(path);
IFLV readflv = (IFLV) service.getStreamableFile(readfile);
readflv.setCache(NoCacheImpl.getInstance());
// Grab a reader for reading a FLV in
ITagReader reader = readflv.getReader();
// Inject MetaData
writeTagsWithInjection(reader, writer);
}
/**
* Write FLV tags and inject Cue Points
* @param reader
* @param writer
* @throws IOException
*/
private void writeTagsWithInjection(ITagReader reader, ITagWriter writer) throws IOException {
IMetaCue cp = new MetaCue<Object, Object>();
cp.setName("cue_1");
cp.setTime(0.01);
cp.setType(ICueType.EVENT);
IMetaCue cp1 = new MetaCue<Object, Object>();
cp1.setName("cue_1");
cp1.setTime(2.01);
cp1.setType(ICueType.EVENT);
// Place in TreeSet for sorting
TreeSet<IMetaCue> ts = new TreeSet<IMetaCue>();
ts.add(cp);
ts.add(cp1);
int cuePointTimeStamp = getTimeInMilliseconds(ts.first());
ITag tag = null;
ITag injectedTag = null;
while (reader.hasMoreTags()) {
tag = reader.readTag();
if (tag.getDataType() != IoConstants.TYPE_METADATA) {
//injectNewMetaData();
} else {
//in
}
// if there are cuePoints in the TreeSet
if (!ts.isEmpty()) {
// If the tag has a greater timestamp than the
// cuePointTimeStamp, then inject the tag
while (tag.getTimestamp() > cuePointTimeStamp) {
injectedTag = injectMetaData(ts.first(), tag);
writer.writeTag(injectedTag);
tag.setPreviousTagSize((injectedTag.getBodySize() + 11));
// Advance to the next CuePoint
ts.remove(ts.first());
if (ts.isEmpty()) {
break;
}
cuePointTimeStamp = getTimeInMilliseconds(ts.first());
}
}
writer.writeTag(tag);
}
}
/**
* Injects metadata (Cue Points) into a tag
* @param cue
* @param writer
* @param tag
* @return ITag tag
*/
private ITag injectMetaData(Object cue, ITag tag) {
IMetaCue cp = (MetaCue<?, ?>) cue;
Output out = new Output(IoBuffer.allocate(1000));
Serializer.serialize(out, "onCuePoint");
Serializer.serialize(out, cp);
IoBuffer tmpBody = out.buf().flip();
int tmpBodySize = out.buf().limit();
int tmpPreviousTagSize = tag.getPreviousTagSize();
byte tmpDataType = IoConstants.TYPE_METADATA;
int tmpTimestamp = getTimeInMilliseconds(cp);
return new Tag(tmpDataType, tmpTimestamp, tmpBodySize, tmpBody, tmpPreviousTagSize);
}
/**
* Returns a timestamp in milliseconds
* @param object
* @return int time
*/
private int getTimeInMilliseconds(Object object) {
IMetaCue cp = (MetaCue<?, ?>) object;
return (int) (cp.getTime() * 1000.00);
}
}