package com.xiaoxuan.demo.tv.widget;

import java.lang.ref.WeakReference;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import android.annotation.TargetApi;
import android.content.Context;
import android.graphics.SurfaceTexture;
import android.os.Build;
import android.util.AttributeSet;
import android.util.Log;
import android.view.Surface;
import android.view.SurfaceHolder;
import android.view.TextureView;
import android.view.View;
import android.view.accessibility.AccessibilityEvent;
import android.view.accessibility.AccessibilityNodeInfo;

import com.xiaoxuan.demo.tv.widget.view.IRenderView;

import tv.danmaku.ijk.media.player.IMediaPlayer;
import tv.danmaku.ijk.media.player.ISurfaceTextureHolder;
import tv.danmaku.ijk.media.player.ISurfaceTextureHost;

@TargetApi(Build.VERSION_CODES.ICE_CREAM_SANDWICH)
public class TextureRenderView extends TextureView implements IRenderView
{
    private static final String TAG = "TextureRenderView";
    
    private MeasureHelper mMeasureHelper;
    
    public TextureRenderView(Context context)
    {
        super(context);
        initView(context);
    }
    
    public TextureRenderView(Context context, AttributeSet attrs)
    {
        super(context, attrs);
        initView(context);
    }
    
    public TextureRenderView(Context context, AttributeSet attrs, int defStyleAttr)
    {
        super(context, attrs, defStyleAttr);
        initView(context);
    }
    
    @TargetApi(Build.VERSION_CODES.LOLLIPOP)
    public TextureRenderView(Context context, AttributeSet attrs, int defStyleAttr, int defStyleRes)
    {
        super(context, attrs, defStyleAttr, defStyleRes);
        initView(context);
    }
    
    private void initView(Context context)
    {
        mMeasureHelper = new MeasureHelper(this);
        mSurfaceCallback = new SurfaceCallback(this);
        setSurfaceTextureListener(mSurfaceCallback);
    }
    
    @Override
    public View getView()
    {
        return this;
    }
    
    @Override
    public boolean shouldWaitForResize()
    {
        return false;
    }
    
    @Override
    protected void onDetachedFromWindow()
    {
        mSurfaceCallback.willDetachFromWindow();
        super.onDetachedFromWindow();
        mSurfaceCallback.didDetachFromWindow();
    }
    
    // --------------------
    // Layout & Measure
    // --------------------
    @Override
    public void setVideoSize(int videoWidth, int videoHeight)
    {
        if (videoWidth > 0 && videoHeight > 0)
        {
            mMeasureHelper.setVideoSize(videoWidth, videoHeight);
            requestLayout();
        }
    }
    
    @Override
    public void setVideoSampleAspectRatio(int videoSarNum, int videoSarDen)
    {
        if (videoSarNum > 0 && videoSarDen > 0)
        {
            mMeasureHelper.setVideoSampleAspectRatio(videoSarNum, videoSarDen);
            requestLayout();
        }
    }
    
    @Override
    public void setVideoRotation(int degree)
    {
        mMeasureHelper.setVideoRotation(degree);
        setRotation(degree);
    }
    
    @Override
    public void setAspectRatio(int aspectRatio)
    {
        mMeasureHelper.setAspectRatio(aspectRatio);
        requestLayout();
    }
    
