From a573d7302badd6f354c429c2e6acc543e7d3e472 Mon Sep 17 00:00:00 2001
From: xiongziliang <771730766@qq.com>
Date: Tue, 14 Jan 2020 18:11:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E9=AB=98rtsp=E5=AE=A2=E6=88=B7?=
 =?UTF-8?q?=E7=AB=AF=E5=85=BC=E5=AE=B9=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Rtsp/Rtsp.cpp | 177 ++++++++++++++++++++++++++--------------------
 src/Rtsp/Rtsp.h   |   2 +-
 2 files changed, 100 insertions(+), 79 deletions(-)

diff --git a/src/Rtsp/Rtsp.cpp b/src/Rtsp/Rtsp.cpp
index bc8dcce4..d6d95258 100644
--- a/src/Rtsp/Rtsp.cpp
+++ b/src/Rtsp/Rtsp.cpp
@@ -85,74 +85,84 @@ string SdpTrack::toString() const {
 	}
 	return _printer;
 }
-void SdpParser::load(const string &sdp) {
-	_track_map.clear();
-	string key;
-	SdpTrack::Ptr track = std::make_shared<SdpTrack>();
 
-	auto lines = split(sdp,"\n");
-	for (auto &line : lines){
-		trim(line);
-		if(line.size() < 2 || line[1] != '='){
-			continue;
-		}
-		char opt = line[0];
-		string opt_val = line.substr(2);
-		switch (opt){
-			case 'o':
-				track->_o = opt_val;
-				break;
-			case 's':
-				track->_s = opt_val;
-				break;
-			case 'i':
-				track->_i = opt_val;
-				break;
-			case 'c':
-				track->_c = opt_val;
-				break;
-			case 't':
-				track->_t = opt_val;
-				break;
-			case 'b':
-				track->_b = opt_val;
-				break;
-			case 'm':{
-				_track_map[key] = track;
-				track = std::make_shared<SdpTrack>();
-				key = FindField(opt_val.data(), nullptr," ");
-				track->_m = opt_val;
-			}
-				break;
-			case 'a':{
-				string attr = FindField(opt_val.data(), nullptr,":");
-				if(attr.empty()){
-					track->_attr[opt_val] = "";
-				}else{
-					track->_attr[attr] = FindField(opt_val.data(),":", nullptr);
-				}
-			}
-				break;
-			default:
-				track->_other[opt] = opt_val;
-				break;
-		}
+static TrackType toTrackType(const string &str) {
+	if (str == "") {
+		return TrackTitle;
 	}
-	_track_map[key] = track;
 
+	if (str == "video") {
+		return TrackVideo;
+	}
 
-	for (auto &pr : _track_map) {
-		auto &track = *pr.second;
-		if (pr.first == "") {
-			track._type = TrackTitle;
-		} else if (pr.first == "video") {
-			track._type = TrackVideo;
-		} else if (pr.first == "audio") {
-			track._type = TrackAudio;
-		} else {
-			track._type = TrackInvalid;
+	if (str == "audio") {
+		return TrackAudio;
+	}
+
+	return TrackInvalid;
+}
+
+void SdpParser::load(const string &sdp) {
+	{
+		_track_vec.clear();
+		string key;
+		SdpTrack::Ptr track = std::make_shared<SdpTrack>();
+
+		auto lines = split(sdp, "\n");
+		for (auto &line : lines) {
+			trim(line);
+			if (line.size() < 2 || line[1] != '=') {
+				continue;
+			}
+			char opt = line[0];
+			string opt_val = line.substr(2);
+			switch (opt) {
+				case 'o':
+					track->_o = opt_val;
+					break;
+				case 's':
+					track->_s = opt_val;
+					break;
+				case 'i':
+					track->_i = opt_val;
+					break;
+				case 'c':
+					track->_c = opt_val;
+					break;
+				case 't':
+					track->_t = opt_val;
+					break;
+				case 'b':
+					track->_b = opt_val;
+					break;
+				case 'm': {
+					track->_type = toTrackType(key);
+					_track_vec.emplace_back(track);
+					track = std::make_shared<SdpTrack>();
+					key = FindField(opt_val.data(), nullptr, " ");
+					track->_m = opt_val;
+				}
+					break;
+				case 'a': {
+					string attr = FindField(opt_val.data(), nullptr, ":");
+					if (attr.empty()) {
+						track->_attr[opt_val] = "";
+					} else {
+						track->_attr[attr] = FindField(opt_val.data(), ":", nullptr);
+					}
+				}
+					break;
+				default:
+					track->_other[opt] = opt_val;
+					break;
+			}
 		}
+		track->_type = toTrackType(key);
+		_track_vec.emplace_back(track);
+	}
 
+	for (auto &track_ptr : _track_vec) {
+		auto &track = *track_ptr;
 		auto it = track._attr.find("range");
 		if (it != track._attr.end()) {
 			char name[16] = {0}, start[16] = {0}, end[16] = {0};
@@ -198,9 +208,9 @@ bool SdpParser::available() const {
 }
 
 SdpTrack::Ptr SdpParser::getTrack(TrackType type) const {
-	for (auto &pr : _track_map){
-		if(pr.second->_type == type){
-			return pr.second;
+	for (auto &track : _track_vec){
+		if(track->_type == type){
+			return track;
 		}
 	}
 	return nullptr;
@@ -208,31 +218,42 @@ SdpTrack::Ptr SdpParser::getTrack(TrackType type) const {
 
 vector<SdpTrack::Ptr> SdpParser::getAvailableTrack() const {
 	vector<SdpTrack::Ptr> ret;
-	auto video = getTrack(TrackVideo);
-	if(video){
-		ret.emplace_back(video);
+	bool audio_added = false;
+	bool video_added = false;
+	for (auto &track : _track_vec){
+		if(track->_type == TrackAudio ){
+			if(!audio_added){
+				ret.emplace_back(track);
+				audio_added = true;
+			}
+			continue;
+		}
+
+		if(track->_type == TrackVideo ){
+			if(!video_added){
+				ret.emplace_back(track);
+				video_added = true;
+			}
+			continue;
+		}
 	}
-	auto audio = getTrack(TrackAudio);
-	if(audio){
-		ret.emplace_back(audio);
-	}
-	return ret;
+	return std::move(ret);
 }
 
 string SdpParser::toString() const {
 	string title,audio,video;
-	for(auto &pr : _track_map){
-		switch (pr.second->_type){
+	for(auto &track : _track_vec){
+		switch (track->_type){
 			case TrackTitle:{
-				title = pr.second->toString();
+				title = track->toString();
 			}
 				break;
 			case TrackVideo:{
-				video = pr.second->toString();
+				video = track->toString();
 			}
 				break;
 			case TrackAudio:{
-				audio = pr.second->toString();
+				audio = track->toString();
 			}
 				break;
 			default:
diff --git a/src/Rtsp/Rtsp.h b/src/Rtsp/Rtsp.h
index c8eb5554..26e2a979 100644
--- a/src/Rtsp/Rtsp.h
+++ b/src/Rtsp/Rtsp.h
@@ -122,7 +122,7 @@ public:
 	vector<SdpTrack::Ptr> getAvailableTrack() const;
 	string toString() const ;
 private:
-	map<string, SdpTrack::Ptr> _track_map;
+	vector<SdpTrack::Ptr> _track_vec;
 };
 
 /**