169 lines
4.0 KiB
Python
169 lines
4.0 KiB
Python
import os.path
|
|
import struct
|
|
import re
|
|
|
|
from hashlib import md5
|
|
from datetime import datetime, timedelta
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
|
|
vertex_pattern = re.compile(r'vertex\s+([0-9.e+-]+)\s+([0-9.e+-]+)\s+([0-9.e+-]+)')
|
|
|
|
def safe_upload_path(base_dir):
|
|
|
|
def generate_path(instance, filename):
|
|
|
|
ext = os.path.splitext(filename)[1]
|
|
|
|
md5sum = md5()
|
|
md5sum.update(instance.name
|
|
+ str(datetime.now())
|
|
+ filename)
|
|
randomname = md5sum.hexdigest()
|
|
|
|
return os.path.join(base_dir,'%s%s' % (randomname, ext))
|
|
|
|
return generate_path
|
|
|
|
|
|
|
|
class Geometry(models.Model):
|
|
|
|
HOUR = 0
|
|
DAY = 1
|
|
WEEK = 2
|
|
MONTH = 3
|
|
FORERVER = 4
|
|
|
|
EXPIRATION_CHOICES = ((HOUR, 'one hour'),
|
|
(DAY, 'one day'),
|
|
(WEEK, 'one week'),
|
|
(MONTH, 'one month'),
|
|
(FORERVER, 'forever'))
|
|
|
|
DELTAS = { HOUR : timedelta(hours = 1),
|
|
DAY : timedelta(days = 1),
|
|
WEEK : timedelta(weeks = 1),
|
|
MONTH : timedelta(weeks = 4)}
|
|
|
|
name = models.CharField(max_length = 128)
|
|
description = models.TextField(blank=True)
|
|
user = models.ForeignKey(User, blank=True, null=True)
|
|
date = models.DateTimeField(auto_now_add=True)
|
|
expiration = models.IntegerField(default = 0, choices = EXPIRATION_CHOICES)
|
|
public = models.BooleanField(default=True)
|
|
polycount = models.IntegerField(blank=True, default=0)
|
|
width = models.FloatField(blank=True, default=0)
|
|
depth = models.FloatField(blank=True, default=0)
|
|
height = models.FloatField(blank=True, default=0)
|
|
file = models.FileField(upload_to=safe_upload_path('models'))
|
|
sourcefile = models.FileField(upload_to=safe_upload_path('sources'), blank=True)
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('geometry_details', kwargs={'id' : self.pk})
|
|
|
|
@classmethod
|
|
def get_latest(cls):
|
|
return cls.objects.all().filter(public = True).order_by('-date')[:10]
|
|
|
|
def get_expiration_date(self):
|
|
for expiration in [self.HOUR, self.DAY, self.WEEK, self.MONTH]:
|
|
if self.expiration == expiration:
|
|
return self.date + self.DELTAS[expiration]
|
|
|
|
return None
|
|
|
|
|
|
def _generate_meta_infos(self):
|
|
print "Generating metainfos %s" % self.name
|
|
|
|
self.file.open()
|
|
|
|
count = 0
|
|
min_coord = [None,None,None]
|
|
max_coord = [None,None,None]
|
|
|
|
if self.file.read(5) != "solid":
|
|
print "binary"
|
|
#Skip header
|
|
self.file.seek(80)
|
|
count = struct.unpack("i",self.file.read(4))[0]
|
|
|
|
done = False
|
|
for pos in range(0,count):
|
|
#Skip normal (3 * 32bit)
|
|
self.file.seek(self.file.tell()+3*4)
|
|
#Loop over each coordinate
|
|
for vert in range(0,3):
|
|
# Loop over each coordinate
|
|
for i in range(0,3):
|
|
x = struct.unpack("<f",self.file.read(4))[0]
|
|
|
|
if min_coord[i] != None :
|
|
min_coord[i] = min(min_coord[i], x)
|
|
else:
|
|
min_coord[i] = x
|
|
|
|
if max_coord[i] != None :
|
|
max_coord[i] = max(max_coord[i], x)
|
|
else:
|
|
max_coord[i] = x
|
|
#Skip attributes (16bit)
|
|
self.file.seek(self.file.tell()+2)
|
|
|
|
else:
|
|
print "ascii"
|
|
|
|
line = self.file.readline()
|
|
while line != "":
|
|
line = line.strip()
|
|
if line.startswith("facet"):
|
|
count = count + 1
|
|
if line.startswith("vertex"):
|
|
coords = vertex_pattern.match(line).groups();
|
|
for i in range(0,3):
|
|
x = float(coords[i])
|
|
|
|
if min_coord[i] != None :
|
|
min_coord[i] = min(min_coord[i], x)
|
|
else:
|
|
min_coord[i] = x
|
|
|
|
if max_coord[i] != None :
|
|
max_coord[i] = max(max_coord[i], x)
|
|
else:
|
|
max_coord[i] = x
|
|
|
|
line = self.file.readline()
|
|
|
|
self.file.close()
|
|
|
|
self.width = max_coord[0] - min_coord[0]
|
|
self.depth = max_coord[1] - min_coord[1]
|
|
self.height = max_coord[2] - min_coord[2]
|
|
self.polycount = count
|
|
|
|
self.save()
|
|
|
|
def get_polycount(self):
|
|
if self.polycount == 0:
|
|
self._generate_meta_infos()
|
|
return self.polycount
|
|
|
|
def get_width(self):
|
|
if self.polycount == 0:
|
|
self._generate_meta_infos()
|
|
return self.width
|
|
|
|
def get_depth(self):
|
|
if self.polycount == 0:
|
|
self._generate_meta_infos()
|
|
return self.depth
|
|
|
|
def get_height(self):
|
|
if self.polycount == 0:
|
|
self._generate_meta_infos()
|
|
return self.height
|