user/src/com/google/gwt/resources/rg/ImageResourceGenerator.java - gwt - Git at Google

 /*
  * Copyright 2008 Google Inc.
  *
  * 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.google.gwt.resources.rg;

 import com.google.gwt.core.ext.TreeLogger;
 import com.google.gwt.core.ext.UnableToCompleteException;
 import com.google.gwt.core.ext.typeinfo.JClassType;
 import com.google.gwt.core.ext.typeinfo.JMethod;
 import com.google.gwt.core.ext.typeinfo.TypeOracle;
 import com.google.gwt.resources.client.ImageResource.ImageOptions;
 import com.google.gwt.resources.client.ImageResource.RepeatStyle;
 import com.google.gwt.resources.client.impl.ImageResourcePrototype;
 import com.google.gwt.resources.ext.AbstractResourceGenerator;
 import com.google.gwt.resources.ext.ClientBundleFields;
 import com.google.gwt.resources.ext.ClientBundleRequirements;
 import com.google.gwt.resources.ext.ResourceContext;
 import com.google.gwt.resources.ext.ResourceGeneratorUtil;
 import com.google.gwt.resources.rg.ImageBundleBuilder.Arranger;
 import com.google.gwt.resources.rg.ImageBundleBuilder.ImageRect;
 import com.google.gwt.user.rebind.SourceWriter;
 import com.google.gwt.user.rebind.StringSourceWriter;

 import java.awt.geom.AffineTransform;
 import java.net.URL;
 import java.util.EnumMap;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.IdentityHashMap;
 import java.util.Map;
 import java.util.Set;

 /**
  * Builds an image strip for all ImageResources defined within an ClientBundle.
  */
 public final class ImageResourceGenerator extends AbstractResourceGenerator {
   private Map<String, ImageRect> imageRectsByName;
   private Map<ImageRect, ImageBundleBuilder> buildersByImageRect;
   private Map<RepeatStyle, ImageBundleBuilder> buildersByRepeatStyle;
   private Map<ImageBundleBuilder, String[]> urlsByBuilder;
   private Map<ImageRect, String[]> urlsByExternalImageRect;
   private Map<ImageRect, ImageBundleBuilder> rtlImages;

   @Override
   public String createAssignment(TreeLogger logger, ResourceContext context,
       JMethod method) throws UnableToCompleteException {
     String name = method.getName();

     SourceWriter sw = new StringSourceWriter();
     sw.println("new " + ImageResourcePrototype.class.getName() + "(");
     sw.indent();
     sw.println('"' + name + "\",");

     ImageRect rect = imageRectsByName.get(name);
     assert rect != null : "No ImageRect ever computed for " + name;

     String[] urlExpressions;
     {
       ImageBundleBuilder builder = buildersByImageRect.get(rect);
       if (builder == null) {
         urlExpressions = urlsByExternalImageRect.get(rect);
       } else {
         urlExpressions = urlsByBuilder.get(builder);
       }
     }
     assert urlExpressions != null : "No URL expression for " + name;
     assert urlExpressions.length == 2;

     if (urlExpressions[1] == null) {
       sw.println(urlExpressions[0] + ",");
     } else {
       sw.println("com.google.gwt.i18n.client.LocaleInfo.getCurrentLocale().isRTL() ?"
           + urlExpressions[1] + " : " + urlExpressions[0] + ",");
     }
     sw.println(rect.getLeft() + ", " + rect.getTop() + ", " + rect.getWidth()
         + ", " + rect.getHeight() + ", " + rect.isAnimated());

     sw.outdent();
     sw.print(")");

     return sw.toString();
   }

   @Override
   public void createFields(TreeLogger logger, ResourceContext context,
       ClientBundleFields fields) throws UnableToCompleteException {

     TypeOracle typeOracle = context.getGeneratorContext().getTypeOracle();
     JClassType stringType = typeOracle.findType(String.class.getName());
     assert stringType != null;

     Map<ImageBundleBuilder, String> prettyNames = new IdentityHashMap<ImageBundleBuilder, String>();

     for (Map.Entry<RepeatStyle, ImageBundleBuilder> entry : buildersByRepeatStyle.entrySet()) {
       RepeatStyle repeatStyle = entry.getKey();
       ImageBundleBuilder builder = entry.getValue();
       Arranger arranger;

       switch (repeatStyle) {
         case None:
           arranger = new ImageBundleBuilder.BestFitArranger();
           break;
         case Horizontal:
           arranger = new ImageBundleBuilder.VerticalArranger();
           break;
         case Vertical:
           arranger = new ImageBundleBuilder.HorizontalArranger();
           break;
         case Both:
           // This is taken care of when writing the external images;
           continue;
         default:
           logger.log(TreeLogger.ERROR, "Unknown RepeatStyle" + repeatStyle);
           throw new UnableToCompleteException();
       }

       String bundleUrlExpression = builder.writeBundledImage(logger.branch(
           TreeLogger.DEBUG, "Writing image strip", null), context, arranger);

       if (bundleUrlExpression == null) {
         continue;
       }

       String prettyName = "imageUrl" + repeatStyle;
       prettyNames.put(builder, prettyName);
       String fieldName = fields.define(stringType, prettyName,
           bundleUrlExpression, true, true);
       String[] strings = {fieldName, null};
       urlsByBuilder.put(builder, strings);
     }

     if (rtlImages.size() > 0) {
       Set<ImageBundleBuilder> rtlBuilders = new HashSet<ImageBundleBuilder>();

       for (Map.Entry<ImageRect, ImageBundleBuilder> entry : rtlImages.entrySet()) {
         ImageRect rtlImage = entry.getKey();

         AffineTransform tx = new AffineTransform();
         tx.setTransform(-1, 0, 0, 1, rtlImage.getWidth(), 0);

         rtlImage.setTransform(tx);

         if (buildersByImageRect.containsKey(rtlImage)) {
           rtlBuilders.add(buildersByImageRect.get(rtlImage));
         } else {
           String[] strings = urlsByExternalImageRect.get(rtlImage);
           assert strings != null;
           byte[] imageBytes = ImageBundleBuilder.toPng(logger, rtlImage);
           strings[1] = context.deploy(rtlImage.getName() + "_rtl.png",
               "image/png", imageBytes, false);
         }
       }

       for (ImageBundleBuilder builder : rtlBuilders) {
         String bundleUrlExpression = builder.writeBundledImage(logger.branch(
             TreeLogger.DEBUG, "Writing image strip", null), context,
             new ImageBundleBuilder.IdentityArranger());

         if (bundleUrlExpression == null) {
           continue;
         }

         String prettyName = prettyNames.get(builder);
         String[] strings = urlsByBuilder.get(builder);
         assert strings != null;

         strings[1] = fields.define(stringType, prettyName + "_rtl",
             bundleUrlExpression, true, true);
       }
     }
   }

   @Override
   public void init(TreeLogger logger, ResourceContext context) {
     imageRectsByName = new HashMap<String, ImageRect>();
     buildersByImageRect = new IdentityHashMap<ImageRect, ImageBundleBuilder>();
     buildersByRepeatStyle = new EnumMap<RepeatStyle, ImageBundleBuilder>(
         RepeatStyle.class);
     rtlImages = new IdentityHashMap<ImageRect, ImageBundleBuilder>();
     urlsByBuilder = new IdentityHashMap<ImageBundleBuilder, String[]>();
     urlsByExternalImageRect = new IdentityHashMap<ImageRect, String[]>();
   }

   @Override
   public void prepare(TreeLogger logger, ResourceContext context,
       ClientBundleRequirements requirements, JMethod method)
       throws UnableToCompleteException {
     URL[] resources = ResourceGeneratorUtil.findResources(logger, context,
         method);

     if (resources.length != 1) {
       logger.log(TreeLogger.ERROR, "Exactly one image may be specified", null);
       throw new UnableToCompleteException();
     }

     ImageBundleBuilder builder = getBuilder(method);
     URL resource = resources[0];
     String name = method.getName();

     ImageRect rect;
     try {
       rect = builder.assimilate(logger, name, resource);
       if (context.supportsDataUrls()
           || getRepeatStyle(method) == RepeatStyle.Both) {
         // Just use the calculated meta-data
         builder.removeMapping(name);
         rect.setPosition(0, 0);
         throw new UnsuitableForStripException(rect);
       }
       buildersByImageRect.put(rect, builder);
     } catch (UnsuitableForStripException e) {
       // Add the image to the output as a separate resource
       rect = e.getImageRect();

       String urlExpression;
       if (rect.isAnimated()) {
         // Can't re-encode animated images, so we emit it as-is
         urlExpression = context.deploy(resource, false);
       } else {
         // Re-encode the image as a PNG to strip random header data
         byte[] imageBytes = ImageBundleBuilder.toPng(logger, rect);
         urlExpression = context.deploy(rect.getName() + ".png", "image/png",
             imageBytes, false);
       }
       urlsByExternalImageRect.put(rect, new String[] {urlExpression, null});
     }

     imageRectsByName.put(name, rect);

     if (getFlipRtl(method)) {
       rtlImages.put(rect, null);
     }
   }

   private ImageBundleBuilder getBuilder(JMethod method) {
     RepeatStyle repeatStyle = getRepeatStyle(method);
     ImageBundleBuilder builder = buildersByRepeatStyle.get(repeatStyle);
     if (builder == null) {
       builder = new ImageBundleBuilder();
       buildersByRepeatStyle.put(repeatStyle, builder);
     }
     return builder;
   }

   private boolean getFlipRtl(JMethod method) {
     ImageOptions options = method.getAnnotation(ImageOptions.class);
     if (options == null) {
       return false;
     } else {
       return options.flipRtl();
     }
   }

   private RepeatStyle getRepeatStyle(JMethod method) {
     ImageOptions options = method.getAnnotation(ImageOptions.class);
     if (options == null) {
       return RepeatStyle.None;
     } else {
       return options.repeatStyle();
     }
   }
 }
	/*
	* Copyright 2008 Google Inc.
	*
	* 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.google.gwt.resources.rg;

	import com.google.gwt.core.ext.TreeLogger;
	import com.google.gwt.core.ext.UnableToCompleteException;
	import com.google.gwt.core.ext.typeinfo.JClassType;
	import com.google.gwt.core.ext.typeinfo.JMethod;
	import com.google.gwt.core.ext.typeinfo.TypeOracle;
	import com.google.gwt.resources.client.ImageResource.ImageOptions;
	import com.google.gwt.resources.client.ImageResource.RepeatStyle;
	import com.google.gwt.resources.client.impl.ImageResourcePrototype;
	import com.google.gwt.resources.ext.AbstractResourceGenerator;
	import com.google.gwt.resources.ext.ClientBundleFields;
	import com.google.gwt.resources.ext.ClientBundleRequirements;
	import com.google.gwt.resources.ext.ResourceContext;
	import com.google.gwt.resources.ext.ResourceGeneratorUtil;
	import com.google.gwt.resources.rg.ImageBundleBuilder.Arranger;
	import com.google.gwt.resources.rg.ImageBundleBuilder.ImageRect;
	import com.google.gwt.user.rebind.SourceWriter;
	import com.google.gwt.user.rebind.StringSourceWriter;

	import java.awt.geom.AffineTransform;
	import java.net.URL;
	import java.util.EnumMap;
	import java.util.HashMap;
	import java.util.HashSet;
	import java.util.IdentityHashMap;
	import java.util.Map;
	import java.util.Set;

	/**
	* Builds an image strip for all ImageResources defined within an ClientBundle.
	*/
	public final class ImageResourceGenerator extends AbstractResourceGenerator {
	private Map<String, ImageRect> imageRectsByName;
	private Map<ImageRect, ImageBundleBuilder> buildersByImageRect;
	private Map<RepeatStyle, ImageBundleBuilder> buildersByRepeatStyle;
	private Map<ImageBundleBuilder, String[]> urlsByBuilder;
	private Map<ImageRect, String[]> urlsByExternalImageRect;
	private Map<ImageRect, ImageBundleBuilder> rtlImages;

	@Override
	public String createAssignment(TreeLogger logger, ResourceContext context,
	JMethod method) throws UnableToCompleteException {
	String name = method.getName();

	SourceWriter sw = new StringSourceWriter();
	sw.println("new " + ImageResourcePrototype.class.getName() + "(");
	sw.indent();
	sw.println('"' + name + "\",");

	ImageRect rect = imageRectsByName.get(name);
	assert rect != null : "No ImageRect ever computed for " + name;

	String[] urlExpressions;
	{
	ImageBundleBuilder builder = buildersByImageRect.get(rect);
	if (builder == null) {
	urlExpressions = urlsByExternalImageRect.get(rect);
	} else {
	urlExpressions = urlsByBuilder.get(builder);
	}
	}
	assert urlExpressions != null : "No URL expression for " + name;
	assert urlExpressions.length == 2;

	if (urlExpressions[1] == null) {
	sw.println(urlExpressions[0] + ",");
	} else {
	sw.println("com.google.gwt.i18n.client.LocaleInfo.getCurrentLocale().isRTL() ?"
	+ urlExpressions[1] + " : " + urlExpressions[0] + ",");
	}
	sw.println(rect.getLeft() + ", " + rect.getTop() + ", " + rect.getWidth()
	+ ", " + rect.getHeight() + ", " + rect.isAnimated());

	sw.outdent();
	sw.print(")");

	return sw.toString();
	}

	@Override
	public void createFields(TreeLogger logger, ResourceContext context,
	ClientBundleFields fields) throws UnableToCompleteException {

	TypeOracle typeOracle = context.getGeneratorContext().getTypeOracle();
	JClassType stringType = typeOracle.findType(String.class.getName());
	assert stringType != null;

	Map<ImageBundleBuilder, String> prettyNames = new IdentityHashMap<ImageBundleBuilder, String>();

	for (Map.Entry<RepeatStyle, ImageBundleBuilder> entry : buildersByRepeatStyle.entrySet()) {
	RepeatStyle repeatStyle = entry.getKey();
	ImageBundleBuilder builder = entry.getValue();
	Arranger arranger;

	switch (repeatStyle) {
	case None:
	arranger = new ImageBundleBuilder.BestFitArranger();
	break;
	case Horizontal:
	arranger = new ImageBundleBuilder.VerticalArranger();
	break;
	case Vertical:
	arranger = new ImageBundleBuilder.HorizontalArranger();
	break;
	case Both:
	// This is taken care of when writing the external images;
	continue;
	default:
	logger.log(TreeLogger.ERROR, "Unknown RepeatStyle" + repeatStyle);
	throw new UnableToCompleteException();
	}

	String bundleUrlExpression = builder.writeBundledImage(logger.branch(
	TreeLogger.DEBUG, "Writing image strip", null), context, arranger);

	if (bundleUrlExpression == null) {
	continue;
	}

	String prettyName = "imageUrl" + repeatStyle;
	prettyNames.put(builder, prettyName);
	String fieldName = fields.define(stringType, prettyName,
	bundleUrlExpression, true, true);
	String[] strings = {fieldName, null};
	urlsByBuilder.put(builder, strings);
	}

	if (rtlImages.size() > 0) {
	Set<ImageBundleBuilder> rtlBuilders = new HashSet<ImageBundleBuilder>();

	for (Map.Entry<ImageRect, ImageBundleBuilder> entry : rtlImages.entrySet()) {
	ImageRect rtlImage = entry.getKey();

	AffineTransform tx = new AffineTransform();
	tx.setTransform(-1, 0, 0, 1, rtlImage.getWidth(), 0);

	rtlImage.setTransform(tx);

	if (buildersByImageRect.containsKey(rtlImage)) {
	rtlBuilders.add(buildersByImageRect.get(rtlImage));
	} else {
	String[] strings = urlsByExternalImageRect.get(rtlImage);
	assert strings != null;
	byte[] imageBytes = ImageBundleBuilder.toPng(logger, rtlImage);
	strings[1] = context.deploy(rtlImage.getName() + "_rtl.png",
	"image/png", imageBytes, false);
	}
	}

	for (ImageBundleBuilder builder : rtlBuilders) {
	String bundleUrlExpression = builder.writeBundledImage(logger.branch(
	TreeLogger.DEBUG, "Writing image strip", null), context,
	new ImageBundleBuilder.IdentityArranger());

	if (bundleUrlExpression == null) {
	continue;
	}

	String prettyName = prettyNames.get(builder);
	String[] strings = urlsByBuilder.get(builder);
	assert strings != null;

	strings[1] = fields.define(stringType, prettyName + "_rtl",
	bundleUrlExpression, true, true);
	}
	}
	}

	@Override
	public void init(TreeLogger logger, ResourceContext context) {
	imageRectsByName = new HashMap<String, ImageRect>();
	buildersByImageRect = new IdentityHashMap<ImageRect, ImageBundleBuilder>();
	buildersByRepeatStyle = new EnumMap<RepeatStyle, ImageBundleBuilder>(
	RepeatStyle.class);
	rtlImages = new IdentityHashMap<ImageRect, ImageBundleBuilder>();
	urlsByBuilder = new IdentityHashMap<ImageBundleBuilder, String[]>();
	urlsByExternalImageRect = new IdentityHashMap<ImageRect, String[]>();
	}

	@Override
	public void prepare(TreeLogger logger, ResourceContext context,
	ClientBundleRequirements requirements, JMethod method)
	throws UnableToCompleteException {
	URL[] resources = ResourceGeneratorUtil.findResources(logger, context,
	method);

	if (resources.length != 1) {
	logger.log(TreeLogger.ERROR, "Exactly one image may be specified", null);
	throw new UnableToCompleteException();
	}

	ImageBundleBuilder builder = getBuilder(method);
	URL resource = resources[0];
	String name = method.getName();

	ImageRect rect;
	try {
	rect = builder.assimilate(logger, name, resource);
	if (context.supportsDataUrls()
	\|\| getRepeatStyle(method) == RepeatStyle.Both) {
	// Just use the calculated meta-data
	builder.removeMapping(name);
	rect.setPosition(0, 0);
	throw new UnsuitableForStripException(rect);
	}
	buildersByImageRect.put(rect, builder);
	} catch (UnsuitableForStripException e) {
	// Add the image to the output as a separate resource
	rect = e.getImageRect();

	String urlExpression;
	if (rect.isAnimated()) {
	// Can't re-encode animated images, so we emit it as-is
	urlExpression = context.deploy(resource, false);
	} else {
	// Re-encode the image as a PNG to strip random header data
	byte[] imageBytes = ImageBundleBuilder.toPng(logger, rect);
	urlExpression = context.deploy(rect.getName() + ".png", "image/png",
	imageBytes, false);
	}
	urlsByExternalImageRect.put(rect, new String[] {urlExpression, null});
	}

	imageRectsByName.put(name, rect);

	if (getFlipRtl(method)) {
	rtlImages.put(rect, null);
	}
	}

	private ImageBundleBuilder getBuilder(JMethod method) {
	RepeatStyle repeatStyle = getRepeatStyle(method);
	ImageBundleBuilder builder = buildersByRepeatStyle.get(repeatStyle);
	if (builder == null) {
	builder = new ImageBundleBuilder();
	buildersByRepeatStyle.put(repeatStyle, builder);
	}
	return builder;
	}

	private boolean getFlipRtl(JMethod method) {
	ImageOptions options = method.getAnnotation(ImageOptions.class);
	if (options == null) {
	return false;
	} else {
	return options.flipRtl();
	}
	}

	private RepeatStyle getRepeatStyle(JMethod method) {
	ImageOptions options = method.getAnnotation(ImageOptions.class);
	if (options == null) {
	return RepeatStyle.None;
	} else {
	return options.repeatStyle();
	}
	}
	}