Examples with PMTSection org.jcodec.containers.mps.psi.PMTSection used on opensource projects

Example 1 with PMTSection

use of org.jcodec.containers.mps.psi.PMTSection in project alliance by codice.

the class MpegTsDecoderImpl method handleProgramMapTable.

private void handleProgramMapTable(MTSPacket mtsPacket) {
    final ByteBuffer payload = mtsPacket.getPayload();
    final int pointer = payload.get() & BYTE_MASK;
    payload.position(payload.position() + pointer);
    final PMTSection pmt = pmtSectionParser.parse(payload);
    for (final PMTSection.PMTStream stream : pmt.getStreams()) {
        programElementaryStreams.put(stream.getPid(), stream);
    }
}

Example 2 with PMTSection

use of org.jcodec.containers.mps.psi.PMTSection in project alliance by codice.

the class MpegTsDecoderTest method testRead.

@Test
public void testRead() {
    MTSUtils.StreamType streamType = MTSUtils.StreamType.VIDEO_H264;
    byte expectedByte1 = 0x01;
    byte expectedByte2 = 0x02;
    byte expectedByte3 = 0x03;
    byte expectedByte4 = 0x04;
    int programMapTableId = 1;
    int videoPacketId = 2;
    MpegTsDecoderImpl decoder = new MpegTsDecoderImpl();
    PATSection patSection = mock(PATSection.class);
    when(patSection.getPrograms()).thenReturn(Collections.singletonMap(1, programMapTableId));
    MpegTsDecoderImpl.PATSectionParser patSectionParser = mock(MpegTsDecoderImpl.PATSectionParser.class);
    when(patSectionParser.parse(any())).thenReturn(patSection);
    decoder.setPatSectionParser(patSectionParser);
    MTSPacket programAssociationTablePacket = mock(MTSPacket.class);
    when(programAssociationTablePacket.getPid()).thenReturn(Constants.PROGRAM_ASSOCIATION_TABLE_PID);
    when(programAssociationTablePacket.isPayloadUnitStartIndicator()).thenReturn(true);
    when(programAssociationTablePacket.getPayload()).thenReturn(ByteBuffer.wrap(new byte[] { 0x00 }));
    PMTSection.PMTStream pmtStream = mock(PMTSection.PMTStream.class);
    when(pmtStream.getStreamType()).thenReturn(streamType);
    when(pmtStream.getPid()).thenReturn(videoPacketId);
    PMTSection pmtSection = mock(PMTSection.class);
    when(pmtSection.getStreams()).thenReturn(new PMTSection.PMTStream[] { pmtStream });
    MpegTsDecoderImpl.PMTSectionParser pmtSectionParser = mock(MpegTsDecoderImpl.PMTSectionParser.class);
    when(pmtSectionParser.parse(any())).thenReturn(pmtSection);
    decoder.setPmtSectionParser(pmtSectionParser);
    MTSPacket programMapTablePacket = mock(MTSPacket.class);
    when(programMapTablePacket.getPid()).thenReturn(programMapTableId);
    when(programMapTablePacket.isPayloadUnitStartIndicator()).thenReturn(true);
    when(programMapTablePacket.getPayload()).thenReturn(ByteBuffer.wrap(new byte[] { 0x00 }));
    MTSPacket elementaryStreamPacket1 = createElementary(true, videoPacketId, expectedByte1);
    MTSPacket elementaryStreamPacket2 = createElementary(false, videoPacketId, expectedByte2);
    MTSPacket elementaryStreamPacket3 = createElementary(false, videoPacketId, expectedByte3);
    MTSPacket elementaryStreamPacket4 = createElementary(false, videoPacketId, expectedByte4);
    MTSPacket elementaryStreamPacket5 = createElementary(true, videoPacketId, (byte) 0x00);
    List<Object> outputList = new LinkedList<>();
    Stream.of(programAssociationTablePacket, programMapTablePacket, elementaryStreamPacket1, elementaryStreamPacket2, elementaryStreamPacket3, elementaryStreamPacket4, elementaryStreamPacket5).forEach(mtsPacket -> {
        try {
            decoder.read(mtsPacket, outputList::add);
        } catch (IOException e) {
            fail();
        }
    });
    assertThat(outputList, hasSize(1));
    assertThat(outputList.get(0), is(instanceOf(PESPacket.class)));
    PESPacket pesPacket = (PESPacket) outputList.get(0);
    assertThat(pesPacket.getPacketId(), is(videoPacketId));
    assertThat(pesPacket.getStreamType(), is(MpegStreamType.lookup(streamType)));
    assertThat(pesPacket.getPayload(), is(new byte[] { expectedByte1, expectedByte2, expectedByte3, expectedByte4 }));
}

Example 3 with PMTSection

use of org.jcodec.containers.mps.psi.PMTSection in project ddf by codice.

the class MpegTransportStreamMetadataExtractor method getProgramMapTable.

private void getProgramMapTable(final MTSPacket packet) {
    final ByteBuffer payload = packet.getPayload();
    final int pointer = payload.get() & 0xff;
    payload.position(payload.position() + pointer);
    final int packetId = packet.getPid();
    final PMTSection pmt = PMTSection.parsePMT(payload);
    programMapTables.put(packetId, pmt);
    for (final PMTStream stream : pmt.getStreams()) {
        programElementaryStreams.put(stream.getPid(), stream);
    }
}