socket通信での画像転送(android)

Question

Wi-Fi Directを使いP2P通信で画像を転送するandroidアプリを作ろうとしています(android 7.0)。
WiFiDirectServiceDiscoveryを参考に接続の構築は実装できたのですが、接続完了後にソケットを通じて画像を送ると受信側には真っ黒の画像が表示されてしまいます。
現在はbitmapをbyte配列に変換した上でstreamに流し込み、受信側で再度bitmapに変換していますがreadやwriteの手順がおかしいのか正常な画像が転送できません。
以下に実装したコードを示します。問題点のご指摘および実装案の提示にご助力お願いいたします。

public class WiFiChatFragment extends Fragment {
--(略)--
    @Override
    public View onCreateView(LayoutInflater inflater, ViewGroup container,Bundle savedInstanceState) {
       --(略)--
       //ボタンのタッチでカメラから画像を取得・送信
        view.findViewById(R.id.button1).setOnClickListener(
                new View.OnClickListener() {

                    @Override
                    public void onClick(View arg0) {
                        Bitmap bmp = null;
                        try {
                            mCaptureSession.stopRepeating(); // プレビューの更新を止める
                            if(mTextureView.isAvailable()) {
                               bmp= mTextureView.getBitmap();//bitmapを取得
                                if(bmp != null && imgManager != null){
                                    ByteArrayOutputStream baos = new ByteArrayOutputStream();
                                    bmp.compress(Bitmap.CompressFormat.JPEG, 100, baos);//bitmap -> byte(jpeg)
                                    byte[] jpgarr = baos.toByteArray();
                                    StrictMode.setThreadPolicy(new StrictMode.ThreadPolicy.Builder().permitAll().build());
                                    chatManager.write(jpgarr);//Streamへ流す

                                }
                            }
                            mTextureView.findViewById(R.id.myTextureView).setVisibility(View.GONE);
                        } catch (CameraAccessException e) {
                            e.printStackTrace();
                        }
                    }
                });
        return view;
    }
}

public class ChatManager implements Runnable{
    private Socket socket = null;
    private Handler handler;
    public ChatManager(Socket socket, Handler handler) {
        this.socket = socket;
        this.handler = handler;
    }
    private InputStream iStream;
    private OutputStream oStream;
    private static final String TAG = "ChatHandler";
    @Override
    public void run() {
        try {
            iStream = socket.getInputStream();
            oStream = socket.getOutputStream();
            byte[] buffer = new byte[1024];
            int bytes;
            while (true) {
                try {
                    // Read from the InputStream
                    bytes = iStream.read(buffer);
                    if (bytes == -1) {
                        break;
                    }
                    // Send the obtained bytes to the UI Activity
                    handler.obtainMessage(WiFiServiceDiscoveryActivity.MESSAGE_READ,bytes, -1, buffer).sendToTarget();
                } catch (IOException e) {
                    Log.e(TAG, "disconnected", e);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public void write(byte[] buffer) {
        try {
            oStream.write(buffer);
        } catch (IOException e) {
            Log.e(TAG, "Exception during write", e);
        }
    }
}

//取得した画像を表示
    @Override
    public boolean handleMessage(Message msg) {
        switch (msg.what) {
            case MESSAGE_READ:
                byte[] readBuf = (byte[]) msg.obj;
                Bitmap readImg = null;
                readImg = BitmapFactory.decodeByteArray(readBuf, 0, readBuf.length);//byte -> bitmap
                
                 (sendFragment).pushMessage(readImg);//List<Bitmap>へデータをプッシュ

                break;

            case MY_HANDLE:
              --(略)--
        }
        return true;
    }

試したこととして、sendFragment内でbitmapをbyte配列に変換後にstreamに流す代わりにその場でbyteからbitmapに戻し描画させたところ、正常な画像が表示されました。このことから通信以外の理由で画像データが壊れた可能性は低いと考えています。

                      try {
                            mCaptureSession.stopRepeating(); // プレビューの更新を止める
                            if(mTextureView.isAvailable()) {
                               bmp= mTextureView.getBitmap();//bitmapを取得
                                if(bmp != null && imgManager != null){
                                    ByteArrayOutputStream baos = new ByteArrayOutputStream();
                                    bmp.compress(Bitmap.CompressFormat.JPEG, 100, baos);//bitmap -> byte(jpeg)
                                    byte[] jpgarr = baos.toByteArray();
                                    StrictMode.setThreadPolicy(new StrictMode.ThreadPolicy.Builder().permitAll().build());
                                    //chatManager.write(jpgarr);
                                    //再度byteからBitmapへ(この場合正しい画像が描画された)
                                    Bitmap readImg = null;
                                    readImg = BitmapFactory.decodeByteArray(jpgarr, 0, jpgarr.length);//byte -> bitmap
                                    pushMessage(readImg);//List<Bitmap>へデータをプッシュ
                                }
                            }

Comments

[jimbe]

受信側で, 送信したデータが全て受信されているか確認はされましたでしょうか?

[Mustard]

jimbeさん、質問ありがとうございます。
送信側の変数jpgarrおよび受信側の変数readImgを確認したところ
jpgarr={byte[1158931]@4672}
0=-1
1=-40
2=-1
3=-32
4=0
5=16
--(略)--

readImg={byte[1024]@4710}
0=25
1=-68
2=-23
3=-98
4=65
5=70
--(略)--
となっており、配列の大きさが全く異なっていました。
受信した黒画像も一枚ではなく複数枚受信していることもわかりましたので、buffer単位で分割されていまっていると考えています。
これらを一つの画像として受信する方法などご存知でしょうか。

Answer 1

obtainMessageの位置を画像データを全て受信した後に呼び出すことで解決いたしました。
下記のように変更したところ正しい画像が転送されました。
方針を決定づける質問をして頂いたjimbeさん本当にありがとうございました。

--(略)--
public void run() {

        try {

            iStream = socket.getInputStream();
            oStream = socket.getOutputStream();
            byte[] buffer = new byte[1024];
            int bytes;

            ByteArrayOutputStream bospv = new ByteArrayOutputStream();
            copyFile(iStream,bospv);
            pvbyte=bospv.toByteArray();
            handler.obtainMessage(WiFiServiceDiscoveryActivity.MESSAGE_READ,pvbyte).sendToTarget();

        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
--(略)--

public static boolean copyFile(InputStream inputStream, OutputStream out) {
        byte buf[] = new byte[1024];
        int len;
        try {
            while ((len = inputStream.read(buf)) != -1) {
                out.write(buf, 0, len);
            }
            out.close();
            inputStream.close();
        } catch (IOException e) {
            return false;
        }
        return true;
    }