org.jcodec.movtool.Cut.java Source code

Introduction

Here is the source code for org.jcodec.movtool.Cut.java

Source

package org.jcodec.movtool;

import static java.lang.Integer.parseInt;
import static java.lang.Math.max;
import static org.apache.commons.io.FilenameUtils.removeExtension;
import static org.jcodec.common.JCodecUtil.bufin;
import static org.jcodec.containers.mp4.MP4Util.createRefMovie;
import static org.jcodec.movtool.Util.forceEditList;

import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;

import org.apache.commons.lang.StringUtils;
import org.jcodec.common.JCodecUtil;
import org.jcodec.common.io.FileRAInputStream;
import org.jcodec.common.io.RAInputStream;
import org.jcodec.containers.mp4.boxes.Edit;
import org.jcodec.containers.mp4.boxes.MovieBox;
import org.jcodec.containers.mp4.boxes.TrakBox;

/**
 * This class is part of JCodec ( www.jcodec.org ) This software is distributed
 * under FreeBSD License
 * 
 * Cut on ref movies
 * 
 * @author The JCodec project
 * 
 */
public class Cut {
    public static void main(String[] args) throws Exception {
        if (args.length < 1) {
            System.out.println("Syntax: cut [-command arg]...[-command arg] [-self] <movie file>\n"
                    + "\tCreates a reference movie out of the file and applies a set of changes specified by the commands to it.");
            System.exit(-1);
        }

        List<Slice> slices = new ArrayList<Slice>();
        List<String> sliceNames = new ArrayList<String>();

        boolean selfContained = false;
        int shift = 0;
        while (true) {
            if ("-cut".equals(args[shift])) {
                String[] pt = StringUtils.split(args[shift + 1], ":");
                slices.add(new Slice(parseInt(pt[0]), parseInt(pt[1])));
                if (pt.length > 2)
                    sliceNames.add(pt[2]);
                else
                    sliceNames.add(null);
                shift += 2;
            } else if ("-self".equals(args[shift])) {
                ++shift;
                selfContained = true;
            } else
                break;
        }
        File source = new File(args[shift]);

        RAInputStream input = null;
        RandomAccessFile out = null;
        List<RandomAccessFile> outs = new ArrayList<RandomAccessFile>();
        try {
            input = bufin(source);
            MovieBox movie = createRefMovie(input, "file://" + source.getCanonicalPath());
            List<MovieBox> slicesMovs;
            if (!selfContained) {
                out = new RandomAccessFile(
                        new File(source.getParentFile(), removeExtension(source.getName()) + ".ref.mov"), "rw");
                slicesMovs = new Cut().cut(movie, slices);
                movie.write(out);
            } else {
                out = new RandomAccessFile(
                        new File(source.getParentFile(), removeExtension(source.getName()) + ".self.mov"), "rw");
                slicesMovs = new Cut().cut(movie, slices);
                new Strip().strip(movie);
                new Flattern().flattern(movie, out);
            }
            saveSlices(slicesMovs, sliceNames, source.getParentFile());
        } finally {
            if (input != null)
                input.close();
            if (out != null)
                out.close();
            for (RandomAccessFile o : outs) {
                o.close();
            }
        }
    }

    private static void saveSlices(List<MovieBox> slices, List<String> names, File parentFile) throws IOException {
        for (int i = 0; i < slices.size(); i++) {
            if (names.get(i) == null)
                continue;
            RandomAccessFile out = null;
            try {
                out = new RandomAccessFile(new File(parentFile, names.get(i)), "rw");
                slices.get(i).write(out);
            } finally {
                if (out != null)
                    out.close();
            }
        }
    }

    public static class Slice {
        private double inSec;
        private double outSec;

        public Slice(double in, double out) {
            super();
            this.inSec = in;
            this.outSec = out;
        }
    }

    public List<MovieBox> cut(MovieBox movie, List<Slice> commands) {

        TrakBox videoTrack = movie.getVideoTrack();
        if (videoTrack != null && videoTrack.getTimescale() != movie.getTimescale())
            movie.fixTimescale(videoTrack.getTimescale());

        TrakBox[] tracks = movie.getTracks();
        for (TrakBox trakBox : tracks) {
            forceEditList(movie, trakBox);
            List<Edit> edits = trakBox.getEdits();
            for (Slice cut : commands) {
                split(edits, cut.inSec, movie, trakBox);
                split(edits, cut.outSec, movie, trakBox);
            }
        }
        ArrayList<MovieBox> result = new ArrayList<MovieBox>();
        for (Slice cut : commands) {
            MovieBox clone = (MovieBox) movie.cloneBox();
            for (TrakBox trakBox : clone.getTracks()) {
                selectInner(trakBox.getEdits(), cut, movie, trakBox);
            }
            result.add(clone);
        }

        long movDuration = 0;
        for (TrakBox trakBox : movie.getTracks()) {
            selectOuter(trakBox.getEdits(), commands, movie, trakBox);
            trakBox.setEdits(trakBox.getEdits());
            movDuration = max(movDuration, trakBox.getDuration());
        }
        movie.setDuration(movDuration);

        return result;
    }

    private void selectOuter(List<Edit> edits, List<Slice> commands, MovieBox movie, TrakBox trakBox) {
        long[] inMv = new long[commands.size()];
        long[] outMv = new long[commands.size()];
        for (int i = 0; i < commands.size(); i++) {
            inMv[i] = (long) (commands.get(i).inSec * movie.getTimescale());
            outMv[i] = (long) (commands.get(i).outSec * movie.getTimescale());
        }
        long editStartMv = 0;
        ListIterator<Edit> lit = edits.listIterator();
        while (lit.hasNext()) {
            Edit edit = lit.next();
            for (int i = 0; i < inMv.length; i++) {
                if (editStartMv + edit.getDuration() > inMv[i] && editStartMv < outMv[i])
                    lit.remove();
            }
            editStartMv += edit.getDuration();
        }
    }

    private void selectInner(List<Edit> edits, Slice cut, MovieBox movie, TrakBox trakBox) {
        long inMv = (long) (movie.getTimescale() * cut.inSec);
        long outMv = (long) (movie.getTimescale() * cut.outSec);

        long editStart = 0;
        ListIterator<Edit> lit = edits.listIterator();
        while (lit.hasNext()) {
            Edit edit = lit.next();
            if (editStart + edit.getDuration() <= inMv || editStart >= outMv)
                lit.remove();
            editStart += edit.getDuration();
        }
    }

    private void split(List<Edit> edits, double sec, MovieBox movie, TrakBox trakBox) {
        Util.split(movie, trakBox, (long) (sec * movie.getTimescale()));
    }
}