    @Override
    protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec)
    {
        mMeasureHelper.doMeasure(widthMeasureSpec, heightMeasureSpec);
        setMeasuredDimension(mMeasureHelper.getMeasuredWidth(), mMeasureHelper.getMeasuredHeight());
    }
    
    // --------------------
    // TextureViewHolder
    // --------------------
    
    public IRenderView.ISurfaceHolder getSurfaceHolder()
    {
        return new InternalSurfaceHolder(this, mSurfaceCallback.mSurfaceTexture, mSurfaceCallback);
    }
    
    private static final class InternalSurfaceHolder implements IRenderView.ISurfaceHolder
    {
        private TextureRenderView mTextureView;
        
        private SurfaceTexture mSurfaceTexture;
        
        private ISurfaceTextureHost mSurfaceTextureHost;
        
        public InternalSurfaceHolder(TextureRenderView textureView, SurfaceTexture surfaceTexture,
            ISurfaceTextureHost surfaceTextureHost)
        {
            mTextureView = textureView;
            mSurfaceTexture = surfaceTexture;
            mSurfaceTextureHost = surfaceTextureHost;
        }
        
        @TargetApi(Build.VERSION_CODES.JELLY_BEAN)
        public void bindToMediaPlayer(IMediaPlayer mp)
        {
            if (mp == null)
                return;
            
            if ((Build.VERSION.SDK_INT >= Build.VERSION_CODES.JELLY_BEAN) && (mp instanceof ISurfaceTextureHolder))
            {
                ISurfaceTextureHolder textureHolder = (ISurfaceTextureHolder)mp;
                mTextureView.mSurfaceCallback.setOwnSurfaceTexture(false);
                
                SurfaceTexture surfaceTexture = textureHolder.getSurfaceTexture();
                if (surfaceTexture != null)
                {
                    mTextureView.setSurfaceTexture(surfaceTexture);
                }
                else
                {
                    textureHolder.setSurfaceTexture(mSurfaceTexture);
                    textureHolder.setSurfaceTextureHost(mTextureView.mSurfaceCallback);
                }
            }
            else
            {
                mp.setSurface(openSurface());
            }
        }
        
        @Override
        public IRenderView getRenderView()
        {
            return mTextureView;
        }
        
        @Override
        public SurfaceHolder getSurfaceHolder()
        {
            return null;
        }
        
        @Override
        public SurfaceTexture getSurfaceTexture()
        {
            return mSurfaceTexture;
        }
        
        @Override
        public Surface openSurface()
        {
            if (mSurfaceTexture == null)
                return null;
            return new Surface(mSurfaceTexture);
        }
    }
    
    // -------------------------
    // SurfaceHolder.Callback
    // -------------------------
    
    @Override
    public void addRenderCallback(IRenderCallback callback)
    {
        mSurfaceCallback.addRenderCallback(callback);
    }
    
    @Override
    public void removeRenderCallback(IRenderCallback callback)
    {
        mSurfaceCallback.removeRenderCallback(callback);
    }
    
    private SurfaceCallback mSurfaceCallback;
    
    private static final class SurfaceCallback implements SurfaceTextureListener, ISurfaceTextureHost
    {
        private SurfaceTexture mSurfaceTexture;
        
        private boolean mIsFormatChanged;
        
        private int mWidth;
        
        private int mHeight;
        
        private boolean mOwnSurfaceTexture = true;
        
        private boolean mWillDetachFromWindow = false;
        
        private boolean mDidDetachFromWindow = false;
        
        private WeakReference<TextureRenderView> mWeakRenderView;
        
        private Map<IRenderCallback, Object> mRenderCallbackMap = new ConcurrentHashMap<IRenderCallback, Object>();
        
        public SurfaceCallback(TextureRenderView renderView)
        {
            mWeakRenderView = new WeakReference<TextureRenderView>(renderView);
        }
        
        public void setOwnSurfaceTexture(boolean ownSurfaceTexture)
        {
            mOwnSurfaceTexture = ownSurfaceTexture;
        }
        
        public void addRenderCallback(IRenderCallback callback)
        {
            mRenderCallbackMap.put(callback, callback);
            
            ISurfaceHolder surfaceHolder = null;
            if (mSurfaceTexture != null)
            {
                if (surfaceHolder == null)
                    surfaceHolder = new InternalSurfaceHolder(mWeakRenderView.get(), mSurfaceTexture, this);
                callback.onSurfaceCreated(surfaceHolder, mWidth, mHeight);
            }
            
            if (mIsFormatChanged)
            {
                if (surfaceHolder == null)
                    surfaceHolder = new InternalSurfaceHolder(mWeakRenderView.get(), mSurfaceTexture, this);
                callback.onSurfaceChanged(surfaceHolder, 0, mWidth, mHeight);
            }
        }
        
        public void removeRenderCallback(IRenderCallback callback)
        {
            mRenderCallbackMap.remove(callback);
        }
        
        @Override
        public void onSurfaceTextureAvailable(SurfaceTexture surface, int width, int height)
        {
            mSurfaceTexture = surface;
            mIsFormatChanged = false;
            mWidth = 0;
            mHeight = 0;
            
            ISurfaceHolder surfaceHolder = new InternalSurfaceHolder(mWeakRenderView.get(), surface, this);
            for (IRenderCallback renderCallback : mRenderCallbackMap.keySet())
            {
                renderCallback.onSurfaceCreated(surfaceHolder, 0, 0);
            }
        }
        
        @Override
        public void onSurfaceTextureSizeChanged(SurfaceTexture surface, int width, int height)
        {
            mSurfaceTexture = surface;
            mIsFormatChanged = true;
            mWidth = width;
            mHeight = height;
            
            ISurfaceHolder surfaceHolder = new InternalSurfaceHolder(mWeakRenderView.get(), surface, this);
            for (IRenderCallback renderCallback : mRenderCallbackMap.keySet())
            {
                renderCallback.onSurfaceChanged(surfaceHolder, 0, width, height);
            }
        }
        
        @Override
        public boolean onSurfaceTextureDestroyed(SurfaceTexture surface)
        {
            mSurfaceTexture = surface;
            mIsFormatChanged = false;
            mWidth = 0;
            mHeight = 0;
            
            ISurfaceHolder surfaceHolder = new InternalSurfaceHolder(mWeakRenderView.get(), surface, this);
            for (IRenderCallback renderCallback : mRenderCallbackMap.keySet())
            {
                renderCallback.onSurfaceDestroyed(surfaceHolder);
            }
            
            Log.d(TAG, "onSurfaceTextureDestroyed: destroy: " + mOwnSurfaceTexture);
            return mOwnSurfaceTexture;
        }
        
        @Override
        public void onSurfaceTextureUpdated(SurfaceTexture surface)
        {
        }
        
        // -------------------------
        // ISurfaceTextureHost
        // -------------------------
        
        @Override
        public void releaseSurfaceTexture(SurfaceTexture surfaceTexture)
        {
            if (surfaceTexture == null)
            {
                Log.d(TAG, "releaseSurfaceTexture: null");
            }
            else if (mDidDetachFromWindow)
            {
                if (surfaceTexture != mSurfaceTexture)
                {
                    Log.d(TAG, "releaseSurfaceTexture: didDetachFromWindow(): release different SurfaceTexture");
                    surfaceTexture.release();
                }
                else if (!mOwnSurfaceTexture)
                {
                    Log.d(TAG, "releaseSurfaceTexture: didDetachFromWindow(): release detached SurfaceTexture");
                    surfaceTexture.release();
                }
                else
                {
                    Log.d(TAG, "releaseSurfaceTexture: didDetachFromWindow(): already released by TextureView");
                }
            }
            else if (mWillDetachFromWindow)
            {
                if (surfaceTexture != mSurfaceTexture)
                {
                    Log.d(TAG, "releaseSurfaceTexture: willDetachFromWindow(): release different SurfaceTexture");
                    surfaceTexture.release();
                }
                else if (!mOwnSurfaceTexture)
                {
                    Log.d(TAG,
                        "releaseSurfaceTexture: willDetachFromWindow(): re-attach SurfaceTexture to TextureView");
                    setOwnSurfaceTexture(true);
                }
                else
                {
                    Log.d(TAG, "releaseSurfaceTexture: willDetachFromWindow(): will released by TextureView");
                }
            }
            else
            {
                if (surfaceTexture != mSurfaceTexture)
                {
                    Log.d(TAG, "releaseSurfaceTexture: alive: release different SurfaceTexture");
                    surfaceTexture.release();
                }
                else if (!mOwnSurfaceTexture)
                {
                    Log.d(TAG, "releaseSurfaceTexture: alive: re-attach SurfaceTexture to TextureView");
                    setOwnSurfaceTexture(true);
                }
                else
                {
                    Log.d(TAG, "releaseSurfaceTexture: alive: will released by TextureView");
                }
            }
        }
        
        public void willDetachFromWindow()
        {
            Log.d(TAG, "willDetachFromWindow()");
            mWillDetachFromWindow = true;
        }
        
        public void didDetachFromWindow()
        {
            Log.d(TAG, "didDetachFromWindow()");
            mDidDetachFromWindow = true;
        }
    }
    
    // --------------------
    // Accessibility
    // --------------------
    
    @Override
    public void onInitializeAccessibilityEvent(AccessibilityEvent event)
    {
        super.onInitializeAccessibilityEvent(event);
        event.setClassName(TextureRenderView.class.getName());
    }
    
    @Override
    public void onInitializeAccessibilityNodeInfo(AccessibilityNodeInfo info)
    {
        super.onInitializeAccessibilityNodeInfo(info);
        info.setClassName(TextureRenderView.class.getName());
    }
}