forked from mxgmn/WaveFunctionCollapse
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTraining.cs
More file actions
171 lines (159 loc) · 5.14 KB
/
Training.cs
File metadata and controls
171 lines (159 loc) · 5.14 KB
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
using System;
using System.IO;
using UnityEngine;
#if UNITY_EDITOR
using UnityEditor;
#endif
using System.Collections.Generic;
class Training : MonoBehaviour{
public int gridsize = 1;
public int width = 12;
public int depth = 12;
public UnityEngine.Object[] tiles = new UnityEngine.Object[0];
public int[] RS = new int[0];
public Dictionary<string, byte> str_tile;
Dictionary<string, int[]> neighbors;
public byte[,] sample;
public static byte Get2DByte(byte[,] ar, int x, int y){
return ar[x, y];
}
public int Card(int n){
return (n%4 + 4)%4;
}
public void RecordNeighbors() {
neighbors = new Dictionary<string, int[]>();
for (int y = 0; y < depth-1; y++){
for (int x = 0; x < width-1; x++){
for (int r = 0; r < 2; r++){
int idx = (int)sample[x, y];
int rot = Card(RS[idx] + r);
int ridx = (int)sample[x+1-r, y+r];
int rrot = Card(RS[ridx] + r);
string key = ""+idx+"."+rot+"|"+ridx+"."+rrot;
if (!neighbors.ContainsKey(key) && tiles[idx] && tiles[ridx]){
neighbors.Add(key, new int[] {idx, rot, ridx, rrot});
Debug.DrawLine(
transform.TransformPoint(new Vector3((x+0f)*gridsize, 1f, (y+0f)*gridsize)),
transform.TransformPoint(new Vector3((x+1f-r)*gridsize, 1f, (y+0f+r)*gridsize)), Color.red, 9.0f, false);
}
}
}
}
System.IO.File.WriteAllText(Application.dataPath+"/"+this.gameObject.name+".xml", NeighborXML());
}
public string AssetPath(UnityEngine.Object o){
return AssetDatabase.GetAssetPath(o).Trim().Replace("Assets/Resources/", "").Replace(".prefab", "");
}
public string NeighborXML(){
Dictionary<UnityEngine.Object,int> counts = new Dictionary<UnityEngine.Object,int>();
string res = "<set>\n <tiles>\n";
for (int i = 0; i < tiles.Length; i++){
UnityEngine.Object o = tiles[i];
if (o && !counts.ContainsKey(o)){
counts[o] = 1;
string assetpath = AssetPath(o);
string sym = "X";
string last = assetpath.Substring(o.name.Length - 1);
if (last == "X" || last == "I" || last == "L" || last == "T" || last == "/"){
sym = last;
}
res += "<tile name=\""+assetpath+"\" symmetry=\""+sym+"\" weight=\"1.0\"/>\n";
}
}
res += " </tiles>\n<neighbors>";
Dictionary<string, int[]>.ValueCollection v = neighbors.Values;
foreach( int[] link in v ) {
res += " <neighbor left=\""+AssetPath(tiles[link[0]])+" "+link[1]+
"\" right=\""+AssetPath(tiles[link[2]])+" "+link[3]+"\"/>\n";
}
return res + " </neighbors>\n</set>";
}
public bool hasWhitespace(){
byte ws = (byte)0;
for (int y = 0; y < depth-1; y++){
for (int x = 0; x < width-1; x++){
if (sample[x, y] == ws){
return true;
}
}
}
return false;
}
public void Compile() {
str_tile = new Dictionary<string, byte>();
sample = new byte[width, depth];
int cnt = this.transform.childCount;
tiles = new UnityEngine.Object[500];
RS = new int[500];
tiles[0] = null;
RS[0] = 0;
for (int i = 0; i < cnt; i++){
GameObject tile = this.transform.GetChild(i).gameObject;
Vector3 tilepos = tile.transform.localPosition;
if ((tilepos.x > -0.55f) && (tilepos.x <= width*gridsize-0.55f) &&
(tilepos.z > -0.55f) && (tilepos.z <= depth*gridsize-0.55f)){
UnityEngine.Object fab = tile;
#if UNITY_EDITOR
fab = PrefabUtility.GetPrefabParent(tile);
if (fab == null){
PrefabUtility.ReconnectToLastPrefab(tile);
fab = PrefabUtility.GetPrefabParent(tile);
}
if (fab == null){
fab = (GameObject)Resources.Load(tile.name);
if (fab){
tile = PrefabUtility.ConnectGameObjectToPrefab(tile, (GameObject)fab);
}else{
fab = tile;}
}
tile.name = fab.name;
#endif
int X = (int)(tilepos.x) / gridsize;
int Y = (int)(tilepos.z) / gridsize;
int R = (int)(tile.transform.localEulerAngles.y)/ 90;
if (R == 4) {R = 0;};
if (!str_tile.ContainsKey(fab.name+R)){
int index = str_tile.Count+1;
str_tile.Add(fab.name+R, (byte)index);
tiles[index] = fab;
RS[index] = R;
sample[X, Y] = str_tile[fab.name+R];
} else {
sample[X, Y] = str_tile[fab.name+R];
}
}
}
tiles = tiles.SubArray(0 , str_tile.Count+1);
RS = RS.SubArray(0 , str_tile.Count+1);
}
void OnDrawGizmos(){
Gizmos.matrix = transform.localToWorldMatrix;
Gizmos.color = Color.magenta;
Gizmos.DrawWireCube(new Vector3((width*gridsize/2f)-gridsize*0.5f, 0, (depth*gridsize/2f)-gridsize*0.5f),
new Vector3(width*gridsize, gridsize, depth*gridsize));
Gizmos.color = Color.cyan;
for (int i = 0; i < this.transform.childCount; i++){
GameObject tile = this.transform.GetChild(i).gameObject;
Vector3 tilepos = tile.transform.localPosition;
if ((tilepos.x > -0.55f) && (tilepos.x <= width*gridsize-0.55f) &&
(tilepos.z > -0.55f) && (tilepos.z <= depth*gridsize-0.55f)){
Gizmos.DrawSphere(tilepos, gridsize*0.2f);
}
}
}
}
#if UNITY_EDITOR
[CustomEditor (typeof(Training))]
public class TrainingEditor : Editor {
public override void OnInspectorGUI () {
Training training = (Training)target;
if(GUILayout.Button("compile")){
training.Compile();
}
if(GUILayout.Button("record neighbors")){
training.RecordNeighbors();
}
DrawDefaultInspector ();
}
}
#endif