Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] View of /branches/vis15/src/compiler/target-cpu/gen.sml
ViewVC logotype

View of /branches/vis15/src/compiler/target-cpu/gen.sml

Parent Directory Parent Directory | Revision Log Revision Log


Revision 3917 - (download) (annotate)
Sat May 28 16:41:39 2016 UTC (3 years ago) by jhr
File size: 9832 byte(s)
  Working on merge: code generation
(* gen.sml
 *
 * Code generation for the sequential and parallel targets.
 *
 * COPYRIGHT (c) 2016 The Diderot Project (http://diderot-language.cs.uchicago.edu)
 * All rights reserved.
 *)

structure Gen : sig

    val exec : TargetSpec.t * TreeIR.program -> unit

    val library : TargetSpec.t * TreeIR.program -> unit

  end = struct

    structure IR = TreeIR
    structure CL = CLang
    structure Env = CodeGenEnv
    structure Out = CodeOutput
    structure RN = CxxNames

    val openCxxOut = Out.openOut {ext = "cxx", ppDecl = PrintAsCxx.output}

    fun mkEnv spec = if TargetSpec.dualState spec
	  then Env.new {
	      global = RN.globalsVar,
	      selfIn = RN.selfInVar,
	      selfOut = RN.selfOutVar,
	      spec = spec
	    }
	  else Env.new {
	      global = RN.globalsVar,
	      selfIn = RN.selfVar,
	      selfOut = RN.selfVar,
	      spec = spec
	    }

  (* create the target-specific substitution list *)
    fun mkSubs (spec, IR.Strand{name, ...}) = [
	    ("CFILE",			OS.Path.joinBaseExt{base= #outBase spec, ext= SOME "c"}),
	    ("CXXFILE",			OS.Path.joinBaseExt{base= #outBase spec, ext= SOME "cxx"}),
	    ("HDRFILE",			OS.Path.joinBaseExt{base= #outBase spec, ext= SOME "h"}),
	    ("PREFIX",			#namespace spec),
	    ("SRCFILE",			#srcFile spec),
	    ("PROG_NAME",		#outBase spec),
	    ("STRAND",			Atom.toString name),
	    ("STRANDTY",		Atom.toString name ^ "_strand"),
	    ("DIDEROTC_CMD",		#diderotc spec),
	    ("DIDEROTC_ARGV",		String.concatWith " " (#argv spec)),
	    ("DIDEROTC_VERSION",	#version spec),
	    ("DIDEROT_FLOAT_PRECISION",	TargetSpec.floatPrecisionDef spec),
	    ("DIDEROT_INT_PRECISION",	TargetSpec.intPrecisionDef spec),
	    ("DIDEROT_TARGET",		TargetSpec.targetDef spec),
	    ("REALTY",			if #double spec then "double" else "float"),
	    ("INTTY",			if #longint spec then "int64_t" else "int32_t")
	  ]

    fun condCons (true, x, xs) = x::xs
      | condCons (false, _, xs) = xs

    fun verbFrag (spec, parFrag, seqFrag, subs) =
	  CL.verbatimDcl [if (TargetSpec.isParallel spec) then parFrag else seqFrag] subs

  (* generate declarations for the types and operations that are used in the program *)
    fun genTypesAndOperations (outS, spec : TargetSpec.t, info) = let
	  fun doType (ty, isPrint) = ()
	  in
	    CollectInfo.applyToTypes doType info
	  end

  (* generate the struct declaration for the global variables *)
    fun genGlobalStruct (spec : TargetSpec.t, globals) =
	  if #hasGlobals spec
	    then let
	      val fields = [] (* FIXME *)
	      in
		[CL.D_StructDef(SOME "globals", fields, NONE)]
	      end
	    else []

  (* generate the struct declaration for the world representation *)
    fun genWorldStruct (spec : TargetSpec.t, strandName) = let
	  fun memberVar (ty, name) = CL.D_Var([], ty, name, NONE)
	  val members = []
	  val statePtrTy = CL.T_Ptr(CL.T_Ptr(CL.T_Named(Atom.toString strandName ^ "_strand")))
	  val members = if TargetSpec.dualState spec
		then memberVar(statePtrTy, "_inState") ::
		  memberVar(statePtrTy, "_inState") :: members
		else memberVar(statePtrTy, "_state") :: members
	  val members = memberVar(CL.T_Ptr CL.uint8, "_status") :: members
	  val members = if #hasGlobals spec
		then memberVar(RN.globalPtrTy, "_globals") :: members
		else members
	  val members = if #exec spec orelse not(#hasInputs spec)
		then members
		else memberVar(CL.T_Named "inputs", "_definedInp") :: members
	  val members = if TargetSpec.isParallel spec
		then memberVar(CL.T_Ptr(CL.T_Named "sched_info"), "_sched") :: members
		else memberVar(CL.T_Named "uint32_t", "_nactive") ::
		  memberVar(CL.T_Named "uint32_t", "_nstable") :: members
	(* add world method decls *)
	  fun memberFun (ty, name, params) = CL.D_Proto([], ty, name, params)
	  val members =
		memberFun (CL.voidTy, "swap_state", []) ::
		memberFun (CL.uint32, "run", [CL.PARAM([], CL.uint32, "max_nsteps")]) ::
		memberFun (CL.voidTy, "initially", []) ::
		CL.D_Destr([], NONE, "world", NONE) ::
		CL.D_Constr([], NONE, "world", [], [], NONE) ::
		members
	  in
	    CL.D_ClassDef{
		name = "world",
		from = SOME "public diderot::world_base",
		public = List.rev members,
		protected = [],
		private = []
	      }
	  end

    fun runFrag (spec, subs) = if TargetSpec.noBSP spec
	    then verbFrag (spec, Fragments.parallelRunNoBSP, Fragments.sequentialRunNoBSP, subs)
	    else verbFrag (spec, Fragments.parallelRun, Fragments.sequentialRun, subs)

    fun compile (spec : TargetSpec.t, basename) = let
	(* generate the C compiler flags *)
	  val cflags = ["-I" ^ Paths.diderotInclude(), "-I" ^ Paths.teemInclude()]
	  val cflags = condCons (TargetSpec.isParallel spec, #pthread Paths.cxxflags, cflags)
	  val cflags = if #debug spec
		then #debug Paths.cxxflags :: cflags
		else #ndebug Paths.cxxflags :: cflags
	  val cflags = #base Paths.cxxflags :: cflags
	  in
	    RunCC.compile (basename, cflags)
	  end

    fun ldFlags (spec : TargetSpec.t) = if #exec spec
	  then let
	    val extraLibs = condCons (TargetSpec.isParallel spec, #pthread Paths.extraLibs, [])
	    val extraLibs = Paths.teemLinkFlags() @ #base Paths.extraLibs :: extraLibs
	    val rtLib = TargetSpec.runtimeLibName spec
	    in
	      condCons (TargetSpec.isParallel spec, #pthread Paths.cxxflags, rtLib :: extraLibs)
	    end
	  else [TargetSpec.runtimeLibName spec]

  (* generate source code that is common to both libraries and standalone executables *)
    fun outputSrc (outS, env, spec, prog, strand, substitutions, genInputCode) = let
	  val IR.Program{globals, create=IR.Create{dim, ...}, ...} = prog
	  val IR.Strand{name=strandName, ...} = strand
	  val ppDecl = Out.decl outS
	  val outputs = OutputUtil.gatherOutputs prog
	  in
	    if (TargetSpec.isParallel spec)
	      then ppDecl (CL.verbatimDcl [Fragments.parallelExtras] substitutions)
	      else ();
	    if TargetSpec.dualState spec
	      then ppDecl (CL.D_Verbatim ["#define DIDEROT_DUAL_STATE\n"])
	      else ();
	    if not(#hasGlobals spec)
	      then ppDecl (CL.D_Verbatim ["#define DIDEROT_NO_GLOBALS\n"])
	      else ();
	    if not(#hasInputs spec)
	      then ppDecl (CL.D_Verbatim ["#define DIDEROT_NO_INPUTS\n"])
	      else ();
(* FIXME: what about constants and inputs? *)
	    List.app ppDecl (genGlobalStruct (spec, globals));
	    ppDecl (genWorldStruct(spec, strandName));
	    List.app ppDecl (genInputCode());
(* FIXME: need to gather/generate topDecls
	    List.app ppDecl (!topDecls);
*)
	    List.app ppDecl (GenStrand.gen (env, strand));
	    List.app ppDecl (GenOutputs.gen (env, dim, outputs));
	    ppDecl (CL.verbatimDcl [Fragments.worldMethods] substitutions);
(* TODO
	    ppDecl (CL.verbatimDcl [InitFrag.text] substitutions);
	    ppDecl (CL.verbatimDcl [AllocFrag.text] substitutions);
	    ppDecl (!initially);
*)
	    ppDecl (runFrag (spec, substitutions))
	  end
	    
(********************
        fun outputLibSrc (baseName, prog as Prog{props, inputs, strands, ...}) = let
              val [strand] = AtomTable.listItems strands
              val substitutions = mkSubs (props, strand)
            (* output to C file *)
              val out = openOut baseName
              fun ppDecl dcl = PrintAsC.output(#ppStrm out, dcl)
              in
                ppDecl (CL.verbatimDcl [LibHdrFrag.text] substitutions);
                List.app ppDecl (GenInputs.genDefinedInpStruct (props, !inputs));
                outputSrc (out, prog, strand, substitutions,
                  fn () => GenInputs.genInputFuns(props, !inputs));
                ppDecl (CL.verbatimDcl [CBodyFrag.text] substitutions);
                closeOut out
              end

        fun generateLib (prog as Prog{props, inputs, strands, ...}) = let
              val {outDir, outBase, exec, double, debug, ...} = props
              val basename = OS.Path.joinDirFile{dir=outDir, file=outBase}
              val [Strand{state, output, ...}] = AtomTable.listItems strands
              in
              (* generate the library .h file *)
                GenLibraryInterface.gen {
                    props = props,
                    rt = if TargetSpec.isParallel spec
                      then SOME LibInterfaceParFrag.text
                      else NONE,
                    inputs = !inputs,
                    outputs = output
                  };
              (* *)
                outputLibSrc (basename, prog);
              (* compile and link *)
                compile (props, basename);
                RunCC.linkLib (basename, ldFlags props)
              end
********************)

    fun exec (spec : TargetSpec.t, prog) = let
	  val IR.Program{inputs, strand, ...} = prog
	  val env = mkEnv spec
	  val baseName = OS.Path.joinDirFile{dir = #outDir spec, file = #outBase spec}
	  val substitutions = mkSubs (spec, strand)
	(* output to C++ file *)
	  val outS = openCxxOut baseName
	  val ppDecl = Out.decl outS
	  val fragment = Out.fragment substitutions outS
	  in
	    GenExecUtil.execHead (outS, substitutions);
	    genTypesAndOperations (outS, spec, CollectInfo.collect prog);
	    outputSrc (outS, env, spec, prog, strand, substitutions,
	      fn () => (
		GenInputs.genInputsStruct (env, inputs) @
		GenInputs.genExecInputFuns (env, prog) @
		OutputUtil.genRegisterOutputOpts (env, OutputUtil.gatherOutputs prog)));
	    ppDecl (verbFrag (spec, Fragments.parallelMain, Fragments.sequentialMain, substitutions));
	    GenExecUtil.execFoot (outS, substitutions);
	    Out.closeOut outS;
	    compile (spec, baseName);
	    RunCC.linkExec (baseName, ldFlags spec)
	  end

    fun library (spec : TargetSpec.t, prog) = let
	  val IR.Program{
		  props, target, consts, inputs, constInit,
		  globals, globalInit, strand, create, update
		} = prog
	  val {outDir, outBase, exec, double, debug, ...} = spec
	  val env = mkEnv spec
	  in
	    raise Fail "FIXME"
	  end

  end

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0