Mercurial > bus-service / annotate
geometry.py@60045b362d71 (annotated)
geometry.py
Wed, 28 Jun 2017 12:20:05 +0300
- author
- Teemu Piippo <teemu@hecknology.net>
- date
- Wed, 28 Jun 2017 12:20:05 +0300
- changeset 31
- 60045b362d71
- parent 19
- 16fa9fb20b32
- child 88
- 3b86597c5a88
- permissions
- -rw-r--r--
- Ajovuoroa ei enää esitetä kahdessa välilehdessä vaan puukuvaimessa
- gtfs.zip ei enää tarvitse avata
| 5 | 1 | from math import sqrt, hypot, radians, sin, cos, atan2 |
| 2 | ||
| 3 | class Sijainti: | |
| 4 | def __init__(self, leveys, pituus): | |
| 5 | self.leveys, self.pituus = leveys, pituus | |
| 6 | def etäisyys(tämä, toinen): | |
| 7 | # https://stackoverflow.com/a/365853 | |
| 8 | a = sin(radians(tämä.leveys - toinen.leveys) / 2) ** 2 | |
| 9 | a += sin(radians(tämä.pituus - toinen.pituus) / 2) ** 2 * cos(radians(tämä.leveys)) * cos(radians(toinen.leveys)) | |
| 10 | return 6371 * 2 * atan2(sqrt(a), sqrt(1 - a)) | |
| 11 | def __repr__(self): | |
| 12 | return '%s(%r, %r)' % (type(self).__name__, self.leveys, self.pituus) | |
| 13 | def __str__(self): | |
| 14 | return '%.5f, %.5f' % (self.leveys, self.pituus) | |
| 15 | @property | |
| 16 | def x(self): | |
| 17 | return self.leveys | |
| 18 | @property | |
| 19 | def y(self): | |
| 20 | return self.pituus | |
|
19
16fa9fb20b32
Lisätty pysäkkiryhmän aikataulunäkymä
Teemu Piippo <teemu@hecknology.net>
parents:
7
diff
changeset
|
21 | @property |
|
16fa9fb20b32
Lisätty pysäkkiryhmän aikataulunäkymä
Teemu Piippo <teemu@hecknology.net>
parents:
7
diff
changeset
|
22 | def link_to_map(self): |
|
16fa9fb20b32
Lisätty pysäkkiryhmän aikataulunäkymä
Teemu Piippo <teemu@hecknology.net>
parents:
7
diff
changeset
|
23 | return 'http://www.openstreetmap.org/#map=19/%f/%f' % (self.leveys, self.pituus) |
| 5 | 24 | |
| 25 | class Rengas: | |
| 26 | def __init__(tämä, säiliö): | |
| 27 | tämä.säiliö = säiliö | |
| 28 | def __getitem__(tämä, indeksi): | |
| 29 | while indeksi < 0: | |
| 30 | indeksi += len(tämä.säiliö) | |
| 31 | while indeksi >= len(tämä.säiliö): | |
| 32 | indeksi -= len(tämä.säiliö) | |
| 33 | return tämä.säiliö[indeksi] | |
| 34 | def __iter__(tämä): | |
| 35 | return iter(tämä.säiliö) | |
| 36 | def __len__(tämä): | |
| 37 | return len(tämä.säiliö) | |
| 38 | ||
| 39 | class Monikulmio: | |
| 40 | def __init__(self, *pisteet): | |
| 41 | self.pisteet = pisteet | |
| 42 | def __repr__(self): | |
| 43 | return '%s(%s)' % (type(self).__name__, ', '.join(map(repr, self.pisteet))) | |
| 44 | def pinta_ala(self): | |
| 45 | rengas = Rengas(self.pisteet) | |
| 46 | return sum( | |
| 47 | rengas[i].x * rengas[i + 1].y - rengas[i + 1].x * rengas[i].y | |
| 48 | for i in range(len(rengas)) | |
| 49 | ) / 2 | |
| 50 | def piiri(self): | |
| 51 | rengas = Rengas(self.pisteet) | |
| 52 | return sum( | |
| 53 | sqrt((rengas[i + 1].x - rengas[i].x)**2 + (rengas[i + 1].y - rengas[i].y)**2) | |
| 54 | for i in range(len(rengas)) | |
| 55 | ) | |
| 56 | def painopiste(self): | |
| 57 | rengas = Rengas(self.pisteet) | |
| 58 | x = sum( | |
| 59 | (rengas[i].x + rengas[i + 1].x) * (rengas[i].x * rengas[i + 1].y - rengas[i + 1].x * rengas[i].y) | |
| 60 | for i in range(len(rengas)) | |
| 61 | ) / 6 / self.pinta_ala() | |
| 62 | y = sum( | |
| 63 | (rengas[i].y + rengas[i + 1].y) * (rengas[i].x * rengas[i + 1].y - rengas[i + 1].x * rengas[i].y) | |
| 64 | for i in range(len(rengas)) | |
| 65 | ) / 6 / self.pinta_ala() | |
| 66 | return self.pistetyyppi()(x, y) | |
| 67 | def pistetyyppi(self): | |
| 68 | if len(self.pisteet): | |
| 69 | return type(self.pisteet[0]) | |
| 70 | else: | |
| 71 | return Piste | |
| 72 | def piiri_janat(self): | |
| 73 | rengas = Rengas(self.pisteet) | |
| 74 | for i in range(len(rengas)): | |
| 75 | yield Jana(rengas[i], rengas[i + 1]) | |
| 76 | def sisältää_pisteen(self, piste): | |
| 77 | ulkopiste = self.pistetyyppi()( | |
| 78 | min(piste.x for piste in self.pisteet) - 1, | |
| 79 | min(piste.y for piste in self.pisteet) - 1 | |
| 80 | ) | |
| 81 | ulkojana = Jana(piste, ulkopiste) | |
| 82 | leikkauksia = 0 | |
| 83 | for jana in self.piiri_janat(): | |
| 84 | if jana.leikkauspiste(ulkojana) is not None: | |
| 85 | leikkauksia += 1 | |
| 86 | return bool(leikkauksia & 1) | |
| 87 | ||
| 88 | class Jana: | |
| 89 | def __init__(self, alkupiste, päätepiste): | |
| 90 | self.alkupiste, self.päätepiste = alkupiste, päätepiste | |
| 91 | def __repr__(self): | |
| 92 | return 'Jana(%r, %r)' % (self.alkupiste, self.päätepiste) | |
| 93 | def pituus(self): | |
| 94 | return hypot(self.alkupiste.x - self.päätepiste.x, self.alkupiste.y - self.päätepiste.y) | |
| 95 | def leikkauspiste(tämä, toinen): | |
| 96 | pistetyyppi = type(tämä.alkupiste) | |
| 97 | x = (tämä.alkupiste.x, tämä.päätepiste.x, toinen.alkupiste.x, toinen.päätepiste.x) | |
| 98 | y = (tämä.alkupiste.y, tämä.päätepiste.y, toinen.alkupiste.y, toinen.päätepiste.y) | |
| 99 | try: | |
| 100 | jakaja = (x[0] - x[1]) * (y[2] - y[3]) - (y[0] - y[1]) * (x[2] - x[3]) | |
| 101 | Px = ((x[0] * y[1] - y[0] * x[1]) * (x[2] - x[3]) - (x[0] - x[1]) * (x[2] * y[3] - y[2] * x[3])) / jakaja | |
| 102 | Py = ((x[0] * y[1] - y[0] * x[1]) * (y[2] - y[3]) - (y[0] - y[1]) * (x[2] * y[3] - y[2] * x[3])) / jakaja | |
| 103 | etäisyys = lambda n: (Px - x[n]) ** 2 + (Py - y[n]) ** 2 | |
| 104 | if max(etäisyys(0), etäisyys(1)) <= tämä.pituus() ** 2 and max(etäisyys(2), etäisyys(3)) <= toinen.pituus() ** 2: | |
| 105 | return pistetyyppi(Px, Py) | |
| 106 | else: | |
| 107 | return None | |
| 108 | except ZeroDivisionError: | |
| 109 | return None | |
| 110 | ||
| 111 | class Piste: | |
| 112 | def __init__(self, x, y): | |
| 113 | self.x, self.y = x, y | |
| 114 | def __repr__(self): | |
| 115 | return 'Piste(%r, %r)' % (self.x, self.y) | |
| 116 | ||
| 117 | A = Monikulmio( | |
| 118 | Piste(2,3), | |
| 119 | Piste(1,1), | |
| 120 | Piste(4,0), | |
| 121 | Piste(6,2), | |
| 122 | Piste(4,4)) | |
| 123 | L1 = Jana(Piste(1, 1), Piste(-1, 5)) | |
| 124 | L2 = Jana(Piste(1, 5), Piste(5, 1)) |
