svn commit: r14506 - branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java

Author: maurelio1234
Date: 2008-04-29 05:57:41-0700
New Revision: 14506

Modified:
   branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java

Log:
implementing fignode and dependencies

Modified: branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java
Url: http://argouml.tigris.org/source/browse/argouml/branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java?view=diff&rev=14506&p1=branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java&p2=branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java&r1=14505&r2=14506
==============================================================================
--- branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java	(original)
+++ branches/gsoc2008_maurelio1234_profiles/profile/plugin/PluginProfileLoaderImpl.java	2008-04-29 05:57:41-0700
@@ -1,14 +1,23 @@
 package org.argouml.profile.plugin;
 
+import java.awt.Image;
+import java.io.BufferedInputStream;
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.HashMap;
+import java.util.List;
+
+import javax.swing.ImageIcon;
 
 import org.argouml.model.Model;
+import org.argouml.profile.FigNodeStrategy;
 import org.argouml.profile.Profile;
 import org.argouml.profile.ProfileException;
+import org.argouml.profile.ProfileFacade;
+import org.argouml.profile.ProfileManager;
 import org.argouml.profile.ProfileModelLoader;
 import org.argouml.profile.ProfileReference;
 import org.argouml.profile.ResourceModelLoader;
@@ -20,123 +29,251 @@
 import org.xml.sax.helpers.XMLReaderFactory;
 
 /**
- * Reads the profile descriptor from a file called "profile.xml" 
- * in the same directory as the reference class
+ * Reads the profile descriptor from a file called "profile.xml" in the same
+ * directory as the reference class
  * 
  * @author maas
  */
