private void printFuncs(AutoTypeImage imageIn, AutoTypeImage imageOut) {
    this.imageIn = imageIn;
    this.imageOut = imageOut;

    if (imageIn.isInteger()) genName = "I32";
    else genName = "F" + imageIn.getNumBits();
    ;

    sumType = imageIn.getSumType();

    printHorizontal();
    printVertical();
    printHorizontalInverse();
    printVerticalInverse();
  }
Esempio n. 2
0
  private void createFile(AutoTypeImage imageType) throws FileNotFoundException {
    String suffix = imageType.getAbbreviatedType();
    suffix = suffix.compareTo("S32") == 0 ? "I32" : suffix;
    className = "Kernel1D_" + suffix;

    String sumType = imageType.getSumType();

    setOutputFile(className);
    out.print(
        "/**\n"
            + " * Floating point 1D convolution kernel that extends {@link Kernel1D}.\n"
            + " *\n"
            + " * <p>\n"
            + " * WARNING: Do not modify.  Automatically generated by {@link "
            + getClass().getName()
            + "}.\n"
            + " * </p>\n"
            + " *\n"
            + " * @author Peter Abeles\n"
            + " */\n"
            + "public class "
            + className
            + " extends Kernel1D {\n"
            + "\n"
            + "\tpublic "
            + sumType
            + " data[];\n"
            + "\n"
            + "\t/**\n"
            + "\t * Creates a new kernel whose initial values are specified by data and width.  The length\n"
            + "\t * of its internal data will be width.  Data must be at least as long as width.\n"
            + "\t *\n"
            + "\t * @param data  The value of the kernel. Not modified.  Reference is not saved.\n"
            + "\t * @param width The kernels width.  Must be odd.\n"
            + "\t */\n"
            + "\tpublic "
            + className
            + "("
            + sumType
            + " data[], int width) {\n"
            + "\t\tthis(data,width/2,width);\n"
            + "\t}\n"
            + "\n"
            + "\t/**\n"
            + "\t * Creates a new kernel whose initial values are specified by data and width.  The length\n"
            + "\t * of its internal data will be width.  Data must be at least as long as width.\n"
            + "\t *\n"
            + "\t * @param data  The value of the kernel. Not modified.  Reference is not saved.\n"
            + "\t * @param width The kernels width.  Must be odd.\n"
            + "\t * @param offset Location of the origin in the array\n"
            + "\t */\n"
            + "\tpublic "
            + className
            + "("
            + sumType
            + " data[], int offset , int width) {\n"
            + "\t\tsuper(offset,width);\n"
            + "\n"
            + "\t\tthis.data = new "
            + sumType
            + "[width];\n"
            + "\t\tSystem.arraycopy(data, 0, this.data, 0, width);\n"
            + "\t}\n"
            + "\n"
            + "\t/**\n"
            + "\t * Create a kernel whose elements are all equal to zero.\n"
            + "\t *\n"
            + "\t * @param width How wide the kernel is.  Must be odd.\n"
            + "\t */\n"
            + "\tpublic "
            + className
            + "(int width) {\n"
            + "\t\tthis(width/2,width);\n"
            + "\t}\n"
            + "\n"
            + "\t/**\n"
            + "\t * Create a kernel whose elements are all equal to zero.\n"
            + "\t *\n"
            + "\t * @param width How wide the kernel is.  Must be odd.\n"
            + "\t * @param offset Location of the origin in the array\n"
            + "\t */\n"
            + "\tpublic "
            + className
            + "(int offset , int width) {\n"
            + "\t\tsuper(offset,width);\n"
            + "\t\tdata = new "
            + sumType
            + "[width];\n"
            + "\t}\n"
            + "\n"
            + "\tprotected "
            + className
            + "() {\n"
            + "\t}\n"
            + "\n"
            + "\t/**\n"
            + "\t * Creates a kernel whose elements are the specified data array and has\n"
            + "\t * the specified width.\n"
            + "\t *\n"
            + "\t * @param data  The array who will be the kernel's data.  Reference is saved.\n"
            + "\t * @param width The kernel's width.\n"
            + "\t * @return A new kernel.\n"
            + "\t */\n"
            + "\tpublic static "
            + className
            + " wrap("
            + sumType
            + " data[], int width) {\n"
            + "\t\t"
            + className
            + " ret = new "
            + className
            + "();\n"
            + "\t\tret.data = data;\n"
            + "\t\tret.width = width;\n"
            + "\t\tret.offset = width/2;\n"
            + "\n"
            + "\t\treturn ret;\n"
            + "\t}\n"
            + "\n"
            + "\t@Override\n"
            + "\tpublic boolean isInteger() {\n"
            + "\t\treturn "
            + imageType.isInteger()
            + ";\n"
            + "\t}\n"
            + "\n"
            + "\tpublic "
            + sumType
            + " get(int i) {\n"
            + "\t\treturn data[i];\n"
            + "\t}\n"
            + "\n"
            + "\tpublic "
            + sumType
            + " computeSum() {\n"
            + "\t\t"
            + sumType
            + " sum = 0;\n"
            + "\t\tfor( int i = 0; i < data.length; i++ ) {\n"
            + "\t\t\tsum += data[i];\n"
            + "\t\t}\n"
            + "\n"
            + "\t\treturn sum;\n"
            + "\t}\n"
            + "\n"
            + "\tpublic "
            + sumType
            + "[] getData() {\n"
            + "\t\treturn data;\n"
            + "\t}\n"
            + "\n"
            + "\tpublic void print() {\n"
            + "\t\tfor (int i = 0; i < width; i++) {\n");
    if (imageType.isInteger()) out.print("\t\t\tSystem.out.printf(\"%6d \", data[i]);\n");
    else if (imageType.isInteger()) out.print("\t\t\tSystem.out.printf(\"%6.3f \", data[i]);\n");

    out.print("\t\t}\n" + "\t\tSystem.out.println();\n" + "\t}\n" + "}\n\n");
  }
  private void printVerticalInverse() {
    out.print(
        "\t/**\n"
            + "\t * Performs a single level inverse wavelet transform along the vertical axis.\n"
            + "\t *\n"
            + "\t * @param inverseCoef Description of wavelet coefficients.\n"
            + "\t * @param input Transformed image. Not modified.\n"
            + "\t * @param output Reconstruction of original image. Modified\n"
            + "\t */\n"
            + "\tpublic static void verticalInverse( BorderIndex1D border , WlBorderCoef<WlCoef_"
            + genName
            + "> inverseCoef , "
            + imageIn.getSingleBandName()
            + " input , "
            + imageIn.getSingleBandName()
            + " output ) {\n"
            + "\n"
            + "\t\tUtilWavelet.checkShape(output,input);\n"
            + "\n"
            + "\t\t"
            + sumType
            + " []trends = new "
            + sumType
            + "[ output.height ];\n"
            + "\t\t"
            + sumType
            + " []details = new "
            + sumType
            + "[ output.height ];\n"
            + "\n"
            + "\t\tboolean isLarger = input.height > output.height;\n"
            + "\t\tint paddedHeight = output.height + output.height%2;\n"
            + "\n"
            + "\t\tfinal int lowerBorder = inverseCoef.getLowerLength()*2;\n"
            + "\t\tfinal int upperBorder = output.height - inverseCoef.getUpperLength()*2;\n"
            + "\n"
            + "\t\tborder.setLength(output.height+output.height%2);\n"
            + "\n");

    if (imageIn.isInteger()) {
      out.print("\t\tWlCoef_" + genName + " coefficients = inverseCoef.getInnerCoefficients();\n");
      out.print(
          "\t\tfinal int e = coefficients.denominatorScaling*2;\n"
              + "\t\tfinal int f = coefficients.denominatorWavelet*2;\n"
              + "\t\tfinal int ef = e*f;\n"
              + "\t\tfinal int ef2 = ef/2;\n");
    } else {
      out.print("\t\tWlCoef_" + genName + " coefficients;\n");
    }
    out.print("\n");
    out.print(
        "\t\tfor( int x = 0; x < output.width; x++) {\n"
            + "\n"
            + "\t\t\tfor( int i = 0; i < details.length; i++ ) {\n"
            + "\t\t\t\tdetails[i] = 0;\n"
            + "\t\t\t\ttrends[i] = 0;\n"
            + "\t\t\t}\n"
            + "\n"
            + "\t\t\tfor( int y = 0; y < output.height; y += 2 ) {\n"
            + "\t\t\t\t"
            + sumType
            + " a = input.get(x,y/2);\n"
            + "\t\t\t\t"
            + sumType
            + " d = input.get(x,y/2+input.height/2);\n"
            + "\n"
            + "\t\t\t\tif( y < lowerBorder ) {\n"
            + "\t\t\t\t\tcoefficients = inverseCoef.getBorderCoefficients(y);\n"
            + "\t\t\t\t} else if( y >= upperBorder ) {\n"
            + "\t\t\t\t\tcoefficients = inverseCoef.getBorderCoefficients(y-paddedHeight);\n"
            + "\t\t\t\t} else {\n"
            + "\t\t\t\t\tcoefficients = inverseCoef.getInnerCoefficients();\n"
            + "\t\t\t\t}\n"
            + "\n"
            + "\t\t\t\tfinal int offsetA = coefficients.offsetScaling;\n"
            + "\t\t\t\tfinal int offsetB = coefficients.offsetWavelet;\n"
            + "\t\t\t\tfinal "
            + sumType
            + "[] alpha = coefficients.scaling;\n"
            + "\t\t\t\tfinal "
            + sumType
            + "[] beta = coefficients.wavelet;\n"
            + "\n"
            + "\t\t\t\t// add the 'average' signal\n"
            + "\t\t\t\tfor( int i = 0; i < alpha.length; i++ ) {\n"
            + "\t\t\t\t\t// if an odd image don't update the outer edge\n"
            + "\t\t\t\t\tint yy = border.getIndex(y+offsetA+i);\n"
            + "\t\t\t\t\tif( isLarger && yy >= output.height )\n"
            + "\t\t\t\t\t\tcontinue;\n"
            + "\t\t\t\t\ttrends[yy] += a*alpha[i];\n"
            + "\t\t\t\t}\n"
            + "\n"
            + "\t\t\t\t// add the detail signal\n"
            + "\t\t\t\tfor( int i = 0; i < beta.length; i++ ) {\n"
            + "\t\t\t\t\tint yy = border.getIndex(y+offsetB+i);\n"
            + "\t\t\t\t\tif( isLarger && yy >= output.height )\n"
            + "\t\t\t\t\t\tcontinue;\n"
            + "\t\t\t\t\tdetails[yy] += d*beta[i];\n"
            + "\t\t\t\t}\n"
            + "\t\t\t}\n"
            + "\n"
            + "\t\t\tfor( int y = 0; y < output.height; y++ ) {\n");
    if (imageIn.isInteger()) {
      out.print("\t\t\t\toutput.set(x,y, UtilWavelet.round(trends[y]*f + details[y]*e,ef2,ef));\n");
    } else {
      out.print("\t\t\t\toutput.set(x,y, trends[y] + details[y]);\n");
    }
    out.print("\t\t\t}\n" + "\t\t}\n" + "\t}\n\n");
  }
 private void printVertical() {
   out.print(
       "\t/**\n"
           + "\t * Performs a single level wavelet transform along the vertical axis.\n"
           + "\t *\n"
           + "\t * @param coefficients Description of wavelet coefficients.\n"
           + "\t * @param input Input image which is being transform. Not modified.\n"
           + "\t * @param output where the output is written to. Modified\n"
           + "\t */\n"
           + "\tpublic static void vertical( BorderIndex1D border , WlCoef_"
           + genName
           + " coefficients ,\n"
           + "\t\t\t\t\t\t\t\t "
           + imageIn.getSingleBandName()
           + " input , "
           + imageIn.getSingleBandName()
           + " output ) {\n"
           + "\n"
           + "\t\tUtilWavelet.checkShape(input,output);\n"
           + "\n"
           + "\t\tfinal int offsetA = coefficients.offsetScaling;\n"
           + "\t\tfinal int offsetB = coefficients.offsetWavelet;\n"
           + "\t\tfinal "
           + sumType
           + "[] alpha = coefficients.scaling;\n"
           + "\t\tfinal "
           + sumType
           + "[] beta = coefficients.wavelet;\n"
           + "\n"
           + "\t\tborder.setLength(input.height+input.height%2);\n"
           + "\n"
           + "\t\tboolean isLarger = output.height > input.height;\n"
           + "\n"
           + "\t\tfor( int x = 0; x < input.width; x++) {\n"
           + "\t\t\tfor( int y = 0; y < input.height; y += 2 ) {\n"
           + "\t\t\t\t"
           + sumType
           + " scale = 0;\n"
           + "\t\t\t\t"
           + sumType
           + " wavelet = 0;\n"
           + "\n"
           + "\t\t\t\tfor( int i = 0; i < alpha.length; i++ ) {\n"
           + "\t\t\t\t\tint yy = border.getIndex(y+i+offsetA);\n"
           + "\t\t\t\t\tif( isLarger && yy >= input.height )\n"
           + "\t\t\t\t\t\tcontinue;\n"
           + "\t\t\t\t\tscale += input.get(x,yy)*alpha[i];\n"
           + "\t\t\t\t}\n"
           + "\t\t\t\tfor( int i = 0; i < beta.length; i++ ) {\n"
           + "\t\t\t\t\tint yy = border.getIndex(y+i+offsetB);\n"
           + "\t\t\t\t\tif( isLarger && yy >= input.height )\n"
           + "\t\t\t\t\t\tcontinue;\n"
           + "\t\t\t\t\twavelet += input.get(x,yy)*beta[i];\n"
           + "\t\t\t\t}\n"
           + "\n"
           + "\t\t\t\tint outY = y/2;\n"
           + "\n");
   if (imageIn.isInteger()) {
     out.print(
         "\t\t\t\tscale = 2*scale/coefficients.denominatorScaling;\n"
             + "\t\t\t\twavelet = 2*wavelet/coefficients.denominatorWavelet;\n"
             + "\n");
   }
   out.print(
       "\t\t\t\toutput.set(x , outY,scale);\n"
           + "\t\t\t\toutput.set(x , output.height/2 + outY , wavelet );\n"
           + "\t\t\t}\n"
           + "\t\t}\n"
           + "\t}\n\n");
 }