CppGenerator.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.record.compiler;
  19. import java.util.ArrayList;
  20. import java.io.File;
  21. import java.io.FileWriter;
  22. import java.io.IOException;
  23. import java.util.Iterator;
  24. /**
  25.  * C++ Code generator front-end for Hadoop record I/O.
  26.  */
  27. class CppGenerator extends CodeGenerator {
  28.   
  29.   CppGenerator() {
  30.   }
  31.   
  32.   /**
  33.    * Generate C++ code. This method only creates the requested file(s)
  34.    * and spits-out file-level elements (such as include statements etc.)
  35.    * record-level code is generated by JRecord.
  36.    */
  37.   void genCode(String name, ArrayList<JFile> ilist,
  38.                ArrayList<JRecord> rlist, String destDir, ArrayList<String> options)
  39.     throws IOException {
  40.     name = new File(destDir, (new File(name)).getName()).getAbsolutePath();
  41.     FileWriter cc = new FileWriter(name+".cc");
  42.     try {
  43.       FileWriter hh = new FileWriter(name+".hh");
  44.       
  45.       try {
  46.         String fileName = (new File(name)).getName();
  47.         hh.write("#ifndef __"+fileName.toUpperCase().replace('.','_')+"__n");
  48.         hh.write("#define __"+fileName.toUpperCase().replace('.','_')+"__n");
  49.         hh.write("#include "recordio.hh"n");
  50.         hh.write("#include "recordTypeInfo.hh"n");
  51.         for (Iterator<JFile> iter = ilist.iterator(); iter.hasNext();) {
  52.           hh.write("#include ""+iter.next().getName()+".hh"n");
  53.         }
  54.         
  55.         cc.write("#include ""+fileName+".hh"n");
  56.         cc.write("#include "utils.hh"n");
  57.         
  58.         for (Iterator<JRecord> iter = rlist.iterator(); iter.hasNext();) {
  59.           iter.next().genCppCode(hh, cc, options);
  60.         }
  61.         
  62.         hh.write("#endif //"+fileName.toUpperCase().replace('.','_')+"__n");
  63.       } finally {
  64.         hh.close();
  65.       }
  66.     } finally {
  67.       cc.close();
  68.     }
  69.   }
  70. }