From 5f6f2f28594f406f2c115e323043bf91f4f0776f Mon Sep 17 00:00:00 2001
From: Chris Allen <chrisallen@users.noreply.github.com>
Date: Wed, 1 Feb 2017 22:22:33 +1300
Subject: [PATCH] Add expanded DTS audio codecs to FileNameBuilder and fix up
 Atmos TrueHD audioCodec string. (#559)

* Add DTS-ES, DTS-HD HRA, DTS-HD MA, DTS-X to audioCodec.

* Cleanup "Atmos / TrueHD" audioCodec string.
---
 .../Organizer/FileNameBuilder.cs              | 25 ++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/src/NzbDrone.Core/Organizer/FileNameBuilder.cs b/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
index c78590015..9e23ed20c 100644
--- a/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
+++ b/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
@@ -730,6 +730,10 @@ namespace NzbDrone.Core.Organizer
                 case "E-AC-3":
                     audioCodec = "EAC3";
                     break;
+                
+                case "Atmos / TrueHD":
+                    audioCodec = "Atmos TrueHD";
+                    break;
 
                 case "MPEG Audio":
                     if (movieFile.MediaInfo.AudioProfile == "Layer 3")
@@ -743,7 +747,26 @@ namespace NzbDrone.Core.Organizer
                     break;
 
                 case "DTS":
-                    audioCodec = movieFile.MediaInfo.AudioFormat;
+                    if (episodeFile.MediaInfo.AudioProfile == "ES Discrete / Core" || episodeFile.MediaInfo.AudioProfile == "ES Matrix / Core")
+                    {
+                        audioCodec = "DTS-ES";
+                    }
+                    else if (episodeFile.MediaInfo.AudioProfile == "MA / Core" || episodeFile.MediaInfo.AudioProfile == "MA / ES Matrix / Core")
+                    {
+                        audioCodec = "DTS-HD MA";
+                    }
+                    else if (episodeFile.MediaInfo.AudioProfile == "HRA / Core")
+                    {
+                        audioCodec = "DTS-HD HRA";
+                    }
+                    else if (episodeFile.MediaInfo.AudioProfile == "X / MA / Core")
+                    {
+                        audioCodec = "DTS-X";
+                    }
+                    else
+                    {
+                        audioCodec = episodeFile.MediaInfo.AudioFormat;
+                    }
                     break;
 
                 default: