astar_pathfinding_node_networks/extract_canvas.py at main · slyautomation/astar_pathfinding_node_networks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import base64
import time
import requests
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import os
from PIL import Image
from PIL import ImageDraw


def test_url():
    options = Options()
    user = "i7 8700" # change this to your windows username
    options.add_argument("user-data-dir=C:\\Users\\" + user "\\AppData\\Local\\Google\\Chrome\\User Data\\Profile 1")
    driver = webdriver.Chrome(executable_path=r"C:\Users\\" + user + "\\Downloads\chromedriver.exe", chrome_options=options)

    driver.get("http://curran.github.io/HTML5Examples/canvas/smileyFace.html")
    canvas = driver.find_element_by_css_selector("#canvas")
    # get the canvas as a PNG base64 string
    canvas_base64 = driver.execute_script("return arguments[0].toDataURL('image/png').substring(21);", canvas)
    # decode
    canvas_png = base64.b64decode(canvas_base64)
    # save to a file
    with open(r"canvas.png", 'wb') as f:
        f.write(canvas_png)

def osrs_canvas_scrape():
    options = Options()
    user = "i7 8700" # change this to your windows username
    options.add_argument("user-data-dir=C:\\Users\\" + user "\\AppData\\Local\\Google\\Chrome\\User Data\\Profile 1")
    driver = webdriver.Chrome(executable_path=r"C:\Users\\" + user + "\\Downloads\chromedriver.exe", chrome_options=options)
    driver.set_window_size(10000, 10000)
    driver.get("https://www.osrsmap.net/")
    canvas = driver.find_element_by_id("map")
    time.sleep(1.5)
    driver.find_element_by_id("show-labels").click()
    # get the canvas as a PNG base64 string
    time.sleep(1.5)
    x = 1000
    i = 1
    while x < 4500:
        y = 2500
        b = 1
        while y < 4500:
            driver.get("https://www.osrsmap.net/#area=main&x=" + str(x) + "&y=" + str(y) + "&zoom=100")
            time.sleep(0.1)
            canvas_base64 = driver.execute_script("return arguments[0].toDataURL('image/png').substring(21);", canvas)
            # decode
            canvas_png = base64.b64decode(canvas_base64)
            # save to a file
            with open(r"images/osrs_map_canvas_" + str(i) + "_" + str(b) + ".png", 'wb') as f:
                f.write(canvas_png)
            y += 200
            b += 1
        x += 600
        i += 1


print('*** Program Started ***')
image_name_output = '02_create_blank_image_01.png'
mode = 'RGB' # for color image “L” (luminance) for greyscale images, “RGB” for true color images, and “CMYK” for pre-press images.
size = (12000, 7250)
color = (0, 0, 0)
im = Image.new(mode, size, color)
im.save(image_name_output )

#im.show()


img0 = Image.open(r"02_create_blank_image_01.png")

x = 2
x_pos = 0
while x < 6:

    y = 9
    y_pos = 0
    while y > 0:
        img = Image.open(r"images/osrs_map_canvas_" + str(x) + "_" + str(y) + ".png")
        img0.paste(img, (x_pos, y_pos))
        y_pos += 800
        print("x:", x, " | y:", y)
        y -= 1
    x_pos += 2400
    x += 1
print('*** Program Ended ***')
img0.save("osrs_map_merged.png")
#img0.show()

#test_url()
#osrs_canvas_scrape()