예제 #1
0
파일: machines.py 프로젝트: Axford/AFRo
def machines():
    print("Machines")
    print("--------")

    temp_name =  "temp.scad"

    #
    # Make the target directories
    #
    view_dir = "../images"
    if not os.path.isdir(view_dir):
        os.makedirs(view_dir)

    # load hardware.json
    jf = open("hardware.json","r")
    jso = json.load(jf)
    jf.close()
    
    # for each machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine':
            print(m['title'])
            
            fn = '../' + m['file']
            
            if (os.path.isfile(fn)):
            
                print("  Checking csg hash")
                h = openscad.get_csg_hash_for(fn);
                
                hashchanged = ('hash' in m and h != m['hash']) or (not 'hash' in m)
                
                # update hash in json
                m['hash'] = h
                    
                
                # Views
                print("  Views")
                for c in m['children']:
                    if type(c) is DictType and c['type'] == 'view':
                        view = c
                        print("    "+view['title'])
                    
                        render_view_using_file(m['title'], fn, view_dir, view, hashchanged, h)
                    
                    
            else:
                print("    Error: scad file not found: "+v['file'])
                        
            
    # Save changes to json
    with open('hardware.json', 'w') as f:
        f.write(json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))
        
    return 0
예제 #2
0
def machines():
    print("Machines")
    print("--------")

    temp_name = "temp.scad"

    #
    # Make the target directories
    #
    view_dir = "../images"
    if not os.path.isdir(view_dir):
        os.makedirs(view_dir)

    # load hardware.json
    jf = open("hardware.json", "r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine':
            print(m['title'])

            fn = '../' + m['file']

            if (os.path.isfile(fn)):

                print("  Checking csg hash")
                h = openscad.get_csg_hash_for(fn)

                hashchanged = ('hash' in m
                               and h != m['hash']) or (not 'hash' in m)

                # update hash in json
                m['hash'] = h

                # Views
                print("  Views")
                for c in m['children']:
                    if type(c) is DictType and c['type'] == 'view':
                        view = c
                        print("    " + view['title'])

                        render_view_using_file(m['title'], fn, view_dir, view,
                                               hashchanged, h)

            else:
                print("    Error: scad file not found: " + v['file'])

    # Save changes to json
    with open('hardware.json', 'w') as f:
        f.write(
            json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0
예제 #3
0
def machines():
    print("Machines")
    print("--------")

    #
    # Make the target directories
    #
    if not os.path.isdir(config.paths['machineimages']):
        os.makedirs(config.paths['machineimages'])

    # load hardware.json
    jf = open(config.paths['json'],"r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if isinstance(m, dict) and m['type'] == 'machine':
            print(m['title'])

            fn = config.paths['root'] + m['file']

            if (os.path.isfile(fn)):

                print("  Checking csg hash")
                h = openscad.get_csg_hash_for(fn);

                hashchanged = ('hash' in m and h != m['hash']) or (not 'hash' in m)

                # update hash in json
                m['hash'] = h


                # Views
                print("  Views")
                for c in m['children']:
                    if isinstance(c, dict) and c['type'] == 'view':
                        view = c
                        print("    "+view['title'])

                        render_view_using_file(m['title'], fn, config.paths['machineimages'], view, hashchanged, h)


            else:
                print("    Error: scad file not found: "+v['file'])


    # Save changes to json
    with open(config.paths['json'], 'w') as f:
        f.write(json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0
예제 #4
0
파일: cut.py 프로젝트: swindonmakers/SCOB
def cut():
    print("Cut Parts")
    print("---------")

    temp_name = "temp.scad"

    # load hardware.json
    jf = open("hardware.json", "r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine':
            print(m['title'])

            al = m['cut']

            # make target directory
            view_dir = "../cutparts/images"
            if not os.path.isdir(view_dir):
                os.makedirs(view_dir)

            for a in al:
                print("  " + a['title'])
                fn = '../' + a['file']
                if (os.path.isfile(fn)):

                    print("    Checking csg hash")
                    h = openscad.get_csg_hash(temp_name, a['call'])
                    os.remove(temp_name)

                    hashchanged = ('hash' in a
                                   and h != a['hash']) or (not 'hash' in a)

                    # update hash in json
                    a['hash'] = h

                    # Generate completed part file
                    f = open(temp_name, "w")
                    f.write("include <../config/config.scad>\n")
                    f.write("DebugConnectors = false;\n")
                    f.write("DebugCoordinateFrames = false;\n")
                    f.write(a['completeCall'] + ";\n")
                    f.close()

                    # Completed views
                    for view in a['views']:
                        render_view_using_file(a['title'], temp_name, view_dir,
                                               view, hashchanged, h)

                    # Fabrication Steps
                    for step in a['steps']:
                        # Generate step file
                        f = open(temp_name, "w")
                        f.write("include <../config/config.scad>\n")
                        f.write("DebugConnectors = false;\n")
                        f.write("DebugCoordinateFrames = false;\n")
                        f.write("$Explode = true;\n")
                        f.write("$ShowStep = " + str(step['num']) + ";\n")
                        f.write(a['call'] + ";\n")
                        f.close()

                        # Views
                        print("      Step " + str(step['num']))
                        for view in step['views']:
                            render_view_using_file(
                                a['title'] + '_step' + str(step['num']),
                                temp_name, view_dir, view, hashchanged, h)

                else:
                    print("    Error: scad file not found: " + a['file'])

    # Save changes to json
    with open('hardware.json', 'w') as f:
        f.write(
            json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0
예제 #5
0
def assemblies():
    print("Assemblies")
    print("----------")

    temp_name =  "temp.scad"

    # load hardware.json
    jf = open("hardware.json","r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine':
            print(m['title'])

            al = m['assemblies']

            # make target directory
            view_dir = "../assemblies/"+machine_dir(m['title'])
            if not os.path.isdir(view_dir):
                os.makedirs(view_dir)

            # for each assembly
            for a in al:
                print("  "+a['title'])
                fn = '../' + a['file']
                if (os.path.isfile(fn)):

                    print("    Checking csg hash")
                    h = openscad.get_csg_hash(temp_name, a['call']);
                    os.remove(temp_name);

                    hashchanged = ('hash' in a and h != a['hash']) or (not 'hash' in a)

                    # update hash in json
                    a['hash'] = h

                    # Steps
                    for step in a['steps']:
                        # Generate step file
                        f = open(temp_name, "w")
                        f.write("include <../config/config.scad>\n")
                        f.write("DebugConnectors = false;\n");
                        f.write("DebugCoordinateFrames = false;\n");
                        f.write("UseSTL=true;\n");
                        f.write("$Explode = true;\n");
                        f.write("$ShowStep = "+ str(step['num']) +";\n");
                        f.write(a['call'] + ";\n");
                        f.close()

                        # Views
                        print("      Step "+str(step['num']))
                        for view in step['views']:
                            render_view_using_file(a['title']+'_step'+str(step['num']), temp_name, view_dir, view, hashchanged, h)

                    # for each animation
                    for anim in a['animations']:
                        print("    Animation: "+anim['title']+", framesPerStep="+str(anim['framesPerStep']))

                        animateAssembly(m['title'], a['title'], anim['title'], anim['framesPerStep'])

                else:
                    print("    Error: scad file not found: "+a['file'])


    # Save changes to json
    with open('hardware.json', 'w') as f:
        f.write(json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0
예제 #6
0
def animateAssembly(mname, aname, prefix, framesPerStep):
    print("Animate Assembly")
    print("----------------")

    print("NB: Make sure you've run parse.py first!")
    print("")

    temp_name = "temp.scad"

    # load hardware.json
    jf = open("hardware.json", "r")
    jso = json.load(jf)
    jf.close()

    # locate required machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine' and m[
                'title'] == mname:
            print("Found machine: " + m['title'])

            al = m['assemblies']

            # make target directory
            view_dir = "../assemblies/" + machine_dir(m['title'])
            if not os.path.isdir(view_dir):
                os.makedirs(view_dir)

            # locate required assembly
            for a in al:
                if a['title'] == aname:
                    print("Found assembly: " + a['title'])
                    fn = '../' + a['file']
                    if (os.path.isfile(fn)):

                        print("  Checking csg hash")
                        h = openscad.get_csg_hash(temp_name, a['call'])
                        os.remove(temp_name)

                        hashchanged = ('hash' in a
                                       and h != a['hash']) or (not 'hash' in a)

                        numSteps = 0
                        frameNum = 0
                        lfn = ""  # last frame filename

                        # Calc number of steps, and grab first view
                        view = {
                            'size': [400, 300],
                            'dist': 140,
                            'rotate': [0, 0, 0],
                            'translate': [0, 0, 0],
                            'title': 'view'
                        }
                        firstView = True
                        for step in a['steps']:
                            print("Step: " + str(step['num']))

                            # generate a transition move?
                            if len(step['views']) > 0:
                                # see if new view is diff from current
                                nv = step['views'][0]

                                if firstView:
                                    view['size'] = nv['size']

                                if ((nv['size'] != view['size']) or
                                    (nv['dist'] != view['dist']) or
                                    (nv['rotate'] != view['rotate']) or
                                    (nv['translate'] !=
                                     view['translate'])) and (not firstView):
                                    print("Generating transition move...")

                                    # prep tween view
                                    tv = copy.deepcopy(view)

                                    # iterate over frames
                                    for frame in range(0, framesPerStep):
                                        t = easeInOut(frame /
                                                      (framesPerStep - 1.0))
                                        # show previous step during transition
                                        ShowStep = step['num'] - 1

                                        # tween between view and nv
                                        tv['dist'] = mapRange(
                                            t, 0, 1.0, view['dist'],
                                            nv['dist'])
                                        tv['translate'][0] = mapRange(
                                            t, 0, 1.0, view['translate'][0],
                                            nv['translate'][0])
                                        tv['translate'][1] = mapRange(
                                            t, 0, 1.0, view['translate'][1],
                                            nv['translate'][1])
                                        tv['translate'][2] = mapRange(
                                            t, 0, 1.0, view['translate'][2],
                                            nv['translate'][2])
                                        tv['rotate'][0] = mapRange(
                                            t, 0, 1.0, view['rotate'][0],
                                            nv['rotate'][0])
                                        tv['rotate'][1] = mapRange(
                                            t, 0, 1.0, view['rotate'][1],
                                            nv['rotate'][1])
                                        tv['rotate'][2] = mapRange(
                                            t, 0, 1.0, view['rotate'][2],
                                            nv['rotate'][2])

                                        #print("t: "+str(t) +", s: "+str(ShowStep)+", a:"+str(AnimateExplodeT))

                                        # Generate step file
                                        f = open(temp_name, "w")
                                        f.write(
                                            "include <../config/config.scad>\n"
                                        )
                                        f.write("DebugConnectors = false;\n")
                                        f.write(
                                            "DebugCoordinateFrames = false;\n")
                                        f.write("$Explode = false;\n")
                                        f.write("UseSTL=true;\n")
                                        f.write("$AnimateExplode = false;\n")
                                        f.write("$ShowStep = " +
                                                str(ShowStep) + ";\n")
                                        f.write(a['call'] + ";\n")
                                        f.close()

                                        # Views
                                        views.PolishTransparentBackground = False
                                        views.PolishCrop = False
                                        lfn = view_dir + "/" + prefix + format(
                                            frameNum, '03'
                                        ) + "_" + view['title'] + ".png"
                                        views.render_view_using_file(
                                            prefix + format(frameNum, '03'),
                                            temp_name, view_dir, tv,
                                            hashchanged, h)
                                        frameNum = frameNum + 1

                                view['dist'] = nv['dist']
                                view['translate'] = nv['translate']
                                view['rotate'] = nv['rotate']
                                firstView = False

                            # iterate over frames to animate the assembly step
                            for frame in range(0, framesPerStep):

                                t = easeInOut(frame / (framesPerStep - 1.0))
                                ShowStep = step['num']
                                AnimateExplodeT = t

                                #print("t: "+str(t) +", s: "+str(ShowStep)+", a:"+str(AnimateExplodeT))

                                # Generate step file
                                f = open(temp_name, "w")
                                f.write("include <../config/config.scad>\n")
                                f.write("DebugConnectors = false;\n")
                                f.write("DebugCoordinateFrames = false;\n")
                                f.write("$Explode = true;\n")
                                f.write("UseSTL=true;\n")
                                f.write("$AnimateExplode = true;\n")
                                f.write("$ShowStep = " + str(ShowStep) + ";\n")
                                f.write("$AnimateExplodeT = " +
                                        str(AnimateExplodeT) + ";\n")
                                f.write(a['call'] + ";\n")
                                f.close()

                                # Views
                                views.PolishTransparentBackground = False
                                views.PolishCrop = False
                                fn = view_dir + "/" + prefix + format(
                                    frameNum,
                                    '03') + "_" + view['title'] + ".png"
                                print("Rendering: " + fn)
                                views.render_view_using_file(
                                    prefix + format(frameNum, '03'), temp_name,
                                    view_dir, view, hashchanged, h)
                                frameNum = frameNum + 1

                                if frame == 0:
                                    # load previous image for cross-fade
                                    if lfn != "":
                                        imgPrev = Image.open(lfn)

                                    # Annotate and extend first frame with step info
                                    img = Image.open(fn)
                                    imgNew = img.copy()

                                    # go back one to overwrite first frame
                                    frameNum = frameNum - 1

                                    # extend a bit
                                    for extra in range(0, 2 * framesPerStep):
                                        if lfn != "":
                                            alpha = extra / (1.0 *
                                                             framesPerStep)
                                            if (alpha > 1.0):
                                                alpha = 1.0
                                            print("Blending: " + str(alpha) +
                                                  ", " + str(frameNum))
                                            imgNew = Image.blend(
                                                imgPrev, img, alpha)

                                        draw = ImageDraw.Draw(imgNew, "RGBA")
                                        font = ImageFont.truetype(
                                            "Tahoma.ttf", 12)
                                        draw.rectangle(
                                            (0, 0, view['size'][0], 30),
                                            fill=(240, 240, 240))
                                        margin = 30
                                        offset = 10
                                        for line in textwrap.wrap(
                                                step['desc'],
                                                width=((view['size'][0] -
                                                        2 * margin) /
                                                       (draw.textsize(
                                                           "abcdefghij")[0] /
                                                        10))):
                                            draw.rectangle(
                                                (0, offset, view['size'][0],
                                                 offset + 20),
                                                fill=(240, 240, 240))
                                            draw.rectangle(
                                                (0, offset + 20,
                                                 view['size'][0], offset + 22),
                                                fill=(200, 200, 200))
                                            draw.text((margin, offset),
                                                      line, (0, 0, 0),
                                                      font=font)
                                            offset += font.getsize(line)[1]

                                        # draw step number
                                        draw.ellipse((5, 5, 25, 25),
                                                     fill=(255, 140, 50),
                                                     outline=(255, 140, 50))
                                        draw.text((11, 8),
                                                  str(step['num']),
                                                  (255, 255, 255),
                                                  font=font)

                                        fn = view_dir + "/" + prefix + format(
                                            frameNum, '03'
                                        ) + "_" + view['title'] + ".png"
                                        frameNum = frameNum + 1
                                        print("Saving: " + fn)
                                        imgNew.save(fn)

                                lfn = fn

                        # final turntable, using last view as starting point
                        print("Generating final turntable")
                        tv = copy.deepcopy(view)
                        for frame in range(0, framesPerStep * 2):
                            t = easeInOut(frame /
                                          ((framesPerStep * 2.0) - 1.0))
                            ShowStep = 100

                            # tween between view and nv
                            r = mapRange(t, 0, 1.0, 0, 360)
                            tv['rotate'][2] = view['rotate'][2] + r
                            if tv['rotate'][2] > 360:
                                tv['rotate'][2] = tv['rotate'][2] - 360

                            #print("t: "+str(t) +", r:"+str(r))

                            # Generate step file
                            f = open(temp_name, "w")
                            f.write("include <../config/config.scad>\n")
                            f.write("DebugConnectors = false;\n")
                            f.write("DebugCoordinateFrames = false;\n")
                            f.write("UseSTL=true;\n")
                            f.write("$Explode = false;\n")
                            f.write("$ShowStep = " + str(ShowStep) + ";\n")
                            f.write(a['call'] + ";\n")
                            f.close()

                            # Views
                            views.PolishTransparentBackground = False
                            views.PolishCrop = False
                            views.render_view_using_file(
                                prefix + format(frameNum, '03'), temp_name,
                                view_dir, tv, hashchanged, h)
                            frameNum = frameNum + 1

                        numFrames = frameNum

                        # build video
                        cmd = "ffmpeg -r " + str(
                            framesPerStep
                        ) + " -y -i " + view_dir + "/" + prefix + "%03d_" + view[
                            'title'] + ".png -vcodec libx264 -pix_fmt yuv420p " + view_dir + "/" + prefix + ".mp4"
                        print("Encoding video with: " + cmd)
                        os.system(cmd)

                        # build animated gif
                        cmd = "convert " + view_dir + "/" + prefix + ".mp4 " + view_dir + "/" + prefix + ".gif"
                        print("Creating GIF with: " + cmd)
                        os.system(cmd)

                        # clean up temporary images
                        #for frame in range(0, numFrames):
                        #    os.remove(view_dir + "/" +prefix + format(frame, '03') + "_" +view['title']+".png");

                        print("Done")

                        try:
                            if sys.platform == "darwin":
                                subprocess.check_output([
                                    'osascript', '-e',
                                    'display notification "Animation Complete" with title "Animation"'
                                ])
                        except:
                            print("Exception running osascript")

                    else:
                        print("    Error: scad file not found: " + a['file'])

    return 0
예제 #7
0
def assemblies():
    print("Assemblies")
    print("----------")

    # load hardware.json
    jf = open(config.paths['json'], "r")
    jso = json.load(jf)
    jf.close()

    # for each machine
    for m in jso:
        if type(m) is DictType and m['type'] == 'machine':
            print(m['title'])

            al = m['assemblies']

            # make target directory
            view_dir = config.paths['assemblies'] + "/" + machine_dir(
                m['title'])
            if not os.path.isdir(view_dir):
                os.makedirs(view_dir)

            # for each assembly
            for a in al:
                print("  " + a['title'])
                fn = config.paths['root'] + a['file']
                if (os.path.isfile(fn)):

                    print("    Checking csg hash")
                    h = openscad.get_csg_hash(config.paths['tempscad'],
                                              a['call'])
                    os.remove(config.paths['tempscad'])

                    hashchanged = ('hash' in a
                                   and h != a['hash']) or (not 'hash' in a)

                    # update hash in json
                    a['hash'] = h

                    # Steps
                    for step in a['steps']:
                        # Generate step file
                        f = open(config.paths['tempscad'], "w")
                        f.write("include <../config/config.scad>\n")
                        f.write("DebugConnectors = false;\n")
                        f.write("DebugCoordinateFrames = false;\n")
                        f.write("UseSTL=true;\n")
                        f.write("$Explode = true;\n")
                        f.write("$ShowStep = " + str(step['num']) + ";\n")
                        f.write(a['call'] + ";\n")
                        f.close()

                        # Views
                        print("      Step " + str(step['num']))
                        for view in step['views']:
                            render_view_using_file(
                                a['title'] + '_step' + str(step['num']),
                                config.paths['tempscad'], view_dir, view,
                                hashchanged, h)

                    # for each animation
                    for anim in a['animations']:
                        print("    Animation: " + anim['title'] +
                              ", framesPerStep=" + str(anim['framesPerStep']))

                        animateAssembly(m['title'], a['title'], anim['title'],
                                        anim['framesPerStep'])

                else:
                    print("    Error: scad file not found: " + a['file'])

    # Save changes to json
    with open(config.paths['json'], 'w') as f:
        f.write(
            json.dumps(jso, sort_keys=False, indent=4, separators=(',', ': ')))

    return 0