cb.generator.java.ClassImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.crazybeans Show documentation
Java library to read, modify or create Rational Rose petal files
The newest version!
/**
 * Copyright (c) 2001 Markus Dahm
 * Copyright (C) 2015-2018 BITPlan GmbH http://www.bitplan.com
 *
 * This source is part of
 * https://github.com/BITPlan/CrazyBeans
 * and the license as outlined there applies
 */
package cb.generator.java;

import java.io.PrintWriter;
import java.util.*;

/**
 * Simple representation of a Java Class.
 *
 * @version $Id: ClassImpl.java,v 1.5 2001/06/27 10:26:03 dahm Exp $
 * @author M. Dahm
 */
public class ClassImpl extends NodeImpl implements Class {
  private Package pack;
  private ArrayList fields = new ArrayList();
  private ArrayList methods = new ArrayList();
  private ArrayList super_classes = new ArrayList();
  private ArrayList interfaces = new ArrayList();
  private ArrayList prefix = new ArrayList();
  private boolean isInterface;
  protected cb.petal.Class clazz;

  /**
   * construct me from the given clazz
   * @param clazz
   */
  public ClassImpl(cb.petal.Class clazz) {
  		super(clazz);
  		setClazz(clazz);
  }

  public void setClazz(cb.petal.Class c) {
    super.setDocumentedObject(c);;
    clazz = c;
  }

  public cb.petal.Class getClazz() {
    return clazz;
  }

  /**
   * set my package
   */
  public void setPackage(Package p) {
    pack = p;
  }

  /**
   * get my package
   */
  public Package getPackage() {
    return pack;
  }

  public void isInterface(boolean i) {
    isInterface = i;
  }

  public boolean isInterface() {
    return isInterface;
  }

  public void addField(Field f) {
    fields.add(f);
  }

  public void removeField(Field f) {
    fields.remove(f);
  }

  public void addMethod(Method f) {
    methods.add(f);
  }

  public void removeMethod(Method f) {
    methods.remove(f);
  }

  public void addSuperClass(String s) {
    super_classes.add(s);
  }

  public void removeSuperClass(String s) {
    super_classes.remove(s);
  }

  public void addImplementedInterface(String s) {
    interfaces.add(s);
  }

  public void removeImplementedInterface(String s) {
    interfaces.remove(s);
  }

  public String getQualifiedName() {
    if (pack != null && !"".equals(pack))
      return pack + "." + name;
    else
      return name;
  }

  public void addPrefixCode(String c) {
    prefix.add(c);
  }

  public Collection getMethods() {
    return methods;
  }

  public Collection getFields() {
    return fields;
  }

  public Collection getSuperClasses() {
    return super_classes;
  }

  public Collection getImplementedInterfaces() {
    return interfaces;
  }

  protected static void print(PrintWriter stream, String pre, String o,
      String post) {
    if ((o != null) && !"".equals(o))
      stream.print(pre + o + post);
  }

  /**
   * Default implementation prints Java code
   */
  public void dump(PrintWriter stream) {
    print(stream, "package ", pack.getQualifiedName(), ";\n");

    for (Iterator i = prefix.iterator(); i.hasNext();)
      stream.println(i.next());

    stream.println("\n/** Created with Generator/"
        + ""
        + "\n * CrazyBeans " + new Date() + "\n *");

    // Print documentation if any
    if (clazz != null) {
      for (String line : this.getDocumentation()) {
        stream.println(" * " + line);
      }
    }

    stream.println(" * @cbversion " + cb.util.Constants.VERSION + "\n */");
    print(stream, "", getAccess(), " ");

    if (isInterface())
      stream.print("interface " + getName() + " ");
    else
      stream.print("class " + getName() + " ");

    if (!super_classes.isEmpty()) {
      stream.print("extends ");

      for (Iterator i = super_classes.iterator(); i.hasNext();) {
        stream.print(i.next());

        if (i.hasNext())
          stream.print(", ");
      }

      stream.print(" ");
    }

    if (!interfaces.isEmpty() && !isInterface()) {
      stream.print("implements ");

      for (Iterator i = interfaces.iterator(); i.hasNext();) {
        stream.print(i.next());
        if (i.hasNext())
          stream.print(", ");
      }

      stream.print(" ");
    }

    stream.println("{");

    for (Iterator i = getFields().iterator(); i.hasNext();)
      (i.next()).dump(stream);

    stream.println();

    for (Iterator i = getMethods().iterator(); i.hasNext();) {
      (i.next()).dump(stream);

      if (i.hasNext())
        stream.println();
    }

    stream.println("}");
  }

  public boolean equals(Object o) {
    if (o instanceof Class) {
      Class c = (Class) o;

      return getQualifiedName().equals(c.getQualifiedName());
    } else
      return false;
  }
}