shadows/framework/src/main/java/org/robolectric/shadows/ShadowPicture.java

package org.robolectric.shadows;

import static android.os.Build.VERSION_CODES.KITKAT_WATCH;
import static android.os.Build.VERSION_CODES.LOLLIPOP;

import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.Picture;
import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;

@Implements(Picture.class)
public class ShadowPicture {

  private int width;
  private int height;

  @Implementation
  protected void __constructor__() {}

  @Implementation(minSdk = LOLLIPOP)
  protected void __constructor__(long nativePicture) {}

  @Implementation(maxSdk = KITKAT_WATCH)
  protected void __constructor__(int nativePicture, boolean fromStream) {}

  @Implementation
  protected void __constructor__(Picture src) {
    width = src.getWidth();
    height = src.getHeight();
  }

  @Implementation
  protected int getWidth() {
    return width;
  }

  @Implementation
  protected int getHeight() {
    return height;
  }

  @Implementation
  protected Canvas beginRecording(int width, int height) {
    this.width = width;
    this.height = height;
    return new Canvas(Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888));
  }
}