From 70153f192cdaffb851313fdc1aebc9ecd131244d Mon Sep 17 00:00:00 2001
From: shikong <919411476@qq.com>
Date: Sun, 7 Jan 2024 19:22:36 +0800
Subject: [PATCH] =?UTF-8?q?GB28181Description=20=E8=A1=A5=E5=85=85=20Strea?=
 =?UTF-8?q?mType=20=E4=B8=BB/=E5=AD=90=E7=A0=81=E6=B5=81=E5=AD=97=E6=AE=B5?=
 =?UTF-8?q?=E5=AE=9A=E4=B9=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../gb28181/sdp/GB28181SDPBuilder.java        | 68 +++++++++++++++++++
 .../cn/skcks/docking/gb28181/sdp/SdpTest.java | 10 +++
 2 files changed, 78 insertions(+)

diff --git a/gb28181-sip/src/main/java/cn/skcks/docking/gb28181/sdp/GB28181SDPBuilder.java b/gb28181-sip/src/main/java/cn/skcks/docking/gb28181/sdp/GB28181SDPBuilder.java
index 00e91cc..0241aab 100644
--- a/gb28181-sip/src/main/java/cn/skcks/docking/gb28181/sdp/GB28181SDPBuilder.java
+++ b/gb28181-sip/src/main/java/cn/skcks/docking/gb28181/sdp/GB28181SDPBuilder.java
@@ -11,6 +11,7 @@ import gov.nist.javax.sdp.fields.ConnectionField;
 import gov.nist.javax.sdp.fields.TimeField;
 import gov.nist.javax.sdp.fields.URIField;
 import lombok.AllArgsConstructor;
+import lombok.Builder;
 import lombok.Getter;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
@@ -42,6 +43,53 @@ public class GB28181SDPBuilder {
         put("125", "profile-level-id=42e01e");
     }};
 
+    public static class StreamType {
+        public interface Attribute<T> {
+            AttributeField stream();
+        }
+
+        public static AttributeField getAttribute(Attribute<?> attribute) {
+            return attribute.stream();
+        }
+
+        @AllArgsConstructor
+        public static class TPLink implements Attribute<String>{
+            private String stream;
+            public static final TPLink MAIN = new TPLink("main");
+
+            public static final TPLink SUB = new TPLink("sub");
+            public AttributeField stream() {
+                return (AttributeField) SdpFactory.getInstance().createAttribute("streamMode", stream);
+            }
+        }
+
+        @AllArgsConstructor
+        public static class StreamProfile implements Attribute<Integer> {
+            private Integer stream;
+            public static final StreamProfile MAIN = new StreamProfile(0);
+            public static final StreamProfile SUB = new StreamProfile(1);
+
+            @Override
+            public AttributeField stream() {
+                return (AttributeField) SdpFactory.getInstance().createAttribute("streamprofile", String.valueOf(stream));
+            }
+        }
+
+        @AllArgsConstructor
+        public static class GB2022 implements Attribute<Integer> {
+            private Integer stream;
+            public static final GB2022 MAIN = new GB2022(0);
+            public static final GB2022 SUB = new GB2022(1);
+
+            public AttributeField stream(){
+                return (AttributeField) SdpFactory.getInstance().createAttribute("streamnumber", String.valueOf(stream));
+            }
+        }
+
+        public static final GB2022 DEFAULT_GB_2022 = GB2022.MAIN;
+        public static final StreamProfile DEFAULT = StreamProfile.MAIN;
+    }
+
     @AllArgsConstructor
     @Getter
     public enum Protocol {
@@ -170,6 +218,26 @@ public class GB28181SDPBuilder {
             return build(Action.PLAY, deviceId, channelId, netType, rtpIp, rtpPort, ssrc, streamMode, timeDescription);
         }
 
+        /**
+         *
+         * @param deviceId 设备id
+         * @param channelId 通道id
+         * @param netType 网络类型
+         * @param rtpIp rtp服务器ip
+         * @param rtpPort rtp端口
+         * @param ssrc ssrc
+         * @param streamMode 网络类型
+         * @param streamType 流类型 (主/子码流)
+         * @return GB28181Description sdp
+         */
+        @SneakyThrows
+        public static GB28181Description play(String deviceId, String channelId, String netType, String rtpIp, int rtpPort, String ssrc, MediaStreamMode streamMode, StreamType.Attribute<?> streamType) {
+            GB28181Description play = play(deviceId, channelId, netType, rtpIp, rtpPort, ssrc, streamMode);
+            MediaDescription m = (MediaDescription)play.getMediaDescriptions(false).get(0);
+            m.addAttribute(StreamType.getAttribute(streamType));
+            return play;
+        }
+
         @SneakyThrows
         public static GB28181Description playback(String deviceId, String channelId, String netType, String rtpIp, int rtpPort, String ssrc, MediaStreamMode streamMode, Date start, Date end) {
             TimeField timeField = new TimeField();
diff --git a/gb28181-sip/src/test/java/cn/skcks/docking/gb28181/sdp/SdpTest.java b/gb28181-sip/src/test/java/cn/skcks/docking/gb28181/sdp/SdpTest.java
index 6ebebdc..92d2ef7 100644
--- a/gb28181-sip/src/test/java/cn/skcks/docking/gb28181/sdp/SdpTest.java
+++ b/gb28181-sip/src/test/java/cn/skcks/docking/gb28181/sdp/SdpTest.java
@@ -70,4 +70,14 @@ public class SdpTest {
         GB28181Description resp = GB28181SDPBuilder.Sender.build(play, sendRtpIp, sendRtpPort,respRtpMap,null);
         log.info("sdp 响应\n{}", resp);
     }
+
+    @Test
+    public void streamType() {
+        log.info("{}", GB28181SDPBuilder.StreamType.getAttribute(GB28181SDPBuilder.StreamType.TPLink.MAIN));
+        log.info("{}", GB28181SDPBuilder.StreamType.getAttribute(GB28181SDPBuilder.StreamType.TPLink.SUB));
+
+        log.info("{}", GB28181SDPBuilder.StreamType.getAttribute(GB28181SDPBuilder.StreamType.GB2022.MAIN));
+        log.info("{}", GB28181SDPBuilder.StreamType.getAttribute(GB28181SDPBuilder.StreamType.GB2022.SUB));
+        log.info("{}", GB28181SDPBuilder.StreamType.getAttribute(new GB28181SDPBuilder.StreamType.GB2022(5)));
+    }
 }