-public class PluginProfileLoaderImpl extends DefaultHandler implements PluginProfileLoader  {
+public class PluginProfileLoaderImpl extends DefaultHandler implements
+        PluginProfileLoader {
+
+    PluginProfileImpl plugin = null;
+
+    Collection profileModel = null;
+
+    ProfileFromPlugin profile = new ProfileFromPlugin();
 
-	PluginProfileImpl plugin = null;
-	Collection profileModel = null;
-	
-	Profile profile = new Profile() {
-		@Override
-		public String getDisplayName() {
-			return plugin.getName();
-		}
-
-		@Override
-		public Collection getProfilePackages() throws ProfileException {
-			return profileModel;
-		}
-	};
-	
-	private Class referenceClass;
-	
-	private interface Tags {
-		static final String PROFILE = "profile";
-		static final String MODEL = "model"; 
-	}
-	
-	private interface Atts {
-		static final String PROFILE_AUTHOR = "author";
-		static final String PROFILE_DESCRIPTION = "description";
-		static final String PROFILE_NAME = "name";
-		static final String PROFILE_VERSION = "version";
-		static final String PROFILE_DOWNLOADSITE = "downloadsite"; 
-		static final String MODEL_XMI = "xmi"; 
-	}
-
-	public PluginProfile loadProfile(Class cl) throws ErrorLoadingProfile {
-		try {
-			this.referenceClass = cl;
-			
-			XMLReader xr = XMLReaderFactory.createXMLReader();
-			
-			xr.setContentHandler(this);
-			xr.setErrorHandler(this);
-			
-			xr.parse(new InputSource(new InputStreamReader(cl.getResourceAsStream("profile.xml"))));
-			
-		} catch (SAXException e) {
-			throw new ErrorLoadingProfile();
-		} catch (IOException e) {
-			throw new ErrorLoadingProfile();
-		}		
-		
-		return plugin;
-	}
-
-	public void startDocument() {
-		plugin = new PluginProfileImpl();
-	}
-		
-	public void startElement(String uri, String name, String qName, Attributes atts) {
-			for(int i=0;i<atts.getLength();++i) {
-				if (name.equalsIgnoreCase(Tags.PROFILE)) {
-					if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_AUTHOR)) {
-						plugin.setAuthor(atts.getValue(i));
-					} else if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_DESCRIPTION)) {
-						plugin.setDescription(atts.getValue(i));
-					} else if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_NAME)) {
-						plugin.setName(atts.getValue(i));
-					} else if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_VERSION)) {
-						plugin.setVersion(atts.getValue(i));
-					} else if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_DOWNLOADSITE)) {
-						plugin.setDownloadSite(atts.getValue(i));
-					}
-				} else if (name.equalsIgnoreCase(Tags.MODEL)) {
-					if (atts.getLocalName(i).equalsIgnoreCase(Atts.MODEL_XMI)) {
-						try {
-							loadProfileModel(atts.getValue(i));
-						} catch (ProfileException e) {
-							e.printStackTrace();
-						}
-					}
-				}
-			}
-	}
-	
-	@SuppressWarnings("unchecked")
-	private void loadProfileModel(String profile) throws ProfileException {
-		ProfileModelLoader profileModelLoader = new ResourceModelLoader(referenceClass);
-		try {
-			profileModel = profileModelLoader.loadModel(new ProfileReference(
-					profile, new URL("http://example.com/argouml/userprofiles/"
-							+ profile)));
-		} catch (Exception e) {
-			e.printStackTrace();
-		}
-
-		if (profileModel == null) {
-			profileModel = new ArrayList();
-			profileModel.add(Model.getModelManagementFactory().createModel());
-		}
-	}
-
-	public void endElement(String uri, String name, String qName) {
-		if (name.equalsIgnoreCase(Tags.PROFILE)) {
-			plugin.setProfile(profile);
-		}
-	}
-
-	private static PluginProfileLoaderImpl instance = null;
-	public static PluginProfileLoader getInstance() {
-		if (instance == null) {
-			instance = new PluginProfileLoaderImpl();
-		}
-		return instance;
-	}
+    private class ProfileFromPlugin extends Profile {
+        
+        private FigNodeStrategyFromPlugin figNodeStrat = new FigNodeStrategyFromPlugin();
+        
+        @Override
+        public String getDisplayName() {
+            return plugin.getName();
+        }
+
+        @Override
+        public Collection getProfilePackages() throws ProfileException {
+            return profileModel;
+        }
+
+        public void addDependency(Profile p) {
+            super.addProfileDependency(p);
+        }
+        
+        public FigNodeStrategy getFigureStrategy() {
+            return figNodeStrat;
+        }
+        
+    };
+
+    private class FigNodeStrategyFromPlugin implements FigNodeStrategy {
+
+        private HashMap<String, Image> images = new HashMap<String, Image>();
+
+        public Image getIconForStereotype(Object stereotype) {
+            return images.get(Model.getFacade().getName(stereotype));
+        }
+
+        public void addDesrciptor(FigNodeDescriptor fnd) {
+            images.put(fnd.stereotype, fnd.img);
+        }
+    }
+
+    private class FigNodeDescriptor {
+        String stereotype;
+
+        Image img;
+
+        String src;
+
+        int length;
+
+        public boolean isValid() {
+            return stereotype != null && src != null && length > 0;
+        }
+    }
+
+    private Class referenceClass;
+
+    private interface Tags {
+        static final String PROFILE = "profile";
+
+        static final String MODEL = "model";
+
+        static final String DEPENDENCY = "dependency";
+
+        static final String FIGNODE = "fignode";
+    }
+
+    private interface Atts {
+        static final String PROFILE_AUTHOR = "author";
+
+        static final String PROFILE_DESCRIPTION = "description";
+
+        static final String PROFILE_NAME = "name";
+
+        static final String PROFILE_VERSION = "version";
+
+        static final String PROFILE_DOWNLOADSITE = "downloadsite";
+
+        static final String MODEL_XMI = "xmi";
+
+        static final String DEPENDENCY_PROFILE = "profile";
+
+        static final String FIGNODE_STEREOTYPE = "stereotype";
+
+        static final String FIGNODE_IMAGE = "image";
+
+        static final String FIGNODE_LENGTH = "length";
+    }
+
+    public PluginProfile loadProfile(Class cl) throws ErrorLoadingProfile {
+        try {
+            this.referenceClass = cl;
+
+            XMLReader xr = XMLReaderFactory.createXMLReader();
+
+            xr.setContentHandler(this);
+            xr.setErrorHandler(this);
+
+            xr.parse(new InputSource(new InputStreamReader(cl
+                    .getResourceAsStream("profile.xml"))));
+
+        } catch (SAXException e) {
+            throw new ErrorLoadingProfile(e);
+        } catch (IOException e) {
+            throw new ErrorLoadingProfile(e);
+        }
+
+        return plugin;
+    }
+
+    public void startDocument() {
+        plugin = new PluginProfileImpl();
+    }
+
+    public void startElement(String uri, String name, String qName,
+            Attributes atts) {
+
+        FigNodeDescriptor desc = null;
+
+        if (name.equalsIgnoreCase(Tags.FIGNODE)) {
+            desc = new FigNodeDescriptor();
+        }
+
+        for (int i = 0; i < atts.getLength(); ++i) {
+            if (name.equalsIgnoreCase(Tags.PROFILE)) {
+                if (atts.getLocalName(i).equalsIgnoreCase(Atts.PROFILE_AUTHOR)) {
+                    plugin.setAuthor(atts.getValue(i));
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.PROFILE_DESCRIPTION)) {
+                    plugin.setDescription(atts.getValue(i));
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.PROFILE_NAME)) {
+                    plugin.setName(atts.getValue(i));
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.PROFILE_VERSION)) {
+                    plugin.setVersion(atts.getValue(i));
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.PROFILE_DOWNLOADSITE)) {
+                    plugin.setDownloadSite(atts.getValue(i));
+                }
+            } else if (name.equalsIgnoreCase(Tags.MODEL)) {
+                if (atts.getLocalName(i).equalsIgnoreCase(Atts.MODEL_XMI)) {
+                    try {
+                        loadProfileModel(atts.getValue(i));
+                    } catch (ProfileException e) {
+                        e.printStackTrace();
+                    }
+                }
+            } else if (name.equalsIgnoreCase(Tags.DEPENDENCY)) {
+                if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.DEPENDENCY_PROFILE)) {
+                    Profile dep = lookForRegisteredProfile(atts.getValue(i));
+
+                    if (dep != null) {
+                        profile.addDependency(dep);
+                    }
+                }
+            } else if (name.equalsIgnoreCase(Tags.FIGNODE)) {
+                if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.FIGNODE_STEREOTYPE)) {
+                    desc.stereotype = atts.getValue(i);
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.FIGNODE_IMAGE)) {
+                    desc.src = atts.getValue(i);
+                } else if (atts.getLocalName(i).equalsIgnoreCase(
+                        Atts.FIGNODE_LENGTH)) {
+                    desc.length = Integer.parseInt(atts.getValue(i));
+                }
+            }
+        }
+
+        if (desc != null && desc.isValid()) {
+            loadImage(desc);
+            profile.figNodeStrat.addDesrciptor(desc);
+        }
+    }
+
+    private void loadImage(FigNodeDescriptor desc) {
+        BufferedInputStream bis = new BufferedInputStream(this.referenceClass
+                .getResourceAsStream(desc.src));
+
+        byte[] buf = new byte[desc.length];
+        try {
+            bis.read(buf);
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        // LOG.info("IMAGE READ: " + fileName);
+        desc.img = new ImageIcon(buf).getImage();// Toolkit.getDefaultToolkit().createImage(buf);
+
+    }
+
+    private Profile lookForRegisteredProfile(String value) {
+        ProfileManager man = ProfileFacade.getManager();
+        List<Profile> regs = man.getRegisteredProfiles();
+
+        for (Profile profile : regs) {
+            if (profile.getDisplayName().equalsIgnoreCase(value)) {
+                return profile;
+            }
+        }
+        return null;
+    }
+
+    @SuppressWarnings("unchecked")
+    private void loadProfileModel(String profile) throws ProfileException {
+        ProfileModelLoader profileModelLoader = new ResourceModelLoader(
+                referenceClass);
+        try {
+            profileModel = profileModelLoader.loadModel(new ProfileReference(
+                    profile, new URL("http://example.com/argouml/userprofiles/"
+                            + profile)));
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
+        if (profileModel == null) {
+            profileModel = new ArrayList();
+            profileModel.add(Model.getModelManagementFactory().createModel());
+        }
+    }
+
+    public void endElement(String uri, String name, String qName) {
+        if (name.equalsIgnoreCase(Tags.PROFILE)) {
+            plugin.setProfile(profile);
+        }
+    }
+
+    private static PluginProfileLoaderImpl instance = null;
+
+    public static PluginProfileLoader getInstance() {
+        if (instance == null) {
+            instance = new PluginProfileLoaderImpl();
+        }
+        return instance;
+    }
 
 }