Java tutorial
/* * Copyright (C) 2014 The Android Open Source Project * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.sien.cpshoot.screencapture; import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.nio.ByteBuffer; import java.text.SimpleDateFormat; import android.app.Activity; import android.content.Context; import android.content.Intent; import android.graphics.Bitmap; import android.graphics.ImageFormat; import android.hardware.display.DisplayManager; import android.hardware.display.VirtualDisplay; import android.media.Image; import android.media.ImageReader; import android.media.projection.MediaProjection; import android.media.projection.MediaProjectionManager; import android.net.Uri; import android.os.Bundle; import android.os.Environment; import android.os.Handler; import android.support.annotation.Nullable; import android.support.v4.app.Fragment; import android.util.DisplayMetrics; import android.view.LayoutInflater; import android.view.Surface; import android.view.SurfaceView; import android.view.View; import android.view.ViewGroup; import android.view.WindowManager; import android.widget.Button; import android.widget.Toast; import com.sien.cpshoot.R; import com.sien.cpshoot.common.logger.Log; import com.sien.cpshoot.croper.Crop; /** * Provides UI for the screen capture. */ public class ScreenCaptureFragment extends Fragment implements View.OnClickListener { private static final String TAG = "ScreenCaptureFragment"; private static final String STATE_RESULT_CODE = "result_code"; private static final String STATE_RESULT_DATA = "result_data"; private static final int REQUEST_MEDIA_PROJECTION = 1; private int mScreenDensity; private int mResultCode; private Intent mResultData; private Surface mSurface; private MediaProjection mMediaProjection; private VirtualDisplay mVirtualDisplay; private MediaProjectionManager mMediaProjectionManager; private Button mButtonToggle; private SurfaceView mSurfaceView; @Override public void onCreate(Bundle savedInstanceState) { super.onCreate(savedInstanceState); if (savedInstanceState != null) { mResultCode = savedInstanceState.getInt(STATE_RESULT_CODE); mResultData = savedInstanceState.getParcelable(STATE_RESULT_DATA); } } @Nullable @Override public View onCreateView(LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) { return inflater.inflate(R.layout.fragment_screen_capture, container, false); } @Override public void onViewCreated(View view, Bundle savedInstanceState) { mSurfaceView = (SurfaceView) view.findViewById(R.id.surface); mSurface = mSurfaceView.getHolder().getSurface(); mButtonToggle = (Button) view.findViewById(R.id.toggle); mButtonToggle.setOnClickListener(this); } @Override public void onActivityCreated(Bundle savedInstanceState) { super.onActivityCreated(savedInstanceState); Activity activity = getActivity(); DisplayMetrics metrics = new DisplayMetrics(); activity.getWindowManager().getDefaultDisplay().getMetrics(metrics); mScreenDensity = metrics.densityDpi; mMediaProjectionManager = (MediaProjectionManager) activity .getSystemService(Context.MEDIA_PROJECTION_SERVICE); } @Override public void onSaveInstanceState(Bundle outState) { super.onSaveInstanceState(outState); if (mResultData != null) { outState.putInt(STATE_RESULT_CODE, mResultCode); outState.putParcelable(STATE_RESULT_DATA, mResultData); } } @Override public void onClick(View v) { switch (v.getId()) { case R.id.toggle: if (mVirtualDisplay == null) { startScreenCapture(); } else { stopScreenCapture(); } break; } } @Override public void onActivityResult(int requestCode, int resultCode, Intent data) { if (requestCode == REQUEST_MEDIA_PROJECTION) { if (resultCode != Activity.RESULT_OK) { Log.i(TAG, "User cancelled"); Toast.makeText(getActivity(), R.string.user_cancelled, Toast.LENGTH_SHORT).show(); return; } Activity activity = getActivity(); if (activity == null) { return; } Log.i(TAG, "Starting screen capture"); mResultCode = resultCode; mResultData = data; setUpMediaProjection(); setUpVirtualDisplay(); } else if (requestCode == Crop.REQUEST_CROP) { handleCrop(resultCode, data); } } private void handleCrop(int resultCode, Intent result) { if (resultCode == Activity.RESULT_OK) { Toast.makeText(getActivity(), " handleCrop: Crop.getOutput(result) " + Crop.getOutput(result), Toast.LENGTH_SHORT).show(); // mImageView.setImageURI( Crop.getOutput(result)); } } @Override public void onPause() { super.onPause(); stopScreenCapture(); } @Override public void onDestroy() { super.onDestroy(); tearDownMediaProjection(); } private void setUpMediaProjection() { mMediaProjection = mMediaProjectionManager.getMediaProjection(mResultCode, mResultData); } private void tearDownMediaProjection() { if (mMediaProjection != null) { mMediaProjection.stop(); mMediaProjection = null; } } private void startScreenCapture() { Activity activity = getActivity(); if (mSurface == null || activity == null) { return; } if (mMediaProjection != null) { setUpVirtualDisplay(); } else if (mResultCode != 0 && mResultData != null) { setUpMediaProjection(); setUpVirtualDisplay(); } else { Log.i(TAG, "Requesting confirmation"); // This initiates a prompt dialog for the user to confirm screen projection. startActivityForResult(mMediaProjectionManager.createScreenCaptureIntent(), REQUEST_MEDIA_PROJECTION); } } private void setUpVirtualDisplay() { Log.i(TAG, "Setting up a VirtualDisplay: " + mSurfaceView.getWidth() + "x" + mSurfaceView.getHeight() + " (" + mScreenDensity + ")"); //---------------cp.add final int w = mSurfaceView.getWidth(); final int h = mSurfaceView.getHeight(); final ImageReader mImageReader = ImageReader.newInstance(w, h, ImageFormat.RGB_565, 2); DisplayMetrics metrics = new DisplayMetrics(); WindowManager mWindowManager = getActivity().getWindowManager(); mWindowManager.getDefaultDisplay().getMetrics(metrics); int mScreenDensity = metrics.densityDpi; //----end mVirtualDisplay = mMediaProjection.createVirtualDisplay("ScreenCapture", mSurfaceView.getWidth(), mSurfaceView.getHeight(), mScreenDensity, DisplayManager.VIRTUAL_DISPLAY_FLAG_AUTO_MIRROR, mImageReader.getSurface(), null, null); //---------------cp.add new Handler().postDelayed(new Runnable() { @Override public void run() { saveImage(mImageReader, w, h); } }, 100); //-----end mButtonToggle.setText(R.string.stop); } private void saveImage(ImageReader mImageReader, int mWidth, int mHeight) { if (mImageReader == null) return; Image image = mImageReader.acquireLatestImage(); if (image == null) return; final Image.Plane[] planes = image.getPlanes(); if (planes.length <= 0) return; final ByteBuffer buffer = planes[0].getBuffer(); int offset = 0; int pixelStride = planes[0].getPixelStride(); int rowStride = planes[0].getRowStride(); int rowPadding = rowStride - pixelStride * mWidth; Bitmap bitmap = Bitmap.createBitmap(mWidth + rowPadding / pixelStride, mHeight, Bitmap.Config.RGB_565); bitmap.copyPixelsFromBuffer(buffer); image.close(); SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy_MM_dd_hh_mm_ss"); String strDate = dateFormat.format(new java.util.Date()); String pathImage = Environment.getExternalStorageDirectory().getPath() + "/Pictures/"; String nameImage = pathImage + strDate + ".png"; if (bitmap != null) { try { File fileImage = new File(nameImage); if (!fileImage.exists()) { fileImage.createNewFile(); } FileOutputStream out = new FileOutputStream(fileImage); if (out != null) { bitmap.compress(Bitmap.CompressFormat.PNG, 100, out); out.flush(); out.close(); Toast.makeText(getActivity(), "get phone's screen succeed", Toast.LENGTH_SHORT).show(); Intent media = new Intent(Intent.ACTION_MEDIA_SCANNER_SCAN_FILE); Uri contentUri = Uri.fromFile(fileImage); media.setData(contentUri); getActivity().sendBroadcast(media); // beginCrop(Uri.fromFile(new File(nameImage))); } } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } } } private void beginCrop(Uri source) { String fileName = "Temp_" + String.valueOf(System.currentTimeMillis()); File mTempDir = new File(Environment.getExternalStorageDirectory(), "Temp"); File cropFile = new File(mTempDir, fileName); Uri outputUri = Uri.fromFile(cropFile); new Crop(source).output(outputUri).setCropType(false).start(getActivity()); } private void stopScreenCapture() { if (mVirtualDisplay == null) { return; } mVirtualDisplay.release(); mVirtualDisplay = null; mButtonToggle.setText(R.string.start); } }