net.sourceforge.jenesis4java.impl.BasicCompilationUnitEncoder Maven / Gradle / Ivy

Go to download
package net.sourceforge.jenesis4java.impl;

/*
 * #%L
 * Jenesis 4 Java Code Generator
 * %%
 * Copyright (C) 2000 - 2015 jenesis4java
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import java.io.*;

import net.sourceforge.jenesis4java.CodeWriter;
import net.sourceforge.jenesis4java.CompilationUnit;
import net.sourceforge.jenesis4java.CompilationUnitEncoder;
import net.sourceforge.jenesis4java.impl.MDeclaration.MCompilationUnit;

import static java.nio.charset.StandardCharsets.UTF_8;

public class BasicCompilationUnitEncoder implements CompilationUnitEncoder {

    @Override
    public void encode(CompilationUnit compilationUnit) {

        // wrap this section for io exceptions
        PrintWriter fout = null;

        // Make a file object which corresponds to the
        // filename which we will write to.
        File file = new File(((MCompilationUnit) compilationUnit).getFileName());

        // Get the parent file such that we can create the
        // directory if necessary. NOTE: JDK1.2 dependency
        // here.
        File dir = file.getParentFile();

        try {

            // Make sure it exists.
            if (!dir.exists() && !dir.mkdirs()) {
                throw new RuntimeException("Could not create directory(s) " + dir.getCanonicalPath());
            }

            // now we want to make the stream...
            Writer fileWriter = new OutputStreamWriter(new FileOutputStream(file), UTF_8);
            fout = new PrintWriter(new BufferedWriter(fileWriter));

            // ...wrap it with a code writer...
            CodeWriter out = new MCodeWriter(fout);

            // ...generate the code...
            compilationUnit.toCode(out);

            // ... and finally close and drop the stream
            fout.close();
            fout = null;

        } catch (IOException ioex) {
            ioex.printStackTrace();
        } finally {
            if (fout != null) {
                try {
                    fout.close();
                } catch (Exception ex) {
                    throw new RuntimeException(ex);
                }
            }
        }
    }

}