source: java2python/src/main/java/tudelft/utilities/j2p/t/PC.java@ 353

package tudelft.utilities.j2p.t;

import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

/**
 * Default {@link PyCode} object
 *
 */
public class PC implements PyCode {
        private Set<String> imp;
        private String cd;

        public PC() {
                this(Collections.emptySet(), "");
        }

        /**
         * a default {@link PyCode} object containing a translation of a java call
         * 
         * @param imports the imports that are needed to evaluate the code
         * @param code    the actual python code
         */
        public PC(Set<String> imports, String code) {
                this.imp = imports;
                this.cd = code;
        }

        @Override
        public Set<String> imports() {
                return imp;
        }

        @Override
        public String code() {
                return cd;
        }

        @Override
        public String toString() {
                // HACK
                return imp.toString() + "\n" + cd;
        }

        public PC add(PyCode othercode) {
                Set<String> newimports = new HashSet<>(imp);
                newimports.addAll(othercode.imports());
                String newcode = cd + othercode.code();

                return new PC(newimports, newcode);
        }

}