#include "evergreend.h"
 #include "atom.h"
 
-extern void dce6_afmt_select_pin(struct drm_encoder *encoder);
-
 /* enable the audio stream */
 static void dce4_audio_enable(struct radeon_device *rdev,
                              struct r600_audio_pin *pin,
 
        /* fglrx sets 0x40 in 0x5f80 here */
 
-       if (ASIC_IS_DCE6(rdev))
-               dce6_afmt_select_pin(encoder);
-
+       radeon_audio_select_pin(encoder);
        radeon_audio_write_sad_regs(encoder);
        radeon_audio_write_latency_fields(encoder, mode);
 
 
                struct drm_connector *connector, struct drm_display_mode *mode);
 struct r600_audio_pin* r600_audio_get_pin(struct radeon_device *rdev);
 struct r600_audio_pin* dce6_audio_get_pin(struct radeon_device *rdev);
+void dce6_afmt_select_pin(struct drm_encoder *encoder);
 
 static const u32 pin_offsets[7] =
 {
 };
 
 static struct radeon_audio_funcs dce6_hdmi_funcs = {
+       .select_pin = dce6_afmt_select_pin,
        .get_pin = dce6_audio_get_pin,
        .write_sad_regs = dce6_afmt_write_sad_regs,
        .write_speaker_allocation = dce6_afmt_hdmi_write_speaker_allocation,
 };
 
 static struct radeon_audio_funcs dce6_dp_funcs = {
+       .select_pin = dce6_afmt_select_pin,
        .get_pin = dce6_audio_get_pin,
        .write_sad_regs = dce6_afmt_write_sad_regs,
        .write_speaker_allocation = dce6_afmt_dp_write_speaker_allocation,
 
        return NULL;
 }
+
+void radeon_audio_select_pin(struct drm_encoder *encoder)
+{
+       struct radeon_encoder *radeon_encoder = to_radeon_encoder(encoder);
+
+       if (radeon_encoder->audio && radeon_encoder->audio->select_pin)
+               radeon_encoder->audio->select_pin(encoder);
+}
 
 
 struct radeon_audio_funcs
 {
+       void (*select_pin)(struct drm_encoder *encoder);
        struct r600_audio_pin* (*get_pin)(struct radeon_device *rdev);
        void (*write_latency_fields)(struct drm_encoder *encoder,
                struct drm_connector *connector, struct drm_display_mode *mode);
 void radeon_audio_write_latency_fields(struct drm_encoder *encoder,
        struct drm_display_mode *mode);
 struct r600_audio_pin *radeon_audio_get_pin(struct drm_encoder *encoder);
+void radeon_audio_select_pin(struct drm_encoder *encoder);
 
 #endif