import static org.jcodec.common.NIOUtils.writableFileChannel;
import java.io.File;
import java.nio.ByteBuffer;
import java.util.Arrays;
import org.jcodec.codecs.raw.V210Encoder;
import org.jcodec.common.model.ColorSpace;
import org.jcodec.common.model.Picture;
import org.jcodec.common.model.Size;
import org.jcodec.containers.mp4.muxer.FramesMP4MuxerTrack;
import org.jcodec.containers.mp4.muxer.MP4Muxer;
import org.jcodec.containers.mp4.MP4Packet;

/**
 * This class is part of JCodec ( www.jcodec.org ) This software is distributed
 * under FreeBSD License
 *
 * This code generates two dark gradients 8bit and 10 bit quantization and saves
 * it into v210 mov
 *
 * @author Jay Codec
 *
 */
public class main {
  public static void main(String[] args) throws Exception {
    if (args.length == 0)
      args = new String[] {"out.mov", "200"};

    if (args.length < 2) {
      System.out.println("<out mov> <n frames>");
      return;
    }

    System.out.println("Writing movie: " + args[0]);
    int width = 640;
    int height = 480;
    Picture pic = Picture.create(width, height, ColorSpace.YUV422_10);
    Arrays.fill(pic.getPlaneData(1), 512);
    Arrays.fill(pic.getPlaneData(2), 512);
    drawGrad(pic.getPlaneData(0), new Size(pic.getWidth(), pic.getHeight()));
    V210Encoder encoder = new V210Encoder();
    MP4Muxer muxer = new MP4Muxer(writableFileChannel(new File(args[0])));
    ByteBuffer out = ByteBuffer.allocate(width * height * 10);
    ByteBuffer frame = encoder.encodeFrame(out, pic);
    FramesMP4MuxerTrack videoTrack = muxer.addVideoTrack("v210", new Size(width, height), "jcodec", 24000);
    for (int i = 0; i < Integer.parseInt(args[1]); i++) {
      videoTrack.addFrame(new MP4Packet(frame, i * 1001, 24000, 1001, i, true, null, i * 1001, 0));
    }
    muxer.writeHeader();
    System.out.println("Done writing movie.");
  }

  private static void drawGrad(int[] y, Size ySize) {
    int blockX = ySize.getWidth() / 10;
    int blockY = ySize.getHeight() / 7;
    fillGrad(y, ySize.getWidth(), blockX, blockY, 9 * blockX, 3 * blockY, 0.2, 0.1, 8);
    fillGrad(y, ySize.getWidth(), blockX, 4 * blockY, 9 * blockX, 6 * blockY, 0.2, 0.1, 10);
  }

  private static void fillGrad(int[] y, int stride, int left, int top, int right, int bottom, double from, double to,
                               int quant) {
    int step = stride + left - right;
    int off = top * stride + left;
    for (int j = top; j < bottom; j++) {
      for (int i = left; i < right; i++) {
        y[off++] = colr(((double) i - left) / (right - left), ((double) j - top) / (bottom - top), from, to,
          quant);
      }
      off += step;
    }
  }

  private static int colr(double i, double j, double from, double to, int quant) {
    int val = ((int) ((1 << quant) * (from + (to - from) * i))) << (10 - quant);
    return val;
  }
}