示例#1
0
def main():
    """Fetch input data and dispatch instructions."""
    # WARNING: you must use send_back() to return data to the parent process.

    instruction = sys.argv[1]
    data = pickle.loads(''.join(sys.stdin.readlines()))

    if instruction == 'build3d':
        from qmt.geometry.freecad.objectConstruction import build
        activate_doc_from(data['current_options'])
        geo_output = build(data['current_options'])
        send_back(geo_output)

    elif instruction == 'writeFCfile':
        pass

    elif instruction == 'region_map_function':
        import fenics  # DON'T TOUCH: this controls fenics mesh module loading order
        from qms.meshing.make_region_marker_wrapper import make_region_marker_function
        new_data = make_region_marker_function(
            data)  # the updated Geo3DData object
        send_back(new_data)

    else:
        raise ValueError('Not a registered FreeCAD QMT instruction')
示例#2
0
def build_3d_geometry(
    input_parts,
    input_file=None,
    xsec_dict=None,
    serialized_input_file=None,
    params=None,
):
    """
    Build a geometry in 3D.

    :param list input_parts: Ordered list of input parts, leftmost items get built first
    :param str input_file: Path to FreeCAD template file. Either this or serialized_input_file
        must be set (but not both).
    :param dict xsec_dict: Dictionary of cross-section specifications. It should be of the
        form {'xsec_name':{'axis':(1,0,0),'distance':0.}}, where the axis parameter is a tuple
        defining the axis that defines the normal of the cross section, and distance is
        the length along the axis used to set the cross section.
    :param bytes serialized_input_file: FreeCAD template file that has been serialized using
        qmt.data.serialize_file. This is useful for passing a
        file into a docker container or other environment that
        doesn't have access to a shared drive. Either this or
        serialized_input_file must be set (but not both).
    :param dict params: Dictionary of parameters to use in FreeCAD.
    :return Geo3DData: A built geometry.
    """
    if input_file is None and serialized_input_file is None:
        raise ValueError("One of input_file or serialized_input_file must be non-none.")
    elif input_file is not None and serialized_input_file is not None:
        raise ValueError("Both input_file and serialized_input_file were non-none.")
    elif input_file is not None:
        serial_fcdoc = serialize_file(input_file)
    else:
        serial_fcdoc = serialized_input_file
    if params is None:
        params = {}
    if xsec_dict is None:
        xsec_dict = {}
    options_dict = {}
    options_dict["serial_fcdoc"] = serial_fcdoc
    options_dict["input_parts"] = input_parts
    options_dict["params"] = params
    options_dict["xsec_dict"] = xsec_dict

    data = Geo3DData()
    data.serial_fcdoc = serial_fcdoc
    data.get_data("fcdoc")
    from qmt.geometry.freecad.objectConstruction import build

    built = build(options_dict)
    FreeCAD.closeDocument("instance")
    return built
示例#3
0
文件: run.py 项目: agarwalrounak/qmt
def main():
    """Fetch input data and dispatch instructions."""
    # WARNING: you must use send_back() to return data to the parent process.

    instruction = sys.argv[1]
    data = pickle.loads(''.join(sys.stdin.readlines()))

    if instruction == 'build3d':
        from qmt.geometry.freecad.objectConstruction import build
        activate_doc_from(data['current_options'])
        geo_output = build(data['current_options'])
        send_back(geo_output)

    elif instruction == 'writeFCfile':
        pass

    else:
        raise ValueError('Not a registered FreeCAD QMT instruction')
示例#4
0
def build_3d_geometry(
    input_parts: List[Geo3DPart],
    input_file: Optional[str] = None,
    xsec_dict: Dict[str, Dict] = None,
    serialized_input_file: Optional[bytes] = None,
    params: Optional[Dict] = None,
) -> Geo3DData:
    """Build a geometry in 3D.

    Parameters
    ----------
    input_parts : list
        Ordered list of input parts, leftmost items get built first
    input_file : str
        Path to FreeCAD template file. Either this or serialized_input_file
        must be set (but not both).
        (Default value = None)
    xsec_dict : dict
        Dictionary of cross-section specifications. It should be of the
        form {'xsec_name':{'axis':(1,0,0),'distance':0.}}, where the axis parameter is a tuple
        defining the axis that defines the normal of the cross section, and distance is
        the length along the axis used to set the cross section.
        (Default value = None)
    serialized_input_file : bytes
        FreeCAD template file that has been serialized using
        qmt.infrastructure.serialize_file. This is useful for passing a
        file into a docker container or other environment that
        doesn't have access to a shared drive. Either this or
        input_file must be set (but not both).
        (Default value = None)
    params : dict
        Dictionary of parameters to use in FreeCAD.
        (Default value = None)
    Returns
    -------
    Geo3DData instance

    """
    if input_file is None and serialized_input_file is None:
        raise ValueError(
            "One of input_file or serialized_input_file must be non-none.")
    elif input_file is not None and serialized_input_file is not None:
        raise ValueError(
            "Both input_file and serialized_input_file were non-none.")
    elif input_file is not None:
        serial_fcdoc = serialize_file(input_file)
    else:
        serial_fcdoc = serialized_input_file
    if params is None:
        params = {}
    if xsec_dict is None:
        xsec_dict = {}
    options_dict = {}
    options_dict["serial_fcdoc"] = serial_fcdoc
    options_dict["input_parts"] = input_parts
    options_dict["params"] = params
    options_dict["xsec_dict"] = xsec_dict

    data = Geo3DData()
    data.serial_fcdoc = serial_fcdoc
    data.get_data("fcdoc")

    try:
        built = build(options_dict)
    except Exception:
        FreeCAD.closeDocument("instance")
        raise
    FreeCAD.closeDocument("instance")
    return built