Error HWUISink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
android::uirenderer::TestWindowContext renderer;
renderer.initialize(src.size().width(), src.size().height());
SkCanvas* canvas = renderer.prepareToDraw();
Error err = src.draw(canvas);
if (!err.isEmpty()) {
return err;
}
renderer.finishDrawing();
renderer.fence();
renderer.capturePixels(dst);
return "";
}
void convert_from(const Src& src)
{
*this = vtable_storage(
src.lookup(
(typename ::boost::type_erasure::detail::rebind_placeholders<
T, Bindings
>::type*)0)...);
}
/** @fn csvToTab
* @brief convert `csv` file into associated table (map) [ (ra,dec) ] = (mag,id)
* @details `csv` file is formated `id,ra,dec,mag`
* @param `std::string fileName` name of csv file
* @param `std::set<long> save` list to save sources
* @return `std::map< Coord<double,double> , Coord<double,long> > tab` (Ra,Dec) => (mag,id) **/
std::map< Coord<double,double> , Coord<double,long> > csvToTab( std::string fileName , std::set<Src> &save )
{
// map returned : [ (ra,dec) ] = (mag,id)
std::map< Coord<double,double> , Coord<double,long> > tab;
std::string s_id, s_ra, s_dec, s_mag; // variables to recover `csv` data
double ra,dec,mag; long id; // data after conversion
// coordinates for map (Ra,Dec) => (mag,id)
Coord<double,double> radec;
Coord<double,long> magid;
Src tmp;
// input file stream
std::ifstream file( fileName.c_str() );
// read first line but doing nothing (for header line)
getline(file, s_id);
// read all data with `getline`, first id, then ra, dec and magnitude
// data are converted directly into id, ra, dec and mag variables
while ( getline(file, s_id, ',') ) {
id = atol(s_id.c_str());
getline(file, s_ra, ','); ra = atof( s_ra.c_str () );
getline(file, s_dec, ','); dec = atof( s_dec.c_str() );
getline(file, s_mag); mag = atof( s_mag.c_str() );
// complete coordinates
radec.xy(ra,dec);
magid.xy(mag,id);
// insert in tab
tab.insert(std::pair<Coord<double,double>, Coord<double,long> >(radec, magid));
// save curent source into a set for test errors
tmp.build(radec,magid);
save.insert(tmp);
}
return tab;
}
Error HWUISink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
// Do all setup in this function because we don't know the size
// for the RenderNode and RenderProxy during the constructor.
// In practice this doesn't seem too expensive.
const SkISize size = src.size();
// Based on android::SurfaceTexture_init()
android::sp<android::IGraphicBufferProducer> producer;
android::sp<android::IGraphicBufferConsumer> consumer;
android::BufferQueue::createBufferQueue(&producer, &consumer);
// Consumer setup
android::sp<android::CpuConsumer> cpuConsumer =
new android::CpuConsumer(consumer, 1);
cpuConsumer->setName(android::String8("SkiaTestClient"));
cpuConsumer->setDefaultBufferSize(size.width(), size.height());
// Producer setup
android::sp<android::Surface> surface = new android::Surface(producer);
native_window_set_buffers_dimensions(surface.get(), size.width(), size.height());
native_window_set_buffers_format(surface.get(), android::PIXEL_FORMAT_RGBA_8888);
native_window_set_usage(surface.get(), GRALLOC_USAGE_SW_READ_OFTEN |
GRALLOC_USAGE_SW_WRITE_NEVER |
GRALLOC_USAGE_HW_RENDER);
// RenderNode setup based on hwui/tests/main.cpp:TreeContentAnimation
SkAutoTDelete<android::uirenderer::RenderNode> rootNode
(new android::uirenderer::RenderNode());
rootNode->incStrong(nullptr);
// Values set here won't be applied until the framework has called
// RenderNode::pushStagingPropertiesChanges() during RenderProxy::syncAndDrawFrame().
rootNode->mutateStagingProperties().setLeftTopRightBottom(0, 0, size.width(), size.height());
rootNode->setPropertyFieldsDirty(android::uirenderer::RenderNode::X |
android::uirenderer::RenderNode::Y);
rootNode->mutateStagingProperties().setClipToBounds(false);
rootNode->setPropertyFieldsDirty(android::uirenderer::RenderNode::GENERIC);
// RenderProxy setup based on hwui/tests/main.cpp:TreeContentAnimation
ContextFactory factory;
SkAutoTDelete<android::uirenderer::renderthread::RenderProxy> proxy
(new android::uirenderer::renderthread::RenderProxy(false, rootNode, &factory));
proxy->loadSystemProperties();
proxy->initialize(surface.get());
float lightX = size.width() / 2.0f;
android::uirenderer::Vector3 lightVector { lightX, dp(-200.0f), dp(800.0f) };
proxy->setup(size.width(), size.height(), lightVector, dp(800.0f), 255 * 0.075f, 255 * 0.15f,
kDensity);
// Do the draw
SkAutoTDelete<android::uirenderer::DisplayListRenderer> renderer
(new android::uirenderer::DisplayListRenderer());
renderer->setViewport(size.width(), size.height());
renderer->prepare();
renderer->clipRect(0, 0, size.width(), size.height(), SkRegion::Op::kReplace_Op);
Error err = src.draw(renderer->asSkCanvas());
if (!err.isEmpty()) {
return err;
}
renderer->finish();
rootNode->setStagingDisplayList(renderer->finishRecording());
proxy->syncAndDrawFrame();
proxy->fence();
// Capture pixels
SkImageInfo destinationConfig =
SkImageInfo::Make(size.width(), size.height(),
kRGBA_8888_SkColorType, kPremul_SkAlphaType);
dst->allocPixels(destinationConfig);
sk_memset32((uint32_t*) dst->getPixels(), SK_ColorRED, size.width() * size.height());
android::CpuConsumer::LockedBuffer nativeBuffer;
android::status_t retval = cpuConsumer->lockNextBuffer(&nativeBuffer);
if (retval == android::BAD_VALUE) {
SkDebugf("HWUISink::draw() got no buffer; returning transparent");
// No buffer ready to read - commonly triggered by dm sending us
// a no-op source, or calling code that doesn't do anything on this
// backend.
dst->eraseColor(SK_ColorTRANSPARENT);
return "";
} else if (retval) {
return SkStringPrintf("Failed to lock buffer to read pixels: %d.", retval);
}
// Move the pixels into the destination SkBitmap
SK_ALWAYSBREAK(nativeBuffer.format == android::PIXEL_FORMAT_RGBA_8888 &&
"Native buffer not RGBA!");
SkImageInfo nativeConfig =
SkImageInfo::Make(nativeBuffer.width, nativeBuffer.height,
kRGBA_8888_SkColorType, kPremul_SkAlphaType);
// Android stride is in pixels, Skia stride is in bytes
SkBitmap nativeWrapper;
bool success =
nativeWrapper.installPixels(nativeConfig, nativeBuffer.data, nativeBuffer.stride * 4);
if (!success) {
return "Failed to wrap HWUI buffer in a SkBitmap";
}
SK_ALWAYSBREAK(dst->colorType() == kRGBA_8888_SkColorType &&
"Destination buffer not RGBA!");
success =
nativeWrapper.readPixels(destinationConfig, dst->getPixels(), dst->rowBytes(), 0, 0);
if (!success) {
return "Failed to extract pixels from HWUI buffer";
}
cpuConsumer->unlockBuffer(nativeBuffer);
return "";
}
void copy(Src& s, Dst& d)
{
copy(s.begin(), s.end(), d.begin());
}
static Target call(Src const& str)
{
return Target(str.c_str());
}
void copy(Src const &src, Dest &des) {
des.assign(src.begin(), src.end());
}
