import math
import argparse

#input config file format: x,y,z,id,filename,angleoffset
#angleoffset is 0 when center of 360 image points to the right, positive rotation means image center in pointing more CCW. value in degrees

parser = argparse.ArgumentParser(
                    prog = 'File Edit Script',
                    description = 'Read content of text file and write to file',
                    epilog = '')

parser.add_argument('filename')           # positional argument
parser.add_argument('-o', '--output')      # option that takes a value
parser.add_argument('-v', '--verbose', action='store_true')  # on/off flag

args = parser.parse_args()
print(args.filename, args.output, args.verbose)
    
def distance(x1,y1,x2,y2):  
    dist = math.sqrt((x2 - x1)**2 + (y2 - y1)**2)  
    return dist  
    

print("Loading Templates")
template_panellum=None
with open("template_panellum.txt",'r') as rf:
    template_panellum=rf.readlines()

template_scenes=None
with open("template_scenes.txt",'r') as rf:
    template_scenes=rf.readlines()

template_hotspots=None
with open("template_hotspots.txt",'r') as rf:
    template_hotspots=rf.readlines()
    
   
linesinput=None
with open(args.filename,'r') as rf:
    linesinput=rf.readlines()
    
linesinput=[x.rstrip() for x in linesinput] #remove newline and whitespaces at end
    
output_scenes=[]

first_sceneId=None
 
for iline1,line1 in enumerate(linesinput):

    x1,y1,z1,id1,filename1,angleoffset1=line1.split(',')
    x1,y1,z1=(float(x1),float(y1),float(z1))
    angleoffset1=float(angleoffset1)

    if first_sceneId is None:
        first_sceneId=id1 #take first entry as start scene

    # ### Set information for scene ###
    current_scene=template_scenes
    current_scene=[x.replace('<id>',id1) for x in current_scene]
    current_scene=[x.replace('<title>',id1) for x in current_scene]
    current_scene=[x.replace('<filename>',filename1) for x in current_scene]

    output_hotspots=[]

    for iline2,line2 in enumerate(linesinput):
        x2,y2,z2,id2,filename2,angleoffset2=line2.split(',')
        x2,y2,z2=(float(x2),float(y2),float(z2))
        angleoffset2=float(angleoffset2)

        

        if id1!=id2:
            print()
            print("from "+str(x1)+", "+str(y1)+" : "+id1)
            print("to   "+str(x2)+", "+str(y2)+" : "+id2)
            
            print("dist="+str(distance(x1,y1,x2,y2)))
            angle=math.degrees(math.atan2(y2-y1, x2-x1)) #in degrees
            print("angle="+str(angle)+" offset1="+str(angleoffset1)+" offset2="+str(angleoffset2))
            angle+=angleoffset1 #correct hotspot position by rotation of scene image
            print("angle final="+str(angle))

            pitch=-math.degrees(math.atan(z1/distance(x1,y1,x2,y2))) #from current looking to hotspot. 0deg=straight, 90deg=down
            pitch=pitch/2 #manual correction
            

            # ### Set information for target/hotspot ###
            current_hotspot=template_hotspots
            current_hotspot=[x.replace('<id>',id2) for x in current_hotspot]
            current_hotspot=[x.replace('<yaw>',str(angle)) for x in current_hotspot]
            current_hotspot=[x.replace('<pitch>',str(pitch)) for x in current_hotspot]

            output_hotspots.append(current_hotspot)

        
    output_hotspots_string=""
    for _ihotspot,_hotspot in enumerate(output_hotspots):
        if _ihotspot>0:
            output_hotspots_string+=",\n" #separate blocks with comma
        for _hotspotline in _hotspot:
            output_hotspots_string+=_hotspotline

    current_scene=[x.replace('<hotspot>',output_hotspots_string) for x in current_scene]

    output_scenes.append(current_scene)




output_scenes_string=""
for _iscene,_scene in enumerate(output_scenes):
    if _iscene>0:
        output_scenes_string+=",\n" #separate blocks with comma
    for _sceneline in _scene:
        output_scenes_string+=_sceneline

output=template_panellum
output=[x.replace('<firstScene>',first_sceneId) for x in output]
output=[x.replace('<scene>',output_scenes_string) for x in output]
    
with open(args.output, 'w') as wf:
    wf.writelines(output)