def process(self, values, stride=0, verbose=False): # Cast all values to float32 for processing result = np.array(values, np.float32) if verbose: print( "Processing %d values. Stride size: %d" % (len(values), stride) ) # Pass the value through each ProcessNode in the ProcessList for i in range(len(self._processes)): processNode = self._processes[i] #print( "processing : %s" % result ) # Process the current value using the ProcessNode if (processNode.getAttribute('bypass') == None or not(getFeatureCompatibility() & featureSets["Autodesk"]) ): result = processNode.process(result, stride, verbose=verbose) if verbose: print( "%s (%s) - result value : %s, result type : %s" % (processNode.getAttribute('name'), processNode.getNodeType(), " ".join(map(lambda x: "%3.6f" % x, result)), type(result) ) ) if (processNode.getAttribute('bypass') != None and not(getFeatureCompatibility() & featureSets["Autodesk"])): print( "%s (%s) - bypass ignored based on feature compatibility setting." % (processNode.getAttribute('name'), processNode.getNodeType()) ) # Bypass this ProcessNode else: if verbose: print( "%s (%s) - bypassing" % (processNode.getAttribute('name'), processNode.getNodeType())) # Handle bit-depth mismatches if i > 0 and i < (len(self._processes)-1): RangeClass = ProcessList.serializableClasses['Range'] inBitDepth = self._processes[i-1].getOutBitDepth() outBitDepth = self._processes[i+1].getInBitDepth() if inBitDepth != outBitDepth: if verbose: print( "%s (%s) - Adding a Range node to adapt bit depth %s to bit depth %s" % ( processNode.getAttribute('name'), processNode.getNodeType(), inBitDepth, outBitDepth)) RangeAdapter = RangeClass(inBitDepth, outBitDepth, "adapter", "adapter", style='noClamp') result = RangeAdapter.process(result, verbose=verbose) if verbose: print( "%s (%s) - result value : %s, result type : %s" % (RangeAdapter.getAttribute('name'), RangeAdapter.getNodeType(), " ".join(map(lambda x: "%3.6f" % x, result)), type(result) ) ) return result
def read(self, element, strict=False): root = element # Store attributes for key, value in root.attrib.iteritems(): self.setAttribute(key, value) # Read child elements for child in root: # Find the Python class associated with this XML element tag elementType = child.tag.replace('-', '') elementType = child.tag.replace('_', '') elementClass = self.getClass(elementType) #print( elementType, elementClass ) if elementClass != None: # Instantiate the class and read XML data element = elementClass() # Add ProcessNodes to the ProcessList data structures processNodeClass = self.getClass("ProcessNode") if issubclass(elementClass, processNodeClass): element.read(child, strict=strict) # Check that a given ProcessNode class is compatible with # the current feature set if ((elementType in ["ExposureContrast", "Gamma", "Log", "Reference"] and not (getFeatureCompatibility() & featureSets["Autodesk"])) or (elementType in ["Group"] and not (getFeatureCompatibility() & featureSets["Duiker Research"]))): msg = "Unsupported feature : ProcessNode type %s" % elementType if strict: raise Errors.UnsupportedExtensionError(msg) else: print("ProcessList::read - %s" % msg) self.addProcess(element) # Add elements to the ProcessList data structures else: element.read(child) self.addElement(element) else: msg = "Ignoring element : %s" % child.tag if strict: raise Errors.UnknownProcessNodeError(msg) else: print("ProcessList::read - %s" % msg)
def read(self, element, strict=False): root = element # Store attributes for key, value in root.attrib.iteritems(): self.setAttribute(key, value) # Read child elements for child in root: # Find the Python class associated with this XML element tag elementType = child.tag.replace('-', '') elementType = child.tag.replace('_', '') elementClass = self.getClass(elementType) #print( elementType, elementClass ) if elementClass != None: # Instantiate the class and read XML data element = elementClass() # Add ProcessNodes to the ProcessList data structures processNodeClass = self.getClass("ProcessNode") if issubclass(elementClass, processNodeClass): element.read(child, strict=strict) # Check that a given ProcessNode class is compatible with # the current feature set if ( (elementType in ["ExposureContrast", "Gamma", "Log", "Reference"] and not (getFeatureCompatibility() & featureSets["Autodesk"])) or (elementType in ["Group"] and not (getFeatureCompatibility() & featureSets["Duiker Research"])) ): msg = "Unsupported feature : ProcessNode type %s" % elementType if strict: raise Errors.UnsupportedExtensionError(msg) else: print( "ProcessList::read - %s" % msg ) self.addProcess( element ) # Add elements to the ProcessList data structures else: element.read(child) self.addElement( element ) else: msg = "Ignoring element : %s" % child.tag if strict: raise Errors.UnknownProcessNodeError(msg) else: print( "ProcessList::read - %s" % msg )
def writeFile(self, clfPath, writeSelfContained=False, writeGzip=False): document = self.write(writeSelfContained=writeSelfContained) # Writing Gzipped XML data if writeGzip: if getFeatureCompatibility() & featureSets["Duiker Research"]: prettyString = self.xmlPrettify(document, clfPath) f = gzip.open(clfPath, 'wb') f.write(prettyString) f.close() else: msg = "Unsupported feature : write gzipped file" #print( "ProcessList::writeFile - %s" % msg ) raise Errors.UnsupportedExtensionError(msg) # Writing XML text else: # Non pretty-saving to disk #document.write(scriptPath) # Pretty saving to to disk prettyString = self.xmlPrettify(document, clfPath) fp = open(clfPath, 'wb') fp.write(prettyString) fp.close() return True
def writeFile(self, clfPath, writeSelfContained=False, writeGzip=False): document = self.write(writeSelfContained=writeSelfContained) # Writing Gzipped XML data if writeGzip: if getFeatureCompatibility() & featureSets["Duiker Research"]: prettyString = self.xmlPrettify(document, clfPath) f = gzip.open(clfPath, 'wb') f.write(prettyString) f.close() else: msg = "Unsupported feature : write gzipped file" #print( "ProcessList::writeFile - %s" % msg ) raise Errors.UnsupportedExtensionError(msg) # Writing XML text else: # Non pretty-saving to disk #document.write(scriptPath) # Pretty saving to to disk prettyString = self.xmlPrettify(document, clfPath) fp = open(clfPath, 'wb') fp.write(prettyString) fp.close() return True
def readFile(self, clfPath, strict=False): # Support for reading gzipped xml data if getFeatureCompatibility() & featureSets["Duiker Research"]: # Try to read as gzipped XML file try: f = gzip.open(clfPath, 'rb') tree = etree.parse(f) root = tree.getroot() self.read(root, strict) f.close() # Read as normal XML otherwise except: tree = etree.parse(clfPath) root = tree.getroot() self.read(root, strict) # Read as XML otherwise else: tree = etree.parse(clfPath) root = tree.getroot() self.read(root, strict)
def readFile(self, clfPath, strict=False): # Support for reading gzipped xml data if getFeatureCompatibility() & featureSets["Duiker Research"]: # Try to read as gzipped XML file try: f = gzip.open(clfPath, 'rb') tree = etree.parse(f) root = tree.getroot() self.read(root, strict) f.close() # Read as normal XML otherwise except: tree = etree.parse(clfPath) root = tree.getroot() self.read(root, strict) # Read as XML otherwise else: tree = etree.parse(clfPath) root = tree.getroot() self.read(root, strict)
def process(self, values, stride=0, verbose=False): # Handle processing of single values if stride == 0: stride = len(values) # Pass through if there aren't at least three channels if stride < 3: return values # Base attributes inBitDepth = self._attributes['inBitDepth'] outBitDepth = self._attributes['outBitDepth'] # Node attributes style = '' if 'style' in self._attributes: style = self._attributes['style'] if (not (getFeatureCompatibility() & featureSets["Autodesk"]) and style in ['v1.2_Fwd', 'noClampFwd', 'v1.2_Rev', 'noClampRev']): msg = "Unsupported feature : Autodesk CDL style keyword %s" % style raise Errors.UnsupportedExtensionError(msg) # Node parameters slope = [1.0, 1.0, 1.0] offset = [0.0, 0.0, 0.0] power = [1.0, 1.0, 1.0] saturation = 1.0 if 'slope' in self._values: slope = self._values['slope'] if 'offset' in self._values: offset = self._values['offset'] if 'power' in self._values: power = self._values['power'] if 'saturation' in self._values: saturation = self._values['saturation'] ''' print( "slope : %s" % slope ) print( "offset : %s" % offset ) print( "power : %s" % power ) print( "saturation : %s" % saturation ) ''' # Initialize the output value outValues = np.zeros(len(values), dtype=np.float32) # Support CLF spec and Autodesk CTF style keywords for p in range(len(values) / stride): value = values[p * stride:(p + 1) * stride] outValue = values[p * stride:(p + 1) * stride] if style == 'Fwd' or style == 'v1.2_Fwd': for i in range(3): outValue[i] = bitDepthToNormalized(value[i], inBitDepth) outValue[i] = pow( clamp(outValue[i] * slope[i] + offset[i]), power[i]) luma = 0.2126 * outValue[0] + 0.7152 * outValue[ 1] + 0.0722 * outValue[2] for i in range(3): outValue[i] = clamp(luma + saturation * (outValue[i] - luma)) outValue[i] = normalizedToBitDepth(outValue[i], outBitDepth) elif style == 'FwdNoClamp' or style == 'noClampFwd': for i in range(3): outValue[i] = bitDepthToNormalized(value[i], inBitDepth) tmp = outValue[i] * slope[i] + offset[i] if tmp < 0: outValue[i] = tmp else: outValue[i] = pow(tmp, power[i]) luma = 0.2126 * outValue[0] + 0.7152 * outValue[ 1] + 0.0722 * outValue[2] for i in range(3): outValue[i] = luma + saturation * (outValue[i] - luma) outValue[i] = normalizedToBitDepth(outValue[i], outBitDepth) elif style == 'Rev' or style == 'v1.2_Rev': for i in range(3): outValue[i] = bitDepthToNormalized(value[i], inBitDepth) outValue[i] = clamp(outValue[i]) luma = 0.2126 * outValue[0] + 0.7152 * outValue[ 1] + 0.0722 * outValue[2] for i in range(3): outSat = luma + (1.0 / saturation) * (outValue[i] - luma) outValue[i] = (pow(clamp(outSat), 1.0 / power[i]) - offset[i]) / slope[i] outValue[i] = clamp(outValue[i]) outValue[i] = normalizedToBitDepth(outValue[i], outBitDepth) elif style == 'RevNoClamp' or style == 'noClampRev': for i in range(3): outValue[i] = bitDepthToNormalized(value[i], inBitDepth) luma = 0.2126 * outValue[0] + 0.7152 * outValue[ 1] + 0.0722 * outValue[2] for i in range(3): outSat = luma + (1.0 / saturation) * (outValue[i] - luma) if outSat < 0: outValue[i] = (clamp(outSat) - offset[i]) / slope[i] else: outValue[i] = (pow(clamp(outSat), 1.0 / power[i]) - offset[i]) / slope[i] outValue[i] = normalizedToBitDepth(outValue[i], outBitDepth) # Copy the extra channels for i in range(min(3, stride), stride): outValue[i] = value[i] # Copy to the output array outValues[p * stride:(p + 1) * stride] = outValue return outValues
def process(self, values, stride=0, verbose=False): # Cast all values to float32 for processing result = np.array(values, np.float32) if verbose: print("Processing %d values. Stride size: %d" % (len(values), stride)) # Pass the value through each ProcessNode in the ProcessList for i in range(len(self._processes)): processNode = self._processes[i] #print( "processing : %s" % result ) # Process the current value using the ProcessNode if (processNode.getAttribute('bypass') == None or not (getFeatureCompatibility() & featureSets["Autodesk"])): result = processNode.process(result, stride, verbose=verbose) if verbose: print("%s (%s) - result value : %s, result type : %s" % (processNode.getAttribute('name'), processNode.getNodeType(), " ".join( map(lambda x: "%3.6f" % x, result)), type(result))) if (processNode.getAttribute('bypass') != None and not (getFeatureCompatibility() & featureSets["Autodesk"])): print( "%s (%s) - bypass ignored based on feature compatibility setting." % (processNode.getAttribute('name'), processNode.getNodeType())) # Bypass this ProcessNode else: if verbose: print("%s (%s) - bypassing" % (processNode.getAttribute('name'), processNode.getNodeType())) # Handle bit-depth mismatches if i > 0 and i < (len(self._processes) - 1): RangeClass = ProcessList.serializableClasses['Range'] inBitDepth = self._processes[i - 1].getOutBitDepth() outBitDepth = self._processes[i + 1].getInBitDepth() if inBitDepth != outBitDepth: if verbose: print( "%s (%s) - Adding a Range node to adapt bit depth %s to bit depth %s" % (processNode.getAttribute('name'), processNode.getNodeType(), inBitDepth, outBitDepth)) RangeAdapter = RangeClass(inBitDepth, outBitDepth, "adapter", "adapter", style='noClamp') result = RangeAdapter.process(result, verbose=verbose) if verbose: print( "%s (%s) - result value : %s, result type : %s" % (RangeAdapter.getAttribute('name'), RangeAdapter.getNodeType(), " ".join( map(lambda x: "%3.6f" % x, result)), type(result))) return result