| import pathlib,zipfile,subprocess,os,urllib.request |
| root=pathlib.Path(__file__).resolve().parent |
| cache=pathlib.Path(os.environ.get('GRADLE_USER_HOME','/workspace/cache/gradle'))/'caches/modules-2/files-2.1' |
| java=pathlib.Path(os.environ.get('JAVA_HOME','/workspace/toolchains/jdk-21')) |
| android=pathlib.Path(os.environ.get('ANDROID_SDK_ROOT','/workspace/toolchains/android-sdk')) |
| kotlin=pathlib.Path(os.environ.get('KOTLIN_HOME','/workspace/toolchains/kotlin-2.2.10')) |
| abi=os.environ.get('VDS_ANDROID_ABI','x86_64') |
| env={**os.environ,'JAVA_HOME':str(java),'PATH':str(java/'bin')+':'+os.environ['PATH']} |
| jars=[] |
| dependencies=[('org.tensorflow','tensorflow-lite','2.16.1'),('org.tensorflow','tensorflow-lite-api','2.16.1'),('org.tensorflow','tensorflow-lite-select-tf-ops','2.16.1')] |
| for group,name,version in dependencies: |
| matches=list((cache/group/name/version).glob('*/*.aar')) |
| aar=matches[0] if matches else root/(name+'-'+version+'.aar') |
| if not aar.exists():urllib.request.urlretrieve(f'https://repo.maven.apache.org/maven2/{group.replace(".","/")}/{name}/{version}/{name}-{version}.aar',aar) |
| with zipfile.ZipFile(aar) as z: |
| jar=root/(name+'.jar');jar.write_bytes(z.read('classes.jar'));jars.append(str(jar)) |
| for item in z.namelist(): |
| if item.startswith('jni/'+abi+'/') and item.endswith('.so'): |
| lib=root/'lib'/pathlib.Path(item).name;lib.parent.mkdir(exist_ok=True);lib.write_bytes(z.read(item)) |
| classes=root/'classes';classes.mkdir(exist_ok=True) |
| sdk=root/'on-device-learning.jar' |
| subprocess.run([str(kotlin/'bin/kotlinc'),*[str(p) for p in root.glob('*.kt')],'-cp',':'.join([str(android/'platforms/android-36/android.jar'),*jars]),'-d',str(sdk)],check=True,env=env) |
| jars.extend([str(sdk),str(kotlin/'lib/kotlin-stdlib.jar')]) |
| subprocess.run([str(java/'bin/javac'),'-source','11','-target','11','-cp',':'.join(jars),'-d',str(classes),*[str(p) for p in root.glob('*.java')]],check=True,env=env) |
| subprocess.run([str(android/'build-tools/36.0.0/d8'),'--min-api','28','--output',str(root/'training-check.jar'),*[str(p) for p in classes.glob('*.class')],*jars],check=True,env=env) |
| print(root/'training-check.jar') |
|
|
